Comments: Move 'fetch_reddit' function to own file + module
This commit is contained in:
parent
1b25737b01
commit
634e913da9
@ -1,41 +1,3 @@
|
|||||||
def fetch_reddit_comments(id, sort_by = "confidence")
|
|
||||||
client = make_client(REDDIT_URL)
|
|
||||||
headers = HTTP::Headers{"User-Agent" => "web:invidious:v#{CURRENT_VERSION} (by github.com/iv-org/invidious)"}
|
|
||||||
|
|
||||||
# TODO: Use something like #479 for a static list of instances to use here
|
|
||||||
query = URI::Params.encode({q: "(url:3D#{id} OR url:#{id}) AND (site:invidio.us OR site:youtube.com OR site:youtu.be)"})
|
|
||||||
search_results = client.get("/search.json?#{query}", headers)
|
|
||||||
|
|
||||||
if search_results.status_code == 200
|
|
||||||
search_results = RedditThing.from_json(search_results.body)
|
|
||||||
|
|
||||||
# For videos that have more than one thread, choose the one with the highest score
|
|
||||||
threads = search_results.data.as(RedditListing).children
|
|
||||||
thread = threads.max_by?(&.data.as(RedditLink).score).try(&.data.as(RedditLink))
|
|
||||||
result = thread.try do |t|
|
|
||||||
body = client.get("/r/#{t.subreddit}/comments/#{t.id}.json?limit=100&sort=#{sort_by}", headers).body
|
|
||||||
Array(RedditThing).from_json(body)
|
|
||||||
end
|
|
||||||
result ||= [] of RedditThing
|
|
||||||
elsif search_results.status_code == 302
|
|
||||||
# Previously, if there was only one result then the API would redirect to that result.
|
|
||||||
# Now, it appears it will still return a listing so this section is likely unnecessary.
|
|
||||||
|
|
||||||
result = client.get(search_results.headers["Location"], headers).body
|
|
||||||
result = Array(RedditThing).from_json(result)
|
|
||||||
|
|
||||||
thread = result[0].data.as(RedditListing).children[0].data.as(RedditLink)
|
|
||||||
else
|
|
||||||
raise NotFoundException.new("Comments not found.")
|
|
||||||
end
|
|
||||||
|
|
||||||
client.close
|
|
||||||
|
|
||||||
comments = result[1]?.try(&.data.as(RedditListing).children)
|
|
||||||
comments ||= [] of RedditThing
|
|
||||||
return comments, thread
|
|
||||||
end
|
|
||||||
|
|
||||||
def template_youtube_comments(comments, locale, thin_mode, is_replies = false)
|
def template_youtube_comments(comments, locale, thin_mode, is_replies = false)
|
||||||
String.build do |html|
|
String.build do |html|
|
||||||
root = comments["comments"].as_a
|
root = comments["comments"].as_a
|
||||||
|
41
src/invidious/comments/reddit.cr
Normal file
41
src/invidious/comments/reddit.cr
Normal file
@ -0,0 +1,41 @@
|
|||||||
|
module Invidious::Comments
|
||||||
|
extend self
|
||||||
|
|
||||||
|
def fetch_reddit(id, sort_by = "confidence")
|
||||||
|
client = make_client(REDDIT_URL)
|
||||||
|
headers = HTTP::Headers{"User-Agent" => "web:invidious:v#{CURRENT_VERSION} (by github.com/iv-org/invidious)"}
|
||||||
|
|
||||||
|
# TODO: Use something like #479 for a static list of instances to use here
|
||||||
|
query = URI::Params.encode({q: "(url:3D#{id} OR url:#{id}) AND (site:invidio.us OR site:youtube.com OR site:youtu.be)"})
|
||||||
|
search_results = client.get("/search.json?#{query}", headers)
|
||||||
|
|
||||||
|
if search_results.status_code == 200
|
||||||
|
search_results = RedditThing.from_json(search_results.body)
|
||||||
|
|
||||||
|
# For videos that have more than one thread, choose the one with the highest score
|
||||||
|
threads = search_results.data.as(RedditListing).children
|
||||||
|
thread = threads.max_by?(&.data.as(RedditLink).score).try(&.data.as(RedditLink))
|
||||||
|
result = thread.try do |t|
|
||||||
|
body = client.get("/r/#{t.subreddit}/comments/#{t.id}.json?limit=100&sort=#{sort_by}", headers).body
|
||||||
|
Array(RedditThing).from_json(body)
|
||||||
|
end
|
||||||
|
result ||= [] of RedditThing
|
||||||
|
elsif search_results.status_code == 302
|
||||||
|
# Previously, if there was only one result then the API would redirect to that result.
|
||||||
|
# Now, it appears it will still return a listing so this section is likely unnecessary.
|
||||||
|
|
||||||
|
result = client.get(search_results.headers["Location"], headers).body
|
||||||
|
result = Array(RedditThing).from_json(result)
|
||||||
|
|
||||||
|
thread = result[0].data.as(RedditListing).children[0].data.as(RedditLink)
|
||||||
|
else
|
||||||
|
raise NotFoundException.new("Comments not found.")
|
||||||
|
end
|
||||||
|
|
||||||
|
client.close
|
||||||
|
|
||||||
|
comments = result[1]?.try(&.data.as(RedditListing).children)
|
||||||
|
comments ||= [] of RedditThing
|
||||||
|
return comments, thread
|
||||||
|
end
|
||||||
|
end
|
@ -345,7 +345,7 @@ module Invidious::Routes::API::V1::Videos
|
|||||||
sort_by ||= "confidence"
|
sort_by ||= "confidence"
|
||||||
|
|
||||||
begin
|
begin
|
||||||
comments, reddit_thread = fetch_reddit_comments(id, sort_by: sort_by)
|
comments, reddit_thread = Comments.fetch_reddit(id, sort_by: sort_by)
|
||||||
rescue ex
|
rescue ex
|
||||||
comments = nil
|
comments = nil
|
||||||
reddit_thread = nil
|
reddit_thread = nil
|
||||||
|
@ -98,7 +98,7 @@ module Invidious::Routes::Watch
|
|||||||
comment_html = JSON.parse(Comments.fetch_youtube(id, nil, "html", locale, preferences.thin_mode, region))["contentHtml"]
|
comment_html = JSON.parse(Comments.fetch_youtube(id, nil, "html", locale, preferences.thin_mode, region))["contentHtml"]
|
||||||
rescue ex
|
rescue ex
|
||||||
if preferences.comments[1] == "reddit"
|
if preferences.comments[1] == "reddit"
|
||||||
comments, reddit_thread = fetch_reddit_comments(id)
|
comments, reddit_thread = Comments.fetch_reddit(id)
|
||||||
comment_html = template_reddit_comments(comments, locale)
|
comment_html = template_reddit_comments(comments, locale)
|
||||||
|
|
||||||
comment_html = fill_links(comment_html, "https", "www.reddit.com")
|
comment_html = fill_links(comment_html, "https", "www.reddit.com")
|
||||||
@ -107,7 +107,7 @@ module Invidious::Routes::Watch
|
|||||||
end
|
end
|
||||||
elsif source == "reddit"
|
elsif source == "reddit"
|
||||||
begin
|
begin
|
||||||
comments, reddit_thread = fetch_reddit_comments(id)
|
comments, reddit_thread = Comments.fetch_reddit(id)
|
||||||
comment_html = template_reddit_comments(comments, locale)
|
comment_html = template_reddit_comments(comments, locale)
|
||||||
|
|
||||||
comment_html = fill_links(comment_html, "https", "www.reddit.com")
|
comment_html = fill_links(comment_html, "https", "www.reddit.com")
|
||||||
|
Loading…
Reference in New Issue
Block a user