Jensen Zhang
4f241f7d32
- Update CMCLIExecutor to detect cm-cli as executable (not just .py script) - Use shutil.which() to find cm-cli in PATH (supports uv/pip installations) - Fall back to .py script detection for legacy installations - Add comprehensive .gitignore for Python, IDE, and ComfyUI artifacts - Expand AGENTS.md with build commands, code style, and architecture patterns - Remove trailing whitespace across all source files
71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
"""REST API route definitions for cm-cli-rest."""
|
|
|
|
from aiohttp import web
|
|
import logging
|
|
|
|
from .handlers.nodes import NodesHandler
|
|
from .handlers.snapshots import SnapshotsHandler
|
|
from .handlers.jobs import JobsHandler
|
|
from ..cm_cli import CMCLIExecutor
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def setup_routes(app: web.Application, executor: CMCLIExecutor) -> None:
|
|
"""
|
|
Register all REST API routes with the aiohttp application.
|
|
|
|
Args:
|
|
app: aiohttp web application (PromptServer.instance.app)
|
|
executor: CMCLIExecutor instance for command execution
|
|
"""
|
|
# Initialize handlers
|
|
nodes_handler = NodesHandler(executor)
|
|
snapshots_handler = SnapshotsHandler(executor)
|
|
jobs_handler = JobsHandler(executor)
|
|
|
|
# Health check endpoint
|
|
app.router.add_get("/cm-cli-rest/health", health_check)
|
|
|
|
# Node management endpoints
|
|
app.router.add_get("/cm-cli-rest/nodes", nodes_handler.list_nodes)
|
|
app.router.add_get("/cm-cli-rest/nodes/{name}", nodes_handler.get_node)
|
|
app.router.add_post("/cm-cli-rest/nodes/install", nodes_handler.install_node)
|
|
app.router.add_post("/cm-cli-rest/nodes/uninstall", nodes_handler.uninstall_node)
|
|
app.router.add_post("/cm-cli-rest/nodes/update", nodes_handler.update_node)
|
|
app.router.add_post("/cm-cli-rest/nodes/update-all", nodes_handler.update_all_nodes)
|
|
app.router.add_post("/cm-cli-rest/nodes/enable", nodes_handler.enable_node)
|
|
app.router.add_post("/cm-cli-rest/nodes/disable", nodes_handler.disable_node)
|
|
app.router.add_post("/cm-cli-rest/nodes/fix", nodes_handler.fix_node)
|
|
app.router.add_post("/cm-cli-rest/nodes/reinstall", nodes_handler.reinstall_node)
|
|
|
|
# Snapshot endpoints
|
|
app.router.add_post("/cm-cli-rest/snapshots/save", snapshots_handler.save_snapshot)
|
|
app.router.add_post("/cm-cli-rest/snapshots/restore", snapshots_handler.restore_snapshot)
|
|
app.router.add_get("/cm-cli-rest/snapshots", snapshots_handler.list_snapshots)
|
|
|
|
# Job tracking endpoints
|
|
app.router.add_get("/cm-cli-rest/jobs", jobs_handler.list_jobs)
|
|
app.router.add_get("/cm-cli-rest/jobs/{id}", jobs_handler.get_job)
|
|
app.router.add_delete("/cm-cli-rest/jobs/{id}", jobs_handler.cleanup_job)
|
|
|
|
logger.info("cm-cli-rest API routes registered")
|
|
|
|
|
|
async def health_check(request: web.Request) -> web.Response:
|
|
"""
|
|
GET /cm-cli-rest/health
|
|
|
|
Health check endpoint.
|
|
"""
|
|
from datetime import datetime
|
|
|
|
return web.json_response(
|
|
{
|
|
"status": "healthy",
|
|
"service": "cm-cli-rest",
|
|
"version": "0.1.0",
|
|
"timestamp": datetime.now().isoformat(),
|
|
}
|
|
)
|