feat: adding ota updates

src/main/events/hardware/get-disk-free-space.ts

@@ -1,10 +1,13 @@
-import disk from "diskusage";
-
+import { DiskUsage } from "@types";
 import { registerEvent } from "../register-event";
+import checkDiskSpace from "check-disk-space";
 
 const getDiskFreeSpace = async (
   _event: Electron.IpcMainInvokeEvent,
   path: string
-) => disk.check(path);
+): Promise<DiskUsage> => {
+  const result = await checkDiskSpace(path);
+  return { free: result.free, total: result.size };
+};
 
 registerEvent("getDiskFreeSpace", getDiskFreeSpace);

src/main/index.ts

@@ -9,6 +9,7 @@ import {
   clearGamesPlaytime,
   WindowManager,
   Lock,
+  Aria2,
 } from "@main/services";
 import resources from "@locales";
 import { PythonRPC } from "./services/python-rpc";
@@ -222,6 +223,7 @@ app.on("before-quit", async (e) => {
     e.preventDefault();
     /* Disconnects libtorrent */
     PythonRPC.kill();
+    Aria2.kill();
     await clearGamesPlaytime();
     canAppBeClosed = true;
     app.quit();

src/main/services/aria2.ts

@@ -1,6 +1,7 @@
 import path from "node:path";
 import cp from "node:child_process";
 import { app } from "electron";
+import { logger } from "./logger";
 
 export class Aria2 {
   private static process: cp.ChildProcess | null = null;
@@ -23,6 +24,9 @@ export class Aria2 {
   }
 
   public static kill() {
-    this.process?.kill();
+    if (this.process) {
+      logger.log("Killing aria2 process");
+      this.process.kill();
+    }
   }
 }

src/main/services/process-watcher.ts

@@ -79,11 +79,18 @@ const findGamePathByProcess = async (
   const executables = gameExecutables[gameId];
 
   for (const executable of executables) {
-    const pathSet = processMap.get(executable.exe);
+    const executablewithoutExtension = executable.exe.replace(/\.exe$/i, "");
+
+    const pathSet =
+      processMap.get(executable.exe) ??
+      processMap.get(executablewithoutExtension);
 
     if (pathSet) {
       for (const path of pathSet) {
-        if (path.toLowerCase().endsWith(executable.name)) {
+        if (
+          path.toLowerCase().endsWith(executable.name) ||
+          path.toLowerCase().endsWith(executablewithoutExtension)
+        ) {
           const gameKey = levelKeys.game("steam", gameId);
           const game = await gamesSublevel.get(gameKey);
 
@@ -124,7 +131,6 @@ const getSystemProcessMap = async () => {
     if (!key || !value) return;
 
     const STEAM_COMPAT_DATA_PATH = process.environ?.STEAM_COMPAT_DATA_PATH;
-
     if (STEAM_COMPAT_DATA_PATH) {
       winePrefixMap.set(value, STEAM_COMPAT_DATA_PATH);
     }

src/main/services/user/get-user-data.ts

@@ -8,58 +8,65 @@ import { levelKeys } from "@main/level/sublevels";
 export const getUserData = async () => {
   return HydraApi.get<UserDetails>(`/profile/me`)
     .then(async (me) => {
-      db.get<string, User>(levelKeys.user, { valueEncoding: "json" }).then(
-        (user) => {
-          return db.put<string, User>(
-            levelKeys.user,
-            {
-              ...user,
-              id: me.id,
-              displayName: me.displayName,
-              profileImageUrl: me.profileImageUrl,
-              backgroundImageUrl: me.backgroundImageUrl,
-              subscription: me.subscription,
-            },
-            { valueEncoding: "json" }
-          );
-        }
-      );
-
+      try {
+        const user = await db.get<string, User>(levelKeys.user, {
+          valueEncoding: "json",
+        });
+        await db.put<string, User>(
+          levelKeys.user,
+          {
+            ...user,
+            id: me.id,
+            displayName: me.displayName,
+            profileImageUrl: me.profileImageUrl,
+            backgroundImageUrl: me.backgroundImageUrl,
+            subscription: me.subscription,
+          },
+          { valueEncoding: "json" }
+        );
+      } catch (error) {
+        logger.error("Failed to update user in DB", error);
+      }
       return me;
     })
     .catch(async (err) => {
       if (err instanceof UserNotLoggedInError) {
         return null;
       }
-      logger.error("Failed to get logged user");
 
-      const loggedUser = await db.get<string, User>(levelKeys.user, {
-        valueEncoding: "json",
-      });
+      logger.error("Failed to get logged user", err);
 
-      if (loggedUser) {
-        return {
-          ...loggedUser,
-          username: "",
-          bio: "",
-          email: null,
-          profileVisibility: "PUBLIC" as ProfileVisibility,
-          quirks: {
-            backupsPerGameLimit: 0,
-          },
-          subscription: loggedUser.subscription
-            ? {
-                id: loggedUser.subscription.id,
-                status: loggedUser.subscription.status,
-                plan: {
-                  id: loggedUser.subscription.plan.id,
-                  name: loggedUser.subscription.plan.name,
-                },
-                expiresAt: loggedUser.subscription.expiresAt,
-              }
-            : null,
-          featurebaseJwt: "",
-        } as UserDetails;
+      try {
+        const loggedUser = await db.get<string, User>(levelKeys.user, {
+          valueEncoding: "json",
+        });
+
+        if (loggedUser) {
+          return {
+            ...loggedUser,
+            username: "",
+            bio: "",
+            email: null,
+            profileVisibility: "PUBLIC" as ProfileVisibility,
+            quirks: {
+              backupsPerGameLimit: 0,
+            },
+            subscription: loggedUser.subscription
+              ? {
+                  id: loggedUser.subscription.id,
+                  status: loggedUser.subscription.status,
+                  plan: {
+                    id: loggedUser.subscription.plan.id,
+                    name: loggedUser.subscription.plan.name,
+                  },
+                  expiresAt: loggedUser.subscription.expiresAt,
+                }
+              : null,
+            featurebaseJwt: "",
+          } as UserDetails;
+        }
+      } catch (dbError) {
+        logger.error("Failed to read user from DB", dbError);
       }
 
       return null;

src/renderer/src/declaration.d.ts

@@ -30,9 +30,9 @@ import type {
   AchievementCustomNotificationPosition,
   AchievementNotificationInfo,
   Game,
+  DiskUsage,
 } from "@types";
 import type { AxiosProgressEvent } from "axios";
-import type disk from "diskusage";
 
 declare global {
   declare module "*.svg" {
@@ -220,7 +220,7 @@ declare global {
     >;
 
     /* Hardware */
-    getDiskFreeSpace: (path: string) => Promise<disk.DiskUsage>;
+    getDiskFreeSpace: (path: string) => Promise<DiskUsage>;
     checkFolderWritePermission: (path: string) => Promise<boolean>;
 
     /* Cloud save */

src/types/index.ts

@@ -10,6 +10,11 @@ export type HydraCloudFeature =
   | "backup"
   | "achievements-points";
 
+export interface DiskUsage {
+  free: number;
+  total: number;
+}
+
 export interface GameRepack {
   id: number;
   title: string;