GameSentenceMiner 2.14.9__py3-none-any.whl → 2.14.10__py3-none-any.whl

GameSentenceMiner/util/communication/__init__.py

@@ -0,0 +1,22 @@
+import os.path
+from dataclasses import dataclass, field
+from typing import Dict, Optional, Any
+
+from dataclasses_json import dataclass_json, Undefined
+from websocket import WebSocket
+
+from GameSentenceMiner.util.configuration import get_app_directory
+
+CONFIG_FILE = os.path.join(get_app_directory(), "shared_config.json")
+websocket: WebSocket = None
+
+@dataclass_json(undefined=Undefined.RAISE)
+@dataclass
+class Message:
+    """
+    Represents a message for inter-process communication.
+    Mimics the structure of IPC or HTTP calls.
+    """
+    function: str
+    data: Dict[str, Any] = field(default_factory=dict)
+    id: Optional[str] = None

GameSentenceMiner/util/communication/send.py

@@ -0,0 +1,7 @@
+import json
+
+from GameSentenceMiner.util.communication.websocket import websocket, Message
+
+async def send_restart_signal():
+    if websocket:
+        await websocket.send(json.dumps(Message(function="restart").to_json()))

GameSentenceMiner/util/communication/websocket.py

@@ -0,0 +1,94 @@
+import asyncio
+import os.path
+
+import websockets
+import json
+from enum import Enum
+
+from websocket import WebSocket
+
+from GameSentenceMiner.util.communication import Message
+from GameSentenceMiner.util.configuration import get_app_directory, logger
+
+CONFIG_FILE = os.path.join(get_app_directory(), "shared_config.json")
+websocket: WebSocket = None
+handle_websocket_message = None
+
+
+class FunctionName(Enum):
+    QUIT = "quit"
+    START = "start"
+    STOP = "stop"
+    QUIT_OBS = "quit_obs"
+    START_OBS = "start_obs"
+    OPEN_SETTINGS = "open_settings"
+    OPEN_TEXTHOOKER = "open_texthooker"
+    OPEN_LOG = "open_log"
+    TOGGLE_REPLAY_BUFFER = "toggle_replay_buffer"
+    RESTART_OBS = "restart_obs"
+    EXIT = "exit"
+    GET_STATUS = "get_status"
+
+
+async def do_websocket_connection(port):
+    """
+    Connects to the WebSocket server running in the Electron app.
+    """
+    global websocket
+
+    uri = f"ws://localhost:{port}"  # Use the port from Electron
+    logger.debug(f"Electron Communication : Connecting to server at {uri}...")
+    try:
+        async with websockets.connect(uri) as websocket:
+            logger.debug(f"Connected to websocket server at {uri}")
+
+            # Send an initial message
+            message = Message(function="on_connect", data={"message": "Hello from Python!"})
+            await websocket.send(message.to_json())
+            logger.debug(f"> Sent: {message}")
+
+            # Receive messages from the server
+            while True:
+                try:
+                    response = await websocket.recv()
+                    if response is None:
+                        break
+                    logger.debug(f"Electron Communication : < Received: {response}")
+                    handle_websocket_message(Message.from_json(response))
+                    await asyncio.sleep(1)  # keep the connection alive
+                except websockets.ConnectionClosedOK:
+                    logger.debug("Electron Communication : Connection closed by server")
+                    break
+                except websockets.ConnectionClosedError as e:
+                    logger.debug(f"Electron Communication : Connection closed with error: {e}")
+                    break
+    except ConnectionRefusedError:
+        logger.debug(f"Electron Communication : Error: Could not connect to server at {uri}.  Electron App not running..")
+    except Exception as e:
+        logger.debug(f"Electron Communication : An error occurred: {e}")
+
+def connect_websocket():
+    """
+    Main function to run the WebSocket client.
+    """
+    # Load the port from the same config.json the Electron app uses
+    try:
+        with open(CONFIG_FILE, "r") as f:
+            config = json.load(f)
+            port = config["port"]
+    except FileNotFoundError:
+        print("Error: shared_config.json not found.  Using default port 8766.  Ensure Electron app creates this file.")
+        port = 8766  # Default port, same as in Electron
+    except json.JSONDecodeError:
+        print("Error: shared_config.json was not valid JSON.  Using default port 8765.")
+        port = 8766
+
+    asyncio.run(do_websocket_connection(port))
+
+def register_websocket_message_handler(handler):
+    global handle_websocket_message
+    handle_websocket_message = handler
+
+
+if __name__ == "__main__":
+    connect_websocket()