From e6d03fa7d3d55f3b1d015d686664d945a1e79860 Mon Sep 17 00:00:00 2001
From: zucham <zucha@zumepro.cz>
Date: Sat, 3 May 2025 15:01:58 +0200
Subject: [PATCH] Implement binary message sending

---
 main.py | 29 +++++++++++++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index c1ea64d..5691f1f 100644
--- a/main.py
+++ b/main.py
@@ -7,6 +7,7 @@ from dataclasses import dataclass
 import json
 import httpx
 import asyncio
+import struct
 
 # Some useful variables
 PEEHAITCHPEA_ENDPOINT = "http://localhost:9000/api.php?cmd=getselectedmessage"
@@ -25,6 +26,7 @@ app.state.auto_polling = False
 app.state.polling_rate = 5
 
 # Store for connected websocket clients
+
 class ConnectionManager:
     def __init__(self):
         self.active_connections: List[WebSocket] = []
@@ -41,6 +43,29 @@ class ConnectionManager:
     async def broadcast(self, message: str):
         for connection in self.active_connections:
             await connection.send_text(message)
+    
+    async def broadcast_binary(self, data_dict: dict):
+        """
+        Broadcasts a message to all connections as binary data.
+        """
+        if not self.active_connections:
+            return
+        
+        json_str = json.dumps(data_dict)
+        json_bytes = json_str.encode('utf-8')
+        json_length = len(json_bytes)
+        
+        # 4-byte unsigned integer (uint32)
+        length_bytes = struct.pack('!I', json_length)
+        
+        message_bytes = length_bytes + json_bytes
+        
+        for connection in self.active_connections:
+            try:
+                await connection.send_bytes(message_bytes)
+                logger.debug(f"Sent binary message ({json_length} bytes) to a client")
+            except Exception as e:
+                logger.error(f"Failed to send binary message: {str(e)}")
 
 manager = ConnectionManager()
 
@@ -126,7 +151,7 @@ async def process_subtitles(request: Request, sub_type: str):
         logger.info(f"Received subtitle text: {subtitle_text}, request type: {sub_type}")
         
         if manager.active_connections:
-            await manager.broadcast(json.dumps({"type": f"subtitle_{sub_type}", "text": subtitle_text}))
+            await manager.broadcast_binary({"type": f"subtitle_{sub_type}", "text": subtitle_text})
         
         return JSONResponse(
             status_code=200,
@@ -167,7 +192,7 @@ async def process_selected_message(message: str):
     """
     logger.info(f"Processing message: {message}")
     if manager.active_connections:
-        await manager.broadcast(json.dumps({"type": "selectedmessage", "message": message}))
+        await manager.broadcast_binary({"type": "selectedmessage", "message": message})
 
 async def periodic_message_check():
     """Periodically checks for new messages."""