agent-bridge-workbuddy 0.0.0-stage → 1.0.0

package/README.md

@@ -1,3 +1,5 @@
-# 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`。

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,69 @@
+"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) {
+        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)}); element?.focus(); return Boolean(element); })()`))
+            throw new Error("WorkBuddy 输入框未找到");
+        await this.cdp("Input.insertText", { text });
+        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 回复超时");
+    }
+    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,122 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkbuddyPlugin = exports.workbuddyManifest = void 0;
+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.0.0", displayName: "WorkBuddy", type: "agent",
+    entry: "dist/index.js", apiVersion: "1", process: "in-process",
+    capabilities: ["chat", "health"], configSchema: "config.schema.json",
+};
+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;
+    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;
+        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 未连接");
+        return this.client.direct(request.task, timeout);
+    }
+    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.0.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.0.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"
+}