Robert Wojciechowski
3058b11ff2
- Remove DistributedListSplitter and DistributedListCollector nodes - Remove /distributed/init_list_queue and /distributed/request_list_item endpoints - Clean up all references from prompt_transform, queue_orchestration, constants, and tests - Refactor and reorganize distributed modules for quality improvements
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
from aiohttp import web
|
|
import server
|
|
|
|
from .endpoint_policy import run_authorized_endpoint
|
|
from ..utils.cloudflare import cloudflare_tunnel_manager
|
|
from ..utils.config import load_config
|
|
|
|
|
|
@server.PromptServer.instance.routes.get("/distributed/tunnel/status")
|
|
async def tunnel_status_endpoint(request: web.Request) -> web.StreamResponse:
|
|
"""Return Cloudflare tunnel status and last known details."""
|
|
|
|
async def _operation() -> web.StreamResponse:
|
|
status = cloudflare_tunnel_manager.get_status()
|
|
config = load_config()
|
|
master_host = (config.get("master") or {}).get("host")
|
|
return web.json_response(
|
|
{
|
|
"status": "success",
|
|
"tunnel": status,
|
|
"master_host": master_host,
|
|
}
|
|
)
|
|
|
|
return await run_authorized_endpoint(request, _operation)
|
|
|
|
|
|
@server.PromptServer.instance.routes.post("/distributed/tunnel/start")
|
|
async def tunnel_start_endpoint(request: web.Request) -> web.StreamResponse:
|
|
"""Start a Cloudflare tunnel pointing at the current ComfyUI server."""
|
|
|
|
async def _operation() -> web.StreamResponse:
|
|
result = await cloudflare_tunnel_manager.start_tunnel()
|
|
config = load_config()
|
|
return web.json_response(
|
|
{
|
|
"status": "success",
|
|
"tunnel": result,
|
|
"master_host": (config.get("master") or {}).get("host"),
|
|
}
|
|
)
|
|
|
|
return await run_authorized_endpoint(request, _operation)
|
|
|
|
|
|
@server.PromptServer.instance.routes.post("/distributed/tunnel/stop")
|
|
async def tunnel_stop_endpoint(request: web.Request) -> web.StreamResponse:
|
|
"""Stop the managed Cloudflare tunnel if running."""
|
|
|
|
async def _operation() -> web.StreamResponse:
|
|
result = await cloudflare_tunnel_manager.stop_tunnel()
|
|
config = load_config()
|
|
return web.json_response(
|
|
{
|
|
"status": "success",
|
|
"tunnel": result,
|
|
"master_host": (config.get("master") or {}).get("host"),
|
|
}
|
|
)
|
|
|
|
return await run_authorized_endpoint(request, _operation)
|