From 7fa50dc5a7423f9b2d1d8696281a3477ded7a2ad Mon Sep 17 00:00:00 2001
From: Chubby Granny Chaser <hydralauncher@proton.me>
Date: Mon, 3 Nov 2025 12:02:52 +0000
Subject: [PATCH] feat: adding ws client back

---
 src/main/main.ts               | 3 ++-
 src/main/services/hydra-api.ts | 5 +++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/main/main.ts b/src/main/main.ts
index ffb8f8a9..6ae927a5 100644
--- a/src/main/main.ts
+++ b/src/main/main.ts
@@ -16,6 +16,7 @@ import {
   Ludusavi,
   Lock,
   DeckyPlugin,
+  WSClient,
 } from "@main/services";
 import { migrateDownloadSources } from "./helpers/migrate-download-sources";
 
@@ -56,7 +57,7 @@ export const loadState = async () => {
 
     const { syncDownloadSourcesFromApi } = await import("./services/user");
     void syncDownloadSourcesFromApi();
-    // WSClient.connect();
+    WSClient.connect();
   });
 
   const downloads = await downloadsSublevel
diff --git a/src/main/services/hydra-api.ts b/src/main/services/hydra-api.ts
index 12090df3..f77a152f 100644
--- a/src/main/services/hydra-api.ts
+++ b/src/main/services/hydra-api.ts
@@ -11,6 +11,7 @@ import { getUserData } from "./user/get-user-data";
 import { db } from "@main/level";
 import { levelKeys } from "@main/level/sublevels";
 import type { Auth, User } from "@types";
+import { WSClient } from "./ws";
 
 export interface HydraApiOptions {
   needsAuth?: boolean;
@@ -103,8 +104,8 @@ export class HydraApi {
       await clearGamesRemoteIds();
       uploadGamesBatch();
 
-      // WSClient.close();
-      // WSClient.connect();
+      WSClient.close();
+      WSClient.connect();
 
       const { syncDownloadSourcesFromApi } = await import("./user");
       syncDownloadSourcesFromApi();