lug/invidious
7
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

improved import algorithm, fixed a referer issue from the playlists page after deleting a playlist

Browse Source 
Signed-off-by: thtmnisamnstr <gavinj1984@gmail.com>
This commit is contained in:
thtmnisamnstr 2023-02-23 15:55:38 -08:00
parent 6f01d6eacf
commit b3eea6ab3e
2 changed files with 30 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
src/invidious/user/imports.cr
View File

@ -30,43 +30,44 @@ struct Invidious::User
return subscriptions
end
def parse_playlist_export_csv(user : User, csv_content : String)
rows = CSV.new(csv_content, headers: true)
row_counter = 0
def parse_playlist_export_csv(user : User, raw_input : String)
playlist = uninitialized InvidiousPlaylist
title = uninitialized String
description = uninitialized String
visibility = uninitialized String
rows.each do |row|
if row_counter == 0
title = row[4]
description = row[5]
visibility = row[6]
privacy = uninitialized PlaylistPrivacy
if visibility.compare("Public", case_insensitive: true) == 0
privacy = PlaylistPrivacy::Public
else
privacy = PlaylistPrivacy::Private
end
# Split the input into head and body content
raw_head, raw_body = raw_input.split("\n\n", limit: 2, remove_empty: true)
if title && privacy && user
playlist = create_playlist(title, privacy, user)
end
# Create the playlist from the head content
csv_head = CSV.new(raw_head, headers: true)
csv_head.next
title = csv_head[4]
description = csv_head[5]
visibility = csv_head[6]
if visibility.compare("Public", case_insensitive: true) == 0
privacy = PlaylistPrivacy::Public
else
privacy = PlaylistPrivacy::Private
end
if playlist && description
Invidious::Database::Playlists.update_description(playlist.id, description)
end
if title && privacy && user
playlist = create_playlist(title, privacy, user)
end
row_counter += 1
end
if row_counter > 0 && row_counter < 3
row_counter += 1
end
if row_counter >= 3
if playlist && description
Invidious::Database::Playlists.update_description(playlist.id, description)
end
# Add each video to the playlist from the body content
CSV.each_row(raw_body) do |row|
if row.size >= 1
video_id = row[0]
if playlist
video_id = row[0]
row_counter += 1
next if !video_id
next if video_id == "Video Id"
begin
video = get_video(video_id)

4
src/invidious/views/feeds/playlists.ecr
View File

@ -10,12 +10,12 @@
</div>
<div class="pure-u-1-3">
<h3 style="text-align:center">
<a href="/create_playlist?referer=<%= URI.encode_www_form(referer) %>"><%= translate(locale, "Create playlist") %></a>
<a href="/create_playlist?referer=<%= URI.encode_www_form("/feed/playlists") %>"><%= translate(locale, "Create playlist") %></a>
</h3>
</div>
<div class="pure-u-1-3">
<h3 style="text-align:right">
<a href="/data_control?referer=<%= URI.encode_www_form(referer) %>">
<a href="/data_control?referer=<%= URI.encode_www_form("/feed/playlists") %>">
<%= translate(locale, "Import/export") %>
</a>
</h3>