feat: add server state

Author: sansyrox

starfyre/js/store.py

@@ -134,3 +134,47 @@ def get_signal(*args, **kwargs):
         return store.get(id, initial_state)
 
     return [use_signal, set_signal, get_signal]
+
+
+import json
+import random
+
+# Assuming connection is already established
+connection = js.WebSocket("ws://localhost:8765")
+
+def server_state():
+    id = random.randint(0, 100000)
+
+    def use_server_signal(element=None):
+        nonlocal id
+        if element:
+            # Send a request to register observer
+            message = json.dumps({"action": "register", "id": id, "element": element})
+            connection.send(message)
+
+        # Request current state for this id
+        connection.send(json.dumps({"action": "get_state", "id": id}))
+
+        # Returning initial state or undefined, as actual state will be set on server response
+        return None
+
+    def set_server_signal(state):
+        nonlocal id
+        # Send the updated state to server
+        connection.send(json.dumps({"action": "set_state", "id": id, "state": state}))
+
+    def get_server_signal():
+        nonlocal id
+        # Request current state for this id
+        connection.send(json.dumps({"action": "get_state", "id": id}))
+        # Returning initial state or undefined, as actual state will be set on server response
+        return None
+
+    return [use_server_signal, set_server_signal, get_server_signal]
+
+# Handle incoming WebSocket messages
+def on_message(event):
+    data = json.loads(event.data)
+    # Process the data, update local state, notify observers, etc.
+
+connection.onmessage = on_message