mirror of
https://github.com/iv-org/invidious.git
synced 2026-03-31 15:18:30 -05:00
Encapsulate helpers constants and functions inside it's own module
This commit is contained in:
parent
fda8d1b528
commit
5df96989a3
@ -127,7 +127,7 @@ def extract_channel_community(items, *, ucid, locale, format, thin_mode, is_sing
|
|||||||
|
|
||||||
reply_count = short_text_to_number(post.dig?("actionButtons", "commentActionButtonsRenderer", "replyButton", "buttonRenderer", "text", "simpleText").try &.as_s || "0")
|
reply_count = short_text_to_number(post.dig?("actionButtons", "commentActionButtonsRenderer", "replyButton", "buttonRenderer", "text", "simpleText").try &.as_s || "0")
|
||||||
|
|
||||||
json.field "content", html_to_content(content_html)
|
json.field "content", Helpers.html_to_content(content_html)
|
||||||
json.field "contentHtml", content_html
|
json.field "contentHtml", content_html
|
||||||
|
|
||||||
json.field "published", published.to_unix
|
json.field "published", published.to_unix
|
||||||
|
|||||||
@ -254,7 +254,7 @@ module Invidious::Comments
|
|||||||
end
|
end
|
||||||
|
|
||||||
content_html = html_content || ""
|
content_html = html_content || ""
|
||||||
json.field "content", html_to_content(content_html)
|
json.field "content", Helpers.html_to_content(content_html)
|
||||||
json.field "contentHtml", content_html
|
json.field "contentHtml", content_html
|
||||||
|
|
||||||
if published_text != nil
|
if published_text != nil
|
||||||
|
|||||||
@ -1,7 +1,5 @@
|
|||||||
require "./macros"
|
require "./macros"
|
||||||
|
|
||||||
TEST_IDS = {"AgbeGFYluEA", "BaW_jenozKc", "a9LDPn-MO4I", "ddFvjfvPnqk", "iqKdEhx-dD4"}
|
|
||||||
|
|
||||||
struct Nonce
|
struct Nonce
|
||||||
include DB::Serializable
|
include DB::Serializable
|
||||||
|
|
||||||
@ -24,6 +22,11 @@ struct Annotation
|
|||||||
property annotations : String
|
property annotations : String
|
||||||
end
|
end
|
||||||
|
|
||||||
|
module Helpers
|
||||||
|
extend self
|
||||||
|
|
||||||
|
private TEST_IDS = {"AgbeGFYluEA", "BaW_jenozKc", "a9LDPn-MO4I", "ddFvjfvPnqk", "iqKdEhx-dD4"}
|
||||||
|
|
||||||
def html_to_content(description_html : String)
|
def html_to_content(description_html : String)
|
||||||
description = description_html.gsub(/(<br>)|(<br\/>)/, {
|
description = description_html.gsub(/(<br>)|(<br\/>)/, {
|
||||||
"<br>": "\n",
|
"<br>": "\n",
|
||||||
@ -200,3 +203,4 @@ def get_playback_statistic
|
|||||||
|
|
||||||
return tracker.as(Hash(String, Int64 | Float64))
|
return tracker.as(Hash(String, Int64 | Float64))
|
||||||
end
|
end
|
||||||
|
end
|
||||||
|
|||||||
@ -53,7 +53,7 @@ struct SearchVideo
|
|||||||
xml.element("img", src: "#{HOST_URL}/vi/#{self.id}/mqdefault.jpg")
|
xml.element("img", src: "#{HOST_URL}/vi/#{self.id}/mqdefault.jpg")
|
||||||
end
|
end
|
||||||
|
|
||||||
xml.element("p", style: "word-break:break-word;white-space:pre-wrap") { xml.text html_to_content(self.description_html) }
|
xml.element("p", style: "word-break:break-word;white-space:pre-wrap") { xml.text Helpers.html_to_content(self.description_html) }
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
@ -63,7 +63,7 @@ struct SearchVideo
|
|||||||
xml.element("media:title") { xml.text self.title }
|
xml.element("media:title") { xml.text self.title }
|
||||||
xml.element("media:thumbnail", url: "#{HOST_URL}/vi/#{self.id}/mqdefault.jpg",
|
xml.element("media:thumbnail", url: "#{HOST_URL}/vi/#{self.id}/mqdefault.jpg",
|
||||||
width: "320", height: "180")
|
width: "320", height: "180")
|
||||||
xml.element("media:description") { xml.text html_to_content(self.description_html) }
|
xml.element("media:description") { xml.text Helpers.html_to_content(self.description_html) }
|
||||||
end
|
end
|
||||||
|
|
||||||
xml.element("media:community") do
|
xml.element("media:community") do
|
||||||
@ -111,7 +111,7 @@ struct SearchVideo
|
|||||||
Invidious::JSONify::APIv1.thumbnails(json, self.id)
|
Invidious::JSONify::APIv1.thumbnails(json, self.id)
|
||||||
end
|
end
|
||||||
|
|
||||||
json.field "description", html_to_content(self.description_html)
|
json.field "description", Helpers.html_to_content(self.description_html)
|
||||||
json.field "descriptionHtml", self.description_html
|
json.field "descriptionHtml", self.description_html
|
||||||
|
|
||||||
json.field "viewCount", self.views
|
json.field "viewCount", self.views
|
||||||
@ -255,7 +255,7 @@ struct SearchChannel
|
|||||||
json.field "videoCount", self.video_count
|
json.field "videoCount", self.video_count
|
||||||
json.field "channelHandle", self.channel_handle
|
json.field "channelHandle", self.channel_handle
|
||||||
|
|
||||||
json.field "description", html_to_content(self.description_html)
|
json.field "description", Helpers.html_to_content(self.description_html)
|
||||||
json.field "descriptionHtml", self.description_html
|
json.field "descriptionHtml", self.description_html
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|||||||
@ -27,7 +27,7 @@ def fetch_mix(rdid, video_id, cookies = nil, locale = nil)
|
|||||||
|
|
||||||
video_id = "CvFH_6DNRCY" if rdid.starts_with? "OLAK5uy_"
|
video_id = "CvFH_6DNRCY" if rdid.starts_with? "OLAK5uy_"
|
||||||
response = YT_POOL.client &.get("/watch?v=#{video_id}&list=#{rdid}&gl=US&hl=en", headers)
|
response = YT_POOL.client &.get("/watch?v=#{video_id}&list=#{rdid}&gl=US&hl=en", headers)
|
||||||
initial_data = extract_initial_data(response.body)
|
initial_data = Helpers.extract_initial_data(response.body)
|
||||||
|
|
||||||
if !initial_data["contents"]["twoColumnWatchNextResults"]["playlist"]?
|
if !initial_data["contents"]["twoColumnWatchNextResults"]["playlist"]?
|
||||||
raise InfoException.new("Could not create mix.")
|
raise InfoException.new("Could not create mix.")
|
||||||
|
|||||||
@ -199,7 +199,7 @@ struct InvidiousPlaylist
|
|||||||
json.field "authorUrl", nil
|
json.field "authorUrl", nil
|
||||||
json.field "authorThumbnails", [] of String
|
json.field "authorThumbnails", [] of String
|
||||||
|
|
||||||
json.field "description", html_to_content(self.description_html)
|
json.field "description", Helpers.html_to_content(self.description_html)
|
||||||
json.field "descriptionHtml", self.description_html
|
json.field "descriptionHtml", self.description_html
|
||||||
json.field "videoCount", self.video_count
|
json.field "videoCount", self.video_count
|
||||||
|
|
||||||
|
|||||||
@ -8,7 +8,7 @@ module Invidious::Routes::API::V1::Authenticated
|
|||||||
# topics = env.params.body["topics"]?.try &.split(",").uniq.first(1000)
|
# topics = env.params.body["topics"]?.try &.split(",").uniq.first(1000)
|
||||||
# topics ||= [] of String
|
# topics ||= [] of String
|
||||||
|
|
||||||
# create_notification_stream(env, topics, connection_channel)
|
# Helpers.create_notification_stream(env, topics, connection_channel)
|
||||||
# end
|
# end
|
||||||
|
|
||||||
def self.get_preferences(env)
|
def self.get_preferences(env)
|
||||||
@ -485,6 +485,6 @@ module Invidious::Routes::API::V1::Authenticated
|
|||||||
topics = raw_topics.try &.split(",").uniq.first(1000)
|
topics = raw_topics.try &.split(",").uniq.first(1000)
|
||||||
topics ||= [] of String
|
topics ||= [] of String
|
||||||
|
|
||||||
create_notification_stream(env, topics, CONNECTION_CHANNEL)
|
Helpers.create_notification_stream(env, topics, CONNECTION_CHANNEL)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|||||||
@ -97,7 +97,7 @@ module Invidious::Routes::API::V1::Channels
|
|||||||
json.field "autoGenerated", channel.auto_generated
|
json.field "autoGenerated", channel.auto_generated
|
||||||
json.field "ageGated", channel.is_age_gated
|
json.field "ageGated", channel.is_age_gated
|
||||||
json.field "isFamilyFriendly", channel.is_family_friendly
|
json.field "isFamilyFriendly", channel.is_family_friendly
|
||||||
json.field "description", html_to_content(channel.description_html)
|
json.field "description", Helpers.html_to_content(channel.description_html)
|
||||||
json.field "descriptionHtml", channel.description_html
|
json.field "descriptionHtml", channel.description_html
|
||||||
|
|
||||||
json.field "allowedRegions", channel.allowed_regions
|
json.field "allowedRegions", channel.allowed_regions
|
||||||
|
|||||||
@ -300,7 +300,7 @@ module Invidious::Routes::API::V1::Videos
|
|||||||
|
|
||||||
annotations = response.body
|
annotations = response.body
|
||||||
|
|
||||||
cache_annotation(id, annotations)
|
Helpers.cache_annotation(id, annotations)
|
||||||
end
|
end
|
||||||
else # "youtube"
|
else # "youtube"
|
||||||
response = YT_POOL.client &.get("/annotations_invideo?video_id=#{id}")
|
response = YT_POOL.client &.get("/annotations_invideo?video_id=#{id}")
|
||||||
|
|||||||
@ -96,7 +96,7 @@ module Invidious::Routes::Images
|
|||||||
break
|
break
|
||||||
end
|
end
|
||||||
|
|
||||||
proxy_file(response, env)
|
Helpers.proxy_file(response, env)
|
||||||
end
|
end
|
||||||
rescue ex
|
rescue ex
|
||||||
end
|
end
|
||||||
@ -148,6 +148,6 @@ module Invidious::Routes::Images
|
|||||||
return env.response.headers.delete("Transfer-Encoding")
|
return env.response.headers.delete("Transfer-Encoding")
|
||||||
end
|
end
|
||||||
|
|
||||||
return proxy_file(response, env)
|
return Helpers.proxy_file(response, env)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|||||||
@ -83,7 +83,7 @@ module Invidious::Routes::VideoPlayback
|
|||||||
# Remove the Range header added previously.
|
# Remove the Range header added previously.
|
||||||
headers.delete("Range") if range_header.nil?
|
headers.delete("Range") if range_header.nil?
|
||||||
|
|
||||||
playback_statistics = get_playback_statistic()
|
playback_statistics = Helpers.get_playback_statistic()
|
||||||
playback_statistics["totalRequests"] += 1
|
playback_statistics["totalRequests"] += 1
|
||||||
|
|
||||||
if response.status_code >= 400
|
if response.status_code >= 400
|
||||||
@ -195,7 +195,7 @@ module Invidious::Routes::VideoPlayback
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
proxy_file(resp, env)
|
Helpers.proxy_file(resp, env)
|
||||||
end
|
end
|
||||||
rescue ex
|
rescue ex
|
||||||
if ex.message != "Error reading socket: Connection reset by peer"
|
if ex.message != "Error reading socket: Connection reset by peer"
|
||||||
|
|||||||
@ -21,7 +21,7 @@ module Invidious::Search
|
|||||||
if response.status_code == 404
|
if response.status_code == 404
|
||||||
response = YT_POOL.client &.get("/user/#{query.channel}")
|
response = YT_POOL.client &.get("/user/#{query.channel}")
|
||||||
response = YT_POOL.client &.get("/c/#{query.channel}") if response.status_code == 404
|
response = YT_POOL.client &.get("/c/#{query.channel}") if response.status_code == 404
|
||||||
initial_data = extract_initial_data(response.body)
|
initial_data = Helpers.extract_initial_data(response.body)
|
||||||
ucid = initial_data.dig?("header", "c4TabbedHeaderRenderer", "channelId").try(&.as_s?)
|
ucid = initial_data.dig?("header", "c4TabbedHeaderRenderer", "channelId").try(&.as_s?)
|
||||||
raise ChannelSearchException.new(query.channel) if !ucid
|
raise ChannelSearchException.new(query.channel) if !ucid
|
||||||
else
|
else
|
||||||
|
|||||||
@ -15,7 +15,7 @@ struct Invidious::User
|
|||||||
playlists.each do |playlist|
|
playlists.each do |playlist|
|
||||||
json.object do
|
json.object do
|
||||||
json.field "title", playlist.title
|
json.field "title", playlist.title
|
||||||
json.field "description", html_to_content(playlist.description_html)
|
json.field "description", Helpers.html_to_content(playlist.description_html)
|
||||||
json.field "privacy", playlist.privacy.to_s
|
json.field "privacy", playlist.privacy.to_s
|
||||||
json.field "videos" do
|
json.field "videos" do
|
||||||
json.array do
|
json.array do
|
||||||
|
|||||||
@ -84,7 +84,7 @@ def extract_video_info(video_id : String)
|
|||||||
|
|
||||||
# Although technically not a call to /videoplayback the fact that YouTube is returning the
|
# Although technically not a call to /videoplayback the fact that YouTube is returning the
|
||||||
# wrong video means that we should count it as a failure.
|
# wrong video means that we should count it as a failure.
|
||||||
get_playback_statistic()["totalRequests"] += 1
|
Helpers.get_playback_statistic()["totalRequests"] += 1
|
||||||
|
|
||||||
return {
|
return {
|
||||||
"version" => JSON::Any.new(Video::SCHEMA_VERSION.to_i64),
|
"version" => JSON::Any.new(Video::SCHEMA_VERSION.to_i64),
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user