om-memory-system 3.6.0-next.39 → 3.6.0-next.42

package/dist/services/web-server.js

@@ -152,6 +152,8 @@ export class WebServer {
     onTakeoverCallback = null;
     onPortsExhaustedCallback = null;
     onStepAsideCallback = null;
+    onPowerActionCallback = null;
+    powerActionRunning = false;
     stepAsideTimer = null;
     holdOffTimer = null;
     portsExhaustedNotified = false;
@@ -211,6 +213,42 @@ export class WebServer {
     setOnStepAside(callback) {
         this.onStepAsideCallback = callback;
     }
+    /** Standalone web apps register this to stop or restart from the page. */
+    setOnPowerAction(callback) {
+        this.onPowerActionCallback = callback;
+    }
+    handlePowerAction(action, remoteAddress) {
+        const record = (outcome) => log("Web server power request", { action, outcome, ownVersion: packageVersion() });
+        if (!isLoopbackAddress(remoteAddress)) {
+            record("refused_not_loopback");
+            return this.jsonResponse({ success: false, error: "Loopback caller required" }, 403);
+        }
+        const callback = this.onPowerActionCallback;
+        if (!callback) {
+            record("unsupported");
+            return this.jsonResponse({ success: false, error: "Not supported by this web app" }, 409);
+        }
+        // A repeated request must not start a second stop or restart while one runs.
+        if (this.powerActionRunning) {
+            record("already_running");
+            return this.jsonResponse({ success: true }, 202);
+        }
+        this.powerActionRunning = true;
+        record(action === "stop" ? "stopping" : "restarting");
+        // Let the 202 reply reach the caller before the web app goes down.
+        setTimeout(async () => {
+            try {
+                await callback(action);
+            }
+            catch (error) {
+                log("Power action callback error", { action, error: String(error) });
+            }
+            finally {
+                this.powerActionRunning = false;
+            }
+        }, STEP_ASIDE_REPLY_GRACE_MS).unref();
+        return this.jsonResponse({ success: true }, 202);
+    }
     scheduleStepAside() {
         if (this.stepAsideTimer)
             return;
@@ -281,6 +319,10 @@ export class WebServer {
                 fetch: this.handleRequest.bind(this),
             });
             this.isOwner = true;
+            // A restart copy is named in the start lock until it owns the port.
+            void import("./web-ensure.js")
+                .then(({ removeStartLockFor }) => removeStartLockFor(process.pid))
+                .catch(() => { });
             // This process serves the Claude Code hooks, so it also retries their failed captures.
             void import("../importer/claude-hook-api.js")
                 .then(({ startClaudeCodeWorker }) => startClaudeCodeWorker())
@@ -477,6 +519,13 @@ export class WebServer {
                         callerVersion: "",
                     });
                 }
+                if (method === "POST" && (path === "/api/web/stop" || path === "/api/web/restart")) {
+                    log("Web server power request", {
+                        action: path === "/api/web/stop" ? "stop" : "restart",
+                        outcome: "refused_auth",
+                        ownVersion: packageVersion(),
+                    });
+                }
                 return (configuredTokenFailure ??
                     this.jsonResponse({ success: false, error: "Unauthorized" }, 401));
             }
@@ -542,6 +591,15 @@ export class WebServer {
                         ? controls.resume(host, cwd)
                         : controls.runNow(host, cwd));
             }
+            if (path === "/api/web/status" && method === "GET") {
+                return this.jsonResponse({
+                    version: packageVersion(),
+                    canControl: isLoopbackAddress(remoteAddress) && this.onPowerActionCallback !== null,
+                });
+            }
+            if ((path === "/api/web/stop" || path === "/api/web/restart") && method === "POST") {
+                return this.handlePowerAction(path === "/api/web/stop" ? "stop" : "restart", remoteAddress);
+            }
             if (path === "/api/web/step-aside" && method === "POST") {
                 const body = await req.json().catch(() => null);
                 return this.handleStepAside(remoteAddress, body);