From 2606decd21a84ac3cba914f327f60a8403398ed9 Mon Sep 17 00:00:00 2001
From: Brahim Hadriche <brahim.hadriche@gmail.com>
Date: Sun, 5 Feb 2023 15:00:11 -0500
Subject: [PATCH] Refactor export function

---
 src/invidious/routes/api/v1/authenticated.cr | 28 +---------------
 src/invidious/routes/subscriptions.cr        | 26 +--------------
 src/invidious/user/exports.cr                | 35 ++++++++++++++++++++
 3 files changed, 37 insertions(+), 52 deletions(-)
 create mode 100644 src/invidious/user/exports.cr

diff --git a/src/invidious/routes/api/v1/authenticated.cr b/src/invidious/routes/api/v1/authenticated.cr
index c6042e40..6b935312 100644
--- a/src/invidious/routes/api/v1/authenticated.cr
+++ b/src/invidious/routes/api/v1/authenticated.cr
@@ -35,33 +35,7 @@ module Invidious::Routes::API::V1::Authenticated
     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
+    return Invidious::User::Export.to_invidious(user)
   end
 
   def self.import_invidious(env)
diff --git a/src/invidious/routes/subscriptions.cr b/src/invidious/routes/subscriptions.cr
index 7b1fa876..3090e026 100644
--- a/src/invidious/routes/subscriptions.cr
+++ b/src/invidious/routes/subscriptions.cr
@@ -106,31 +106,7 @@ module Invidious::Routes::Subscriptions
         env.response.headers["content-disposition"] = "attachment"
         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
+        return Invidious::User::Export.to_invidious(user)
       else
         env.response.content_type = "application/xml"
         env.response.headers["content-disposition"] = "attachment"
diff --git a/src/invidious/user/exports.cr b/src/invidious/user/exports.cr
new file mode 100644
index 00000000..32be0ca2
--- /dev/null
+++ b/src/invidious/user/exports.cr
@@ -0,0 +1,35 @@
+struct Invidious::User
+  module Export
+    extend self
+
+    def to_invidious(user : 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
+  end # module
+end