From db2e31b8ccfd5b23f33bac0f3e6f77fd666ccfbd Mon Sep 17 00:00:00 2001
From: Zamitto <167933696+zamitto@users.noreply.github.com>
Date: Sat, 11 Jan 2025 17:52:48 -0300
Subject: [PATCH] feat: add torbox migration

---
 src/main/entity/user-preferences.entity.ts      |  3 +++
 src/main/knex-client.ts                         |  2 ++
 src/main/main.ts                                |  6 ++++--
 ...0250111182229_add_torbox_api_token_column.ts | 17 +++++++++++++++++
 src/main/services/download/torbox.ts            |  2 +-
 src/types/index.ts                              |  1 +
 6 files changed, 28 insertions(+), 3 deletions(-)
 create mode 100644 src/main/migrations/20250111182229_add_torbox_api_token_column.ts

diff --git a/src/main/entity/user-preferences.entity.ts b/src/main/entity/user-preferences.entity.ts
index a850b42f..109ede5f 100644
--- a/src/main/entity/user-preferences.entity.ts
+++ b/src/main/entity/user-preferences.entity.ts
@@ -20,6 +20,9 @@ export class UserPreferences {
   @Column("text", { nullable: true })
   realDebridApiToken: string | null;
 
+  @Column("text", { nullable: true })
+  torBoxApiToken: string | null;
+
   @Column("boolean", { default: false })
   downloadNotificationsEnabled: boolean;
 
diff --git a/src/main/knex-client.ts b/src/main/knex-client.ts
index 821efc80..c816c7c7 100644
--- a/src/main/knex-client.ts
+++ b/src/main/knex-client.ts
@@ -17,6 +17,7 @@ import { AddShouldSeedColumn } from "./migrations/20241108200154_add_should_seed
 import { AddSeedAfterDownloadColumn } from "./migrations/20241108201806_add_seed_after_download";
 import { AddHiddenAchievementDescriptionColumn } from "./migrations/20241216140633_add_hidden_achievement_description_column ";
 import { AddLaunchOptionsColumnToGame } from "./migrations/20241226044022_add_launch_options_column_to_game";
+import { AddTorBoxApiToken } from "./migrations/20250111182229_add_torbox_api_token_column";
 
 export type HydraMigration = Knex.Migration & { name: string };
 
@@ -39,6 +40,7 @@ class MigrationSource implements Knex.MigrationSource<HydraMigration> {
       AddSeedAfterDownloadColumn,
       AddHiddenAchievementDescriptionColumn,
       AddLaunchOptionsColumnToGame,
+      AddTorBoxApiToken,
     ]);
   }
   getMigrationName(migration: HydraMigration): string {
diff --git a/src/main/main.ts b/src/main/main.ts
index 81916174..d27f0cbd 100644
--- a/src/main/main.ts
+++ b/src/main/main.ts
@@ -19,10 +19,12 @@ const loadState = async (userPreferences: UserPreferences | null) => {
   Aria2.spawn();
 
   if (userPreferences?.realDebridApiToken) {
-    RealDebridClient.authorize(userPreferences?.realDebridApiToken);
+    RealDebridClient.authorize(userPreferences.realDebridApiToken);
   }
 
-  TorBoxClient.authorize("7371d5ec-52fa-4b87-9052-0c8c96d947cc");
+  if (userPreferences?.torBoxApiToken) {
+    TorBoxClient.authorize(userPreferences?.torBoxApiToken);
+  }
 
   Ludusavi.addManifestToLudusaviConfig();
 
diff --git a/src/main/migrations/20250111182229_add_torbox_api_token_column.ts b/src/main/migrations/20250111182229_add_torbox_api_token_column.ts
new file mode 100644
index 00000000..fc1904fd
--- /dev/null
+++ b/src/main/migrations/20250111182229_add_torbox_api_token_column.ts
@@ -0,0 +1,17 @@
+import type { HydraMigration } from "@main/knex-client";
+import type { Knex } from "knex";
+
+export const AddTorBoxApiToken: HydraMigration = {
+  name: "AddTorBoxApiToken",
+  up: (knex: Knex) => {
+    return knex.schema.alterTable("user_preferences", (table) => {
+      return table.string("torBoxApiToken").nullable();
+    });
+  },
+
+  down: async (knex: Knex) => {
+    return knex.schema.alterTable("user_preferences", (table) => {
+      return table.dropColumn("torBoxApiToken");
+    });
+  },
+};
diff --git a/src/main/services/download/torbox.ts b/src/main/services/download/torbox.ts
index 7e0c9089..f0af52eb 100644
--- a/src/main/services/download/torbox.ts
+++ b/src/main/services/download/torbox.ts
@@ -19,7 +19,7 @@ export class TorBoxClient {
         Authorization: `Bearer ${apiToken}`,
       },
     });
-    this.apiToken = "7371d5ec-52fa-4b87-9052-0c8c96d947cc";
+    this.apiToken = apiToken;
   }
 
   static async addMagnet(magnet: string) {
diff --git a/src/types/index.ts b/src/types/index.ts
index 345893a5..92cc566e 100644
--- a/src/types/index.ts
+++ b/src/types/index.ts
@@ -163,6 +163,7 @@ export interface UserPreferences {
   repackUpdatesNotificationsEnabled: boolean;
   achievementNotificationsEnabled: boolean;
   realDebridApiToken: string | null;
+  torboxApiToken: string | null;
   preferQuitInsteadOfHiding: boolean;
   runAtStartup: boolean;
   startMinimized: boolean;