From 1921e7ec56b39d4a6af111a75cf74c495800abc6 Mon Sep 17 00:00:00 2001
From: AAGaming <aa@mail.catvibers.me>
Date: Mon, 10 Jul 2023 18:41:56 -0400
Subject: JS -> Python WS now functional

---
 backend/decky_loader/loader.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

(limited to 'backend/decky_loader/loader.py')

diff --git a/backend/decky_loader/loader.py b/backend/decky_loader/loader.py
index 85b5de58..8721ea05 100644
--- a/backend/decky_loader/loader.py
+++ b/backend/decky_loader/loader.py
@@ -70,7 +70,7 @@ class FileChangeHandler(RegexMatchingEventHandler):
         self.maybe_reload(src_path)
 
 class Loader:
-    def __init__(self, server_instance: PluginManager, plugin_path: str, loop: AbstractEventLoop, live_reload: bool = False) -> None:
+    def __init__(self, server_instance: PluginManager, ws: WSRouter, plugin_path: str, loop: AbstractEventLoop, live_reload: bool = False) -> None:
         self.loop = loop
         self.logger = getLogger("Loader")
         self.plugin_path = plugin_path
@@ -88,10 +88,7 @@ class Loader:
             self.observer.start()
             self.loop.create_task(self.enable_reload_wait())
 
-        self.ws = WSRouter()
-
-        server_instance.web_app.add_routes([
-            web.get("/ws", self.ws.handle),
+        server_instance.add_routes([
             web.get("/frontend/{path:.*}", self.handle_frontend_assets),
             web.get("/locales/{path:.*}", self.handle_frontend_locales),
             web.get("/plugins", self.get_plugins),
@@ -101,6 +98,15 @@ class Loader:
             web.post("/plugins/{plugin_name}/reload", self.handle_backend_reload_request)
         ])
 
+        ws.add_route("test", self.test_method)
+
+    async def test_method():
+        await sleep(2)
+
+        return {
+            "test data": True
+        }
+
     async def enable_reload_wait(self):
         if self.live_reload:
             await sleep(10)
-- 
cgit v1.2.3