From 855202e40e09af1cb5fb372d4a2d05a61b3a9bb2 Mon Sep 17 00:00:00 2001
From: Gavin Johnson <gavinj1984@gmail.com>
Date: Mon, 16 Jan 2023 15:40:38 -0800
Subject: [PATCH] added youtube playlist import; initial commit

Signed-off-by: Gavin Johnson <gavinj1984@gmail.com>
---
 locales/en-US.json                        |  1 +
 src/invidious/user/imports.cr             | 85 +++++++++++++++++++++++
 src/invidious/views/user/data_control.ecr |  5 ++
 3 files changed, 91 insertions(+)

diff --git a/locales/en-US.json b/locales/en-US.json
index 12955665..c30a90db 100644
--- a/locales/en-US.json
+++ b/locales/en-US.json
@@ -33,6 +33,7 @@
     "Import": "Import",
     "Import Invidious data": "Import Invidious JSON data",
     "Import YouTube subscriptions": "Import YouTube/OPML subscriptions",
+    "Import YouTube playlist": "Import YouTube playlist (.csv)",
     "Import FreeTube subscriptions (.db)": "Import FreeTube subscriptions (.db)",
     "Import NewPipe subscriptions (.json)": "Import NewPipe subscriptions (.json)",
     "Import NewPipe data (.zip)": "Import NewPipe data (.zip)",
diff --git a/src/invidious/user/imports.cr b/src/invidious/user/imports.cr
index 20ae0d47..870d083e 100644
--- a/src/invidious/user/imports.cr
+++ b/src/invidious/user/imports.cr
@@ -30,6 +30,75 @@ struct Invidious::User
       return subscriptions
     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
     # -------------------
@@ -149,6 +218,22 @@ struct Invidious::User
       return true
     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
     # -------------------
diff --git a/src/invidious/views/user/data_control.ecr b/src/invidious/views/user/data_control.ecr
index a451159f..0f8e8dae 100644
--- a/src/invidious/views/user/data_control.ecr
+++ b/src/invidious/views/user/data_control.ecr
@@ -21,6 +21,11 @@
                 <input type="file" id="import_youtube" name="import_youtube">
             </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">
                 <label for="import_freetube"><%= translate(locale, "Import FreeTube subscriptions (.db)") %></label>
                 <input type="file" id="import_freetube" name="import_freetube">