cli-wechat-bridge 1.1.6 → 1.1.8

package/dist/companion/local-companion.js

@@ -6,6 +6,7 @@ import { LOCAL_COMPANION_RECONNECT_GRACE_MS, } from "../bridge/bridge-adapters.s
 import { runCodexRemoteClientFromEndpoint } from "./codex-remote-client.js";
 import { attachLocalCompanionMessageListener, readLocalCompanionEndpoint, sendLocalCompanionMessage, } from "./local-companion-link.js";
 import { migrateLegacyChannelFiles } from "../wechat/channel-config.js";
+import { isDirectModuleRun } from "../core/direct-run.js";
 export const LOCAL_COMPANION_RECONNECT_RETRY_MS = 250;
 function log(adapter, message) {
     process.stderr.write(`[${adapter}-companion] ${message}\n`);
@@ -230,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;
@@ -439,7 +457,7 @@ export async function main(argv = process.argv.slice(2)) {
         process.exit(1);
     }
 }
-const isDirectRun = Boolean(import.meta.main);
+const isDirectRun = isDirectModuleRun(import.meta.url, process.argv, import.meta.main);
 if (isDirectRun) {
     void main();
 }

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/channel-types.js

@@ -1 +1,3 @@
-export {};
+export function normalizeBridgeChannelId(value) {
+    return value === "wecom" ? "wecom" : "wechat";
+}

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;
+}

package/dist/core/direct-run.js

@@ -0,0 +1,17 @@
+import path from "node:path";
+import { fileURLToPath } from "node:url";
+function normalizeComparablePath(filePath) {
+    const normalized = path.resolve(filePath);
+    return process.platform === "win32" ? normalized.toLowerCase() : normalized;
+}
+export function isDirectModuleRun(importMetaUrl, argv = process.argv, importMetaMain) {
+    if (typeof importMetaMain === "boolean") {
+        return importMetaMain;
+    }
+    const entryPath = argv[1];
+    if (!entryPath) {
+        return false;
+    }
+    return normalizeComparablePath(fileURLToPath(importMetaUrl)) ===
+        normalizeComparablePath(entryPath);
+}

package/dist/core/inbound-message-claims.js

@@ -0,0 +1,61 @@
+import crypto from "node:crypto";
+import fs from "node:fs";
+import path from "node:path";
+export const DEFAULT_INBOUND_MESSAGE_CLAIM_TTL_MS = 10 * 60 * 1000;
+export function buildInboundMessageClaimPath(messageKey, claimsDir) {
+    const fileName = `${crypto.createHash("sha1").update(messageKey).digest("hex")}.json`;
+    return path.join(claimsDir, fileName);
+}
+export function tryClaimChannelInboundMessage(messageKey, options) {
+    if (!messageKey) {
+        return false;
+    }
+    const nowMs = options.nowMs ?? Date.now();
+    const ttlMs = options.ttlMs ?? DEFAULT_INBOUND_MESSAGE_CLAIM_TTL_MS;
+    const claimPath = buildInboundMessageClaimPath(messageKey, options.claimsDir);
+    const attemptClaim = () => {
+        fs.mkdirSync(options.claimsDir, { recursive: true });
+        const handle = fs.openSync(claimPath, "wx");
+        try {
+            fs.writeFileSync(handle, JSON.stringify({
+                key: messageKey,
+                claimedAt: new Date(nowMs).toISOString(),
+                pid: process.pid,
+            }, null, 2), "utf-8");
+        }
+        finally {
+            fs.closeSync(handle);
+        }
+        return true;
+    };
+    try {
+        return attemptClaim();
+    }
+    catch (error) {
+        const code = typeof error === "object" &&
+            error !== null &&
+            "code" in error &&
+            typeof error.code === "string"
+            ? error.code
+            : "";
+        if (code !== "EEXIST") {
+            return false;
+        }
+    }
+    try {
+        const stat = fs.statSync(claimPath);
+        if (Number.isFinite(stat.mtimeMs) && nowMs - stat.mtimeMs > ttlMs) {
+            fs.rmSync(claimPath, { force: true });
+            return attemptClaim();
+        }
+    }
+    catch {
+        try {
+            return attemptClaim();
+        }
+        catch {
+            return false;
+        }
+    }
+    return false;
+}

package/dist/daemon/daemon-link.js

@@ -23,6 +23,9 @@ function normalizeDaemonEndpoint(value) {
         token: record.token,
         cwd: record.cwd,
         startedAt: record.startedAt,
+        ...(record.channelId === "wechat" || record.channelId === "wecom"
+            ? { channelId: record.channelId }
+            : {}),
     };
 }
 export function buildDaemonToken() {