hashirama/invidious-mod-JP
1
0
Fork 0
Code Issues Pull requests Actions 2 Projects Releases Wiki Activity

DB: playlists: make that 'insert' never raises

Browse source
This commit is contained in:
Samantaz Fox 2022-01-07 18:05:59 +01:00
parent c78f84d5c6
commit 714a001332
No known key found for this signature in database
GPG key ID: F42821059186176E
2 changed files with 6 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/invidious/database/playlists.cr
View file

@ -94,17 +94,13 @@ module Invidious::Database::Playlists
# Salect # Salect
# ------------------- # -------------------
def select(*, id : String, raise_on_fail : Bool = false) : InvidiousPlaylist? def select(*, id : String) : InvidiousPlaylist?
request = <<-SQL request = <<-SQL
SELECT * FROM playlists SELECT * FROM playlists
WHERE id = $1 WHERE id = $1
SQL SQL
if raise_on_fail return PG_DB.query_one?(request, id, as: InvidiousPlaylist)
return PG_DB.query_one(request, id, as: InvidiousPlaylist)
else
return PG_DB.query_one?(request, id, as: InvidiousPlaylist)
end
end end
def select_all(*, author : String) : Array(InvidiousPlaylist) def select_all(*, author : String) : Array(InvidiousPlaylist)

16
src/invidious/routes/playlists.cr
View file

@ -151,12 +151,8 @@ module Invidious::Routes::Playlists
page = env.params.query["page"]?.try &.to_i? page = env.params.query["page"]?.try &.to_i?
page ||= 1 page ||= 1
begin playlist = Invidious::Database::Playlists.select(id: plid)
playlist = Invidious::Database::Playlists.select(id: plid, raise_on_fail: true) if !playlist || playlist.author != user.email
if !playlist || playlist.author != user.email
return env.redirect referer
end
rescue ex
return env.redirect referer return env.redirect referer
end end
@ -235,12 +231,8 @@ module Invidious::Routes::Playlists
page = env.params.query["page"]?.try &.to_i? page = env.params.query["page"]?.try &.to_i?
page ||= 1 page ||= 1
begin playlist = Invidious::Database::Playlists.select(id: plid)
playlist = Invidious::Database::Playlists.select(id: plid, raise_on_fail: true) if !playlist || playlist.author != user.email
if !playlist || playlist.author != user.email
return env.redirect referer
end
rescue ex
return env.redirect referer return env.redirect referer
end end