added youtube playlist import; initial commit

Signed-off-by: Gavin Johnson <gavinj1984@gmail.com>
This commit is contained in:
Gavin Johnson 2023-01-16 15:40:38 -08:00
parent 05258d56bd
commit 855202e40e
3 changed files with 91 additions and 0 deletions

View File

@ -33,6 +33,7 @@
"Import": "Import", "Import": "Import",
"Import Invidious data": "Import Invidious JSON data", "Import Invidious data": "Import Invidious JSON data",
"Import YouTube subscriptions": "Import YouTube/OPML subscriptions", "Import YouTube subscriptions": "Import YouTube/OPML subscriptions",
"Import YouTube playlist": "Import YouTube playlist (.csv)",
"Import FreeTube subscriptions (.db)": "Import FreeTube subscriptions (.db)", "Import FreeTube subscriptions (.db)": "Import FreeTube subscriptions (.db)",
"Import NewPipe subscriptions (.json)": "Import NewPipe subscriptions (.json)", "Import NewPipe subscriptions (.json)": "Import NewPipe subscriptions (.json)",
"Import NewPipe data (.zip)": "Import NewPipe data (.zip)", "Import NewPipe data (.zip)": "Import NewPipe data (.zip)",

View File

@ -30,6 +30,75 @@ struct Invidious::User
return subscriptions return subscriptions
end end
# Parse a youtube CSV playlist file and create the playlist
#NEW - Done
def parse_playlist_export_csv(user : User, csv_content : String)
rows = CSV.new(csv_content, headers: false)
if rows.size >= 2
title = rows[1][4]?.try &.as_s?.try
descripton = rows[1][5]?.try &.as_s?.try
visibility = rows[1][6]?.try &.as_s?.try
if visibility.compare("Public", case_insensitive: true) == 0
privacy = PlaylistPrivacy:Public
else
privacy = PlaylistPrivacy:Private
end
if title && privacy && user
playlist = create_playlist(title, privacy, user)
end
if playlist && descripton
Invidious::Database::Playlists.update_description(playlist.id, description)
end
end
return playlist
end
# Parse a youtube CSV playlist file and add videos from it to a playlist
#NEW - done
def parse_playlist_videos_export_csv(playlist : Playlist, csv_content : String)
rows = CSV.new(csv_content, headers: false)
if rows.size >= 5
offset = env.params.query["index"]?.try &.to_i? || 0
row_counter = 0
rows.each do |row|
if row_counter >= 4
video_id = row[0]?.try &.as_s?.try
end
row_counter += 1
next if !video_id
begin
video = get_video(video_id)
rescue ex
next
end
playlist_video = PlaylistVideo.new({
title: video.title,
id: video.id,
author: video.author,
ucid: video.ucid,
length_seconds: video.length_seconds,
published: video.published,
plid: playlist.id,
live_now: video.live_now,
index: Random::Secure.rand(0_i64..Int64::MAX),
})
Invidious::Database::PlaylistVideos.insert(playlist_video)
Invidious::Database::Playlists.update_video_added(playlist.id, playlist_video.index)
end
videos = get_playlist_videos(playlist, offset: offset)
end
return videos
end
# ------------------- # -------------------
# Invidious # Invidious
# ------------------- # -------------------
@ -149,6 +218,22 @@ struct Invidious::User
return true return true
end end
# Import playlist from Youtube
# Returns success status
#NEW
def from_youtube_pl(user : User, body : String, filename : String, type : String) : Bool
extension = filename.split(".").last
if extension == "csv" || type == "text/csv"
playlist = parse_playlist_export_csv(user, body)
playlist = parse_playlist_videos_export_csv(playlist, body)
else
return false
end
return true
end
# ------------------- # -------------------
# Freetube # Freetube
# ------------------- # -------------------

View File

@ -21,6 +21,11 @@
<input type="file" id="import_youtube" name="import_youtube"> <input type="file" id="import_youtube" name="import_youtube">
</div> </div>
<div class="pure-control-group">
<label for="import_youtube_pl"><%= translate(locale, "Import YouTube playlists") %></label>
<input type="file" id="import_youtube_pl" name="import_youtube_pl">
</div>
<div class="pure-control-group"> <div class="pure-control-group">
<label for="import_freetube"><%= translate(locale, "Import FreeTube subscriptions (.db)") %></label> <label for="import_freetube"><%= translate(locale, "Import FreeTube subscriptions (.db)") %></label>
<input type="file" id="import_freetube" name="import_freetube"> <input type="file" id="import_freetube" name="import_freetube">