From 69fd2e03cac3a744dbe647750e2e33091d57cc53 Mon Sep 17 00:00:00 2001
From: RyanOnTheInside <7623207+ryanontheinside@users.noreply.github.com>
Date: Tue, 22 Apr 2025 11:32:17 -0400
Subject: [PATCH] routes for managing state

---
 server.py | 68 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 server.py

diff --git a/server.py b/server.py
new file mode 100644
index 0000000..a6b80b9
--- /dev/null
+++ b/server.py
@@ -0,0 +1,68 @@
+"""Server routes for the RealtimeNodes extension"""
+import json
+from aiohttp import web
+from server import PromptServer
+
+# Dictionary to store workflow-specific keys
+# Format: { workflow_id: { key: value, ... }, ... }
+workflow_states = {}
+
+@PromptServer.instance.routes.get('/realtimenodes/state/{workflow_id}')
+async def get_workflow_state(request):
+    """Get the current state for a specific workflow"""
+    workflow_id = request.match_info['workflow_id']
+    state = workflow_states.get(workflow_id, {})
+    return web.json_response(state)
+
+@PromptServer.instance.routes.get('/realtimenodes/state/{workflow_id}/{key}')
+async def get_state_value(request):
+    """Get a specific state value for a workflow"""
+    workflow_id = request.match_info['workflow_id']
+    key = request.match_info['key']
+    
+    workflow_state = workflow_states.get(workflow_id, {})
+    value = workflow_state.get(key, None)
+    
+    return web.json_response({"value": value})
+
+@PromptServer.instance.routes.post('/realtimenodes/state/{workflow_id}/{key}')
+async def set_state_value(request):
+    """Set a specific state value for a workflow"""
+    workflow_id = request.match_info['workflow_id']
+    key = request.match_info['key']
+    
+    # Get request body
+    body = await request.json()
+    value = body.get("value")
+    
+    # Ensure workflow exists in states
+    if workflow_id not in workflow_states:
+        workflow_states[workflow_id] = {}
+    
+    # Store the value
+    workflow_states[workflow_id][key] = value
+    
+    return web.json_response({"success": True})
+
+@PromptServer.instance.routes.delete('/realtimenodes/state/{workflow_id}/{key}')
+async def delete_state_value(request):
+    """Delete a specific state value for a workflow"""
+    workflow_id = request.match_info['workflow_id']
+    key = request.match_info['key']
+    
+    if workflow_id in workflow_states and key in workflow_states[workflow_id]:
+        del workflow_states[workflow_id][key]
+        return web.json_response({"success": True})
+    
+    return web.json_response({"success": False, "error": "Key not found"}, status=404)
+
+@PromptServer.instance.routes.delete('/realtimenodes/state/{workflow_id}')
+async def clear_workflow_state(request):
+    """Clear all state for a specific workflow"""
+    workflow_id = request.match_info['workflow_id']
+    
+    if workflow_id in workflow_states:
+        workflow_states[workflow_id] = {}
+        return web.json_response({"success": True})
+    
+    return web.json_response({"success": False, "error": "Workflow not found"}, status=404) 
\ No newline at end of file