summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorOmar Roth <omarroth@protonmail.com>2020-01-08 20:27:21 -0500
committerOmar Roth <omarroth@protonmail.com>2020-01-08 20:27:21 -0500
commit7b88d0efe3db6d1c288bf04991f3386c61e445d1 (patch)
tree6615dcca9df57db6dd8f5e4aff98752532c38e19
parent4aada65dae72d1f563a6f3f816afd2299d66a3dc (diff)
downloadinvidious-7b88d0efe3db6d1c288bf04991f3386c61e445d1.tar.gz
invidious-7b88d0efe3db6d1c288bf04991f3386c61e445d1.tar.bz2
invidious-7b88d0efe3db6d1c288bf04991f3386c61e445d1.zip
Minor refactor
-rw-r--r--src/invidious.cr14
-rw-r--r--src/invidious/videos.cr5
2 files changed, 7 insertions, 12 deletions
diff --git a/src/invidious.cr b/src/invidious.cr
index 5d011a5c..5ab779de 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -3841,10 +3841,10 @@ get "/api/v1/captions/:id" do |env|
env.response.content_type = "text/vtt; charset=UTF-8"
- caption = captions.select { |caption| caption.name.simpleText == label }
-
if lang
caption = captions.select { |caption| caption.languageCode == lang }
+ else
+ caption = captions.select { |caption| caption.name.simpleText == label }
end
if caption.empty?
@@ -5155,7 +5155,7 @@ get "/api/manifest/dash/id/:id" do |env|
# Since some implementations create playlists based on resolution regardless of different codecs,
# we can opt to only add a source to a representation if it has a unique height within that representation
- unique_res = env.params.query["unique_res"]? && (env.params.query["unique_res"] == "true" || env.params.query["unique_res"] == "1")
+ unique_res = env.params.query["unique_res"]?.try { |q| (q == "true" || q == "1").to_unsafe }
begin
video = get_video(id, PG_DB, region: region)
@@ -5167,7 +5167,7 @@ get "/api/manifest/dash/id/:id" do |env|
end
if dashmpd = video.player_response["streamingData"]?.try &.["dashManifestUrl"]?.try &.as_s
- manifest = YT_POOL.client &.get(dashmpd).body
+ manifest = YT_POOL.client &.get(URI.parse(dashmpd).full_path).body
manifest = manifest.gsub(/<BaseURL>[^<]+<\/BaseURL>/) do |baseurl|
url = baseurl.lchop("<BaseURL>")
@@ -5192,7 +5192,7 @@ get "/api/manifest/dash/id/:id" do |env|
end
audio_streams = video.audio_streams(adaptive_fmts)
- video_streams = video.video_streams(adaptive_fmts).sort_by { |stream| stream["fps"].to_i }.reverse
+ video_streams = video.video_streams(adaptive_fmts).sort_by { |stream| {stream["size"].split("x")[0].to_i, stream["fps"].to_i} }.reverse
XML.build(indent: " ", encoding: "UTF-8") do |xml|
xml.element("MPD", "xmlns": "urn:mpeg:dash:schema:mpd:2011",
@@ -5230,9 +5230,7 @@ get "/api/manifest/dash/id/:id" do |env|
{"video/mp4", "video/webm"}.each do |mime_type|
mime_streams = video_streams.select { |stream| stream["type"].starts_with? mime_type }
- if mime_streams.empty?
- next
- end
+ next if mime_streams.empty?
heights = [] of Int32
xml.element("AdaptationSet", id: i, mimeType: mime_type, startWithSAP: 1, subsegmentAlignment: true, scanType: "progressive") do
diff --git a/src/invidious/videos.cr b/src/invidious/videos.cr
index e9aee092..1c7599f8 100644
--- a/src/invidious/videos.cr
+++ b/src/invidious/videos.cr
@@ -621,10 +621,7 @@ struct Video
if fmts = player_response["streamingData"]?.try &.["adaptiveFormats"]?
fmts.as_a.each do |adaptive_fmt|
- if !adaptive_fmt.as_h?
- next
- end
-
+ next if !adaptive_fmt.as_h?
fmt = {} of String => String
if init = adaptive_fmt["initRange"]?