feat: migration to leveldb

electron.vite.config.ts

@@ -38,6 +38,13 @@ export default defineConfig(({ mode }) => {
       build: {
         sourcemap: true,
       },
+      css: {
+        preprocessorOptions: {
+          scss: {
+            api: "modern",
+          },
+        },
+      },
       resolve: {
         alias: {
           "@renderer": resolve("src/renderer/src"),

package.json

@@ -62,6 +62,7 @@
     "jsdom": "^24.0.0",
     "jsonwebtoken": "^9.0.2",
     "knex": "^3.1.0",
+    "level": "^9.0.0",
     "lodash-es": "^4.17.21",
     "parse-torrent": "^11.0.17",
     "piscina": "^4.7.0",

src/main/constants.ts

@@ -7,13 +7,18 @@ export const defaultDownloadsPath = app.getPath("downloads");
 
 export const isStaging = import.meta.env.MAIN_VITE_API_URL.includes("staging");
 
+export const levelDatabasePath = path.join(
+  app.getPath("userData"),
+  `hydra-db${isStaging ? "-staging" : ""}`
+);
+
 export const databaseDirectory = path.join(app.getPath("appData"), "hydra");
 export const databasePath = path.join(
   databaseDirectory,
   isStaging ? "hydra_test.db" : "hydra.db"
 );
 
-export const logsPath = path.join(app.getPath("appData"), "hydra", "logs");
+export const logsPath = path.join(app.getPath("userData"), "hydra", "logs");
 
 export const seedsPath = app.isPackaged
   ? path.join(process.resourcesPath, "seeds")

src/main/data-source.ts

@@ -4,9 +4,7 @@ import {
   Game,
   GameShopCache,
   UserPreferences,
-  UserAuth,
   GameAchievement,
-  UserSubscription,
 } from "@main/entity";
 
 import { databasePath } from "./constants";
@@ -15,9 +13,7 @@ export const dataSource = new DataSource({
   type: "better-sqlite3",
   entities: [
     Game,
-    UserAuth,
     UserPreferences,
-    UserSubscription,
     GameShopCache,
     DownloadQueue,
     GameAchievement,

src/main/entity/index.ts

@@ -1,7 +1,5 @@
 export * from "./game.entity";
-export * from "./user-auth.entity";
 export * from "./user-preferences.entity";
-export * from "./user-subscription.entity";
 export * from "./game-shop-cache.entity";
 export * from "./game.entity";
 export * from "./game-achievements.entity";

src/main/events/auth/get-session-hash.ts

@@ -1,13 +1,20 @@
 import jwt from "jsonwebtoken";
 
-import { userAuthRepository } from "@main/repository";
 import { registerEvent } from "../register-event";
+import { db } from "@main/level";
+import type { Auth } from "@types";
+import { levelKeys } from "@main/level/sublevels/keys";
+import { Crypto } from "@main/services";
 
 const getSessionHash = async (_event: Electron.IpcMainInvokeEvent) => {
-  const auth = await userAuthRepository.findOne({ where: { id: 1 } });
+  const auth = await db.get<string, Auth>(levelKeys.auth, {
+    valueEncoding: "json",
+  });
 
   if (!auth) return null;
-  const payload = jwt.decode(auth.accessToken) as jwt.JwtPayload;
+  const payload = jwt.decode(
+    Crypto.decrypt(auth.accessToken)
+  ) as jwt.JwtPayload;
 
   if (!payload) return null;
 

src/main/events/auth/sign-out.ts

@@ -1,8 +1,10 @@
 import { registerEvent } from "../register-event";
 import { DownloadManager, HydraApi, gamesPlaytime } from "@main/services";
 import { dataSource } from "@main/data-source";
-import { DownloadQueue, Game, UserAuth, UserSubscription } from "@main/entity";
+import { DownloadQueue, Game } from "@main/entity";
 import { PythonRPC } from "@main/services/python-rpc";
+import { db } from "@main/level";
+import { levelKeys } from "@main/level/sublevels/keys";
 
 const signOut = async (_event: Electron.IpcMainInvokeEvent) => {
   const databaseOperations = dataSource
@@ -11,13 +13,16 @@ const signOut = async (_event: Electron.IpcMainInvokeEvent) => {
 
       await transactionalEntityManager.getRepository(Game).delete({});
 
-      await transactionalEntityManager
-        .getRepository(UserAuth)
-        .delete({ id: 1 });
-
-      await transactionalEntityManager
-        .getRepository(UserSubscription)
-        .delete({ id: 1 });
+      await db.batch([
+        {
+          type: "del",
+          key: levelKeys.auth,
+        },
+        {
+          type: "del",
+          key: levelKeys.user,
+        },
+      ]);
     })
     .then(() => {
       /* Removes all games being played */

src/main/events/misc/open-checkout.ts

@@ -1,17 +1,21 @@
 import { shell } from "electron";
 import { registerEvent } from "../register-event";
-import { userAuthRepository } from "@main/repository";
-import { HydraApi } from "@main/services";
+import { Crypto, HydraApi } from "@main/services";
+import { db } from "@main/level";
+import type { Auth } from "@types";
+import { levelKeys } from "@main/level/sublevels/keys";
 
 const openCheckout = async (_event: Electron.IpcMainInvokeEvent) => {
-  const userAuth = await userAuthRepository.findOne({ where: { id: 1 } });
+  const auth = await db.get<string, Auth>(levelKeys.auth, {
+    valueEncoding: "json",
+  });
 
-  if (!userAuth) {
+  if (!auth) {
     return;
   }
 
   const paymentToken = await HydraApi.post("/auth/payment", {
-    refreshToken: userAuth.refreshToken,
+    refreshToken: Crypto.decrypt(auth.refreshToken),
   }).then((response) => response.accessToken);
 
   const params = new URLSearchParams({

src/main/events/user/get-user-friends.ts

@@ -1,16 +1,19 @@
-import { userAuthRepository } from "@main/repository";
+import { db } from "@main/level";
 import { registerEvent } from "../register-event";
 import { HydraApi } from "@main/services";
-import type { UserFriends } from "@types";
+import type { User, UserFriends } from "@types";
+import { levelKeys } from "@main/level/sublevels/keys";
 
 export const getUserFriends = async (
   userId: string,
   take: number,
   skip: number
 ): Promise<UserFriends> => {
-  const loggedUser = await userAuthRepository.findOne({ where: { id: 1 } });
+  const user = await db.get<string, User>(levelKeys.user, {
+    valueEncoding: "json",
+  });
 
-  if (loggedUser?.userId === userId) {
+  if (user?.id === userId) {
     return HydraApi.get(`/profile/friends`, { take, skip });
   }
 

src/main/level/index.ts

@@ -0,0 +1,3 @@
+export { db } from "./level";
+
+export * from "./sublevels";

src/main/level/level.ts

@@ -0,0 +1,4 @@
+import { levelDatabasePath } from "@main/constants";
+import { Level } from "level";
+
+export const db = new Level(levelDatabasePath, { valueEncoding: "json" });

src/main/level/sublevels/games.ts

@@ -0,0 +1,7 @@
+import { Game } from "@types";
+import { db } from "../level";
+import { levelKeys } from "./keys";
+
+export const gamesSublevel = db.sublevel<string, Game>(levelKeys.games, {
+  valueEncoding: "json",
+});

src/main/level/sublevels/index.ts

@@ -0,0 +1 @@
+export * from "./games";

src/main/level/sublevels/keys.ts

@@ -0,0 +1,8 @@
+import type { GameShop } from "@types";
+
+export const levelKeys = {
+  games: "games",
+  game: (shop: GameShop, objectId: string) => `${shop}:${objectId}`,
+  user: "user",
+  auth: "auth",
+};

src/main/repository.ts

@@ -4,9 +4,7 @@ import {
   Game,
   GameShopCache,
   UserPreferences,
-  UserAuth,
   GameAchievement,
-  UserSubscription,
 } from "@main/entity";
 
 export const gameRepository = dataSource.getRepository(Game);
@@ -18,10 +16,5 @@ export const gameShopCacheRepository = dataSource.getRepository(GameShopCache);
 
 export const downloadQueueRepository = dataSource.getRepository(DownloadQueue);
 
-export const userAuthRepository = dataSource.getRepository(UserAuth);
-
-export const userSubscriptionRepository =
-  dataSource.getRepository(UserSubscription);
-
 export const gameAchievementRepository =
   dataSource.getRepository(GameAchievement);

src/main/services/crypto.ts

@@ -0,0 +1,28 @@
+import { safeStorage } from "electron";
+import { logger } from "./logger";
+
+export class Crypto {
+  public static encrypt(str: string) {
+    if (safeStorage.isEncryptionAvailable()) {
+      return safeStorage.encryptString(str).toString("base64");
+    } else {
+      logger.warn(
+        "Encrypt method returned raw string because encryption is not available"
+      );
+
+      return str;
+    }
+  }
+
+  public static decrypt(b64: string) {
+    if (safeStorage.isEncryptionAvailable()) {
+      return safeStorage.decryptString(Buffer.from(b64, "base64"));
+    } else {
+      logger.warn(
+        "Decrypt method returned raw string because encryption is not available"
+      );
+
+      return b64;
+    }
+  }
+}

src/main/services/hosters/datanodes.ts

@@ -33,7 +33,8 @@ export class DatanodesApi {
           "User-Agent":
             "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/125.0.0.0 Safari/537.36",
         },
-        maxRedirects: 0,        validateStatus: (status: number) => status === 302 || status < 400,
+        maxRedirects: 0,
+        validateStatus: (status: number) => status === 302 || status < 400,
       }
     );