From b2589c74bedb73a1ab6e0afe1a921b97f80c4b8e Mon Sep 17 00:00:00 2001
From: Brahim Hadriche <brahim.hadriche@gmail.com>
Date: Thu, 2 Feb 2023 19:14:02 -0500
Subject: [PATCH] Add API for import/export

---
 src/invidious/routes/api/v1/authenticated.cr | 49 ++++++++++++++++++++
 src/invidious/routing.cr                     |  3 ++
 2 files changed, 52 insertions(+)

diff --git a/src/invidious/routes/api/v1/authenticated.cr b/src/invidious/routes/api/v1/authenticated.cr
index 421355bb..c6042e40 100644
--- a/src/invidious/routes/api/v1/authenticated.cr
+++ b/src/invidious/routes/api/v1/authenticated.cr
@@ -31,6 +31,55 @@ module Invidious::Routes::API::V1::Authenticated
     env.response.status_code = 204
   end
 
+  def self.export_invidious(env)
+    env.response.content_type = "application/json"
+    user = env.get("user").as(User)
+
+    playlists = Invidious::Database::Playlists.select_like_iv(user.email)
+
+    return JSON.build do |json|
+      json.object do
+        json.field "subscriptions", user.subscriptions
+        json.field "watch_history", user.watched
+        json.field "preferences", user.preferences
+        json.field "playlists" do
+          json.array do
+            playlists.each do |playlist|
+              json.object do
+                json.field "title", playlist.title
+                json.field "description", html_to_content(playlist.description_html)
+                json.field "privacy", playlist.privacy.to_s
+                json.field "videos" do
+                  json.array do
+                    Invidious::Database::PlaylistVideos.select_ids(playlist.id, playlist.index, limit: 500).each do |video_id|
+                      json.string video_id
+                    end
+                  end
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def self.import_invidious(env)
+    user = env.get("user").as(User)
+
+    begin
+      if body = env.request.body
+        body = env.request.body.not_nil!.gets_to_end
+      else
+        body = "{}"
+      end
+      Invidious::User::Import.from_invidious(user, body)
+    rescue
+    end
+
+    env.response.status_code = 204
+  end
+
   def self.feed(env)
     env.response.content_type = "application/json"
 
diff --git a/src/invidious/routing.cr b/src/invidious/routing.cr
index 157e6de7..d5766b90 100644
--- a/src/invidious/routing.cr
+++ b/src/invidious/routing.cr
@@ -254,6 +254,9 @@ module Invidious::Routing
       get "/api/v1/auth/preferences", {{namespace}}::Authenticated, :get_preferences
       post "/api/v1/auth/preferences", {{namespace}}::Authenticated, :set_preferences
 
+      get "/api/v1/auth/export/invidious", {{namespace}}::Authenticated, :export_invidious
+      post "/api/v1/auth/import/invidious", {{namespace}}::Authenticated, :import_invidious
+
       get "/api/v1/auth/feed", {{namespace}}::Authenticated, :feed
 
       get "/api/v1/auth/subscriptions", {{namespace}}::Authenticated, :get_subscriptions