cli-wechat-bridge 1.1.7 → 1.1.8

package/dist/companion/codex-visible-client-link.js

@@ -23,12 +23,37 @@ export function parseCodexVisibleSwitchRequest(value) {
         threadId: value.threadId,
     };
 }
+export function parseCodexVisibleControlRequest(value) {
+    const switchRequest = parseCodexVisibleSwitchRequest(value);
+    if (switchRequest) {
+        return switchRequest;
+    }
+    if (!isRecord(value)) {
+        return null;
+    }
+    if (value.type !== "shutdown" ||
+        typeof value.id !== "string" ||
+        typeof value.token !== "string") {
+        return null;
+    }
+    return {
+        type: "shutdown",
+        id: value.id,
+        token: value.token,
+    };
+}
 export function sendCodexVisibleSwitchResponse(socket, response) {
     if (socket.destroyed || socket.writableEnded) {
         return;
     }
     socket.end(`${JSON.stringify(response)}\n`);
 }
+export function sendCodexVisibleShutdownResponse(socket, response) {
+    if (socket.destroyed || socket.writableEnded) {
+        return;
+    }
+    socket.end(`${JSON.stringify(response)}\n`);
+}
 function resolveCodexControlEndpoint(cwd, instanceId) {
     const endpoint = readLocalCompanionEndpoint(cwd, { adapter: "codex" });
     if (!endpoint || endpoint.kind !== "codex" || endpoint.instanceId !== instanceId) {
@@ -120,3 +145,79 @@ export async function requestCodexVisibleThreadSwitch(params) {
         });
     });
 }
+export async function requestCodexVisibleClientShutdown(params) {
+    const endpoint = resolveCodexControlEndpoint(params.cwd, params.instanceId);
+    const requestId = crypto.randomUUID();
+    const timeoutMs = params.timeoutMs ?? 5_000;
+    await new Promise((resolve, reject) => {
+        const socket = net.connect({
+            host: CODEX_VISIBLE_CONTROL_HOST,
+            port: endpoint.codexControlPort,
+        });
+        let buffer = "";
+        let settled = false;
+        const finish = (error) => {
+            if (settled) {
+                return;
+            }
+            settled = true;
+            socket.destroy();
+            if (error) {
+                reject(error);
+            }
+            else {
+                resolve();
+            }
+        };
+        socket.setEncoding("utf8");
+        socket.setTimeout(timeoutMs);
+        socket.once("connect", () => {
+            const request = {
+                type: "shutdown",
+                id: requestId,
+                token: endpoint.codexControlToken,
+            };
+            socket.write(`${JSON.stringify(request)}\n`);
+        });
+        socket.on("data", (chunk) => {
+            buffer += chunk;
+            if (buffer.length > 65_536) {
+                finish(new Error("The visible Codex client returned an oversized shutdown response."));
+                return;
+            }
+            const newlineIndex = buffer.indexOf("\n");
+            if (newlineIndex < 0) {
+                return;
+            }
+            const line = buffer.slice(0, newlineIndex).trim();
+            let parsed;
+            try {
+                parsed = JSON.parse(line);
+            }
+            catch {
+                finish(new Error("The visible Codex client returned an invalid shutdown response."));
+                return;
+            }
+            if (!isRecord(parsed) || parsed.type !== "shutdown_result" || parsed.id !== requestId) {
+                finish(new Error("The visible Codex client returned an unexpected shutdown response."));
+                return;
+            }
+            if (parsed.ok !== true) {
+                finish(new Error(typeof parsed.error === "string"
+                    ? parsed.error
+                    : "The visible Codex client could not shut down."));
+                return;
+            }
+            finish();
+        });
+        socket.once("timeout", () => {
+            finish(new Error("Timed out waiting for the visible Codex client to shut down."));
+        });
+        socket.once("error", (error) => finish(error));
+        socket.once("close", () => {
+            if (!settled) {
+                finish(new Error("The visible Codex client closed the shutdown connection."));
+            }
+        });
+    });
+}

package/dist/companion/local-companion-start.js

@@ -39,8 +39,13 @@ export function formatSwitchFailureMessage(cwd) {
 export function formatRestartUnhealthyMessage(cwd) {
     return `Detected unhealthy companion state for ${cwd}. Restarting bridge...`;
 }
-export function defaultSessionStartMode(adapter) {
-    return adapter === "codex" ? "restore" : "new";
+export function defaultSessionStartMode(_adapter) {
+    // Codex 0.155+ persists app-server threads lazily, so launching the visible
+    // client with a freshly prepared (not yet persisted) thread id crashes its
+    // bootstrap with "no rollout found". Start every adapter fresh by default;
+    // explicit --session-start-mode restore remains available for deliberate
+    // restores of already persisted threads.
+    return "new";
 }
 export function decideLaunchAction(input) {
     if (input.lockShouldAutoReclaim) {

package/dist/companion/local-companion.js

@@ -231,6 +231,23 @@ export async function runLocalCompanion(options) {
                     publishState();
                     sendResponse(socket, message.id, true);
                     break;
+                case "list_models":
+                    if (!adapter.listModels)
+                        throw new Error("Model listing is not supported by this companion.");
+                    sendResponse(socket, message.id, true, await adapter.listModels());
+                    break;
+                case "select_model":
+                    if (!adapter.selectModel)
+                        throw new Error("Model selection is not supported by this companion.");
+                    sendResponse(socket, message.id, true, await adapter.selectModel(message.payload.modelId));
+                    publishState();
+                    break;
+                case "set_plan_mode":
+                    if (!adapter.setPlanMode)
+                        throw new Error("Plan mode is not supported by this companion.");
+                    sendResponse(socket, message.id, true, await adapter.setPlanMode(message.payload.enabled));
+                    publishState();
+                    break;
                 case "interrupt":
                     sendResponse(socket, message.id, true, await adapter.interrupt());
                     break;

package/dist/companion/pi-tui-bridge-extension.js

@@ -167,7 +167,7 @@ export default function piTuiBridgeExtension(pi) {
             }
         },
     });
-    const executeCommand = (command) => {
+    const executeCommand = async (command) => {
         try {
             if (command.type === "prompt" && typeof command.text === "string") {
                 if (!latestContext?.isIdle()) {
@@ -180,6 +180,59 @@ export default function piTuiBridgeExtension(pi) {
                 latestContext?.abort();
                 writeFrame({ type: "response", id: command.id, success: true });
             }
+            else if (command.type === "list_models") {
+                if (!latestContext) {
+                    throw new Error("Pi session context is not ready.");
+                }
+                const models = latestContext.scopedModels.length > 0
+                    ? latestContext.scopedModels.map((entry) => entry.model)
+                    : latestContext.modelRegistry.getAvailable();
+                writeFrame({
+                    type: "response",
+                    id: command.id,
+                    success: true,
+                    data: {
+                        models: models.map((model) => ({
+                            id: `${model.provider}/${model.id}`,
+                            displayName: model.name
+                                ? `${model.name} (${model.provider})`
+                                : `${model.id} (${model.provider})`,
+                            isCurrent: latestContext?.model?.provider === model.provider &&
+                                latestContext.model.id === model.id,
+                        })),
+                    },
+                });
+            }
+            else if (command.type === "select_model" && typeof command.modelId === "string") {
+                if (!latestContext?.isIdle()) {
+                    throw new Error("Pi TUI is already processing a turn.");
+                }
+                const models = latestContext.scopedModels.length > 0
+                    ? latestContext.scopedModels.map((entry) => entry.model)
+                    : latestContext.modelRegistry.getAvailable();
+                const model = models.find((candidate) => `${candidate.provider}/${candidate.id}` === command.modelId);
+                if (!model) {
+                    throw new Error(`Pi model is no longer available: ${command.modelId}`);
+                }
+                const selected = await pi.setModel(model);
+                if (!selected) {
+                    throw new Error(`Pi could not switch to model: ${command.modelId}`);
+                }
+                writeFrame({
+                    type: "response",
+                    id: command.id,
+                    success: true,
+                    data: {
+                        model: {
+                            id: `${model.provider}/${model.id}`,
+                            displayName: model.name
+                                ? `${model.name} (${model.provider})`
+                                : `${model.id} (${model.provider})`,
+                            isCurrent: true,
+                        },
+                    },
+                });
+            }
             else if (command.type === "new_session" && typeof command.id === "string") {
                 pi.sendUserMessage(`/${INTERNAL_NEW_COMMAND} ${command.id}`, {
                     expandPromptTemplates: true,
@@ -245,7 +298,7 @@ export default function piTuiBridgeExtension(pi) {
             try {
                 const parsed = JSON.parse(line);
                 if (isRecord(parsed)) {
-                    dispatchCommand(parsed);
+                    void dispatchCommand(parsed);
                 }
             }
             catch {
@@ -260,7 +313,7 @@ export default function piTuiBridgeExtension(pi) {
         latestContext = context;
         writeSessionState(context, typeof event.reason === "string" ? event.reason : "startup");
         for (const command of queuedCommands.splice(0)) {
-            executeCommand(command);
+            void executeCommand(command);
         }
     });
     pi.on("input", (event, context) => {

package/dist/core/bridge-control.js

@@ -26,6 +26,14 @@ export function parseBridgeControlCommand(text) {
             return { type: "deny" };
         case "/answer":
             return argument ? { type: "answer", raw: argument } : null;
+        case "/model":
+            return { type: "model", target: argument || undefined };
+        case "/plan":
+            if (!argument || argument.toLowerCase() === "on")
+                return { type: "plan", enabled: true };
+            if (argument.toLowerCase() === "off")
+                return { type: "plan", enabled: false };
+            return null;
         default:
             return null;
     }

package/dist/core/conversation-routing.js

@@ -0,0 +1,48 @@
+import { AsyncLocalStorage } from "node:async_hooks";
+/**
+ * Keeps the reply target scoped to one inbound async call chain. Different
+ * conversations may run concurrently without overwriting each other's target.
+ */
+export class InboundConversationContext {
+    storage = new AsyncLocalStorage();
+    run(conversation, callback) {
+        return this.storage.run(conversation, callback);
+    }
+    get() {
+        return this.storage.getStore();
+    }
+}
+export function tryBeginTurn(state, task, conversation) {
+    if (state.activeTask) {
+        return null;
+    }
+    const lease = {
+        task,
+        previousConversation: state.activeConversation,
+    };
+    state.activeTask = task;
+    if (conversation) {
+        state.activeConversation = conversation;
+        state.lastConversation = conversation;
+    }
+    return lease;
+}
+export function rollbackTurn(state, lease) {
+    if (state.activeTask !== lease.task) {
+        return false;
+    }
+    state.activeTask = null;
+    state.activeConversation = lease.previousConversation;
+    return true;
+}
+export function clearTurn(state, expectedTask) {
+    if (expectedTask && state.activeTask !== expectedTask) {
+        return false;
+    }
+    state.activeTask = null;
+    state.activeConversation = undefined;
+    return true;
+}
+export function resolveConversationTarget(params) {
+    return params.active ?? params.last ?? params.fallback;
+}