viber-channel 0.8.39 → 0.8.41

package/viber-codex-bridge.ts

@@ -119,6 +119,7 @@ import {
 import { startInstanceHeartbeat } from "./lib/heartbeat.ts";
 import { type PresenceGate, REASON_NO_RUNNER as REASON_STARTING, startCoverageWatch } from "./lib/coverage_watch.ts";
 import { startPresenceRecord } from "./lib/presence_record.ts";
+import { CodexRuntimeWriter, runtimeFilePath } from "./lib/codex_runtime.ts";
 import { startBridgeToolHost, type BridgeToolHost } from "./lib/bridge_tool_host.ts";
 import type { AgentToolsContext } from "./lib/agent_tools.ts";
 import type { OpenDmResult } from "./lib/peers.ts";
@@ -693,6 +694,14 @@ export class CodexAppServer {
   /** Set true once we intentionally `close()` so the exit handler can tell a
    * deliberate shutdown from an unexpected child death (#298). */
   private closing = false;
+  /** #708: model, effort and context left for the runner. Unset = not recorded. */
+  private runtime?: CodexRuntimeWriter;
+  /** #708: threads this bridge opened — a usage for any other thread is not this agent's. */
+  private readonly ownThreads = new Set<string>();
+
+  setRuntime(writer: CodexRuntimeWriter): void {
+    this.runtime = writer;
+  }
 
   /**
    * @param onUnexpectedExit fired ONLY when the Codex child dies on its own (not
@@ -840,6 +849,8 @@ export class CodexAppServer {
         // leaves this try (it must not fall back to a fresh thread and hide the problem).
         resumeMismatch = pairMismatch(this.pair, resumed, "thread/resume");
         if (resumeMismatch === undefined) {
+          this.ownThreads.add(resumedId);
+          this.runtime?.onThread(resumed);
           logTiming("ensureThread", nowMs() - t0, "outcome=resume");
           return resumedId;
         }
@@ -873,6 +884,8 @@ export class CodexAppServer {
     // #665: same check on a fresh thread — what codex SAYS it runs, not what we asked.
     const startMismatch = pairMismatch(this.pair, started, "thread/start");
     if (startMismatch !== undefined) this.stopOnPairMismatch(startMismatch);
+    this.ownThreads.add(threadId);
+    this.runtime?.onThread(started);
     state.threads[key] = {
       thread_id: threadId,
       conversation_id: options.conversationId,
@@ -998,6 +1011,12 @@ export class CodexAppServer {
       return;
     }
 
+    if (msg.method === "thread/tokenUsage/updated") {
+      const threadId = (msg.params as { threadId?: unknown } | undefined)?.threadId;
+      if (typeof threadId === "string" && this.ownThreads.has(threadId)) this.runtime?.onTokenUsage(msg.params);
+      return;
+    }
+
     if (msg.method === "turn/completed") {
       const params = msg.params as { threadId?: string; turn?: { id?: string; status?: unknown; error?: unknown } };
       const turnId = params.turn?.id;
@@ -1444,6 +1463,8 @@ function setupCodexRuntime(deps: {
   pauseReason: () => string | null;
   /** #665: the template's model/effort, or undefined. */
   pair: CodexPair | undefined;
+  /** #708: this instance's id — the runtime file the runner reads is keyed on it. */
+  instanceKey: string;
 }): { codex: CodexAppServer; toolHost: BridgeToolHost } {
   const { tracker } = deps;
   const bridgeToolsCtx: AgentToolsContext = {
@@ -1481,6 +1502,12 @@ function setupCodexRuntime(deps: {
     { url: toolHost.url, bearerEnvVar: "VIBER_MCP_TOKEN", bearerValue: toolHost.bearerToken },
     deps.pair,
   );
+  try {
+    codex.setRuntime(new CodexRuntimeWriter(runtimeFilePath(IDENTITY_URL, deps.instanceKey, LOCK_DIR()), deps.instanceKey));
+  } catch (err) {
+    process.stderr.write(`${LOG_PREFIX} runtime info disabled: ${safeErrorMessage(err)}
+`);
+  }
   return { codex, toolHost };
 }
 
@@ -1606,6 +1633,7 @@ export async function main(argv: string[] = process.argv.slice(2)): Promise<void
           requestShutdown: helpers.requestShutdown,
           pauseReason: helpers.pauseReason,
           pair,
+          instanceKey: identity.instanceKey,
         });
         codex = rt.codex;
         toolHost = rt.toolHost;
@@ -1674,6 +1702,7 @@ export async function main(argv: string[] = process.argv.slice(2)): Promise<void
       },
       pauseReason: () => gateRef.gate?.reason() ?? REASON_STARTING,
       pair,
+      instanceKey: res.instanceKey,
     });
     codex = rt.codex;
     toolHost = rt.toolHost;