agent-bridge-workbuddy 0.0.0-stage → 1.1.0

package/README.md

@@ -1,3 +1,7 @@
-# Temporary Holding Version
+# WorkBuddy 插件
 
-This version is a temporary placeholder for this package. An operational version to replace this has been submitted for review and is awaiting a staged release.
+插件保留旧桥的 CodeBuddy CLI 调用，并支持 WorkBuddy 桌面端 CDP。默认 `auto` 优先使用 CLI；CLI 未登录时直接返回登录错误，不切换桌面端账号。没有 CLI 时才使用 CDP。未连接桌面端时默认自动启动 `WorkBuddy.exe`，CDP 默认端口为 `9229`。
+
+将本目录作为本地插件安装，或发布后安装 `agent-bridge-workbuddy`；在 AUN 通道配置中把 `agent_plugin` 设为 `workbuddy`。插件串行处理消息；CLI 按 AUN 对端续接会话，CDP 使用当前桌面会话。可用 `transport` 固定为 `cli` 或 `cdp`。
+
+当提示中以“只要、调用、使用、找、用”等表达指定专家（例如“只要 AI 短剧剧本专家，帮我写开场”）时，插件会切换到 WorkBuddy 桌面端，在专家中心按名称搜索并召唤唯一匹配的专家，再发送原始提示。专家请求需要桌面端 CDP；普通提示继续优先使用 CLI。简称没有唯一匹配时会提示用户指定完整专家名称。

package/config.schema.json

@@ -0,0 +1,14 @@
+{
+  "type": "object",
+  "properties": {
+    "transport": { "type": "string", "enum": ["auto", "cli", "cdp"], "default": "auto" },
+    "cliPath": { "type": "string", "description": "CodeBuddy CLI 路径" },
+    "model": { "type": "string", "description": "CLI 模型 ID" },
+    "executablePath": { "type": "string", "description": "WorkBuddy.exe 路径，省略时自动查找" },
+    "cdpPort": { "type": "integer", "minimum": 1, "maximum": 65535, "default": 9229 },
+    "autoStart": { "type": "boolean", "default": true, "description": "未连接时自动启动 WorkBuddy" },
+    "startupTimeoutMs": { "type": "integer", "minimum": 1000, "default": 30000 },
+    "timeoutMs": { "type": "integer", "minimum": 1, "default": 240000, "description": "单次回复超时时间（毫秒）" }
+  },
+  "additionalProperties": false
+}

package/dist/src-ts/agent.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentError = void 0;
+class AgentError extends Error {
+    constructor(message, options) {
+        super(message, options);
+        this.name = "AgentError";
+    }
+}
+exports.AgentError = AgentError;

package/dist/src-ts/plugins/workbuddy/cdp-client.js

@@ -0,0 +1,118 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkbuddyCdpClient = exports.assistantSelector = exports.editorSelector = void 0;
+function sleep(ms) { return new Promise((resolve) => setTimeout(resolve, ms)); }
+exports.editorSelector = '[data-slate-editor][contenteditable="true"]';
+exports.assistantSelector = '.cr-agent[data-message-id] [data-assistant-content]';
+class WorkbuddyCdpClient {
+    socket;
+    sequence = 0;
+    pending = new Map();
+    port;
+    timeoutMs;
+    fetchImpl;
+    constructor(options = {}) { this.port = options.port ?? Number(process.env.WORKBUDDY_CDP_PORT ?? "9229"); this.timeoutMs = options.timeoutMs ?? 30_000; this.fetchImpl = options.fetchImpl ?? fetch; }
+    connected() { return Boolean(this.socket); }
+    async connect() {
+        if (this.socket)
+            return;
+        const response = await this.fetchImpl(`http://127.0.0.1:${this.port}/json`);
+        if (!response.ok)
+            throw new Error(`WorkBuddy CDP /json HTTP ${response.status}`);
+        const targets = await response.json();
+        const page = targets.find((target) => target.type === "page" && /workbuddy/i.test(`${target.title} ${target.url}`));
+        if (!page?.webSocketDebuggerUrl)
+            throw new Error(`未找到 WorkBuddy 页面 target（端口 ${this.port}）`);
+        this.socket = new WebSocket(page.webSocketDebuggerUrl);
+        this.socket.onmessage = (event) => this.onMessage(String(event.data));
+        try {
+            await new Promise((resolve, reject) => { const socket = this.socket; socket.onopen = () => resolve(); socket.onerror = () => reject(new Error("WorkBuddy CDP WebSocket 连接失败")); });
+        }
+        catch (error) {
+            this.close();
+            throw error;
+        }
+    }
+    close() { this.socket?.close(); this.socket = undefined; for (const pending of this.pending.values())
+        pending.reject(new Error("WorkBuddy CDP 已关闭")); this.pending.clear(); }
+    async direct(text, timeoutMs = 240_000, expert, newTask = false) {
+        if (expert)
+            await this.selectExpert(expert);
+        else if (newTask) {
+            if (!await this.evaluate(`(() => { const button = [...document.querySelectorAll('button,[role="button"]')].find((item) => item.textContent?.trim() === '新建任务'); button?.click(); return Boolean(button); })()`))
+                throw new Error("WorkBuddy 新建任务入口未找到");
+        }
+        await this.waitFor(`Boolean(document.querySelector(${JSON.stringify(exports.editorSelector)}))`, "WorkBuddy 输入框未就绪");
+        const before = JSON.parse(String(await this.evaluate(`JSON.stringify([...document.querySelectorAll(${JSON.stringify(exports.assistantSelector)})].map(element => ({ id: element.closest('.cr-agent')?.getAttribute('data-message-id'), text: element.innerText })))`)));
+        const beforeIds = new Set(before.map((item) => item.id).filter((id) => Boolean(id)));
+        if (!await this.evaluate(`(() => { const element = document.querySelector(${JSON.stringify(exports.editorSelector)}); if (!element) return false; element.focus(); const selection = window.getSelection(); const range = document.createRange(); range.selectNodeContents(element); selection?.removeAllRanges(); selection?.addRange(range); return true; })()`))
+            throw new Error("WorkBuddy 输入框未找到");
+        await this.cdp("Input.dispatchKeyEvent", { type: "keyDown", key: "Backspace", code: "Backspace" });
+        await this.cdp("Input.dispatchKeyEvent", { type: "keyUp", key: "Backspace", code: "Backspace" });
+        await this.cdp("Input.insertText", { text });
+        if (!await this.evaluate(`document.querySelector(${JSON.stringify(exports.editorSelector)})?.innerText.trim() === ${JSON.stringify(text.trim())}`))
+            throw new Error("WorkBuddy 输入内容校验失败，未发送消息");
+        if (!await this.evaluate(`(() => { const button = document.querySelector('button[aria-label="发送"]'); if (!button || button.disabled) return false; button.click(); return true; })()`))
+            throw new Error("WorkBuddy 发送按钮不可用");
+        const deadline = Date.now() + timeoutMs;
+        while (Date.now() < deadline) {
+            await sleep(1_000);
+            const current = JSON.parse(String(await this.evaluate(`JSON.stringify([...document.querySelectorAll('.cr-agent[data-message-id]')].map(element => ({ id: element.getAttribute('data-message-id'), text: element.querySelector('[data-assistant-content]')?.innerText, finished: Boolean(element.querySelector('[data-testid="conversation-finished-footer"]')) })))`)));
+            const reply = current.filter((item) => item.id && !beforeIds.has(item.id) && item.text?.trim() && item.finished).at(-1)?.text?.trim();
+            if (reply)
+                return reply;
+        }
+        throw new Error("WorkBuddy UI 回复超时");
+    }
+    async selectExpert(name) {
+        if (!await this.evaluate(`(() => { const element = [...document.querySelectorAll('button,[role="button"],a')].find((item) => item.textContent?.trim() === '专家·技能·连接器'); if (!element) return false; element.click(); return true; })()`))
+            throw new Error("WorkBuddy 专家入口未找到");
+        await this.waitFor(`Boolean(document.querySelector('input[placeholder="搜索专家"]'))`, "WorkBuddy 专家搜索框未就绪");
+        if (!await this.evaluate(`(() => { const element = document.querySelector('input[placeholder="搜索专家"]'); if (!element) return false; element.focus(); return true; })()`))
+            throw new Error("WorkBuddy 专家搜索框未找到");
+        for (const query of [...new Set([name, name.replace(/专家(?:团)?$/, "")])]) {
+            if (!await this.evaluate(`(() => { const element = document.querySelector('input[placeholder="搜索专家"]'); if (!element) return false; const setter = Object.getOwnPropertyDescriptor(HTMLInputElement.prototype, 'value')?.set; setter?.call(element, ''); element.dispatchEvent(new Event('input', { bubbles: true })); element.focus(); return true; })()`))
+                throw new Error("WorkBuddy 专家搜索框未找到");
+            await this.cdp("Input.insertText", { text: query });
+            await this.waitFor(`(() => { const input = document.querySelector('input[placeholder="搜索专家"]'); const cards = document.querySelectorAll('article[role="button"] button[data-track-id="expert_summon_click"]'); return input?.value === ${JSON.stringify(query)} && cards.length > 0; })()`, "WorkBuddy 专家搜索超时");
+            const result = await this.evaluate(`(() => {
+        const cards = [...document.querySelectorAll('article[role="button"]')].filter(card => card.querySelector('button[data-track-id="expert_summon_click"]'));
+        const title = card => card.querySelector('[class*="_role_"]')?.textContent?.trim() || card.querySelector('img')?.alt || card.innerText.split('\\n').filter(Boolean)[1];
+        const exact = cards.filter(card => title(card) === ${JSON.stringify(name)});
+        const alias = cards.filter(card => card.querySelector('img')?.alt === ${JSON.stringify(name)} || card.querySelector('button[data-track-id="expert_summon_click"]')?.getAttribute('aria-label') === ${JSON.stringify(`召唤${name}`)});
+        const related = cards.filter(card => title(card)?.includes(${JSON.stringify(name)}) || ${JSON.stringify(name)}.includes(title(card) || ''));
+        const matches = exact.length ? exact : related.length === 1 ? related : alias.length === 1 ? alias : related;
+        if (matches.length !== 1) return { names: matches.map(title) };
+        const button = matches[0].querySelector('button[data-track-id="expert_summon_click"]');
+        button.click(); return { selected: true };
+      })()`);
+            if (result.selected)
+                return;
+            if (result.names?.length)
+                throw new Error(`WorkBuddy 专家匹配不唯一，请指定完整名称：${result.names.join("、")}`);
+        }
+        throw new Error(`未找到 WorkBuddy 专家：${name}`);
+    }
+    async waitFor(expression, message) {
+        const deadline = Date.now() + this.timeoutMs;
+        while (Date.now() < deadline) {
+            if (await this.evaluate(expression))
+                return;
+            await sleep(200);
+        }
+        throw new Error(message);
+    }
+    onMessage(raw) { let message; try {
+        message = JSON.parse(raw);
+    }
+    catch {
+        return;
+    } if (message.id === undefined)
+        return; const pending = this.pending.get(message.id); if (!pending)
+        return; this.pending.delete(message.id); pending.resolve(message); }
+    async cdp(method, params = {}) { if (!this.socket)
+        throw new Error("WorkBuddy CDP 尚未连接"); const id = ++this.sequence; const result = new Promise((resolve, reject) => this.pending.set(id, { resolve, reject })); this.socket.send(JSON.stringify({ id, method, params })); return Promise.race([result, sleep(this.timeoutMs).then(() => { throw new Error(`CDP ${method} 超时`); })]); }
+    async evaluate(expression) { const response = await this.cdp("Runtime.evaluate", { expression, returnByValue: true, awaitPromise: true }); if (response.result?.exceptionDetails)
+        throw new Error(`WorkBuddy 页面 JS 异常：${JSON.stringify(response.result.exceptionDetails).slice(0, 300)}`); return response.result?.result?.value; }
+}
+exports.WorkbuddyCdpClient = WorkbuddyCdpClient;

package/dist/src-ts/plugins/workbuddy/cli-client.js

@@ -0,0 +1,39 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.findCli = findCli;
+exports.callCli = callCli;
+const node_child_process_1 = require("node:child_process");
+const node_fs_1 = require("node:fs");
+const node_os_1 = require("node:os");
+const node_path_1 = require("node:path");
+const node_util_1 = require("node:util");
+const execFileAsync = (0, node_util_1.promisify)(node_child_process_1.execFile);
+function findCli(configured) {
+    if (configured)
+        return (0, node_fs_1.existsSync)((0, node_path_1.resolve)(configured)) ? (0, node_path_1.resolve)(configured) : undefined;
+    const roots = [process.env.CODEBUDDY_NPM_DIR, (0, node_path_1.join)(process.env.APPDATA ?? (0, node_path_1.join)((0, node_os_1.homedir)(), "AppData", "Roaming"), "npm", "node_modules", "@tencent-ai", "codebuddy-code"), (0, node_path_1.join)((0, node_os_1.homedir)(), ".npm", "node_modules", "@tencent-ai", "codebuddy-code")];
+    return roots.filter((root) => Boolean(root)).map((root) => (0, node_path_1.join)(root, "bin", "codebuddy")).find(node_fs_1.existsSync);
+}
+async function callCli(cliPath, text, sessionId, timeoutMs, model) {
+    const id = (sessionId.replace(/[^A-Za-z0-9_-]/g, "-").replace(/^-+|-+$/g, "") || "wb").slice(0, 80);
+    const args = [cliPath, "-p", text, "--session-id", id, "--output-format", "text"];
+    if (model)
+        args.push("--model", model);
+    let stdout;
+    let stderr;
+    try {
+        ({ stdout, stderr } = await execFileAsync(process.execPath, args, { cwd: (0, node_path_1.dirname)((0, node_path_1.dirname)(cliPath)), timeout: timeoutMs, maxBuffer: 4 * 1024 * 1024, encoding: "utf8", windowsHide: true }));
+    }
+    catch (error) {
+        const failure = error;
+        if (failure.killed)
+            throw new Error("CodeBuddy 回复超时");
+        throw new Error(`CodeBuddy 调用失败：${failure.stderr?.trim().slice(-300) || failure.message}`);
+    }
+    if (/Authentication required\. Please use \/login command to sign in to your account/.test(`${stdout}\n${stderr}`))
+        throw new Error("CodeBuddy CLI 尚未登录，请先执行 /login");
+    const reply = stdout.split(/\r?\n/).filter((line) => !(line.startsWith("[") && line.slice(0, 40).includes("]["))).join("\n").trim();
+    if (!reply)
+        throw new Error("CodeBuddy 无输出");
+    return reply;
+}

package/dist/src-ts/plugins/workbuddy/index.js

@@ -0,0 +1,139 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkbuddyPlugin = exports.workbuddyManifest = void 0;
+exports.extractExpertQuery = extractExpertQuery;
+const node_child_process_1 = require("node:child_process");
+const node_fs_1 = require("node:fs");
+const node_os_1 = require("node:os");
+const node_path_1 = require("node:path");
+const cdp_client_1 = require("./cdp-client");
+const cli_client_1 = require("./cli-client");
+exports.workbuddyManifest = {
+    id: "workbuddy", version: "1.1.0", displayName: "WorkBuddy", type: "agent",
+    entry: "dist/index.js", apiVersion: "1", process: "in-process",
+    capabilities: ["chat", "health"], configSchema: "config.schema.json",
+};
+function extractExpertQuery(task) {
+    const text = task.normalize("NFKC");
+    const match = text.match(/^(?:请)?(?:只要|调用|使用|帮我找|找|用|选择|切换到)\s*[“「"]?([^，。！？,!?\n“「"]{2,30}?专家(?:团)?)(?!技能)/)
+        ?? text.match(/^([A-Za-z\u4e00-\u9fff]{2,24}?专家(?:团)?)(?=[，。！,!:：\s]|$)/);
+    return match?.[1]?.trim() || undefined;
+}
+function findExecutable(configured) {
+    if (configured)
+        return (0, node_fs_1.existsSync)((0, node_path_1.resolve)(configured)) ? (0, node_path_1.resolve)(configured) : undefined;
+    const local = process.env.LOCALAPPDATA ?? (0, node_path_1.join)((0, node_os_1.homedir)(), "AppData", "Local");
+    return [(0, node_path_1.join)(local, "Programs", "WorkBuddy", "WorkBuddy.exe"), "D:\\Programs\\WorkBuddy\\WorkBuddy.exe"].find(node_fs_1.existsSync);
+}
+class WorkbuddyPlugin {
+    options;
+    manifest = exports.workbuddyManifest;
+    client;
+    process;
+    cliPath;
+    model;
+    transport = "auto";
+    timeoutMs = 240_000;
+    queue = Promise.resolve();
+    failure;
+    expertSession = false;
+    constructor(options = {}) {
+        this.options = options;
+    }
+    async activate(context) {
+        const config = context.config;
+        if (config.transport !== undefined && !["auto", "cli", "cdp"].includes(String(config.transport)))
+            throw new Error("transport 只能是 auto、cli 或 cdp");
+        this.transport = config.transport ?? "auto";
+        this.timeoutMs = typeof config.timeoutMs === "number" && config.timeoutMs > 0 ? config.timeoutMs : 240_000;
+        this.model = typeof config.model === "string" && config.model.trim() ? config.model : undefined;
+        this.cliPath = (0, cli_client_1.findCli)(typeof config.cliPath === "string" ? config.cliPath : undefined);
+        this.failure = undefined;
+        if (this.transport === "cli") {
+            if (!this.cliPath)
+                throw new Error("未找到 CodeBuddy CLI，请配置 cliPath");
+            return;
+        }
+        const port = typeof config.cdpPort === "number" ? config.cdpPort : this.options.cdpPort ?? 9229;
+        this.client = this.options.client ?? new cdp_client_1.WorkbuddyCdpClient({ port });
+        try {
+            await this.connectOrStart(config, context.logger, port);
+        }
+        catch (error) {
+            this.failure = error instanceof Error ? error.message : String(error);
+            this.client.close();
+            this.client = undefined;
+            if (this.transport === "cdp" || !this.cliPath)
+                throw error;
+        }
+    }
+    async deactivate(_reason) {
+        await this.queue;
+        this.client?.close();
+        this.client = undefined;
+        // WorkBuddy 是用户桌面应用，插件停止时不结束桌面进程。
+        this.process = undefined;
+    }
+    async health() {
+        if (this.client?.connected())
+            return { status: "ready", details: { transport: "cdp" } };
+        if (this.cliPath && (0, node_fs_1.existsSync)(this.cliPath))
+            return { status: "ready", details: { transport: "cli" } };
+        return { status: "failed", message: this.failure ?? "WorkBuddy 未连接", details: { transport: this.transport } };
+    }
+    async handle(request) {
+        if (!request.task.trim())
+            throw new Error("消息内容不能为空");
+        const run = this.queue.then(() => this.complete(request), () => this.complete(request));
+        this.queue = run.then(() => undefined, () => undefined);
+        return { text: await run, sessionId: request.sessionId };
+    }
+    async complete(request) {
+        const timeout = request.timeoutMs ?? this.timeoutMs;
+        const expert = extractExpertQuery(request.task);
+        if (expert) {
+            if (!this.client?.connected())
+                throw new Error(`专家请求需要已连接的 WorkBuddy 桌面端：${expert}`);
+            this.expertSession = true;
+            return this.client.direct(request.task, timeout, expert);
+        }
+        if (this.cliPath && this.transport !== "cdp")
+            return (0, cli_client_1.callCli)(this.cliPath, request.task, request.sessionId ?? "local-test", timeout, this.model);
+        if (!this.client?.connected())
+            throw new Error("WorkBuddy CDP 未连接");
+        const reset = this.expertSession;
+        this.expertSession = false;
+        return this.client.direct(request.task, timeout, undefined, reset);
+    }
+    async connectOrStart(config, logger, port) {
+        try {
+            await this.client.connect();
+            return;
+        }
+        catch (first) {
+            if (config.autoStart === false || this.options.client)
+                throw first;
+            const executable = findExecutable(typeof config.executablePath === "string" ? config.executablePath : this.options.executablePath);
+            if (!executable)
+                throw new Error("未找到 WorkBuddy.exe，请配置 executablePath");
+            this.process = (0, node_child_process_1.spawn)(executable, [`--remote-debugging-port=${port}`], { detached: true, stdio: "ignore", windowsHide: false });
+            this.process.unref();
+            logger.info(`WorkBuddy 已启动，等待 CDP ${port}`);
+            const deadline = Date.now() + (typeof config.startupTimeoutMs === "number" ? config.startupTimeoutMs : 30_000);
+            let last = first;
+            while (Date.now() < deadline) {
+                await new Promise((resolve) => setTimeout(resolve, 500));
+                try {
+                    await this.client.connect();
+                    return;
+                }
+                catch (error) {
+                    last = error;
+                }
+            }
+            throw last instanceof Error ? last : new Error("WorkBuddy CDP 启动超时");
+        }
+    }
+}
+exports.WorkbuddyPlugin = WorkbuddyPlugin;
+exports.default = WorkbuddyPlugin;

package/package.json

@@ -1,6 +1,12 @@
 {
   "name": "agent-bridge-workbuddy",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+  "version": "1.1.0",
+  "description": "agent-bridge 的 CodeBuddy CLI 插件。",
+  "keywords": ["mu-agent-bridge-plugin", "mu-agent-bridge-agent"],
+  "repository": { "type": "git", "url": "https://codeup.aliyun.com/6808da56fb3da14b899afd54/agent_bridge.git", "directory": "packages/workbuddy" },
+  "main": "dist/src-ts/plugins/workbuddy/index.js",
+  "files": ["dist/", "plugin.json", "config.schema.json", "README.md"],
+  "agentBridge": { "plugin": "./plugin.json", "type": "agent" },
+  "publishConfig": { "access": "public", "registry": "https://registry.npmjs.org" },
+  "engines": { "node": ">=22" }
+}

package/plugin.json

@@ -0,0 +1,11 @@
+{
+  "id": "workbuddy",
+  "version": "1.1.0",
+  "type": "agent",
+  "displayName": "WorkBuddy（CodeBuddy CLI）",
+  "entry": "dist/src-ts/plugins/workbuddy/index.js",
+  "apiVersion": "1",
+  "process": "in-process",
+  "capabilities": ["chat", "health"],
+  "configSchema": "config.schema.json"
+}