From 0c1a75eedd4ba89e73e6f536a7af594e6a189eec Mon Sep 17 00:00:00 2001
From: Zamitto <167933696+zamitto@users.noreply.github.com>
Date: Mon, 1 Jul 2024 20:23:39 -0300
Subject: [PATCH 1/6] poc: psutil for process watcher

---
 requirements.txt       |  1 +
 torrent-client/main.py | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/requirements.txt b/requirements.txt
index 6cee730a..b1488003 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,3 +3,4 @@ cx_Freeze
 cx_Logging; sys_platform == 'win32'
 lief; sys_platform == 'win32'
 pywin32; sys_platform == 'win32'
+psutil
diff --git a/torrent-client/main.py b/torrent-client/main.py
index 9a5737a4..a0d2a57e 100644
--- a/torrent-client/main.py
+++ b/torrent-client/main.py
@@ -2,7 +2,9 @@ import sys
 from http.server import HTTPServer, BaseHTTPRequestHandler
 import json
 import urllib.parse
+import psutil
 from downloader import Downloader
+from time import time
 
 torrent_port = sys.argv[1]
 http_port = sys.argv[2]
@@ -33,6 +35,15 @@ class Handler(BaseHTTPRequestHandler):
         if self.path == "/healthcheck":
             self.send_response(200)
             self.end_headers()
+        
+        if self.path == "/process":
+            start_time = time()
+            process_path = set([proc.info["exe"] for proc in psutil.process_iter(['exe'])])
+            self.send_response(200)
+            self.send_header("Content-type", "application/json")
+            self.end_headers()
+            self.wfile.write(json.dumps(list(process_path)).encode('utf-8'))
+            print(time() - start_time)
     
     def do_POST(self):
         if self.path == "/action":

From 75c8f69e816033223e56eb9064b76e2ecc4b89b1 Mon Sep 17 00:00:00 2001
From: Zamitto <167933696+zamitto@users.noreply.github.com>
Date: Wed, 3 Jul 2024 11:23:50 -0300
Subject: [PATCH 2/6] feat: get process list from rpc

---
 src/main/events/auth/sign-out.ts              |  4 +--
 src/main/index.ts                             |  4 +--
 src/main/main.ts                              | 16 +++++++++---
 .../services/download/download-manager.ts     | 12 ++++-----
 src/main/services/download/index.ts           |  2 +-
 .../{torrent-downloader.ts => rpc-manager.ts} | 20 +++++++++------
 src/main/services/download/torrent-client.ts  |  4 +--
 src/main/services/process-watcher.ts          | 17 +++----------
 .../game-details/hero/hero-panel-actions.tsx  |  2 +-
 torrent-client/main.py                        | 25 ++++++++++++-------
 10 files changed, 57 insertions(+), 49 deletions(-)
 rename src/main/services/download/{torrent-downloader.ts => rpc-manager.ts} (88%)

diff --git a/src/main/events/auth/sign-out.ts b/src/main/events/auth/sign-out.ts
index 9c2ce2e0..0eb8cf07 100644
--- a/src/main/events/auth/sign-out.ts
+++ b/src/main/events/auth/sign-out.ts
@@ -1,6 +1,6 @@
 import { registerEvent } from "../register-event";
 import * as Sentry from "@sentry/electron/main";
-import { HydraApi, TorrentDownloader, gamesPlaytime } from "@main/services";
+import { HydraApi, RPCManager, gamesPlaytime } from "@main/services";
 import { dataSource } from "@main/data-source";
 import { DownloadQueue, Game, UserAuth } from "@main/entity";
 
@@ -24,7 +24,7 @@ const signOut = async (_event: Electron.IpcMainInvokeEvent) => {
   Sentry.setUser(null);
 
   /* Disconnects libtorrent */
-  TorrentDownloader.kill();
+  RPCManager.kill();
 
   await Promise.all([
     databaseOperations,
diff --git a/src/main/index.ts b/src/main/index.ts
index cb20733a..1372539d 100644
--- a/src/main/index.ts
+++ b/src/main/index.ts
@@ -5,7 +5,7 @@ import i18n from "i18next";
 import path from "node:path";
 import url from "node:url";
 import { electronApp, optimizer } from "@electron-toolkit/utils";
-import { logger, TorrentDownloader, WindowManager } from "@main/services";
+import { logger, RPCManager, WindowManager } from "@main/services";
 import { dataSource } from "@main/data-source";
 import * as resources from "@locales";
 import { userPreferencesRepository } from "@main/repository";
@@ -116,7 +116,7 @@ app.on("window-all-closed", () => {
 
 app.on("before-quit", () => {
   /* Disconnects libtorrent */
-  TorrentDownloader.kill();
+  RPCManager.kill();
 });
 
 app.on("activate", () => {
diff --git a/src/main/main.ts b/src/main/main.ts
index 1abd8c2f..759b1714 100644
--- a/src/main/main.ts
+++ b/src/main/main.ts
@@ -1,4 +1,9 @@
-import { DownloadManager, RepacksManager, startMainLoop } from "./services";
+import {
+  DownloadManager,
+  RepacksManager,
+  RPCManager,
+  startMainLoop,
+} from "./services";
 import {
   downloadQueueRepository,
   repackRepository,
@@ -12,8 +17,6 @@ import { MoreThan } from "typeorm";
 import { HydraApi } from "./services/hydra-api";
 import { uploadGamesBatch } from "./services/library-sync";
 
-startMainLoop();
-
 const loadState = async (userPreferences: UserPreferences | null) => {
   await RepacksManager.updateRepacks();
 
@@ -35,8 +38,13 @@ const loadState = async (userPreferences: UserPreferences | null) => {
     },
   });
 
-  if (nextQueueItem?.game.status === "active")
+  if (nextQueueItem?.game.status === "active") {
     DownloadManager.startDownload(nextQueueItem.game);
+  } else {
+    RPCManager.spawn();
+  }
+
+  startMainLoop();
 
   const now = new Date();
 
diff --git a/src/main/services/download/download-manager.ts b/src/main/services/download/download-manager.ts
index f7553c9c..8d402e5a 100644
--- a/src/main/services/download/download-manager.ts
+++ b/src/main/services/download/download-manager.ts
@@ -1,6 +1,6 @@
 import { Game } from "@main/entity";
 import { Downloader } from "@shared";
-import { TorrentDownloader } from "./torrent-downloader";
+import { RPCManager } from "./rpc-manager";
 import { WindowManager } from "../window-manager";
 import { downloadQueueRepository, gameRepository } from "@main/repository";
 import { publishDownloadCompleteNotification } from "../notifications";
@@ -16,7 +16,7 @@ export class DownloadManager {
     if (this.currentDownloader === Downloader.RealDebrid) {
       status = await RealDebridDownloader.getStatus();
     } else {
-      status = await TorrentDownloader.getStatus();
+      status = await RPCManager.getStatus();
     }
 
     if (status) {
@@ -65,7 +65,7 @@ export class DownloadManager {
     if (this.currentDownloader === Downloader.RealDebrid) {
       RealDebridDownloader.pauseDownload();
     } else {
-      await TorrentDownloader.pauseDownload();
+      await RPCManager.pauseDownload();
     }
 
     WindowManager.mainWindow?.setProgressBar(-1);
@@ -77,7 +77,7 @@ export class DownloadManager {
       RealDebridDownloader.startDownload(game);
       this.currentDownloader = Downloader.RealDebrid;
     } else {
-      TorrentDownloader.startDownload(game);
+      RPCManager.startDownload(game);
       this.currentDownloader = Downloader.Torrent;
     }
   }
@@ -86,7 +86,7 @@ export class DownloadManager {
     if (this.currentDownloader === Downloader.RealDebrid) {
       RealDebridDownloader.cancelDownload();
     } else {
-      TorrentDownloader.cancelDownload(gameId);
+      RPCManager.cancelDownload(gameId);
     }
 
     WindowManager.mainWindow?.setProgressBar(-1);
@@ -98,7 +98,7 @@ export class DownloadManager {
       RealDebridDownloader.startDownload(game);
       this.currentDownloader = Downloader.RealDebrid;
     } else {
-      TorrentDownloader.startDownload(game);
+      RPCManager.startDownload(game);
       this.currentDownloader = Downloader.Torrent;
     }
   }
diff --git a/src/main/services/download/index.ts b/src/main/services/download/index.ts
index 6244b81a..0298f7d8 100644
--- a/src/main/services/download/index.ts
+++ b/src/main/services/download/index.ts
@@ -1,2 +1,2 @@
 export * from "./download-manager";
-export * from "./torrent-downloader";
+export * from "./rpc-manager";
diff --git a/src/main/services/download/torrent-downloader.ts b/src/main/services/download/rpc-manager.ts
similarity index 88%
rename from src/main/services/download/torrent-downloader.ts
rename to src/main/services/download/rpc-manager.ts
index e35bb617..12d40a28 100644
--- a/src/main/services/download/torrent-downloader.ts
+++ b/src/main/services/download/rpc-manager.ts
@@ -15,8 +15,8 @@ import {
   LibtorrentPayload,
 } from "./types";
 
-export class TorrentDownloader {
-  private static torrentClient: cp.ChildProcess | null = null;
+export class RPCManager {
+  private static rpcProcess: cp.ChildProcess | null = null;
   private static downloadingGameId = -1;
 
   private static rpc = axios.create({
@@ -26,18 +26,22 @@ export class TorrentDownloader {
     },
   });
 
-  private static spawn(args: StartDownloadPayload) {
-    this.torrentClient = startTorrentClient(args);
+  public static spawn(args?: StartDownloadPayload) {
+    this.rpcProcess = startTorrentClient(args);
   }
 
   public static kill() {
-    if (this.torrentClient) {
-      this.torrentClient.kill();
-      this.torrentClient = null;
+    if (this.rpcProcess) {
+      this.rpcProcess.kill();
+      this.rpcProcess = null;
       this.downloadingGameId = -1;
     }
   }
 
+  public static async getProccessList() {
+    return (await this.rpc.get<string[] | null>("/process-list")).data;
+  }
+
   public static async getStatus() {
     if (this.downloadingGameId === -1) return null;
 
@@ -113,7 +117,7 @@ export class TorrentDownloader {
   }
 
   static async startDownload(game: Game) {
-    if (!this.torrentClient) {
+    if (!this.rpcProcess) {
       this.spawn({
         game_id: game.id,
         magnet: game.uri!,
diff --git a/src/main/services/download/torrent-client.ts b/src/main/services/download/torrent-client.ts
index 176ec664..93d20b7f 100644
--- a/src/main/services/download/torrent-client.ts
+++ b/src/main/services/download/torrent-client.ts
@@ -15,12 +15,12 @@ export const BITTORRENT_PORT = "5881";
 export const RPC_PORT = "8084";
 export const RPC_PASSWORD = crypto.randomBytes(32).toString("hex");
 
-export const startTorrentClient = (args: StartDownloadPayload) => {
+export const startTorrentClient = (args?: StartDownloadPayload) => {
   const commonArgs = [
     BITTORRENT_PORT,
     RPC_PORT,
     RPC_PASSWORD,
-    encodeURIComponent(JSON.stringify(args)),
+    args ? encodeURIComponent(JSON.stringify(args)) : "",
   ];
 
   if (app.isPackaged) {
diff --git a/src/main/services/process-watcher.ts b/src/main/services/process-watcher.ts
index f20b4128..4dc12bc1 100644
--- a/src/main/services/process-watcher.ts
+++ b/src/main/services/process-watcher.ts
@@ -1,11 +1,9 @@
-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";
 import { createGame, updateGamePlaytime } from "./library-sync";
 import { GameRunning } from "@types";
+import { RPCManager } from "./download";
 
 export const gamesPlaytime = new Map<
   number,
@@ -22,22 +20,13 @@ export const watchProcesses = async () => {
 
   if (games.length === 0) return;
 
-  const processes = await getProcesses();
+  const processes = (await RPCManager.getProccessList()) || [];
 
   for (const game of games) {
     const executablePath = game.executablePath!;
-    const basename = path.win32.basename(executablePath);
-    const basenameWithoutExtension = path.win32.basename(
-      executablePath,
-      path.extname(executablePath)
-    );
 
     const gameProcess = processes.find((runningProcess) => {
-      if (process.platform === "win32") {
-        return runningProcess.name === basename;
-      }
-
-      return [basename, basenameWithoutExtension].includes(runningProcess.name);
+      return executablePath == runningProcess;
     });
 
     if (gameProcess) {
diff --git a/src/renderer/src/pages/game-details/hero/hero-panel-actions.tsx b/src/renderer/src/pages/game-details/hero/hero-panel-actions.tsx
index 4741668e..795aa8af 100644
--- a/src/renderer/src/pages/game-details/hero/hero-panel-actions.tsx
+++ b/src/renderer/src/pages/game-details/hero/hero-panel-actions.tsx
@@ -122,7 +122,7 @@ export function HeroPanelActions() {
         <Button
           onClick={() => setShowGameOptionsModal(true)}
           theme="outline"
-          disabled={deleting || isGameRunning}
+          disabled={deleting}
           className={styles.heroPanelAction}
         >
           <GearIcon />
diff --git a/torrent-client/main.py b/torrent-client/main.py
index a0d2a57e..eb04b77c 100644
--- a/torrent-client/main.py
+++ b/torrent-client/main.py
@@ -4,16 +4,17 @@ import json
 import urllib.parse
 import psutil
 from downloader import Downloader
-from time import time
 
 torrent_port = sys.argv[1]
 http_port = sys.argv[2]
 rpc_password = sys.argv[3]
-initial_download = json.loads(urllib.parse.unquote(sys.argv[4]))
 
-downloader = Downloader(torrent_port)
+downloader = None
 
-downloader.start_download(initial_download['game_id'], initial_download['magnet'], initial_download['save_path'])
+if sys.argv[4]:
+    initial_download = json.loads(urllib.parse.unquote(sys.argv[4]))
+    downloader = Downloader(torrent_port)
+    downloader.start_download(initial_download['game_id'], initial_download['magnet'], initial_download['save_path'])
 
 class Handler(BaseHTTPRequestHandler):
     rpc_password_header = 'x-hydra-rpc-password'
@@ -36,14 +37,17 @@ class Handler(BaseHTTPRequestHandler):
             self.send_response(200)
             self.end_headers()
         
-        if self.path == "/process":
-            start_time = time()
-            process_path = set([proc.info["exe"] for proc in psutil.process_iter(['exe'])])
+        if self.path == "/process-list":
+            if self.headers.get(self.rpc_password_header) != rpc_password:
+                self.send_response(401)
+                self.end_headers()
+                return
+            
+            process_path = list(set([proc.info["exe"] for proc in psutil.process_iter(['exe'])]))
             self.send_response(200)
             self.send_header("Content-type", "application/json")
             self.end_headers()
-            self.wfile.write(json.dumps(list(process_path)).encode('utf-8'))
-            print(time() - start_time)
+            self.wfile.write(json.dumps(process_path).encode('utf-8'))
     
     def do_POST(self):
         if self.path == "/action":
@@ -56,6 +60,9 @@ class Handler(BaseHTTPRequestHandler):
             post_data = self.rfile.read(content_length)
             data = json.loads(post_data.decode('utf-8'))
 
+            if downloader is None:
+                downloader = Downloader(torrent_port)
+
             if data['action'] == 'start':
                 downloader.start_download(data['game_id'], data['magnet'], data['save_path'])
             elif data['action'] == 'pause':

From 0f5db4f34e713ed41381dadff4e76b63ffc2f467 Mon Sep 17 00:00:00 2001
From: Zamitto <167933696+zamitto@users.noreply.github.com>
Date: Wed, 3 Jul 2024 12:06:26 -0300
Subject: [PATCH 3/6] feat: crete kill-torrent

---
 src/main/events/auth/sign-out.ts          |  2 +-
 src/main/services/download/rpc-manager.ts | 17 ++++++++++++++---
 src/main/services/process-watcher.ts      |  2 +-
 torrent-client/downloader.py              |  9 +++++++++
 torrent-client/main.py                    | 14 +++++++++++---
 5 files changed, 36 insertions(+), 8 deletions(-)

diff --git a/src/main/events/auth/sign-out.ts b/src/main/events/auth/sign-out.ts
index 0eb8cf07..a38e9d9e 100644
--- a/src/main/events/auth/sign-out.ts
+++ b/src/main/events/auth/sign-out.ts
@@ -24,7 +24,7 @@ const signOut = async (_event: Electron.IpcMainInvokeEvent) => {
   Sentry.setUser(null);
 
   /* Disconnects libtorrent */
-  RPCManager.kill();
+  RPCManager.killTorrent();
 
   await Promise.all([
     databaseOperations,
diff --git a/src/main/services/download/rpc-manager.ts b/src/main/services/download/rpc-manager.ts
index 12d40a28..9ada3e54 100644
--- a/src/main/services/download/rpc-manager.ts
+++ b/src/main/services/download/rpc-manager.ts
@@ -1,7 +1,11 @@
 import cp from "node:child_process";
 
 import { Game } from "@main/entity";
-import { RPC_PASSWORD, RPC_PORT, startTorrentClient } from "./torrent-client";
+import {
+  RPC_PASSWORD,
+  RPC_PORT,
+  startTorrentClient as startRPCClient,
+} from "./torrent-client";
 import { gameRepository } from "@main/repository";
 import { DownloadProgress } from "@types";
 import { QueryDeepPartialEntity } from "typeorm/query-builder/QueryPartialEntity";
@@ -27,7 +31,7 @@ export class RPCManager {
   });
 
   public static spawn(args?: StartDownloadPayload) {
-    this.rpcProcess = startTorrentClient(args);
+    this.rpcProcess = startRPCClient(args);
   }
 
   public static kill() {
@@ -38,8 +42,15 @@ export class RPCManager {
     }
   }
 
+  public static killTorrent() {
+    if (this.rpcProcess) {
+      this.rpc.post("/action", { action: "kill-torrent" });
+      this.downloadingGameId = -1;
+    }
+  }
+
   public static async getProccessList() {
-    return (await this.rpc.get<string[] | null>("/process-list")).data;
+    return (await this.rpc.get<string[] | null>("/process-list")).data || [];
   }
 
   public static async getStatus() {
diff --git a/src/main/services/process-watcher.ts b/src/main/services/process-watcher.ts
index 4dc12bc1..8512f1d8 100644
--- a/src/main/services/process-watcher.ts
+++ b/src/main/services/process-watcher.ts
@@ -20,7 +20,7 @@ export const watchProcesses = async () => {
 
   if (games.length === 0) return;
 
-  const processes = (await RPCManager.getProccessList()) || [];
+  const processes = await RPCManager.getProccessList();
 
   for (const game of games) {
     const executablePath = game.executablePath!;
diff --git a/torrent-client/downloader.py b/torrent-client/downloader.py
index 5ddbcf67..2368ff17 100644
--- a/torrent-client/downloader.py
+++ b/torrent-client/downloader.py
@@ -30,6 +30,15 @@ class Downloader:
             self.torrent_handles[game_id] = None
             self.downloading_game_id = -1
 
+    def cancel_all_downloads(self):
+        for game_id in self.torrent_handles:
+            torrent_handle = self.torrent_handles[game_id]
+            torrent_handle.pause()
+            self.session.remove_torrent(torrent_handle)
+
+        self.torrent_handles = {}
+        self.downloading_game_id = -1
+
     def get_download_status(self):
         if self.downloading_game_id == -1:
             return None
diff --git a/torrent-client/main.py b/torrent-client/main.py
index eb04b77c..1099af91 100644
--- a/torrent-client/main.py
+++ b/torrent-client/main.py
@@ -33,23 +33,28 @@ class Handler(BaseHTTPRequestHandler):
             status = downloader.get_download_status()
 
             self.wfile.write(json.dumps(status).encode('utf-8'))
-        if self.path == "/healthcheck":
+
+        elif self.path == "/healthcheck":
             self.send_response(200)
             self.end_headers()
         
-        if self.path == "/process-list":
+        elif self.path == "/process-list":
             if self.headers.get(self.rpc_password_header) != rpc_password:
                 self.send_response(401)
                 self.end_headers()
                 return
             
             process_path = list(set([proc.info["exe"] for proc in psutil.process_iter(['exe'])]))
+
             self.send_response(200)
             self.send_header("Content-type", "application/json")
             self.end_headers()
+
             self.wfile.write(json.dumps(process_path).encode('utf-8'))
     
     def do_POST(self):
+        global downloader
+        
         if self.path == "/action":
             if self.headers.get(self.rpc_password_header) != rpc_password:
                 self.send_response(401)
@@ -69,7 +74,10 @@ class Handler(BaseHTTPRequestHandler):
                 downloader.pause_download(data['game_id'])
             elif data['action'] == 'cancel':
                 downloader.cancel_download(data['game_id'])
-        
+            elif data['action'] == 'kill-torrent':
+                downloader.cancel_all_downloads()
+                downloader = None
+
             self.send_response(200)
             self.end_headers()
 

From 1397e3932d53c855e8cbe2edbaf9841426f85872 Mon Sep 17 00:00:00 2001
From: Zamitto <167933696+zamitto@users.noreply.github.com>
Date: Wed, 3 Jul 2024 15:29:59 -0300
Subject: [PATCH 4/6] feat: remove pslist and use sudo-prompt to close game if
 needed

---
 .gitignore                                    |  1 -
 electron-builder.yml                          |  5 --
 package.json                                  |  2 +-
 resources/hydralauncher.vbs                   |  3 --
 src/main/constants.ts                         |  9 ----
 src/main/events/library/close-game.ts         | 48 +++++++++++--------
 .../events/user-preferences/auto-launch.ts    | 22 ++-------
 src/main/helpers/index.ts                     |  1 -
 src/main/helpers/ps.ts                        | 41 ----------------
 src/main/services/download/rpc-manager.ts     |  7 ++-
 src/main/services/download/types.ts           |  5 ++
 src/main/services/process-watcher.ts          |  5 +-
 torrent-client/main.py                        |  6 +--
 yarn.lock                                     | 10 ++--
 14 files changed, 51 insertions(+), 114 deletions(-)
 delete mode 100644 resources/hydralauncher.vbs
 delete mode 100644 src/main/helpers/ps.ts

diff --git a/.gitignore b/.gitignore
index fb4badd7..017a9141 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,7 +1,6 @@
 .vscode
 node_modules
 hydra-download-manager/
-fastlist.exe
 __pycache__
 dist
 out
diff --git a/electron-builder.yml b/electron-builder.yml
index fec128a6..cfdafe7d 100644
--- a/electron-builder.yml
+++ b/electron-builder.yml
@@ -5,10 +5,7 @@ directories:
 extraResources:
   - hydra-download-manager
   - seeds
-  - from: node_modules/ps-list/vendor/fastlist-0.3.0-x64.exe
-    to: fastlist.exe
   - from: node_modules/create-desktop-shortcuts/src/windows.vbs
-  - from: resources/hydralauncher.vbs
 files:
   - "!**/.vscode/*"
   - "!src/*"
@@ -20,7 +17,6 @@ asarUnpack:
   - resources/**
 win:
   executableName: Hydra
-  requestedExecutionLevel: requireAdministrator
   target:
     - nsis
     - portable
@@ -33,7 +29,6 @@ nsis:
   allowToChangeInstallationDirectory: true
 portable:
   artifactName: ${name}-${version}-portable.${ext}
-  requestExecutionLevel: admin
 mac:
   entitlementsInherit: build/entitlements.mac.plist
   extendInfo:
diff --git a/package.json b/package.json
index c029e4db..549f7a1b 100644
--- a/package.json
+++ b/package.json
@@ -65,11 +65,11 @@
     "lottie-react": "^2.4.0",
     "parse-torrent": "^11.0.16",
     "piscina": "^4.5.1",
-    "ps-list": "^8.1.1",
     "react-i18next": "^14.1.0",
     "react-loading-skeleton": "^3.4.0",
     "react-redux": "^9.1.1",
     "react-router-dom": "^6.22.3",
+    "sudo-prompt": "^9.2.1",
     "typeorm": "^0.3.20",
     "user-agents": "^1.1.193",
     "yaml": "^2.4.1",
diff --git a/resources/hydralauncher.vbs b/resources/hydralauncher.vbs
deleted file mode 100644
index ff611acf..00000000
--- a/resources/hydralauncher.vbs
+++ /dev/null
@@ -1,3 +0,0 @@
-Set WshShell = CreateObject("WScript.Shell" ) 
-WshShell.Run """%localappdata%\Programs\Hydra\Hydra.exe""", 0 'Must quote command if it has spaces; must escape quotes
-Set WshShell = Nothing
diff --git a/src/main/constants.ts b/src/main/constants.ts
index 62713a2c..850c9ada 100644
--- a/src/main/constants.ts
+++ b/src/main/constants.ts
@@ -14,12 +14,3 @@ export const logsPath = path.join(app.getPath("appData"), "hydra", "logs");
 export const seedsPath = app.isPackaged
   ? path.join(process.resourcesPath, "seeds")
   : path.join(__dirname, "..", "..", "seeds");
-
-export const windowsStartupPath = path.join(
-  app.getPath("appData"),
-  "Microsoft",
-  "Windows",
-  "Start Menu",
-  "Programs",
-  "Startup"
-);
diff --git a/src/main/events/library/close-game.ts b/src/main/events/library/close-game.ts
index ccf446d3..7f3b74a5 100644
--- a/src/main/events/library/close-game.ts
+++ b/src/main/events/library/close-game.ts
@@ -1,39 +1,45 @@
-import path from "node:path";
-
 import { gameRepository } from "@main/repository";
-import { getProcesses } from "@main/helpers";
-
 import { registerEvent } from "../register-event";
+import { RPCManager, logger } from "@main/services";
+import sudo from "sudo-prompt";
+import { app } from "electron";
+
+const getKillCommand = (pid: number) => {
+  if (process.platform == "win32") {
+    return `taskkill /PID ${pid}`;
+  }
+
+  return `kill -9 ${pid}`;
+};
 
 const closeGame = async (
   _event: Electron.IpcMainInvokeEvent,
   gameId: number
 ) => {
-  const processes = await getProcesses();
+  const processes = await RPCManager.getProcessList();
   const game = await gameRepository.findOne({
     where: { id: gameId, isDeleted: false },
   });
 
-  if (!game) return false;
-
-  const executablePath = game.executablePath!;
-
-  const basename = path.win32.basename(executablePath);
-  const basenameWithoutExtension = path.win32.basename(
-    executablePath,
-    path.extname(executablePath)
-  );
+  if (!game) return;
 
   const gameProcess = processes.find((runningProcess) => {
-    if (process.platform === "win32") {
-      return runningProcess.name === basename;
-    }
-
-    return [basename, basenameWithoutExtension].includes(runningProcess.name);
+    return runningProcess.exe === game.executablePath;
   });
 
-  if (gameProcess) return process.kill(gameProcess.pid);
-  return false;
+  if (gameProcess) {
+    try {
+      process.kill(gameProcess.pid);
+    } catch (err) {
+      sudo.exec(
+        getKillCommand(gameProcess.pid),
+        { name: app.getName() },
+        (error, _stdout, _stderr) => {
+          logger.error(error);
+        }
+      );
+    }
+  }
 };
 
 registerEvent("closeGame", closeGame);
diff --git a/src/main/events/user-preferences/auto-launch.ts b/src/main/events/user-preferences/auto-launch.ts
index cb40a969..ade7833d 100644
--- a/src/main/events/user-preferences/auto-launch.ts
+++ b/src/main/events/user-preferences/auto-launch.ts
@@ -1,9 +1,6 @@
-import { windowsStartupPath } from "@main/constants";
 import { registerEvent } from "../register-event";
 import AutoLaunch from "auto-launch";
 import { app } from "electron";
-import fs from "node:fs";
-import path from "node:path";
 
 const autoLaunch = async (
   _event: Electron.IpcMainInvokeEvent,
@@ -15,23 +12,10 @@ const autoLaunch = async (
     name: app.getName(),
   });
 
-  if (process.platform == "win32") {
-    const destination = path.join(windowsStartupPath, "Hydra.vbs");
-
-    if (enabled) {
-      const scriptPath = path.join(process.resourcesPath, "hydralauncher.vbs");
-
-      fs.copyFileSync(scriptPath, destination);
-    } else {
-      appLauncher.disable().catch();
-      fs.rmSync(destination);
-    }
+  if (enabled) {
+    appLauncher.enable().catch();
   } else {
-    if (enabled) {
-      appLauncher.enable().catch();
-    } else {
-      appLauncher.disable().catch();
-    }
+    appLauncher.disable().catch();
   }
 };
 
diff --git a/src/main/helpers/index.ts b/src/main/helpers/index.ts
index db1d13ff..902b927d 100644
--- a/src/main/helpers/index.ts
+++ b/src/main/helpers/index.ts
@@ -57,5 +57,4 @@ export const requestWebPage = async (url: string) => {
     .then((response) => response.data);
 };
 
-export * from "./ps";
 export * from "./download-source";
diff --git a/src/main/helpers/ps.ts b/src/main/helpers/ps.ts
deleted file mode 100644
index 5b326dfa..00000000
--- a/src/main/helpers/ps.ts
+++ /dev/null
@@ -1,41 +0,0 @@
-import psList from "ps-list";
-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") {
-    const binaryPath = app.isPackaged
-      ? path.join(process.resourcesPath, "fastlist.exe")
-      : path.join(
-          __dirname,
-          "..",
-          "..",
-          "node_modules",
-          "ps-list",
-          "vendor",
-          "fastlist-0.3.0-x64.exe"
-        );
-
-    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();
-  }
-};
diff --git a/src/main/services/download/rpc-manager.ts b/src/main/services/download/rpc-manager.ts
index 9ada3e54..58cc1daa 100644
--- a/src/main/services/download/rpc-manager.ts
+++ b/src/main/services/download/rpc-manager.ts
@@ -17,6 +17,7 @@ import {
   PauseDownloadPayload,
   LibtorrentStatus,
   LibtorrentPayload,
+  ProcessPayload,
 } from "./types";
 
 export class RPCManager {
@@ -49,8 +50,10 @@ export class RPCManager {
     }
   }
 
-  public static async getProccessList() {
-    return (await this.rpc.get<string[] | null>("/process-list")).data || [];
+  public static async getProcessList() {
+    return (
+      (await this.rpc.get<ProcessPayload[] | null>("/process-list")).data || []
+    );
   }
 
   public static async getStatus() {
diff --git a/src/main/services/download/types.ts b/src/main/services/download/types.ts
index 516cd84f..fd8009a2 100644
--- a/src/main/services/download/types.ts
+++ b/src/main/services/download/types.ts
@@ -31,3 +31,8 @@ export interface LibtorrentPayload {
   status: LibtorrentStatus;
   gameId: number;
 }
+
+export interface ProcessPayload {
+  exe: string;
+  pid: number;
+}
diff --git a/src/main/services/process-watcher.ts b/src/main/services/process-watcher.ts
index 8512f1d8..ec7d8e20 100644
--- a/src/main/services/process-watcher.ts
+++ b/src/main/services/process-watcher.ts
@@ -19,14 +19,13 @@ export const watchProcesses = async () => {
   });
 
   if (games.length === 0) return;
-
-  const processes = await RPCManager.getProccessList();
+  const processes = await RPCManager.getProcessList();
 
   for (const game of games) {
     const executablePath = game.executablePath!;
 
     const gameProcess = processes.find((runningProcess) => {
-      return executablePath == runningProcess;
+      return executablePath == runningProcess.exe;
     });
 
     if (gameProcess) {
diff --git a/torrent-client/main.py b/torrent-client/main.py
index 1099af91..8f57de8a 100644
--- a/torrent-client/main.py
+++ b/torrent-client/main.py
@@ -44,17 +44,17 @@ class Handler(BaseHTTPRequestHandler):
                 self.end_headers()
                 return
             
-            process_path = list(set([proc.info["exe"] for proc in psutil.process_iter(['exe'])]))
+            process_list = [proc.info for proc in psutil.process_iter(['exe', 'pid', 'username'])]
 
             self.send_response(200)
             self.send_header("Content-type", "application/json")
             self.end_headers()
 
-            self.wfile.write(json.dumps(process_path).encode('utf-8'))
+            self.wfile.write(json.dumps(process_list).encode('utf-8'))
     
     def do_POST(self):
         global downloader
-        
+
         if self.path == "/action":
             if self.headers.get(self.rpc_password_header) != rpc_password:
                 self.send_response(401)
diff --git a/yarn.lock b/yarn.lock
index 9fce82e7..30dfe715 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -6311,11 +6311,6 @@ proxy-from-env@^1.1.0:
   resolved "https://registry.npmjs.org/proxy-from-env/-/proxy-from-env-1.1.0.tgz"
   integrity sha512-D+zkORCbA9f1tdWRK0RaCR3GPv50cMxcrz4X8k5LTSUD1Dkw47mKJEZQNunItRTkWwgtaUSo1RVFRIG9ZXiFYg==
 
-ps-list@^8.1.1:
-  version "8.1.1"
-  resolved "https://registry.npmjs.org/ps-list/-/ps-list-8.1.1.tgz"
-  integrity sha512-OPS9kEJYVmiO48u/B9qneqhkMvgCxT+Tm28VCEJpheTpl8cJ0ffZRRNgS5mrQRTrX5yRTpaJ+hRDeefXYmmorQ==
-
 psl@^1.1.33:
   version "1.9.0"
   resolved "https://registry.npmjs.org/psl/-/psl-1.9.0.tgz"
@@ -6999,6 +6994,11 @@ strtok3@^7.0.0:
     "@tokenizer/token" "^0.3.0"
     peek-readable "^5.0.0"
 
+sudo-prompt@^9.2.1:
+  version "9.2.1"
+  resolved "https://registry.yarnpkg.com/sudo-prompt/-/sudo-prompt-9.2.1.tgz#77efb84309c9ca489527a4e749f287e6bdd52afd"
+  integrity sha512-Mu7R0g4ig9TUuGSxJavny5Rv0egCEtpZRNMrZaYS1vxkiIxGiGUwoezU3LazIQ+KE04hTrTfNPgxU5gzi7F5Pw==
+
 sumchecker@^3.0.1:
   version "3.0.1"
   resolved "https://registry.npmjs.org/sumchecker/-/sumchecker-3.0.1.tgz"

From ae3daa4c7966c7aba46b2d0cedcc22e90e1d69a0 Mon Sep 17 00:00:00 2001
From: Zamitto <167933696+zamitto@users.noreply.github.com>
Date: Wed, 3 Jul 2024 15:52:25 -0300
Subject: [PATCH 5/6] fix: remove symbols from name before creating game
 shortcut

---
 src/main/events/library/create-game-shortcut.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/main/events/library/create-game-shortcut.ts b/src/main/events/library/create-game-shortcut.ts
index 30dfbf7c..a27317f0 100644
--- a/src/main/events/library/create-game-shortcut.ts
+++ b/src/main/events/library/create-game-shortcut.ts
@@ -4,6 +4,7 @@ import { IsNull, Not } from "typeorm";
 import createDesktopShortcut from "create-desktop-shortcuts";
 import path from "node:path";
 import { app } from "electron";
+import { removeSymbolsFromName } from "@shared";
 
 const createGameShortcut = async (
   _event: Electron.IpcMainInvokeEvent,
@@ -22,7 +23,7 @@ const createGameShortcut = async (
 
     const options = {
       filePath,
-      name: game.title,
+      name: removeSymbolsFromName(game.title),
     };
 
     return createDesktopShortcut({

From 153291f89fdb0d23c474ea2caac792ba0b87a672 Mon Sep 17 00:00:00 2001
From: Zamitto <167933696+zamitto@users.noreply.github.com>
Date: Wed, 3 Jul 2024 16:10:23 -0300
Subject: [PATCH 6/6] feat: apply suggestions

---
 src/main/events/auth/sign-out.ts                 |  4 ++--
 src/main/events/library/close-game.ts            |  4 ++--
 src/main/index.ts                                |  4 ++--
 src/main/main.ts                                 |  4 ++--
 src/main/services/download/download-manager.ts   | 12 ++++++------
 src/main/services/download/index.ts              |  2 +-
 .../{rpc-manager.ts => python-instance.ts}       | 16 ++++++++--------
 src/main/services/process-watcher.ts             |  4 ++--
 torrent-client/downloader.py                     |  5 +++--
 torrent-client/main.py                           |  7 ++++---
 10 files changed, 32 insertions(+), 30 deletions(-)
 rename src/main/services/download/{rpc-manager.ts => python-instance.ts} (92%)

diff --git a/src/main/events/auth/sign-out.ts b/src/main/events/auth/sign-out.ts
index a38e9d9e..0ba23e2b 100644
--- a/src/main/events/auth/sign-out.ts
+++ b/src/main/events/auth/sign-out.ts
@@ -1,6 +1,6 @@
 import { registerEvent } from "../register-event";
 import * as Sentry from "@sentry/electron/main";
-import { HydraApi, RPCManager, gamesPlaytime } from "@main/services";
+import { HydraApi, PythonInstance, gamesPlaytime } from "@main/services";
 import { dataSource } from "@main/data-source";
 import { DownloadQueue, Game, UserAuth } from "@main/entity";
 
@@ -24,7 +24,7 @@ const signOut = async (_event: Electron.IpcMainInvokeEvent) => {
   Sentry.setUser(null);
 
   /* Disconnects libtorrent */
-  RPCManager.killTorrent();
+  PythonInstance.killTorrent();
 
   await Promise.all([
     databaseOperations,
diff --git a/src/main/events/library/close-game.ts b/src/main/events/library/close-game.ts
index 7f3b74a5..9c431d06 100644
--- a/src/main/events/library/close-game.ts
+++ b/src/main/events/library/close-game.ts
@@ -1,6 +1,6 @@
 import { gameRepository } from "@main/repository";
 import { registerEvent } from "../register-event";
-import { RPCManager, logger } from "@main/services";
+import { PythonInstance, logger } from "@main/services";
 import sudo from "sudo-prompt";
 import { app } from "electron";
 
@@ -16,7 +16,7 @@ const closeGame = async (
   _event: Electron.IpcMainInvokeEvent,
   gameId: number
 ) => {
-  const processes = await RPCManager.getProcessList();
+  const processes = await PythonInstance.getProcessList();
   const game = await gameRepository.findOne({
     where: { id: gameId, isDeleted: false },
   });
diff --git a/src/main/index.ts b/src/main/index.ts
index 1372539d..7b7c6390 100644
--- a/src/main/index.ts
+++ b/src/main/index.ts
@@ -5,7 +5,7 @@ import i18n from "i18next";
 import path from "node:path";
 import url from "node:url";
 import { electronApp, optimizer } from "@electron-toolkit/utils";
-import { logger, RPCManager, WindowManager } from "@main/services";
+import { logger, PythonInstance, WindowManager } from "@main/services";
 import { dataSource } from "@main/data-source";
 import * as resources from "@locales";
 import { userPreferencesRepository } from "@main/repository";
@@ -116,7 +116,7 @@ app.on("window-all-closed", () => {
 
 app.on("before-quit", () => {
   /* Disconnects libtorrent */
-  RPCManager.kill();
+  PythonInstance.kill();
 });
 
 app.on("activate", () => {
diff --git a/src/main/main.ts b/src/main/main.ts
index 759b1714..c80f0368 100644
--- a/src/main/main.ts
+++ b/src/main/main.ts
@@ -1,7 +1,7 @@
 import {
   DownloadManager,
   RepacksManager,
-  RPCManager,
+  PythonInstance,
   startMainLoop,
 } from "./services";
 import {
@@ -41,7 +41,7 @@ const loadState = async (userPreferences: UserPreferences | null) => {
   if (nextQueueItem?.game.status === "active") {
     DownloadManager.startDownload(nextQueueItem.game);
   } else {
-    RPCManager.spawn();
+    PythonInstance.spawn();
   }
 
   startMainLoop();
diff --git a/src/main/services/download/download-manager.ts b/src/main/services/download/download-manager.ts
index 8d402e5a..1a1b4ca8 100644
--- a/src/main/services/download/download-manager.ts
+++ b/src/main/services/download/download-manager.ts
@@ -1,6 +1,6 @@
 import { Game } from "@main/entity";
 import { Downloader } from "@shared";
-import { RPCManager } from "./rpc-manager";
+import { PythonInstance } from "./python-instance";
 import { WindowManager } from "../window-manager";
 import { downloadQueueRepository, gameRepository } from "@main/repository";
 import { publishDownloadCompleteNotification } from "../notifications";
@@ -16,7 +16,7 @@ export class DownloadManager {
     if (this.currentDownloader === Downloader.RealDebrid) {
       status = await RealDebridDownloader.getStatus();
     } else {
-      status = await RPCManager.getStatus();
+      status = await PythonInstance.getStatus();
     }
 
     if (status) {
@@ -65,7 +65,7 @@ export class DownloadManager {
     if (this.currentDownloader === Downloader.RealDebrid) {
       RealDebridDownloader.pauseDownload();
     } else {
-      await RPCManager.pauseDownload();
+      await PythonInstance.pauseDownload();
     }
 
     WindowManager.mainWindow?.setProgressBar(-1);
@@ -77,7 +77,7 @@ export class DownloadManager {
       RealDebridDownloader.startDownload(game);
       this.currentDownloader = Downloader.RealDebrid;
     } else {
-      RPCManager.startDownload(game);
+      PythonInstance.startDownload(game);
       this.currentDownloader = Downloader.Torrent;
     }
   }
@@ -86,7 +86,7 @@ export class DownloadManager {
     if (this.currentDownloader === Downloader.RealDebrid) {
       RealDebridDownloader.cancelDownload();
     } else {
-      RPCManager.cancelDownload(gameId);
+      PythonInstance.cancelDownload(gameId);
     }
 
     WindowManager.mainWindow?.setProgressBar(-1);
@@ -98,7 +98,7 @@ export class DownloadManager {
       RealDebridDownloader.startDownload(game);
       this.currentDownloader = Downloader.RealDebrid;
     } else {
-      RPCManager.startDownload(game);
+      PythonInstance.startDownload(game);
       this.currentDownloader = Downloader.Torrent;
     }
   }
diff --git a/src/main/services/download/index.ts b/src/main/services/download/index.ts
index 0298f7d8..26d21799 100644
--- a/src/main/services/download/index.ts
+++ b/src/main/services/download/index.ts
@@ -1,2 +1,2 @@
 export * from "./download-manager";
-export * from "./rpc-manager";
+export * from "./python-instance";
diff --git a/src/main/services/download/rpc-manager.ts b/src/main/services/download/python-instance.ts
similarity index 92%
rename from src/main/services/download/rpc-manager.ts
rename to src/main/services/download/python-instance.ts
index 58cc1daa..c534e41d 100644
--- a/src/main/services/download/rpc-manager.ts
+++ b/src/main/services/download/python-instance.ts
@@ -20,8 +20,8 @@ import {
   ProcessPayload,
 } from "./types";
 
-export class RPCManager {
-  private static rpcProcess: cp.ChildProcess | null = null;
+export class PythonInstance {
+  private static pythonProcess: cp.ChildProcess | null = null;
   private static downloadingGameId = -1;
 
   private static rpc = axios.create({
@@ -32,19 +32,19 @@ export class RPCManager {
   });
 
   public static spawn(args?: StartDownloadPayload) {
-    this.rpcProcess = startRPCClient(args);
+    this.pythonProcess = startRPCClient(args);
   }
 
   public static kill() {
-    if (this.rpcProcess) {
-      this.rpcProcess.kill();
-      this.rpcProcess = null;
+    if (this.pythonProcess) {
+      this.pythonProcess.kill();
+      this.pythonProcess = null;
       this.downloadingGameId = -1;
     }
   }
 
   public static killTorrent() {
-    if (this.rpcProcess) {
+    if (this.pythonProcess) {
       this.rpc.post("/action", { action: "kill-torrent" });
       this.downloadingGameId = -1;
     }
@@ -131,7 +131,7 @@ export class RPCManager {
   }
 
   static async startDownload(game: Game) {
-    if (!this.rpcProcess) {
+    if (!this.pythonProcess) {
       this.spawn({
         game_id: game.id,
         magnet: game.uri!,
diff --git a/src/main/services/process-watcher.ts b/src/main/services/process-watcher.ts
index ec7d8e20..4ee78334 100644
--- a/src/main/services/process-watcher.ts
+++ b/src/main/services/process-watcher.ts
@@ -3,7 +3,7 @@ import { gameRepository } from "@main/repository";
 import { WindowManager } from "./window-manager";
 import { createGame, updateGamePlaytime } from "./library-sync";
 import { GameRunning } from "@types";
-import { RPCManager } from "./download";
+import { PythonInstance } from "./download";
 
 export const gamesPlaytime = new Map<
   number,
@@ -19,7 +19,7 @@ export const watchProcesses = async () => {
   });
 
   if (games.length === 0) return;
-  const processes = await RPCManager.getProcessList();
+  const processes = await PythonInstance.getProcessList();
 
   for (const game of games) {
     const executablePath = game.executablePath!;
diff --git a/torrent-client/downloader.py b/torrent-client/downloader.py
index 2368ff17..142da020 100644
--- a/torrent-client/downloader.py
+++ b/torrent-client/downloader.py
@@ -30,12 +30,13 @@ class Downloader:
             self.torrent_handles[game_id] = None
             self.downloading_game_id = -1
 
-    def cancel_all_downloads(self):
+    def abort_session(self):
         for game_id in self.torrent_handles:
             torrent_handle = self.torrent_handles[game_id]
             torrent_handle.pause()
             self.session.remove_torrent(torrent_handle)
-
+            
+        self.session.abort()
         self.torrent_handles = {}
         self.downloading_game_id = -1
 
diff --git a/torrent-client/main.py b/torrent-client/main.py
index 8f57de8a..004cd108 100644
--- a/torrent-client/main.py
+++ b/torrent-client/main.py
@@ -8,11 +8,12 @@ from downloader import Downloader
 torrent_port = sys.argv[1]
 http_port = sys.argv[2]
 rpc_password = sys.argv[3]
+start_download_payload = sys.argv[4]
 
 downloader = None
 
-if sys.argv[4]:
-    initial_download = json.loads(urllib.parse.unquote(sys.argv[4]))
+if start_download_payload:
+    initial_download = json.loads(urllib.parse.unquote(start_download_payload))
     downloader = Downloader(torrent_port)
     downloader.start_download(initial_download['game_id'], initial_download['magnet'], initial_download['save_path'])
 
@@ -75,7 +76,7 @@ class Handler(BaseHTTPRequestHandler):
             elif data['action'] == 'cancel':
                 downloader.cancel_download(data['game_id'])
             elif data['action'] == 'kill-torrent':
-                downloader.cancel_all_downloads()
+                downloader.abort_session()
                 downloader = None
 
             self.send_response(200)