Remove useless arguments from playlist-related functions

This commit is contained in:
Samantaz Fox 2022-01-19 18:34:35 +01:00
parent 2ae074a9a4
commit 5e3c9cf290
No known key found for this signature in database
GPG key ID: F42821059186176E
6 changed files with 30 additions and 38 deletions

View file

@ -90,7 +90,7 @@ struct Playlist
property updated : Time property updated : Time
property thumbnail : String? property thumbnail : String?
def to_json(offset, locale, json : JSON::Builder, video_id : String? = nil) def to_json(offset, json : JSON::Builder, video_id : String? = nil)
json.object do json.object do
json.field "type", "playlist" json.field "type", "playlist"
json.field "title", self.title json.field "title", self.title
@ -125,7 +125,7 @@ struct Playlist
json.field "videos" do json.field "videos" do
json.array do json.array do
videos = get_playlist_videos(self, offset: offset, locale: locale, video_id: video_id) videos = get_playlist_videos(self, offset: offset, video_id: video_id)
videos.each do |video| videos.each do |video|
video.to_json(json) video.to_json(json)
end end
@ -134,13 +134,9 @@ struct Playlist
end end
end end
def to_json(offset, locale, json : JSON::Builder? = nil, video_id : String? = nil) def to_json(offset, _json : Nil = nil, video_id : String? = nil)
if json JSON.build do |json|
to_json(offset, locale, json, video_id: video_id) to_json(offset, json, video_id: video_id)
else
JSON.build do |json|
to_json(offset, locale, json, video_id: video_id)
end
end end
end end
@ -179,7 +175,7 @@ struct InvidiousPlaylist
end end
end end
def to_json(offset, locale, json : JSON::Builder, video_id : String? = nil) def to_json(offset, json : JSON::Builder, video_id : String? = nil)
json.object do json.object do
json.field "type", "invidiousPlaylist" json.field "type", "invidiousPlaylist"
json.field "title", self.title json.field "title", self.title
@ -205,7 +201,7 @@ struct InvidiousPlaylist
offset = self.index.index(index) || 0 offset = self.index.index(index) || 0
end end
videos = get_playlist_videos(self, offset: offset, locale: locale, video_id: video_id) videos = get_playlist_videos(self, offset: offset, video_id: video_id)
videos.each_with_index do |video, index| videos.each_with_index do |video, index|
video.to_json(json, offset + index) video.to_json(json, offset + index)
end end
@ -214,13 +210,9 @@ struct InvidiousPlaylist
end end
end end
def to_json(offset, locale, json : JSON::Builder? = nil, video_id : String? = nil) def to_json(offset, _json : Nil = nil, video_id : String? = nil)
if json JSON.build do |json|
to_json(offset, locale, json, video_id: video_id) to_json(offset, json, video_id: video_id)
else
JSON.build do |json|
to_json(offset, locale, json, video_id: video_id)
end
end end
end end
@ -320,7 +312,7 @@ def produce_playlist_continuation(id, index)
return continuation return continuation
end end
def get_playlist(plid, locale, refresh = true, force_refresh = false) def get_playlist(plid : String)
if plid.starts_with? "IV" if plid.starts_with? "IV"
if playlist = Invidious::Database::Playlists.select(id: plid) if playlist = Invidious::Database::Playlists.select(id: plid)
return playlist return playlist
@ -328,11 +320,11 @@ def get_playlist(plid, locale, refresh = true, force_refresh = false)
raise InfoException.new("Playlist does not exist.") raise InfoException.new("Playlist does not exist.")
end end
else else
return fetch_playlist(plid, locale) return fetch_playlist(plid)
end end
end end
def fetch_playlist(plid, locale) def fetch_playlist(plid : String)
if plid.starts_with? "UC" if plid.starts_with? "UC"
plid = "UU#{plid.lchop("UC")}" plid = "UU#{plid.lchop("UC")}"
end end
@ -402,7 +394,7 @@ def fetch_playlist(plid, locale)
}) })
end end
def get_playlist_videos(playlist, offset, locale = nil, video_id = nil) def get_playlist_videos(playlist : InvidiousPlaylist | Playlist, offset : Int32, video_id = nil)
# Show empy playlist if requested page is out of range # Show empy playlist if requested page is out of range
# (e.g, when a new playlist has been created, offset will be negative) # (e.g, when a new playlist has been created, offset will be negative)
if offset >= playlist.video_count || offset < 0 if offset >= playlist.video_count || offset < 0

View file

@ -125,7 +125,7 @@ module Invidious::Routes::API::V1::Authenticated
JSON.build do |json| JSON.build do |json|
json.array do json.array do
playlists.each do |playlist| playlists.each do |playlist|
playlist.to_json(0, locale, json) playlist.to_json(0, json)
end end
end end
end end

View file

@ -14,7 +14,7 @@ module Invidious::Routes::API::V1::Misc
# APIv1 currently uses the same logic for both # APIv1 currently uses the same logic for both
# user playlists and Invidious playlists. This means that we can't # user playlists and Invidious playlists. This means that we can't
# reasonably split them yet. This should be addressed in APIv2 # reasonably split them yet. This should be addressed in APIv2
def self.get_playlist(env) def self.get_playlist(env : HTTP::Server::Context)
locale = env.get("preferences").as(Preferences).locale locale = env.get("preferences").as(Preferences).locale
env.response.content_type = "application/json" env.response.content_type = "application/json"
@ -34,7 +34,7 @@ module Invidious::Routes::API::V1::Misc
end end
begin begin
playlist = get_playlist(plid, locale) playlist = get_playlist(plid)
rescue ex : InfoException rescue ex : InfoException
return error_json(404, ex) return error_json(404, ex)
rescue ex rescue ex
@ -49,7 +49,7 @@ module Invidious::Routes::API::V1::Misc
# includes into the playlist a maximum of 20 videos, before the offset # includes into the playlist a maximum of 20 videos, before the offset
if offset > 0 if offset > 0
lookback = offset < 50 ? offset : 50 lookback = offset < 50 ? offset : 50
response = playlist.to_json(offset - lookback, locale) response = playlist.to_json(offset - lookback)
json_response = JSON.parse(response) json_response = JSON.parse(response)
else else
# Unless the continuation is really the offset 0, it becomes expensive. # Unless the continuation is really the offset 0, it becomes expensive.
@ -58,13 +58,13 @@ module Invidious::Routes::API::V1::Misc
# it shouldn't happen often though # it shouldn't happen often though
lookback = 0 lookback = 0
response = playlist.to_json(offset, locale, video_id: video_id) response = playlist.to_json(offset, video_id: video_id)
json_response = JSON.parse(response) json_response = JSON.parse(response)
if json_response["videos"].as_a[0]["index"] != offset if json_response["videos"].as_a[0]["index"] != offset
offset = json_response["videos"].as_a[0]["index"].as_i offset = json_response["videos"].as_a[0]["index"].as_i
lookback = offset < 50 ? offset : 50 lookback = offset < 50 ? offset : 50
response = playlist.to_json(offset - lookback, locale) response = playlist.to_json(offset - lookback)
json_response = JSON.parse(response) json_response = JSON.parse(response)
end end
end end

View file

@ -6,9 +6,9 @@ module Invidious::Routes::Embed
if plid = env.params.query["list"]?.try &.gsub(/[^a-zA-Z0-9_-]/, "") if plid = env.params.query["list"]?.try &.gsub(/[^a-zA-Z0-9_-]/, "")
begin begin
playlist = get_playlist(plid, locale: locale) playlist = get_playlist(plid)
offset = env.params.query["index"]?.try &.to_i? || 0 offset = env.params.query["index"]?.try &.to_i? || 0
videos = get_playlist_videos(playlist, offset: offset, locale: locale) videos = get_playlist_videos(playlist, offset: offset)
rescue ex rescue ex
return error_template(500, ex) return error_template(500, ex)
end end
@ -60,9 +60,9 @@ module Invidious::Routes::Embed
if plid if plid
begin begin
playlist = get_playlist(plid, locale: locale) playlist = get_playlist(plid)
offset = env.params.query["index"]?.try &.to_i? || 0 offset = env.params.query["index"]?.try &.to_i? || 0
videos = get_playlist_videos(playlist, offset: offset, locale: locale) videos = get_playlist_videos(playlist, offset: offset)
rescue ex rescue ex
return error_template(500, ex) return error_template(500, ex)
end end

View file

@ -265,7 +265,7 @@ module Invidious::Routes::Feeds
if plid.starts_with? "IV" if plid.starts_with? "IV"
if playlist = Invidious::Database::Playlists.select(id: plid) if playlist = Invidious::Database::Playlists.select(id: plid)
videos = get_playlist_videos(playlist, offset: 0, locale: locale) videos = get_playlist_videos(playlist, offset: 0)
return XML.build(indent: " ", encoding: "UTF-8") do |xml| return XML.build(indent: " ", encoding: "UTF-8") do |xml|
xml.element("feed", "xmlns:yt": "http://www.youtube.com/xml/schemas/2015", xml.element("feed", "xmlns:yt": "http://www.youtube.com/xml/schemas/2015",

View file

@ -66,7 +66,7 @@ module Invidious::Routes::Playlists
user = user.as(User) user = user.as(User)
playlist_id = env.params.query["list"] playlist_id = env.params.query["list"]
playlist = get_playlist(playlist_id, locale) playlist = get_playlist(playlist_id)
subscribe_playlist(user, playlist) subscribe_playlist(user, playlist)
env.redirect "/playlist?list=#{playlist.id}" env.redirect "/playlist?list=#{playlist.id}"
@ -161,7 +161,7 @@ module Invidious::Routes::Playlists
end end
begin begin
videos = get_playlist_videos(playlist, offset: (page - 1) * 100, locale: locale) videos = get_playlist_videos(playlist, offset: (page - 1) * 100)
rescue ex rescue ex
videos = [] of PlaylistVideo videos = [] of PlaylistVideo
end end
@ -314,7 +314,7 @@ module Invidious::Routes::Playlists
begin begin
playlist_id = env.params.query["playlist_id"] playlist_id = env.params.query["playlist_id"]
playlist = get_playlist(playlist_id, locale).as(InvidiousPlaylist) playlist = get_playlist(playlist_id).as(InvidiousPlaylist)
raise "Invalid user" if playlist.author != user.email raise "Invalid user" if playlist.author != user.email
rescue ex rescue ex
if redirect if redirect
@ -405,7 +405,7 @@ module Invidious::Routes::Playlists
end end
begin begin
playlist = get_playlist(plid, locale) playlist = get_playlist(plid)
rescue ex rescue ex
return error_template(500, ex) return error_template(500, ex)
end end
@ -422,7 +422,7 @@ module Invidious::Routes::Playlists
end end
begin begin
videos = get_playlist_videos(playlist, offset: (page - 1) * 100, locale: locale) videos = get_playlist_videos(playlist, offset: (page - 1) * 100)
rescue ex rescue ex
return error_template(500, "Error encountered while retrieving playlist videos.<br>#{ex.message}") return error_template(500, "Error encountered while retrieving playlist videos.<br>#{ex.message}")
end end