forked from midou/invidious
2f335b3d2c
This allows us to not pass file name ("title") in the form
data and to enforce some sanity checks
109 lines
3.4 KiB
Crystal
109 lines
3.4 KiB
Crystal
module Invidious::Frontend::WatchPage
|
|
extend self
|
|
|
|
# A handy structure to pass many elements at
|
|
# once to the download widget function
|
|
struct VideoAssets
|
|
getter full_videos : Array(Hash(String, JSON::Any))
|
|
getter video_streams : Array(Hash(String, JSON::Any))
|
|
getter audio_streams : Array(Hash(String, JSON::Any))
|
|
getter captions : Array(Caption)
|
|
|
|
def initialize(
|
|
@full_videos,
|
|
@video_streams,
|
|
@audio_streams,
|
|
@captions
|
|
)
|
|
end
|
|
end
|
|
|
|
def download_widget(locale : String, video : Video, video_assets : VideoAssets) : String
|
|
if CONFIG.disabled?("downloads")
|
|
return "<p id=\"download\">#{translate(locale, "Download is disabled.")}</p>"
|
|
end
|
|
|
|
return String.build(4000) do |str|
|
|
str << "<form"
|
|
str << " class=\"pure-form pure-form-stacked\""
|
|
str << " action='/download'"
|
|
str << " method='post'"
|
|
str << " rel='noopener'"
|
|
str << " target='_blank'>"
|
|
str << '\n'
|
|
|
|
# Hidden inputs for video id and title
|
|
str << "<input type='hidden' name='id' value='" << video.id << "'/>\n"
|
|
str << "<input type='hidden' name='title' value='" << HTML.escape(video.title) << "'/>\n"
|
|
|
|
str << "\t<div class=\"pure-control-group\">\n"
|
|
|
|
str << "\t\t<label for='download_widget'>"
|
|
str << translate(locale, "Download as: ")
|
|
str << "</label>\n"
|
|
|
|
# TODO: remove inline style
|
|
str << "\t\t<select style=\"width:100%\" name='download_widget' id='download_widget'>\n"
|
|
|
|
# Non-DASH videos (audio+video)
|
|
|
|
video_assets.full_videos.each do |option|
|
|
mimetype = option["mimeType"].as_s.split(";")[0]
|
|
|
|
height = itag_to_metadata?(option["itag"]).try &.["height"]?
|
|
|
|
value = {"itag": option["itag"], "ext": mimetype.split("/")[1]}.to_json
|
|
|
|
str << "\t\t\t<option value='" << value << "'>"
|
|
str << (height || "~240") << "p - " << mimetype
|
|
str << "</option>\n"
|
|
end
|
|
|
|
# DASH video streams
|
|
|
|
video_assets.video_streams.each do |option|
|
|
mimetype = option["mimeType"].as_s.split(";")[0]
|
|
|
|
value = {"itag": option["itag"], "ext": mimetype.split("/")[1]}.to_json
|
|
|
|
str << "\t\t\t<option value='" << value << "'>"
|
|
str << option["qualityLabel"] << " - " << mimetype << " @ " << option["fps"] << "fps - video only"
|
|
str << "</option>\n"
|
|
end
|
|
|
|
# DASH audio streams
|
|
|
|
video_assets.audio_streams.each do |option|
|
|
mimetype = option["mimeType"].as_s.split(";")[0]
|
|
|
|
value = {"itag": option["itag"], "ext": mimetype.split("/")[1]}.to_json
|
|
|
|
str << "\t\t\t<option value='" << value << "'>"
|
|
str << mimetype << " @ " << (option["bitrate"]?.try &.as_i./ 1000) << "k - audio only"
|
|
str << "</option>\n"
|
|
end
|
|
|
|
# Subtitles (a.k.a "closed captions")
|
|
|
|
video_assets.captions.each do |caption|
|
|
value = {"label": caption.name, "ext": "#{caption.language_code}.vtt"}.to_json
|
|
|
|
str << "\t\t\t<option value='" << value << "'>"
|
|
str << translate(locale, "download_subtitles", translate(locale, caption.name))
|
|
str << "</option>\n"
|
|
end
|
|
|
|
# End of form
|
|
|
|
str << "\t\t</select>\n"
|
|
str << "\t</div>\n"
|
|
|
|
str << "\t<button type=\"submit\" class=\"pure-button pure-button-primary\">\n"
|
|
str << "\t\t<b>" << translate(locale, "Download") << "</b>\n"
|
|
str << "\t</button>\n"
|
|
|
|
str << "</form>\n"
|
|
end
|
|
end
|
|
end
|