feature: add management api for Comfystream mgmt api

nodes/api/__init__.py

@@ -8,6 +8,7 @@
 import aiohttp
 from ..server_manager import LocalComfyStreamServer
 from .. import settings_storage
+import subprocess
 
 routes = None
 server_manager = None
@@ -215,3 +216,45 @@ async def manage_configuration(request):
             logging.error(f"Error managing configuration: {str(e)}")
             return web.json_response({"error": str(e)}, status=500)
 
+    @routes.post('/comfystream/settings/manage')
+    async def manage_comfystream(request):
+        """Manage ComfyStream server settings"""
+        #check if server is running
+        server_status = server_manager.get_status()
+        if not server_status["running"]:
+            return web.json_response({"error": "ComfyStream Server is not running"}, status=503)
+
+        try:
+            data = await request.json()
+            action_type = data.get("action_type")
+            action = data.get("action")
+            payload = data.get("payload")
+            url_host = server_status.get("host", "localhost")
+            url_port = server_status.get("port", "8889")
+            mgmt_url = f"http://{url_host}:{url_port}/settings/{action_type}/{action}"
+
+            async with aiohttp.ClientSession() as session:
+                async with session.post(
+                    mgmt_url,
+                    json=payload,
+                    headers={"Content-Type": "application/json"}
+                ) as response:
+                    if not response.ok:
+                        return web.json_response(
+                            {"error": f"Server error: {response.status}"}, 
+                            status=response.status
+                        )
+                    return web.json_response(await response.json())
+        except Exception as e:
+            logging.error(f"Error managing ComfyStream: {str(e)}")
+            return web.json_response({"error": str(e)}, status=500)
+
+    @routes.post('/comfyui/restart')
+    async def manage_configuration(request):
+        server_status = server_manager.get_status()
+        if server_status["running"]:
+            await server_manager.stop()
+        logging.info("Restarting ComfyUI...")
+        subprocess.run(["supervisorctl", "restart", "comfyui"])
+        logging.info("Restarting ComfyUI...in process")
+        return web.json_response({"success": True}, status=200)

nodes/web/js/launcher.js

@@ -98,6 +98,38 @@ document.addEventListener('comfy-extension-registered', (event) => {
     }
 });
 
+async function restartComfyUI() {
+    try {
+        const response = await fetch('/comfyui/restart', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                'Accept': 'application/json'
+            },
+            body: "" // No body needed
+        });
+
+        if (!response.ok) {
+            const errorText = await response.text();
+            console.error("[ComfyStream] ComfyUI restart returned error response:", response.status, errorText);
+            try {
+                const errorData = JSON.parse(errorText);
+                throw new Error(errorData.error || `Server error: ${response.status}`);
+            } catch (e) {
+                throw new Error(`Server error: ${response.status} - ${errorText}`);
+            }
+        }
+
+        const data = await response.json();
+
+        return data;
+    } catch (error) {
+        console.error('[ComfyStream] Error restarting ComfyUI:', error);
+        app.ui.dialog.show('Error', error.message || 'Failed to restart ComfyUI');
+        throw error;
+    }
+}
+
 async function controlServer(action) {
     try {
         // Get settings from the settings manager
@@ -256,6 +288,12 @@ const extension = {
             icon: "pi pi-cog",
             label: "Server Settings",
             function: openSettings
+        },
+        {
+            id: "ComfyStream.RestartComfyUI",
+            icon: "pi pi-refresh",
+            label: "Restart ComfyUI",
+            function: restartComfyUI
         }
     ],
 
@@ -270,7 +308,9 @@ const extension = {
                 "ComfyStream.StopServer", 
                 "ComfyStream.RestartServer",
                 null, // Separator
-                "ComfyStream.Settings"
+                "ComfyStream.Settings",
+                null, // Separator
+                "ComfyStream.RestartComfyUI"
             ]
         }
     ],
@@ -300,6 +340,8 @@ const extension = {
             comfyStreamMenu.addItem("Restart Server", () => controlServer('restart'), { icon: "pi pi-refresh" });
             comfyStreamMenu.addSeparator();
             comfyStreamMenu.addItem("Server Settings", openSettings, { icon: "pi pi-cog" });
+            comfyStreamMenu.addSeparator();
+            comfyStreamMenu.addItem("Restart ComfyUI", () => restartComfyUI(), { icon: "pi pi-refresh" });
         }
         // New menu system is handled automatically by the menuCommands registration