@moxt-ai/mobius 0.0.292 → 0.0.293

package/dist/runtime.js

@@ -22572,7 +22572,7 @@ import { fetch as fetch3 } from "undici";
 // package.json
 var package_default = {
   name: "@moxt-ai/mobius",
-  version: "0.0.292",
+  version: "0.0.293",
   description: "Mobius CLI for pairing and running local AI agent services.",
   type: "module",
   bin: {
@@ -86536,6 +86536,50 @@ var ExecutionReceipts = class {
   };
 };
 
+// src/session/turn/runtime-turn-activity.ts
+var RuntimeTurnActivity = class {
+  #turns = /* @__PURE__ */ new Set();
+  #admission = { kind: "open" };
+  acceptsPrompts = () => this.#admission.kind !== "reserved";
+  count = () => this.#turns.size;
+  start = (turn) => {
+    this.#turns.add(turn);
+  };
+  finish = (turn) => {
+    this.#turns.delete(turn);
+    if (this.#turns.size === 0 && this.#admission.kind === "waiting") {
+      this.#admission.reserve();
+    }
+  };
+  reserveWhenIdle = (signal) => {
+    signal.throwIfAborted();
+    if (this.#admission.kind !== "open") {
+      throw new Error("A daemon upgrade already owns admission");
+    }
+    return new Promise((resolve18, reject2) => {
+      const abort = () => {
+        this.#admission = { kind: "open" };
+        signal.removeEventListener("abort", abort);
+        reject2(signal.reason);
+      };
+      const reserve = () => {
+        this.#admission = { kind: "reserved" };
+        signal.removeEventListener("abort", abort);
+        resolve18({
+          [Symbol.dispose]: () => {
+            this.#admission = { kind: "open" };
+          }
+        });
+      };
+      this.#admission = { kind: "waiting", reserve };
+      signal.addEventListener("abort", abort, { once: true });
+      if (this.#turns.size === 0) {
+        reserve();
+      }
+    });
+  };
+};
+
 // src/session/turn/active-turn.ts
 import { randomUUID as randomUUID14 } from "node:crypto";
 var ActiveRuntimeTurn = class {
@@ -86752,9 +86796,12 @@ var SessionExecutionState = class {
 var SessionTurns = class {
   #bySessionId = /* @__PURE__ */ new Map();
   #operations;
-  constructor(operations) {
+  #activity;
+  constructor(operations, activity2 = new RuntimeTurnActivity()) {
     this.#operations = operations;
+    this.#activity = activity2;
   }
+  acceptsPrompts = () => this.#activity.acceptsPrompts();
   session = (sessionId) => {
     let session = this.#bySessionId.get(sessionId);
     if (session === void 0) {
@@ -86764,10 +86811,15 @@ var SessionTurns = class {
     return session;
   };
   get = (commandId, sessionId) => this.#bySessionId.get(sessionId)?.get(commandId);
-  start = (command, attemptId) => this.session(command.sessionId).start(command, attemptId);
+  start = (command, attemptId) => {
+    const turn = this.session(command.sessionId).start(command, attemptId);
+    this.#activity.start(turn);
+    return turn;
+  };
   finish = (turn) => {
     this.#bySessionId.get(turn.command.sessionId)?.finish(turn);
     this.sweep();
+    this.#activity.finish(turn);
   };
   recordEvent = (event) => "sessionId" in event ? this.#bySessionId.get(event.sessionId)?.recordEvent(event) ?? event : event;
   recordSessionEvent = (event) => {
@@ -87009,7 +87061,7 @@ function buildSessionLayer(options) {
   const sessionStore = new SqliteSessionStore(new AcpSessionTable(sessionDatabase), events);
   const externalContent = new LocalExternalContentStore(stateDirectory);
   const operations = new ExecutionReceipts();
-  const activeSessions = new SessionTurns(operations);
+  const activeSessions = new SessionTurns(operations, options.turnActivity);
   const directories = new WorkspaceDirectoryService(bindingId, defaultWorkspace);
   const manager = new SessionManager2({
     observer: new SessionLayerObserver((event) => eventProjection.record(event), sink.publishInfo),
@@ -89070,6 +89122,16 @@ var sessionPromptCommand = {
     if (message.kind !== SESSION_PROMPT_KIND) {
       throw new Error("Command kind mismatch");
     }
+    if (!context.sessionTurns.acceptsPrompts()) {
+      send(context.socket, {
+        code: "runtime_upgrading",
+        requestId: message.commandId,
+        kind: "session.operation.failed",
+        message: "The Runtime is upgrading. Retry after it reconnects.",
+        sessionId: message.sessionId
+      });
+      return;
+    }
     const receipt = context.operations.claimExecution(
       message.commandId,
       message.sessionId,
@@ -89875,6 +89937,7 @@ async function runRelayConnection({
         reject2(failure2);
       };
       const stop = () => {
+        publisher.flush();
         diagnostics.connectionClosed(connectionEpoch, 1e3, "Daemon stopped", "daemon_stop");
         closeRelaySocket(socket, 1e3, "Daemon stopped");
         resolveConnection();
@@ -90030,6 +90093,11 @@ var RuntimeSessionEventPublisher = class {
     this.#socket = null;
     this.dropPendingEvents();
   };
+  flush = () => {
+    if (this.#socket !== null) {
+      this.flushEvents(this.#socket);
+    }
+  };
   publishOperationFailure = (sessionId, requestId, code, message) => {
     this.#send({ kind: "session.operation.failed", sessionId, requestId, code, message });
   };
@@ -90163,6 +90231,7 @@ async function runRuntime(config2, parentSignal, timer = systemTimer, options =
   );
   const publisher = new RuntimeSessionEventPublisher(timer);
   const layer = buildSessionLayer({
+    ...options.turnActivity === void 0 ? {} : { turnActivity: options.turnActivity },
     bindingId: config2.bindingId,
     channelId: config2.channelId,
     defaultWorkspace: config2.defaultWorkspace,
@@ -90406,6 +90475,7 @@ var MobiusRuntimeEngine = class {
     this.#state = new RunningMobiusRuntimeState();
     try {
       await runRuntime(this.#config, signal, void 0, {
+        ...this.#options.turnActivity === void 0 ? {} : { turnActivity: this.#options.turnActivity },
         ...this.#options.onConnectionStatus === void 0 ? {} : { onConnectionStatus: this.#options.onConnectionStatus },
         ...this.#options.reportUpgradeAvailability === void 0 ? {} : { reportUpgradeAvailability: this.#options.reportUpgradeAvailability },
         ...this.#options.processLogsDirectory === void 0 ? {} : { processLogsDirectory: this.#options.processLogsDirectory }