Add backtraces to errors (#1498)

Error handling has been reworked to always go through the new `error_template`,
`error_json` and `error_atom` macros.
They all accept a status code followed by a string message or an exception
object. `error_json` accepts a hash with additional fields as third argument.

If the second argument is an exception a backtrace will be printed, if it is a
string only the string is printed. Since up till now only the exception message
was printed a new `InfoException` class was added for situations where no
backtrace is intended but a string cannot be used.

`error_template` with a string message automatically localizes the message.
Missing error translations have been collected in https://github.com/iv-org/invidious/issues/1497
`error_json` with a string message does not localize the message. This is the
same as previous behavior. If translations are desired for `error_json` they
can be added easily but those error messages have not been collected yet.

Uncaught exceptions previously only printed a generic message ("Looks like
you've found a bug in Invidious. [...]"). They still print that message
but now also include a backtrace.
This commit is contained in:
saltycrys 2020-11-30 10:59:21 +01:00 committed by GitHub
parent fe73eccb90
commit 3dac33ffba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
11 changed files with 250 additions and 378 deletions

View File

@ -367,9 +367,7 @@ get "/search" do |env|
begin begin
search_query, count, videos = process_search_query(query, page, user, region: nil) search_query, count, videos = process_search_query(query, page, user, region: nil)
rescue ex rescue ex
error_message = ex.message next error_template(500, ex)
env.response.status_code = 500
next templated "error"
end end
env.set "search", query env.set "search", query
@ -387,9 +385,7 @@ get "/login" do |env|
end end
if !config.login_enabled if !config.login_enabled
error_message = "Login has been disabled by administrator." next error_template(400, "Login has been disabled by administrator.")
env.response.status_code = 400
next templated "error"
end end
referer = get_referer(env, "/feed/subscriptions") referer = get_referer(env, "/feed/subscriptions")
@ -416,9 +412,7 @@ post "/login" do |env|
referer = get_referer(env, "/feed/subscriptions") referer = get_referer(env, "/feed/subscriptions")
if !config.login_enabled if !config.login_enabled
error_message = "Login has been disabled by administrator." next error_template(403, "Login has been disabled by administrator.")
env.response.status_code = 403
next templated "error"
end end
# https://stackoverflow.com/a/574698 # https://stackoverflow.com/a/574698
@ -499,9 +493,7 @@ post "/login" do |env|
headers["Cookie"] = URI.decode_www_form(headers["Cookie"]) headers["Cookie"] = URI.decode_www_form(headers["Cookie"])
if challenge_results[0][3]?.try &.== 7 if challenge_results[0][3]?.try &.== 7
error_message = translate(locale, "Account has temporarily been disabled") next error_template(423, "Account has temporarily been disabled")
env.response.status_code = 423
next templated "error"
end end
if token = challenge_results[0][-1]?.try &.[-1]?.try &.as_h?.try &.["5001"]?.try &.[-1].as_a?.try &.[-1].as_s if token = challenge_results[0][-1]?.try &.[-1]?.try &.as_h?.try &.["5001"]?.try &.[-1].as_a?.try &.[-1].as_s
@ -515,9 +507,7 @@ post "/login" do |env|
end end
if challenge_results[0][-1]?.try &.[5] == "INCORRECT_ANSWER_ENTERED" if challenge_results[0][-1]?.try &.[5] == "INCORRECT_ANSWER_ENTERED"
error_message = translate(locale, "Incorrect password") next error_template(401, "Incorrect password")
env.response.status_code = 401
next templated "error"
end end
prompt_type = challenge_results[0][-1]?.try &.[0].as_a?.try &.[0][2]? prompt_type = challenge_results[0][-1]?.try &.[0].as_a?.try &.[0][2]?
@ -550,9 +540,7 @@ post "/login" do |env|
end end
if tfa[5] == "QUOTA_EXCEEDED" if tfa[5] == "QUOTA_EXCEEDED"
error_message = translate(locale, "Quota exceeded, try again in a few hours") next error_template(423, "Quota exceeded, try again in a few hours")
env.response.status_code = 423
next templated "error"
end end
if !tfa_code if !tfa_code
@ -608,9 +596,7 @@ post "/login" do |env|
}, },
}.to_json }.to_json
else else
error_message = translate(locale, "Unable to log in, make sure two-factor authentication (Authenticator or SMS) is turned on.") next error_template(500, "Unable to log in, make sure two-factor authentication (Authenticator or SMS) is turned on.")
env.response.status_code = 500
next templated "error"
end end
traceback << "Submitting challenge..." traceback << "Submitting challenge..."
@ -621,9 +607,7 @@ post "/login" do |env|
if (challenge_results[0][-1]?.try &.[5] == "INCORRECT_ANSWER_ENTERED") || if (challenge_results[0][-1]?.try &.[5] == "INCORRECT_ANSWER_ENTERED") ||
(challenge_results[0][-1]?.try &.[5] == "INVALID_INPUT") (challenge_results[0][-1]?.try &.[5] == "INVALID_INPUT")
error_message = translate(locale, "Invalid TFA code") next error_template(401, "Invalid TFA code")
env.response.status_code = 401
next templated "error"
end end
traceback << "done.<br/>" traceback << "done.<br/>"
@ -702,29 +686,22 @@ post "/login" do |env|
traceback.rewind traceback.rewind
# error_message = translate(locale, "Login failed. This may be because two-factor authentication is not turned on for your account.") # error_message = translate(locale, "Login failed. This may be because two-factor authentication is not turned on for your account.")
error_message = %(#{ex.message}<br/>Traceback:<br/><div style="padding-left:2em" id="traceback">#{traceback.gets_to_end}</div>) error_message = %(#{ex.message}<br/>Traceback:<br/><div style="padding-left:2em" id="traceback">#{traceback.gets_to_end}</div>)
env.response.status_code = 500 next error_template(500, error_message)
next templated "error"
end end
when "invidious" when "invidious"
if !email if !email
error_message = translate(locale, "User ID is a required field") next error_template(401, "User ID is a required field")
env.response.status_code = 401
next templated "error"
end end
if !password if !password
error_message = translate(locale, "Password is a required field") next error_template(401, "Password is a required field")
env.response.status_code = 401
next templated "error"
end end
user = PG_DB.query_one?("SELECT * FROM users WHERE email = $1", email, as: User) user = PG_DB.query_one?("SELECT * FROM users WHERE email = $1", email, as: User)
if user if user
if !user.password if !user.password
error_message = translate(locale, "Please sign in using 'Log in with Google'") next error_template(400, "Please sign in using 'Log in with Google'")
env.response.status_code = 400
next templated "error"
end end
if Crypto::Bcrypt::Password.new(user.password.not_nil!).verify(password.byte_slice(0, 55)) if Crypto::Bcrypt::Password.new(user.password.not_nil!).verify(password.byte_slice(0, 55))
@ -745,9 +722,7 @@ post "/login" do |env|
secure: secure, http_only: true) secure: secure, http_only: true)
end end
else else
error_message = translate(locale, "Wrong username or password") next error_template(401, "Wrong username or password")
env.response.status_code = 401
next templated "error"
end end
# Since this user has already registered, we don't want to overwrite their preferences # Since this user has already registered, we don't want to overwrite their preferences
@ -758,22 +733,16 @@ post "/login" do |env|
end end
else else
if !config.registration_enabled if !config.registration_enabled
error_message = "Registration has been disabled by administrator." next error_template(400, "Registration has been disabled by administrator.")
env.response.status_code = 400
next templated "error"
end end
if password.empty? if password.empty?
error_message = translate(locale, "Password cannot be empty") next error_template(401, "Password cannot be empty")
env.response.status_code = 401
next templated "error"
end end
# See https://security.stackexchange.com/a/39851 # See https://security.stackexchange.com/a/39851
if password.bytesize > 55 if password.bytesize > 55
error_message = translate(locale, "Password should not be longer than 55 characters") next error_template(400, "Password cannot be longer than 55 characters")
env.response.status_code = 400
next templated "error"
end end
password = password.byte_slice(0, 55) password = password.byte_slice(0, 55)
@ -815,28 +784,28 @@ post "/login" do |env|
begin begin
validate_request(tokens[0], answer, env.request, HMAC_KEY, PG_DB, locale) validate_request(tokens[0], answer, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
error_message = ex.message next error_template(400, ex)
env.response.status_code = 400
next templated "error"
end end
else # "text" else # "text"
answer = Digest::MD5.hexdigest(answer.downcase.strip) answer = Digest::MD5.hexdigest(answer.downcase.strip)
found_valid_captcha = false if tokens.empty?
next error_template(500, "Erroneous CAPTCHA")
end
error_message = translate(locale, "Erroneous CAPTCHA") found_valid_captcha = false
error_exception = Exception.new
tokens.each_with_index do |token, i| tokens.each_with_index do |token, i|
begin begin
validate_request(token, answer, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, answer, env.request, HMAC_KEY, PG_DB, locale)
found_valid_captcha = true found_valid_captcha = true
rescue ex rescue ex
error_message = ex.message error_exception = ex
end end
end end
if !found_valid_captcha if !found_valid_captcha
env.response.status_code = 500 next error_template(500, error_exception)
next templated "error"
end end
end end
end end
@ -902,9 +871,7 @@ post "/signout" do |env|
begin begin
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
error_message = ex.message next error_template(400, ex)
env.response.status_code = 400
next templated "error"
end end
PG_DB.exec("DELETE FROM session_ids * WHERE id = $1", sid) PG_DB.exec("DELETE FROM session_ids * WHERE id = $1", sid)
@ -1182,9 +1149,7 @@ post "/watch_ajax" do |env|
if redirect if redirect
next env.redirect referer next env.redirect referer
else else
error_message = {"error" => "No such user"}.to_json next error_json(403, "No such user")
env.response.status_code = 403
next error_message
end end
end end
@ -1201,13 +1166,10 @@ post "/watch_ajax" do |env|
begin begin
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
env.response.status_code = 400
if redirect if redirect
error_message = ex.message next error_template(400, ex)
next templated "error"
else else
error_message = {"error" => ex.message}.to_json next error_json(400, ex)
next error_message
end end
end end
@ -1227,9 +1189,7 @@ post "/watch_ajax" do |env|
when "action_mark_unwatched" when "action_mark_unwatched"
PG_DB.exec("UPDATE users SET watched = array_remove(watched, $1) WHERE email = $2", id, user.email) PG_DB.exec("UPDATE users SET watched = array_remove(watched, $1) WHERE email = $2", id, user.email)
else else
error_message = {"error" => "Unsupported action #{action}"}.to_json next error_json(400, "Unsupported action #{action}")
env.response.status_code = 400
next error_message
end end
if redirect if redirect
@ -1259,9 +1219,7 @@ get "/modify_notifications" do |env|
if redirect if redirect
next env.redirect referer next env.redirect referer
else else
error_message = {"error" => "No such user"}.to_json next error_json(403, "No such user")
env.response.status_code = 403
next error_message
end end
end end
@ -1334,9 +1292,7 @@ post "/subscription_ajax" do |env|
if redirect if redirect
next env.redirect referer next env.redirect referer
else else
error_message = {"error" => "No such user"}.to_json next error_json(403, "No such user")
env.response.status_code = 403
next error_message
end end
end end
@ -1348,13 +1304,9 @@ post "/subscription_ajax" do |env|
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
if redirect if redirect
error_message = ex.message next error_template(400, ex)
env.response.status_code = 400
next templated "error"
else else
error_message = {"error" => ex.message}.to_json next error_json(400, ex)
env.response.status_code = 400
next error_message
end end
end end
@ -1384,9 +1336,7 @@ post "/subscription_ajax" do |env|
when "action_remove_subscriptions" when "action_remove_subscriptions"
PG_DB.exec("UPDATE users SET feed_needs_update = true, subscriptions = array_remove(subscriptions, $1) WHERE email = $2", channel_id, email) PG_DB.exec("UPDATE users SET feed_needs_update = true, subscriptions = array_remove(subscriptions, $1) WHERE email = $2", channel_id, email)
else else
error_message = {"error" => "Unsupported action #{action}"}.to_json next error_json(400, "Unsupported action #{action}")
env.response.status_code = 400
next error_message
end end
if redirect if redirect
@ -1569,7 +1519,7 @@ post "/data_control" do |env|
PG_DB.exec("UPDATE playlists SET description = $1 WHERE id = $2", description, playlist.id) PG_DB.exec("UPDATE playlists SET description = $1 WHERE id = $2", description, playlist.id)
videos = item["videos"]?.try &.as_a?.try &.each_with_index do |video_id, idx| videos = item["videos"]?.try &.as_a?.try &.each_with_index do |video_id, idx|
raise "Playlist cannot have more than 500 videos" if idx > 500 raise InfoException.new("Playlist cannot have more than 500 videos") if idx > 500
video_id = video_id.try &.as_s? video_id = video_id.try &.as_s?
next if !video_id next if !video_id
@ -1706,51 +1656,37 @@ post "/change_password" do |env|
# We don't store passwords for Google accounts # We don't store passwords for Google accounts
if !user.password if !user.password
error_message = "Cannot change password for Google accounts" next error_template(400, "Cannot change password for Google accounts")
env.response.status_code = 400
next templated "error"
end end
begin begin
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
error_message = ex.message next error_template(400, ex)
env.response.status_code = 400
next templated "error"
end end
password = env.params.body["password"]? password = env.params.body["password"]?
if !password if !password
error_message = translate(locale, "Password is a required field") next error_template(401, "Password is a required field")
env.response.status_code = 401
next templated "error"
end end
new_passwords = env.params.body.select { |k, v| k.match(/^new_password\[\d+\]$/) }.map { |k, v| v } new_passwords = env.params.body.select { |k, v| k.match(/^new_password\[\d+\]$/) }.map { |k, v| v }
if new_passwords.size <= 1 || new_passwords.uniq.size != 1 if new_passwords.size <= 1 || new_passwords.uniq.size != 1
error_message = translate(locale, "New passwords must match") next error_template(400, "New passwords must match")
env.response.status_code = 400
next templated "error"
end end
new_password = new_passwords.uniq[0] new_password = new_passwords.uniq[0]
if new_password.empty? if new_password.empty?
error_message = translate(locale, "Password cannot be empty") next error_template(401, "Password cannot be empty")
env.response.status_code = 401
next templated "error"
end end
if new_password.bytesize > 55 if new_password.bytesize > 55
error_message = translate(locale, "Password should not be longer than 55 characters") next error_template(400, "Password cannot be longer than 55 characters")
env.response.status_code = 400
next templated "error"
end end
if !Crypto::Bcrypt::Password.new(user.password.not_nil!).verify(password.byte_slice(0, 55)) if !Crypto::Bcrypt::Password.new(user.password.not_nil!).verify(password.byte_slice(0, 55))
error_message = translate(locale, "Incorrect password") next error_template(401, "Incorrect password")
env.response.status_code = 401
next templated "error"
end end
new_password = Crypto::Bcrypt::Password.create(new_password, cost: 10) new_password = Crypto::Bcrypt::Password.create(new_password, cost: 10)
@ -1795,9 +1731,7 @@ post "/delete_account" do |env|
begin begin
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
error_message = ex.message next error_template(400, ex)
env.response.status_code = 400
next templated "error"
end end
view_name = "subscriptions_#{sha256(user.email)}" view_name = "subscriptions_#{sha256(user.email)}"
@ -1849,9 +1783,7 @@ post "/clear_watch_history" do |env|
begin begin
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
error_message = ex.message next error_template(400, ex)
env.response.status_code = 400
next templated "error"
end end
PG_DB.exec("UPDATE users SET watched = '{}' WHERE email = $1", user.email) PG_DB.exec("UPDATE users SET watched = '{}' WHERE email = $1", user.email)
@ -1904,9 +1836,7 @@ post "/authorize_token" do |env|
begin begin
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
error_message = ex.message next error_template(400, ex)
env.response.status_code = 400
next templated "error"
end end
scopes = env.params.body.select { |k, v| k.match(/^scopes\[\d+\]$/) }.map { |k, v| v } scopes = env.params.body.select { |k, v| k.match(/^scopes\[\d+\]$/) }.map { |k, v| v }
@ -1969,9 +1899,7 @@ post "/token_ajax" do |env|
if redirect if redirect
next env.redirect referer next env.redirect referer
else else
error_message = {"error" => "No such user"}.to_json next error_json(403, "No such user")
env.response.status_code = 403
next error_message
end end
end end
@ -1983,13 +1911,9 @@ post "/token_ajax" do |env|
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
if redirect if redirect
error_message = ex.message next error_template(400, ex)
env.response.status_code = 400
next templated "error"
else else
error_message = {"error" => ex.message}.to_json next error_json(400, ex)
env.response.status_code = 400
next error_message
end end
end end
@ -2006,9 +1930,7 @@ post "/token_ajax" do |env|
when .starts_with? "action_revoke_token" when .starts_with? "action_revoke_token"
PG_DB.exec("DELETE FROM session_ids * WHERE id = $1 AND email = $2", session, user.email) PG_DB.exec("DELETE FROM session_ids * WHERE id = $1 AND email = $2", session, user.email)
else else
error_message = {"error" => "Unsupported action #{action}"}.to_json next error_json(400, "Unsupported action #{action}")
env.response.status_code = 400
next error_message
end end
if redirect if redirect
@ -2048,9 +1970,7 @@ get "/feed/trending" do |env|
begin begin
trending, plid = fetch_trending(trending_type, region, locale) trending, plid = fetch_trending(trending_type, region, locale)
rescue ex rescue ex
error_message = "#{ex.message}" next error_template(500, ex)
env.response.status_code = 500
next templated "error"
end end
templated "trending" templated "trending"
@ -2145,9 +2065,7 @@ get "/feed/channel/:ucid" do |env|
rescue ex : ChannelRedirect rescue ex : ChannelRedirect
next env.redirect env.request.resource.gsub(ucid, ex.channel_id) next env.redirect env.request.resource.gsub(ucid, ex.channel_id)
rescue ex rescue ex
error_message = ex.message next error_atom(500, ex)
env.response.status_code = 500
next error_message
end end
response = YT_POOL.client &.get("/feeds/videos.xml?channel_id=#{channel.ucid}") response = YT_POOL.client &.get("/feeds/videos.xml?channel_id=#{channel.ucid}")
@ -2558,9 +2476,7 @@ get "/channel/:ucid" do |env|
rescue ex : ChannelRedirect rescue ex : ChannelRedirect
next env.redirect env.request.resource.gsub(ucid, ex.channel_id) next env.redirect env.request.resource.gsub(ucid, ex.channel_id)
rescue ex rescue ex
error_message = ex.message next error_template(500, ex)
env.response.status_code = 500
next templated "error"
end end
if channel.auto_generated if channel.auto_generated
@ -2627,9 +2543,7 @@ get "/channel/:ucid/playlists" do |env|
rescue ex : ChannelRedirect rescue ex : ChannelRedirect
next env.redirect env.request.resource.gsub(ucid, ex.channel_id) next env.redirect env.request.resource.gsub(ucid, ex.channel_id)
rescue ex rescue ex
error_message = ex.message next error_template(500, ex)
env.response.status_code = 500
next templated "error"
end end
if channel.auto_generated if channel.auto_generated
@ -2667,9 +2581,7 @@ get "/channel/:ucid/community" do |env|
rescue ex : ChannelRedirect rescue ex : ChannelRedirect
next env.redirect env.request.resource.gsub(ucid, ex.channel_id) next env.redirect env.request.resource.gsub(ucid, ex.channel_id)
rescue ex rescue ex
error_message = ex.message next error_template(500, ex)
env.response.status_code = 500
next templated "error"
end end
if !channel.tabs.includes? "community" if !channel.tabs.includes? "community"
@ -2678,9 +2590,11 @@ get "/channel/:ucid/community" do |env|
begin begin
items = JSON.parse(fetch_channel_community(ucid, continuation, locale, "json", thin_mode)) items = JSON.parse(fetch_channel_community(ucid, continuation, locale, "json", thin_mode))
rescue ex rescue ex : InfoException
env.response.status_code = 500 env.response.status_code = 500
error_message = ex.message error_message = ex.message
rescue ex
next error_template(500, ex)
end end
env.set "search", "channel:#{channel.ucid} " env.set "search", "channel:#{channel.ucid} "
@ -2690,12 +2604,11 @@ end
# API Endpoints # API Endpoints
get "/api/v1/stats" do |env| get "/api/v1/stats" do |env|
locale = LOCALES[env.get("preferences").as(Preferences).locale]?
env.response.content_type = "application/json" env.response.content_type = "application/json"
if !config.statistics_enabled if !config.statistics_enabled
error_message = {"error" => "Statistics are not enabled."}.to_json next error_json(400, "Statistics are not enabled.")
env.response.status_code = 400
next error_message
end end
Invidious::Jobs::StatisticsRefreshJob::STATISTICS.to_json Invidious::Jobs::StatisticsRefreshJob::STATISTICS.to_json
@ -2715,10 +2628,8 @@ get "/api/v1/storyboards/:id" do |env|
begin begin
video = get_video(id, PG_DB, region: region) video = get_video(id, PG_DB, region: region)
rescue ex : VideoRedirect rescue ex : VideoRedirect
error_message = {"error" => "Video is unavailable", "videoId" => ex.video_id}.to_json
env.response.status_code = 302
env.response.headers["Location"] = env.request.resource.gsub(id, ex.video_id) env.response.headers["Location"] = env.request.resource.gsub(id, ex.video_id)
next error_message next error_json(302, "Video is unavailable", {"videoId" => ex.video_id})
rescue ex rescue ex
env.response.status_code = 500 env.response.status_code = 500
next next
@ -2803,10 +2714,8 @@ get "/api/v1/captions/:id" do |env|
begin begin
video = get_video(id, PG_DB, region: region) video = get_video(id, PG_DB, region: region)
rescue ex : VideoRedirect rescue ex : VideoRedirect
error_message = {"error" => "Video is unavailable", "videoId" => ex.video_id}.to_json
env.response.status_code = 302
env.response.headers["Location"] = env.request.resource.gsub(id, ex.video_id) env.response.headers["Location"] = env.request.resource.gsub(id, ex.video_id)
next error_message next error_json(302, "Video is unavailable", {"videoId" => ex.video_id})
rescue ex rescue ex
env.response.status_code = 500 env.response.status_code = 500
next next
@ -2938,9 +2847,7 @@ get "/api/v1/comments/:id" do |env|
begin begin
comments = fetch_youtube_comments(id, PG_DB, continuation, format, locale, thin_mode, region, sort_by: sort_by) comments = fetch_youtube_comments(id, PG_DB, continuation, format, locale, thin_mode, region, sort_by: sort_by)
rescue ex rescue ex
error_message = {"error" => ex.message}.to_json next error_json(500, ex)
env.response.status_code = 500
next error_message
end end
next comments next comments
@ -2983,13 +2890,7 @@ end
get "/api/v1/insights/:id" do |env| get "/api/v1/insights/:id" do |env|
locale = LOCALES[env.get("preferences").as(Preferences).locale]? locale = LOCALES[env.get("preferences").as(Preferences).locale]?
next error_json(410, "YouTube has removed publicly available analytics.")
id = env.params.url["id"]
env.response.content_type = "application/json"
error_message = {"error" => "YouTube has removed publicly available analytics."}.to_json
env.response.status_code = 410
error_message
end end
get "/api/v1/annotations/:id" do |env| get "/api/v1/annotations/:id" do |env|
@ -3078,14 +2979,10 @@ get "/api/v1/videos/:id" do |env|
begin begin
video = get_video(id, PG_DB, region: region) video = get_video(id, PG_DB, region: region)
rescue ex : VideoRedirect rescue ex : VideoRedirect
error_message = {"error" => "Video is unavailable", "videoId" => ex.video_id}.to_json
env.response.status_code = 302
env.response.headers["Location"] = env.request.resource.gsub(id, ex.video_id) env.response.headers["Location"] = env.request.resource.gsub(id, ex.video_id)
next error_message next error_json(302, "Video is unavailable", {"videoId" => ex.video_id})
rescue ex rescue ex
error_message = {"error" => ex.message}.to_json next error_json(500, ex)
env.response.status_code = 500
next error_message
end end
video.to_json(locale) video.to_json(locale)
@ -3102,9 +2999,7 @@ get "/api/v1/trending" do |env|
begin begin
trending, plid = fetch_trending(trending_type, region, locale) trending, plid = fetch_trending(trending_type, region, locale)
rescue ex rescue ex
error_message = {"error" => ex.message}.to_json next error_json(500, ex)
env.response.status_code = 500
next error_message
end end
videos = JSON.build do |json| videos = JSON.build do |json|
@ -3151,14 +3046,10 @@ get "/api/v1/channels/:ucid" do |env|
begin begin
channel = get_about_info(ucid, locale) channel = get_about_info(ucid, locale)
rescue ex : ChannelRedirect rescue ex : ChannelRedirect
error_message = {"error" => "Channel is unavailable", "authorId" => ex.channel_id}.to_json
env.response.status_code = 302
env.response.headers["Location"] = env.request.resource.gsub(ucid, ex.channel_id) env.response.headers["Location"] = env.request.resource.gsub(ucid, ex.channel_id)
next error_message next error_json(302, "Channel is unavailable", {"authorId" => ex.channel_id})
rescue ex rescue ex
error_message = {"error" => ex.message}.to_json next error_json(500, ex)
env.response.status_code = 500
next error_message
end end
page = 1 page = 1
@ -3169,9 +3060,7 @@ get "/api/v1/channels/:ucid" do |env|
begin begin
count, videos = get_60_videos(channel.ucid, channel.author, page, channel.auto_generated, sort_by) count, videos = get_60_videos(channel.ucid, channel.author, page, channel.auto_generated, sort_by)
rescue ex rescue ex
error_message = {"error" => ex.message}.to_json next error_json(500, ex)
env.response.status_code = 500
next error_message
end end
end end
@ -3286,22 +3175,16 @@ end
begin begin
channel = get_about_info(ucid, locale) channel = get_about_info(ucid, locale)
rescue ex : ChannelRedirect rescue ex : ChannelRedirect
error_message = {"error" => "Channel is unavailable", "authorId" => ex.channel_id}.to_json
env.response.status_code = 302
env.response.headers["Location"] = env.request.resource.gsub(ucid, ex.channel_id) env.response.headers["Location"] = env.request.resource.gsub(ucid, ex.channel_id)
next error_message next error_json(302, "Channel is unavailable", {"authorId" => ex.channel_id})
rescue ex rescue ex
error_message = {"error" => ex.message}.to_json next error_json(500, ex)
env.response.status_code = 500
next error_message
end end
begin begin
count, videos = get_60_videos(channel.ucid, channel.author, page, channel.auto_generated, sort_by) count, videos = get_60_videos(channel.ucid, channel.author, page, channel.auto_generated, sort_by)
rescue ex rescue ex
error_message = {"error" => ex.message}.to_json next error_json(500, ex)
env.response.status_code = 500
next error_message
end end
JSON.build do |json| JSON.build do |json|
@ -3325,9 +3208,7 @@ end
begin begin
videos = get_latest_videos(ucid) videos = get_latest_videos(ucid)
rescue ex rescue ex
error_message = {"error" => ex.message}.to_json next error_json(500, ex)
env.response.status_code = 500
next error_message
end end
JSON.build do |json| JSON.build do |json|
@ -3355,14 +3236,10 @@ end
begin begin
channel = get_about_info(ucid, locale) channel = get_about_info(ucid, locale)
rescue ex : ChannelRedirect rescue ex : ChannelRedirect
error_message = {"error" => "Channel is unavailable", "authorId" => ex.channel_id}.to_json
env.response.status_code = 302
env.response.headers["Location"] = env.request.resource.gsub(ucid, ex.channel_id) env.response.headers["Location"] = env.request.resource.gsub(ucid, ex.channel_id)
next error_message next error_json(302, "Channel is unavailable", {"authorId" => ex.channel_id})
rescue ex rescue ex
error_message = {"error" => ex.message}.to_json next error_json(500, ex)
env.response.status_code = 500
next error_message
end end
items, continuation = fetch_channel_playlists(channel.ucid, channel.author, channel.auto_generated, continuation, sort_by) items, continuation = fetch_channel_playlists(channel.ucid, channel.author, channel.auto_generated, continuation, sort_by)
@ -3403,9 +3280,7 @@ end
begin begin
fetch_channel_community(ucid, continuation, locale, format, thin_mode) fetch_channel_community(ucid, continuation, locale, format, thin_mode)
rescue ex rescue ex
env.response.status_code = 400 next error_json(500, ex)
error_message = {"error" => ex.message}.to_json
next error_message
end end
end end
end end
@ -3463,9 +3338,7 @@ get "/api/v1/search" do |env|
begin begin
search_params = produce_search_params(sort_by, date, content_type, duration, features) search_params = produce_search_params(sort_by, date, content_type, duration, features)
rescue ex rescue ex
env.response.status_code = 400 next error_json(400, ex)
error_message = {"error" => ex.message}.to_json
next error_message
end end
count, search_results = search(query, page, search_params, region).as(Tuple) count, search_results = search(query, page, search_params, region).as(Tuple)
@ -3508,9 +3381,7 @@ get "/api/v1/search/suggestions" do |env|
end end
end end
rescue ex rescue ex
env.response.status_code = 500 next error_json(500, ex)
error_message = {"error" => ex.message}.to_json
next error_message
end end
end end
@ -3537,16 +3408,12 @@ end
begin begin
playlist = get_playlist(PG_DB, plid, locale) playlist = get_playlist(PG_DB, plid, locale)
rescue ex rescue ex
env.response.status_code = 404 next error_json(404, "Playlist does not exist.")
error_message = {"error" => "Playlist does not exist."}.to_json
next error_message
end end
user = env.get?("user").try &.as(User) user = env.get?("user").try &.as(User)
if !playlist || playlist.privacy.private? && playlist.author != user.try &.email if !playlist || playlist.privacy.private? && playlist.author != user.try &.email
env.response.status_code = 404 next error_json(404, "Playlist does not exist.")
error_message = {"error" => "Playlist does not exist."}.to_json
next error_message
end end
response = playlist.to_json(offset, locale, continuation: continuation) response = playlist.to_json(offset, locale, continuation: continuation)
@ -3590,9 +3457,7 @@ get "/api/v1/mixes/:rdid" do |env|
mix.videos = mix.videos[index..-1] mix.videos = mix.videos[index..-1]
rescue ex rescue ex
error_message = {"error" => ex.message}.to_json next error_json(500, ex)
env.response.status_code = 500
next error_message
end end
response = JSON.build do |json| response = JSON.build do |json|
@ -3794,22 +3659,16 @@ post "/api/v1/auth/playlists" do |env|
title = env.params.json["title"]?.try &.as(String).delete("<>").byte_slice(0, 150) title = env.params.json["title"]?.try &.as(String).delete("<>").byte_slice(0, 150)
if !title if !title
error_message = {"error" => "Invalid title."}.to_json next error_json(400, "Invalid title.")
env.response.status_code = 400
next error_message
end end
privacy = env.params.json["privacy"]?.try { |privacy| PlaylistPrivacy.parse(privacy.as(String).downcase) } privacy = env.params.json["privacy"]?.try { |privacy| PlaylistPrivacy.parse(privacy.as(String).downcase) }
if !privacy if !privacy
error_message = {"error" => "Invalid privacy setting."}.to_json next error_json(400, "Invalid privacy setting.")
env.response.status_code = 400
next error_message
end end
if PG_DB.query_one("SELECT count(*) FROM playlists WHERE author = $1", user.email, as: Int64) >= 100 if PG_DB.query_one("SELECT count(*) FROM playlists WHERE author = $1", user.email, as: Int64) >= 100
error_message = {"error" => "User cannot have more than 100 playlists."}.to_json next error_json(400, "User cannot have more than 100 playlists.")
env.response.status_code = 400
next error_message
end end
playlist = create_playlist(PG_DB, title, privacy, user) playlist = create_playlist(PG_DB, title, privacy, user)
@ -3831,15 +3690,11 @@ patch "/api/v1/auth/playlists/:plid" do |env|
playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist) playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist)
if !playlist || playlist.author != user.email && playlist.privacy.private? if !playlist || playlist.author != user.email && playlist.privacy.private?
env.response.status_code = 404 next error_json(404, "Playlist does not exist.")
error_message = {"error" => "Playlist does not exist."}.to_json
next error_message
end end
if playlist.author != user.email if playlist.author != user.email
env.response.status_code = 403 next error_json(403, "Invalid user")
error_message = {"error" => "Invalid user"}.to_json
next error_message
end end
title = env.params.json["title"].try &.as(String).delete("<>").byte_slice(0, 150) || playlist.title title = env.params.json["title"].try &.as(String).delete("<>").byte_slice(0, 150) || playlist.title
@ -3859,6 +3714,8 @@ patch "/api/v1/auth/playlists/:plid" do |env|
end end
delete "/api/v1/auth/playlists/:plid" do |env| delete "/api/v1/auth/playlists/:plid" do |env|
locale = LOCALES[env.get("preferences").as(Preferences).locale]?
env.response.content_type = "application/json" env.response.content_type = "application/json"
user = env.get("user").as(User) user = env.get("user").as(User)
@ -3866,15 +3723,11 @@ delete "/api/v1/auth/playlists/:plid" do |env|
playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist) playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist)
if !playlist || playlist.author != user.email && playlist.privacy.private? if !playlist || playlist.author != user.email && playlist.privacy.private?
env.response.status_code = 404 next error_json(404, "Playlist does not exist.")
error_message = {"error" => "Playlist does not exist."}.to_json
next error_message
end end
if playlist.author != user.email if playlist.author != user.email
env.response.status_code = 403 next error_json(403, "Invalid user")
error_message = {"error" => "Invalid user"}.to_json
next error_message
end end
PG_DB.exec("DELETE FROM playlist_videos * WHERE plid = $1", plid) PG_DB.exec("DELETE FROM playlist_videos * WHERE plid = $1", plid)
@ -3893,36 +3746,26 @@ post "/api/v1/auth/playlists/:plid/videos" do |env|
playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist) playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist)
if !playlist || playlist.author != user.email && playlist.privacy.private? if !playlist || playlist.author != user.email && playlist.privacy.private?
env.response.status_code = 404 next error_json(404, "Playlist does not exist.")
error_message = {"error" => "Playlist does not exist."}.to_json
next error_message
end end
if playlist.author != user.email if playlist.author != user.email
env.response.status_code = 403 next error_json(403, "Invalid user")
error_message = {"error" => "Invalid user"}.to_json
next error_message
end end
if playlist.index.size >= 500 if playlist.index.size >= 500
env.response.status_code = 400 next error_json(400, "Playlist cannot have more than 500 videos")
error_message = {"error" => "Playlist cannot have more than 500 videos"}.to_json
next error_message
end end
video_id = env.params.json["videoId"].try &.as(String) video_id = env.params.json["videoId"].try &.as(String)
if !video_id if !video_id
env.response.status_code = 403 next error_json(403, "Invalid videoId")
error_message = {"error" => "Invalid videoId"}.to_json
next error_message
end end
begin begin
video = get_video(video_id, PG_DB) video = get_video(video_id, PG_DB)
rescue ex rescue ex
error_message = {"error" => ex.message}.to_json next error_json(500, ex)
env.response.status_code = 500
next error_message
end end
playlist_video = PlaylistVideo.new({ playlist_video = PlaylistVideo.new({
@ -3949,6 +3792,8 @@ post "/api/v1/auth/playlists/:plid/videos" do |env|
end end
delete "/api/v1/auth/playlists/:plid/videos/:index" do |env| delete "/api/v1/auth/playlists/:plid/videos/:index" do |env|
locale = LOCALES[env.get("preferences").as(Preferences).locale]?
env.response.content_type = "application/json" env.response.content_type = "application/json"
user = env.get("user").as(User) user = env.get("user").as(User)
@ -3957,21 +3802,15 @@ delete "/api/v1/auth/playlists/:plid/videos/:index" do |env|
playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist) playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist)
if !playlist || playlist.author != user.email && playlist.privacy.private? if !playlist || playlist.author != user.email && playlist.privacy.private?
env.response.status_code = 404 next error_json(404, "Playlist does not exist.")
error_message = {"error" => "Playlist does not exist."}.to_json
next error_message
end end
if playlist.author != user.email if playlist.author != user.email
env.response.status_code = 403 next error_json(403, "Invalid user")
error_message = {"error" => "Invalid user"}.to_json
next error_message
end end
if !playlist.index.includes? index if !playlist.index.includes? index
env.response.status_code = 404 next error_json(404, "Playlist does not contain index")
error_message = {"error" => "Playlist does not contain index"}.to_json
next error_message
end end
PG_DB.exec("DELETE FROM playlist_videos * WHERE index = $1", index) PG_DB.exec("DELETE FROM playlist_videos * WHERE index = $1", index)
@ -4017,9 +3856,7 @@ post "/api/v1/auth/tokens/register" do |env|
callback_url = env.params.json["callbackUrl"]?.try &.as(String) callback_url = env.params.json["callbackUrl"]?.try &.as(String)
expire = env.params.json["expire"]?.try &.as(Int64) expire = env.params.json["expire"]?.try &.as(Int64)
else else
error_message = {"error" => "Invalid or missing header 'Content-Type'"}.to_json next error_json(400, "Invalid or missing header 'Content-Type'")
env.response.status_code = 400
next error_message
end end
if callback_url && callback_url.empty? if callback_url && callback_url.empty?
@ -4069,6 +3906,7 @@ post "/api/v1/auth/tokens/register" do |env|
end end
post "/api/v1/auth/tokens/unregister" do |env| post "/api/v1/auth/tokens/unregister" do |env|
locale = LOCALES[env.get("preferences").as(Preferences).locale]?
env.response.content_type = "application/json" env.response.content_type = "application/json"
user = env.get("user").as(User) user = env.get("user").as(User)
scopes = env.get("scopes").as(Array(String)) scopes = env.get("scopes").as(Array(String))
@ -4082,9 +3920,7 @@ post "/api/v1/auth/tokens/unregister" do |env|
elsif scopes_include_scope(scopes, "GET:tokens") elsif scopes_include_scope(scopes, "GET:tokens")
PG_DB.exec("DELETE FROM session_ids * WHERE id = $1", session) PG_DB.exec("DELETE FROM session_ids * WHERE id = $1", session)
else else
error_message = {"error" => "Cannot revoke session #{session}"}.to_json next error_json(400, "Cannot revoke session #{session}")
env.response.status_code = 400
next error_message
end end
env.response.status_code = 204 env.response.status_code = 204
@ -4408,6 +4244,7 @@ get "/videoplayback/*" do |env|
end end
get "/videoplayback" do |env| get "/videoplayback" do |env|
locale = LOCALES[env.get("preferences").as(Preferences).locale]?
query_params = env.params.query query_params = env.params.query
fvip = query_params["fvip"]? || "3" fvip = query_params["fvip"]? || "3"
@ -4474,9 +4311,7 @@ get "/videoplayback" do |env|
if url.includes? "&file=seg.ts" if url.includes? "&file=seg.ts"
if CONFIG.disabled?("livestreams") if CONFIG.disabled?("livestreams")
env.response.status_code = 403 next error_template(403, "Administrator has disabled this endpoint.")
error_message = "Administrator has disabled this endpoint."
next templated "error"
end end
begin begin
@ -4508,9 +4343,7 @@ get "/videoplayback" do |env|
else else
if query_params["title"]? && CONFIG.disabled?("downloads") || if query_params["title"]? && CONFIG.disabled?("downloads") ||
CONFIG.disabled?("dash") CONFIG.disabled?("dash")
env.response.status_code = 403 next error_template(403, "Administrator has disabled this endpoint.")
error_message = "Administrator has disabled this endpoint."
next templated "error"
end end
content_length = nil content_length = nil
@ -4851,14 +4684,9 @@ error 404 do |env|
halt env, status_code: 302 halt env, status_code: 302
end end
error 500 do |env| error 500 do |env, ex|
error_message = <<-END_HTML locale = LOCALES[env.get("preferences").as(Preferences).locale]?
Looks like you've found a bug in Invidious. Feel free to open a new issue error_template(500, ex)
<a href="https://github.com/iv-org/invidious/issues">here</a>
or send an email to
<a href="mailto:#{CONFIG.admin_email}">#{CONFIG.admin_email}</a>.
END_HTML
templated "error"
end end
static_headers do |response, filepath, filestat| static_headers do |response, filepath, filestat|

View File

@ -208,7 +208,7 @@ def fetch_channel(ucid, db, pull_all_videos = true, locale = nil)
author = rss.xpath_node(%q(//feed/title)) author = rss.xpath_node(%q(//feed/title))
if !author if !author
raise translate(locale, "Deleted or invalid channel") raise InfoException.new("Deleted or invalid channel")
end end
author = author.content author = author.content
@ -226,13 +226,14 @@ def fetch_channel(ucid, db, pull_all_videos = true, locale = nil)
videos = [] of SearchVideo videos = [] of SearchVideo
begin begin
initial_data = JSON.parse(response.body).as_a.find &.["response"]? initial_data = JSON.parse(response.body).as_a.find &.["response"]?
raise "Could not extract JSON" if !initial_data raise InfoException.new("Could not extract channel JSON") if !initial_data
videos = extract_videos(initial_data.as_h, author, ucid) videos = extract_videos(initial_data.as_h, author, ucid)
rescue ex rescue ex
if response.body.includes?("To continue with your YouTube experience, please fill out the form below.") || if response.body.includes?("To continue with your YouTube experience, please fill out the form below.") ||
response.body.includes?("https://www.google.com/sorry/index") response.body.includes?("https://www.google.com/sorry/index")
raise "Could not extract channel info. Instance is likely blocked." raise InfoException.new("Could not extract channel info. Instance is likely blocked.")
end end
raise ex
end end
rss.xpath_nodes("//feed/entry").each do |entry| rss.xpath_nodes("//feed/entry").each do |entry|
@ -287,7 +288,7 @@ def fetch_channel(ucid, db, pull_all_videos = true, locale = nil)
loop do loop do
response = get_channel_videos_response(ucid, page, auto_generated: auto_generated) response = get_channel_videos_response(ucid, page, auto_generated: auto_generated)
initial_data = JSON.parse(response.body).as_a.find &.["response"]? initial_data = JSON.parse(response.body).as_a.find &.["response"]?
raise "Could not extract JSON" if !initial_data raise InfoException.new("Could not extract channel JSON") if !initial_data
videos = extract_videos(initial_data.as_h, author, ucid) videos = extract_videos(initial_data.as_h, author, ucid)
count = videos.size count = videos.size
@ -507,8 +508,7 @@ def fetch_channel_community(ucid, continuation, locale, format, thin_mode)
end end
if response.status_code != 200 if response.status_code != 200
error_message = translate(locale, "This channel does not exist.") raise InfoException.new("This channel does not exist.")
raise error_message
end end
ucid = response.body.match(/https:\/\/www.youtube.com\/channel\/(?<ucid>UC[a-zA-Z0-9_-]{22})/).not_nil!["ucid"] ucid = response.body.match(/https:\/\/www.youtube.com\/channel\/(?<ucid>UC[a-zA-Z0-9_-]{22})/).not_nil!["ucid"]
@ -518,7 +518,7 @@ def fetch_channel_community(ucid, continuation, locale, format, thin_mode)
body = initial_data["contents"]?.try &.["twoColumnBrowseResultsRenderer"]["tabs"].as_a.select { |tab| tab["tabRenderer"]?.try &.["selected"].as_bool.== true }[0]? body = initial_data["contents"]?.try &.["twoColumnBrowseResultsRenderer"]["tabs"].as_a.select { |tab| tab["tabRenderer"]?.try &.["selected"].as_bool.== true }[0]?
if !body if !body
raise "Could not extract community tab." raise InfoException.new("Could not extract community tab.")
end end
body = body["tabRenderer"]["content"]["sectionListRenderer"]["contents"][0]["itemSectionRenderer"] body = body["tabRenderer"]["content"]["sectionListRenderer"]["contents"][0]["itemSectionRenderer"]
@ -540,7 +540,7 @@ def fetch_channel_community(ucid, continuation, locale, format, thin_mode)
body["response"]["continuationContents"]["backstageCommentsContinuation"]? body["response"]["continuationContents"]["backstageCommentsContinuation"]?
if !body if !body
raise "Could not extract continuation." raise InfoException.new("Could not extract continuation.")
end end
end end
@ -551,7 +551,7 @@ def fetch_channel_community(ucid, continuation, locale, format, thin_mode)
error_message = (message["text"]["simpleText"]? || error_message = (message["text"]["simpleText"]? ||
message["text"]["runs"]?.try &.[0]?.try &.["text"]?) message["text"]["runs"]?.try &.[0]?.try &.["text"]?)
.try &.as_s || "" .try &.as_s || ""
raise error_message raise InfoException.new(error_message)
end end
response = JSON.build do |json| response = JSON.build do |json|
@ -786,21 +786,19 @@ def get_about_info(ucid, locale)
end end
if result.status_code != 200 if result.status_code != 200
error_message = translate(locale, "This channel does not exist.") raise InfoException.new("This channel does not exist.")
raise error_message
end end
about = XML.parse_html(result.body) about = XML.parse_html(result.body)
if about.xpath_node(%q(//div[contains(@class, "channel-empty-message")])) if about.xpath_node(%q(//div[contains(@class, "channel-empty-message")]))
error_message = translate(locale, "This channel does not exist.") raise InfoException.new("This channel does not exist.")
raise error_message
end end
initdata = extract_initial_data(result.body) initdata = extract_initial_data(result.body)
if initdata.empty? if initdata.empty?
error_message = about.xpath_node(%q(//div[@class="yt-alert-content"])).try &.content.strip error_message = about.xpath_node(%q(//div[@class="yt-alert-content"])).try &.content.strip
error_message ||= translate(locale, "Could not get channel info.") error_message ||= translate(locale, "Could not get channel info.")
raise error_message raise InfoException.new(error_message)
end end
author = about.xpath_node(%q(//meta[@name="title"])).not_nil!["content"] author = about.xpath_node(%q(//meta[@name="title"])).not_nil!["content"]

View File

@ -92,7 +92,7 @@ def fetch_youtube_comments(id, db, cursor, format, locale, thin_mode, region, so
response = JSON.parse(response.body) response = JSON.parse(response.body)
if !response["response"]["continuationContents"]? if !response["response"]["continuationContents"]?
raise translate(locale, "Could not fetch comments") raise InfoException.new("Could not fetch comments")
end end
response = response["response"]["continuationContents"] response = response["response"]["continuationContents"]
@ -266,7 +266,7 @@ def fetch_reddit_comments(id, sort_by = "confidence")
thread = result[0].data.as(RedditListing).children[0].data.as(RedditLink) thread = result[0].data.as(RedditListing).children[0].data.as(RedditLink)
else else
raise "Got error code #{search_results.status_code}" raise InfoException.new("Could not fetch comments")
end end
comments = result[1].data.as(RedditListing).children comments = result[1].data.as(RedditListing).children

View File

@ -0,0 +1,90 @@
# InfoExceptions are for displaying information to the user.
#
# An InfoException might or might not indicate that something went wrong.
# Historically Invidious didn't differentiate between these two options, so to
# maintain previous functionality InfoExceptions do not print backtraces.
class InfoException < Exception
end
macro error_template(*args)
error_template_helper(env, config, locale, {{*args}})
end
def error_template_helper(env : HTTP::Server::Context, config : Config, locale : Hash(String, JSON::Any) | Nil, status_code : Int32, exception : Exception)
if exception.is_a?(InfoException)
return error_template_helper(env, config, locale, status_code, exception.message || "")
end
env.response.status_code = status_code
error_message = <<-END_HTML
Looks like you've found a bug in Invidious. Feel free to open a new issue
<a href="https://github.com/iv-org/invidious/issues">here</a>
or send an email to
<a href="mailto:#{CONFIG.admin_email}">#{CONFIG.admin_email}</a>.
<br>
<br>
<br>
Please include the following text in your message:
<pre style="padding: 20px; background: rgba(0, 0, 0, 0.12345);">#{exception.inspect_with_backtrace}</pre>
END_HTML
return templated "error"
end
def error_template_helper(env : HTTP::Server::Context, config : Config, locale : Hash(String, JSON::Any) | Nil, status_code : Int32, message : String)
env.response.status_code = status_code
error_message = translate(locale, message)
return templated "error"
end
macro error_atom(*args)
error_atom_helper(env, config, locale, {{*args}})
end
def error_atom_helper(env : HTTP::Server::Context, config : Config, locale : Hash(String, JSON::Any) | Nil, status_code : Int32, exception : Exception)
if exception.is_a?(InfoException)
return error_atom_helper(env, config, locale, status_code, exception.message || "")
end
env.response.content_type = "application/atom+xml"
env.response.status_code = status_code
return "<error>#{exception.inspect_with_backtrace}</error>"
end
def error_atom_helper(env : HTTP::Server::Context, config : Config, locale : Hash(String, JSON::Any) | Nil, status_code : Int32, message : String)
env.response.content_type = "application/atom+xml"
env.response.status_code = status_code
return "<error>#{message}</error>"
end
macro error_json(*args)
error_json_helper(env, config, locale, {{*args}})
end
def error_json_helper(env : HTTP::Server::Context, config : Config, locale : Hash(String, JSON::Any) | Nil, status_code : Int32, exception : Exception, additional_fields : Hash(String, Object) | Nil)
if exception.is_a?(InfoException)
return error_json_helper(env, config, locale, status_code, exception.message || "", additional_fields)
end
env.response.content_type = "application/json"
env.response.status_code = status_code
error_message = {"error" => exception.message, "errorBacktrace" => exception.inspect_with_backtrace}
if additional_fields
error_message = error_message.merge(additional_fields)
end
return error_message.to_json
end
def error_json_helper(env : HTTP::Server::Context, config : Config, locale : Hash(String, JSON::Any) | Nil, status_code : Int32, exception : Exception)
return error_json_helper(env, config, locale, status_code, exception, nil)
end
def error_json_helper(env : HTTP::Server::Context, config : Config, locale : Hash(String, JSON::Any) | Nil, status_code : Int32, message : String, additional_fields : Hash(String, Object) | Nil)
env.response.content_type = "application/json"
env.response.status_code = status_code
error_message = {"error" => message}
if additional_fields
error_message = error_message.merge(additional_fields)
end
return error_message.to_json
end
def error_json_helper(env : HTTP::Server::Context, config : Config, locale : Hash(String, JSON::Any) | Nil, status_code : Int32, message : String)
error_json_helper(env, config, locale, status_code, message, nil)
end

View File

@ -70,33 +70,33 @@ def validate_request(token, session, request, key, db, locale = nil)
when JSON::Any when JSON::Any
token = token.as_h token = token.as_h
when Nil when Nil
raise translate(locale, "Hidden field \"token\" is a required field") raise InfoException.new("Hidden field \"token\" is a required field")
end end
expire = token["expire"]?.try &.as_i expire = token["expire"]?.try &.as_i
if expire.try &.< Time.utc.to_unix if expire.try &.< Time.utc.to_unix
raise translate(locale, "Token is expired, please try again") raise InfoException.new("Token is expired, please try again")
end end
if token["session"] != session if token["session"] != session
raise translate(locale, "Erroneous token") raise InfoException.new("Erroneous token")
end end
scopes = token["scopes"].as_a.map { |v| v.as_s } scopes = token["scopes"].as_a.map { |v| v.as_s }
scope = "#{request.method}:#{request.path.lchop("/api/v1/auth/").lstrip("/")}" scope = "#{request.method}:#{request.path.lchop("/api/v1/auth/").lstrip("/")}"
if !scopes_include_scope(scopes, scope) if !scopes_include_scope(scopes, scope)
raise translate(locale, "Invalid scope") raise InfoException.new("Invalid scope")
end end
if !Crypto::Subtle.constant_time_compare(token["signature"].to_s, sign_token(key, token)) if !Crypto::Subtle.constant_time_compare(token["signature"].to_s, sign_token(key, token))
raise translate(locale, "Invalid signature") raise InfoException.new("Invalid signature")
end end
if token["nonce"]? && (nonce = db.query_one?("SELECT * FROM nonces WHERE nonce = $1", token["nonce"], as: {String, Time})) if token["nonce"]? && (nonce = db.query_one?("SELECT * FROM nonces WHERE nonce = $1", token["nonce"], as: {String, Time}))
if nonce[1] > Time.utc if nonce[1] > Time.utc
db.exec("UPDATE nonces SET expire = $1 WHERE nonce = $2", Time.utc(1990, 1, 1), nonce[0]) db.exec("UPDATE nonces SET expire = $1 WHERE nonce = $2", Time.utc(1990, 1, 1), nonce[0])
else else
raise translate(locale, "Erroneous token") raise InfoException.new("Erroneous token")
end end
end end

View File

@ -30,7 +30,7 @@ def fetch_mix(rdid, video_id, cookies = nil, locale = nil)
initial_data = extract_initial_data(response.body) initial_data = extract_initial_data(response.body)
if !initial_data["contents"]["twoColumnWatchNextResults"]["playlist"]? if !initial_data["contents"]["twoColumnWatchNextResults"]["playlist"]?
raise translate(locale, "Could not create mix.") raise InfoException.new("Could not create mix.")
end end
playlist = initial_data["contents"]["twoColumnWatchNextResults"]["playlist"]["playlist"] playlist = initial_data["contents"]["twoColumnWatchNextResults"]["playlist"]["playlist"]

View File

@ -338,7 +338,7 @@ def get_playlist(db, plid, locale, refresh = true, force_refresh = false)
if playlist = db.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist) if playlist = db.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist)
return playlist return playlist
else else
raise "Playlist does not exist." raise InfoException.new("Playlist does not exist.")
end end
else else
return fetch_playlist(plid, locale) return fetch_playlist(plid, locale)
@ -353,16 +353,16 @@ def fetch_playlist(plid, locale)
response = YT_POOL.client &.get("/playlist?list=#{plid}&hl=en") response = YT_POOL.client &.get("/playlist?list=#{plid}&hl=en")
if response.status_code != 200 if response.status_code != 200
if response.headers["location"]?.try &.includes? "/sorry/index" if response.headers["location"]?.try &.includes? "/sorry/index"
raise "Could not extract playlist info. Instance is likely blocked." raise InfoException.new("Could not extract playlist info. Instance is likely blocked.")
else else
raise translate(locale, "Not a playlist.") raise InfoException.new("Not a playlist.")
end end
end end
initial_data = extract_initial_data(response.body) initial_data = extract_initial_data(response.body)
playlist_info = initial_data["sidebar"]?.try &.["playlistSidebarRenderer"]?.try &.["items"]?.try &.[0]["playlistSidebarPrimaryInfoRenderer"]? playlist_info = initial_data["sidebar"]?.try &.["playlistSidebarRenderer"]?.try &.["items"]?.try &.[0]["playlistSidebarPrimaryInfoRenderer"]?
raise "Could not extract playlist info" if !playlist_info raise InfoException.new("Could not extract playlist info") if !playlist_info
title = playlist_info["title"]?.try &.["runs"][0]?.try &.["text"]?.try &.as_s || "" title = playlist_info["title"]?.try &.["runs"][0]?.try &.["text"]?.try &.as_s || ""
desc_item = playlist_info["description"]? desc_item = playlist_info["description"]?
@ -390,7 +390,7 @@ def fetch_playlist(plid, locale)
author_info = initial_data["sidebar"]?.try &.["playlistSidebarRenderer"]?.try &.["items"]?.try &.[1]["playlistSidebarSecondaryInfoRenderer"]? author_info = initial_data["sidebar"]?.try &.["playlistSidebarRenderer"]?.try &.["items"]?.try &.[1]["playlistSidebarSecondaryInfoRenderer"]?
.try &.["videoOwner"]["videoOwnerRenderer"]? .try &.["videoOwner"]["videoOwnerRenderer"]?
raise "Could not extract author info" if !author_info raise InfoException.new("Could not extract author info") if !author_info
author_thumbnail = author_info["thumbnail"]["thumbnails"][0]["url"]?.try &.as_s || "" author_thumbnail = author_info["thumbnail"]["thumbnails"][0]["url"]?.try &.as_s || ""
author = author_info["title"]["runs"][0]["text"]?.try &.as_s || "" author = author_info["title"]["runs"][0]["text"]?.try &.as_s || ""

View File

@ -8,9 +8,7 @@ class Invidious::Routes::Embed::Index < Invidious::Routes::BaseRoute
offset = env.params.query["index"]?.try &.to_i? || 0 offset = env.params.query["index"]?.try &.to_i? || 0
videos = get_playlist_videos(PG_DB, playlist, offset: offset, locale: locale) videos = get_playlist_videos(PG_DB, playlist, offset: offset, locale: locale)
rescue ex rescue ex
error_message = ex.message return error_template(500, ex)
env.response.status_code = 500
return templated "error"
end end
url = "/embed/#{videos[0].id}?#{env.params.query}" url = "/embed/#{videos[0].id}?#{env.params.query}"

View File

@ -38,9 +38,7 @@ class Invidious::Routes::Embed::Show < Invidious::Routes::BaseRoute
offset = env.params.query["index"]?.try &.to_i? || 0 offset = env.params.query["index"]?.try &.to_i? || 0
videos = get_playlist_videos(PG_DB, playlist, offset: offset, locale: locale) videos = get_playlist_videos(PG_DB, playlist, offset: offset, locale: locale)
rescue ex rescue ex
error_message = ex.message return error_template(500, ex)
env.response.status_code = 500
return templated "error"
end end
url = "/embed/#{videos[0].id}" url = "/embed/#{videos[0].id}"
@ -63,8 +61,7 @@ class Invidious::Routes::Embed::Show < Invidious::Routes::BaseRoute
env.params.query.delete_all("channel") env.params.query.delete_all("channel")
if !video_id || video_id == "live_stream" if !video_id || video_id == "live_stream"
error_message = "Video is unavailable." return error_template(500, "Video is unavailable.")
return templated "error"
end end
url = "/embed/#{video_id}" url = "/embed/#{video_id}"
@ -100,9 +97,7 @@ class Invidious::Routes::Embed::Show < Invidious::Routes::BaseRoute
rescue ex : VideoRedirect rescue ex : VideoRedirect
return env.redirect env.request.resource.gsub(id, ex.video_id) return env.redirect env.request.resource.gsub(id, ex.video_id)
rescue ex rescue ex
error_message = ex.message return error_template(500, ex)
env.response.status_code = 500
return templated "error"
end end
if preferences.annotations_subscribed && if preferences.annotations_subscribed &&

View File

@ -56,26 +56,21 @@ class Invidious::Routes::Playlists < Invidious::Routes::BaseRoute
begin begin
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
error_message = ex.message return error_template(400, ex)
env.response.status_code = 400
return templated "error"
end end
title = env.params.body["title"]?.try &.as(String) title = env.params.body["title"]?.try &.as(String)
if !title || title.empty? if !title || title.empty?
error_message = "Title cannot be empty." return error_template(400, "Title cannot be empty.")
return templated "error"
end end
privacy = PlaylistPrivacy.parse?(env.params.body["privacy"]?.try &.as(String) || "") privacy = PlaylistPrivacy.parse?(env.params.body["privacy"]?.try &.as(String) || "")
if !privacy if !privacy
error_message = "Invalid privacy setting." return error_template(400, "Invalid privacy setting.")
return templated "error"
end end
if PG_DB.query_one("SELECT count(*) FROM playlists WHERE author = $1", user.email, as: Int64) >= 100 if PG_DB.query_one("SELECT count(*) FROM playlists WHERE author = $1", user.email, as: Int64) >= 100
error_message = "User cannot have more than 100 playlists." return error_template(400, "User cannot have more than 100 playlists.")
return templated "error"
end end
playlist = create_playlist(PG_DB, title, privacy, user) playlist = create_playlist(PG_DB, title, privacy, user)
@ -142,9 +137,7 @@ class Invidious::Routes::Playlists < Invidious::Routes::BaseRoute
begin begin
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
error_message = ex.message return error_template(400, ex)
env.response.status_code = 400
return templated "error"
end end
playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist) playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist)
@ -217,9 +210,7 @@ class Invidious::Routes::Playlists < Invidious::Routes::BaseRoute
begin begin
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
error_message = ex.message return error_template(400, ex)
env.response.status_code = 400
return templated "error"
end end
playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist) playlist = PG_DB.query_one?("SELECT * FROM playlists WHERE id = $1", plid, as: InvidiousPlaylist)
@ -306,9 +297,7 @@ class Invidious::Routes::Playlists < Invidious::Routes::BaseRoute
if redirect if redirect
return env.redirect referer return env.redirect referer
else else
error_message = {"error" => "No such user"}.to_json return error_json(403, "No such user")
env.response.status_code = 403
return error_message
end end
end end
@ -320,13 +309,9 @@ class Invidious::Routes::Playlists < Invidious::Routes::BaseRoute
validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale) validate_request(token, sid, env.request, HMAC_KEY, PG_DB, locale)
rescue ex rescue ex
if redirect if redirect
error_message = ex.message return error_template(400, ex)
env.response.status_code = 400
return templated "error"
else else
error_message = {"error" => ex.message}.to_json return error_json(400, ex)
env.response.status_code = 400
return error_message
end end
end end
@ -353,13 +338,9 @@ class Invidious::Routes::Playlists < Invidious::Routes::BaseRoute
raise "Invalid user" if playlist.author != user.email raise "Invalid user" if playlist.author != user.email
rescue ex rescue ex
if redirect if redirect
error_message = ex.message return error_template(400, ex)
env.response.status_code = 400
return templated "error"
else else
error_message = {"error" => ex.message}.to_json return error_json(400, ex)
env.response.status_code = 400
return error_message
end end
end end
@ -374,13 +355,10 @@ class Invidious::Routes::Playlists < Invidious::Routes::BaseRoute
# TODO: Playlist stub # TODO: Playlist stub
when "action_add_video" when "action_add_video"
if playlist.index.size >= 500 if playlist.index.size >= 500
env.response.status_code = 400
if redirect if redirect
error_message = "Playlist cannot have more than 500 videos" return error_template(400, "Playlist cannot have more than 500 videos")
return templated "error"
else else
error_message = {"error" => "Playlist cannot have more than 500 videos"}.to_json return error_json(400, "Playlist cannot have more than 500 videos")
return error_message
end end
end end
@ -389,13 +367,10 @@ class Invidious::Routes::Playlists < Invidious::Routes::BaseRoute
begin begin
video = get_video(video_id, PG_DB) video = get_video(video_id, PG_DB)
rescue ex rescue ex
env.response.status_code = 500
if redirect if redirect
error_message = ex.message return error_template(500, ex)
return templated "error"
else else
error_message = {"error" => ex.message}.to_json return error_json(500, ex)
return error_message
end end
end end
@ -423,9 +398,7 @@ class Invidious::Routes::Playlists < Invidious::Routes::BaseRoute
when "action_move_video_before" when "action_move_video_before"
# TODO: Playlist stub # TODO: Playlist stub
else else
error_message = {"error" => "Unsupported action #{action}"}.to_json return error_json(400, "Unsupported action #{action}")
env.response.status_code = 400
return error_message
end end
if redirect if redirect
@ -457,15 +430,11 @@ class Invidious::Routes::Playlists < Invidious::Routes::BaseRoute
begin begin
playlist = get_playlist(PG_DB, plid, locale) playlist = get_playlist(PG_DB, plid, locale)
rescue ex rescue ex
error_message = ex.message return error_template(500, ex)
env.response.status_code = 500
return templated "error"
end end
if playlist.privacy == PlaylistPrivacy::Private && playlist.author != user.try &.email if playlist.privacy == PlaylistPrivacy::Private && playlist.author != user.try &.email
error_message = "This playlist is private." return error_template(403, "This playlist is private.")
env.response.status_code = 403
return templated "error"
end end
begin begin
@ -495,9 +464,7 @@ class Invidious::Routes::Playlists < Invidious::Routes::BaseRoute
begin begin
mix = fetch_mix(rdid, continuation, locale: locale) mix = fetch_mix(rdid, continuation, locale: locale)
rescue ex rescue ex
error_message = ex.message return error_template(500, ex)
env.response.status_code = 500
return templated "error"
end end
templated "mix" templated "mix"

View File

@ -12,9 +12,7 @@ class Invidious::Routes::Watch < Invidious::Routes::BaseRoute
id = env.params.query["v"] id = env.params.query["v"]
if env.params.query["v"].empty? if env.params.query["v"].empty?
error_message = "Invalid parameters." return error_template(400, "Invalid parameters.")
env.response.status_code = 400
return templated "error"
end end
if id.size > 11 if id.size > 11
@ -56,10 +54,8 @@ class Invidious::Routes::Watch < Invidious::Routes::BaseRoute
rescue ex : VideoRedirect rescue ex : VideoRedirect
return env.redirect env.request.resource.gsub(id, ex.video_id) return env.redirect env.request.resource.gsub(id, ex.video_id)
rescue ex rescue ex
error_message = ex.message
env.response.status_code = 500
logger.puts("#{id} : #{ex.message}") logger.puts("#{id} : #{ex.message}")
return templated "error" return error_template(500, ex)
end end
if preferences.annotations_subscribed && if preferences.annotations_subscribed &&