From f88697541caf337bf068aa5e7db297a015336d3d Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Sat, 13 Oct 2018 20:01:58 -0500
Subject: [PATCH] Add author_thumbnail to '/api/v1/videos'

---
 config/sql/videos.sql   |  1 +
 src/invidious.cr        | 15 +++++++++++++++
 src/invidious/videos.cr | 31 +++++++++++++++++++++++++------
 3 files changed, 41 insertions(+), 6 deletions(-)

diff --git a/config/sql/videos.sql b/config/sql/videos.sql
index 2a02ccdfe..b94405e9b 100644
--- a/config/sql/videos.sql
+++ b/config/sql/videos.sql
@@ -23,6 +23,7 @@ CREATE TABLE public.videos
     genre_url text COLLATE pg_catalog."default",
     license text COLLATE pg_catalog."default",
     sub_count_text text COLLATE pg_catalog."default",
+    author_thumbnail text COLLATE pg_catalog."default",
     CONSTRAINT videos_pkey PRIMARY KEY (id)
 )
 WITH (
diff --git a/src/invidious.cr b/src/invidious.cr
index 74c708209..b5312e5b9 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -2298,6 +2298,21 @@ get "/api/v1/videos/:id" do |env|
       json.field "author", video.author
       json.field "authorId", video.ucid
       json.field "authorUrl", "/channel/#{video.ucid}"
+
+      json.field "authorThumbnails" do
+        json.array do
+          qualities = [32, 48, 76, 100, 176, 512]
+
+          qualities.each do |quality|
+            json.object do
+              json.field "url", video.author_thumbnail.gsub("=s48-", "=s#{quality}-")
+              json.field "width", quality
+              json.field "height", quality
+            end
+          end
+        end
+      end
+
       json.field "subCountText", video.sub_count_text
 
       json.field "lengthSeconds", video.info["length_seconds"].to_i
diff --git a/src/invidious/videos.cr b/src/invidious/videos.cr
index 864466a3c..28b0922b8 100644
--- a/src/invidious/videos.cr
+++ b/src/invidious/videos.cr
@@ -457,9 +457,10 @@ class Video
     genre:              String,
     genre_url:          String,
     license:            String,
-    sub_count_text:     {
+    sub_count_text:     String,
+    author_thumbnail:   {
       type:    String,
-      default: "0",
+      default: "",
     },
   })
 end
@@ -491,12 +492,20 @@ def get_video(id, db, proxies = {} of String => Array({ip: String, port: Int32})
         video = fetch_video(id, proxies)
         video_array = video.to_a
 
+        # Migration point
+        video_array = video_array[0..-2]
+
         args = arg_array(video_array[1..-1], 2)
 
+        # Migration point
+        # db.exec("UPDATE videos SET (info,updated,title,views,likes,dislikes,wilson_score,\
+        #   published,description,language,author,ucid,allowed_regions,is_family_friendly,\
+        #   genre,genre_url,license,sub_count_text,author_thumbnail)\
+        #   = (#{args}) WHERE id = $1", video_array)
         db.exec("UPDATE videos SET (info,updated,title,views,likes,dislikes,wilson_score,\
-          published,description,language,author,ucid,allowed_regions,is_family_friendly,\
-          genre,genre_url,license,sub_count_text)\
-          = (#{args}) WHERE id = $1", video_array)
+        published,description,language,author,ucid,allowed_regions,is_family_friendly,\
+        genre,genre_url,license,sub_count_text)\
+        = (#{args}) WHERE id = $1", video_array)
       rescue ex
         db.exec("DELETE FROM videos * WHERE id = $1", id)
         raise ex
@@ -506,6 +515,9 @@ def get_video(id, db, proxies = {} of String => Array({ip: String, port: Int32})
     video = fetch_video(id, proxies)
     video_array = video.to_a
 
+    # Migration point
+    video_array = video_array[0..-2]
+
     args = arg_array(video_array)
 
     db.exec("INSERT INTO videos VALUES (#{args}) ON CONFLICT (id) DO NOTHING", video_array)
@@ -673,8 +685,15 @@ def fetch_video(id, proxies)
     sub_count_text = "0"
   end
 
+  author_thumbnail = html.xpath_node(%(//img[@alt="#{author}"]))
+  if author_thumbnail
+    author_thumbnail = author_thumbnail["data-thumb"]
+  else
+    author_thumbnail = ""
+  end
+
   video = Video.new(id, info, Time.now, title, views, likes, dislikes, wilson_score, published, description,
-    nil, author, ucid, allowed_regions, is_family_friendly, genre, genre_url, license, sub_count_text)
+    nil, author, ucid, allowed_regions, is_family_friendly, genre, genre_url, license, sub_count_text, author_thumbnail)
 
   return video
 end