Merge branch 'main' into main

src/declaration.d.ts

@@ -1,13 +0,0 @@
-declare module "tasklist" {
-  interface Task {
-    imageName: string;
-    pid: number;
-    sessionName: string;
-    sessionNumber: number;
-    memUsage: number;
-  }
-
-  function tasklist(): Promise<Task[]>;
-
-  export { tasklist };
-}

src/main/helpers/ps.ts

@@ -1,12 +1,33 @@
 import psList from "ps-list";
-import { tasklist } from "tasklist";
+import path from "node:path";
+import childProcess from "node:child_process";
+import { promisify } from "node:util";
+import { app } from "electron";
+
+const TEN_MEGABYTES = 1000 * 1000 * 10;
+const execFile = promisify(childProcess.execFile);
 
 export const getProcesses = async () => {
-  if (process.platform === "win32") {
-    return tasklist().then((tasks) =>
-      tasks.map((task) => ({ ...task, name: task.imageName }))
-    );
-  }
+  if (process.platform == "win32") {
+    const binaryPath = app.isPackaged
+      ? path.join(process.resourcesPath, "dist", "fastlist.exe")
+      : path.join(__dirname, "..", "..", "resources", "dist", "fastlist.exe");
 
-  return psList();
+    const { stdout } = await execFile(binaryPath, {
+      maxBuffer: TEN_MEGABYTES,
+      windowsHide: true,
+    });
+
+    return stdout
+      .trim()
+      .split("\r\n")
+      .map((line) => line.split("\t"))
+      .map(([pid, ppid, name]) => ({
+        pid: Number.parseInt(pid, 10),
+        ppid: Number.parseInt(ppid, 10),
+        name,
+      }));
+  } else {
+    return psList();
+  }
 };

src/main/services/process-watcher.ts

@@ -1,7 +1,6 @@
 import path from "node:path";
 
 import { IsNull, Not } from "typeorm";
-
 import { gameRepository } from "@main/repository";
 import { getProcesses } from "@main/helpers";
 import { WindowManager } from "./window-manager";
@@ -9,27 +8,33 @@ import { WindowManager } from "./window-manager";
 const sleep = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));
 
 export const startProcessWatcher = async () => {
-  const sleepTime = 100;
+  const sleepTime = 300;
   const gamesPlaytime = new Map<number, number>();
 
   // eslint-disable-next-line no-constant-condition
   while (true) {
+    await sleep(sleepTime);
+
     const games = await gameRepository.find({
       where: {
         executablePath: Not(IsNull()),
       },
     });
 
+    if (games.length == 0) {
+      continue;
+    }
+
     const processes = await getProcesses();
 
     for (const game of games) {
-      const gameProcess = processes.find((runningProcess) => {
-        const basename = path.win32.basename(game.executablePath);
-        const basenameWithoutExtension = path.win32.basename(
-          game.executablePath,
-          path.extname(game.executablePath)
-        );
+      const basename = path.win32.basename(game.executablePath);
+      const basenameWithoutExtension = path.win32.basename(
+        game.executablePath,
+        path.extname(game.executablePath)
+      );
 
+      const gameProcess = processes.find((runningProcess) => {
         if (process.platform === "win32") {
           return runningProcess.name === basename;
         }
@@ -55,26 +60,15 @@ export const startProcessWatcher = async () => {
           gameRepository.update(game.id, {
             lastTimePlayed: new Date().toUTCString(),
           });
-
-          gamesPlaytime.set(game.id, performance.now());
-          await sleep(sleepTime);
-          continue;
         }
 
         gamesPlaytime.set(game.id, performance.now());
-
-        await sleep(sleepTime);
-        continue;
-      }
-
-      if (gamesPlaytime.has(game.id)) {
+      } else if (gamesPlaytime.has(game.id)) {
         gamesPlaytime.delete(game.id);
         if (WindowManager.mainWindow) {
           WindowManager.mainWindow.webContents.send("on-game-close", game.id);
         }
       }
-
-      await sleep(sleepTime);
     }
   }
 };