shennian 0.2.88 → 0.2.90

package/dist/src/channels/wecom.js

@@ -1,357 +1,4 @@
-// @arch docs/features/manager-agent.md
-// @test src/__tests__/manager-runtime.test.ts
-import { randomUUID } from 'node:crypto';
-import WebSocket from 'ws';
-import { ChannelSecretRegistry } from './secret-registry.js';
-const DEFAULT_WS_URL = 'wss://openws.work.weixin.qq.com';
-const APP_CMD_SUBSCRIBE = 'aibot_subscribe';
-const APP_CMD_CALLBACK = 'aibot_msg_callback';
-const APP_CMD_LEGACY_CALLBACK = 'aibot_callback';
-const APP_CMD_SEND = 'aibot_send_msg';
-const APP_CMD_RESPONSE = 'aibot_respond_msg';
-const APP_CMD_PING = 'ping';
-const RECONNECT_BACKOFF_MS = [2_000, 5_000, 10_000, 30_000];
-const MAX_MESSAGE_LENGTH = 4_000;
-export class WeComChannelAdapter {
-    onMessage;
-    type = 'wecom';
-    secrets = new ChannelSecretRegistry();
-    connections = new Map();
-    constructor(onMessage) {
-        this.onMessage = onMessage;
-    }
-    async connect(config) {
-        if (!config.enabled)
-            return;
-        const conn = this.ensureConnection(config);
-        if (conn.connecting)
-            return conn.connecting;
-        conn.stopped = false;
-        conn.config = config;
-        conn.connecting = this.openConnection(conn).finally(() => {
-            conn.connecting = null;
-        });
-        return conn.connecting;
-    }
-    async disconnect(config) {
-        const conn = this.connections.get(config.id);
-        if (!conn)
-            return;
-        conn.stopped = true;
-        if (conn.reconnectTimer)
-            clearTimeout(conn.reconnectTimer);
-        if (conn.pingTimer)
-            clearInterval(conn.pingTimer);
-        conn.reconnectTimer = null;
-        conn.pingTimer = null;
-        for (const pending of conn.pending.values()) {
-            clearTimeout(pending.timer);
-            pending.reject(new Error('WeCom adapter disconnected'));
-        }
-        conn.pending.clear();
-        const socket = conn.socket;
-        conn.socket = null;
-        if (socket)
-            await new Promise((resolve) => {
-                socket.once('close', () => resolve());
-                socket.close();
-                setTimeout(resolve, 500);
-            });
-        this.connections.delete(config.id);
-    }
-    async send(config, reply) {
-        const conn = this.ensureConnection(config);
-        await this.connect(config);
-        const text = reply.text.trim();
-        if (!text)
-            throw new Error('Reply text is required');
-        const replyReqId = reply.messageId ? conn.replyReqIds.get(reply.messageId) : undefined;
-        const fallbackReplyReqId = conn.lastChatReqIds.get(reply.conversationId);
-        const requestBody = replyReqId || fallbackReplyReqId
-            ? {
-                msg: {
-                    msgtype: 'markdown',
-                    markdown: { content: text.slice(0, MAX_MESSAGE_LENGTH) },
-                },
-                req_id: (replyReqId || fallbackReplyReqId),
-            }
-            : {
-                chatid: reply.conversationId,
-                msgtype: 'markdown',
-                markdown: { content: text.slice(0, MAX_MESSAGE_LENGTH) },
-            };
-        const cmd = replyReqId || fallbackReplyReqId ? APP_CMD_RESPONSE : APP_CMD_SEND;
-        const response = await this.sendRequest(conn, cmd, requestBody);
-        this.raiseForWeComError(response, 'send message');
-    }
-    async health(config) {
-        const secret = this.readSecret(config);
-        if (!secret?.botId || !secret.secret) {
-            return { ok: false, message: 'WeCom adapter requires local credentials' };
-        }
-        const conn = this.connections.get(config.id);
-        return conn?.socket?.readyState === WebSocket.OPEN
-            ? { ok: true }
-            : { ok: false, message: 'WeCom websocket disconnected' };
-    }
-    ensureConnection(config) {
-        let conn = this.connections.get(config.id);
-        if (!conn) {
-            conn = {
-                config,
-                socket: null,
-                connecting: null,
-                deviceId: randomUUID().replace(/-/g, ''),
-                stopped: false,
-                reconnectAttempt: 0,
-                reconnectTimer: null,
-                pingTimer: null,
-                pending: new Map(),
-                replyReqIds: new Map(),
-                lastChatReqIds: new Map(),
-                dedup: new Set(),
-                dedupQueue: [],
-                seenContextLineKeys: new Set(),
-                seenContextLineQueue: [],
-            };
-            this.connections.set(config.id, conn);
-        }
-        conn.config = config;
-        return conn;
-    }
-    readSecret(config) {
-        const secret = this.secrets.get(config.secretRef);
-        if (!secret || secret.type !== 'wecom' || !secret.botId || !secret.secret) {
-            throw new Error('WeCom outbound adapter is not configured on this daemon');
-        }
-        return secret;
-    }
-    async openConnection(conn) {
-        const secret = this.readSecret(conn.config);
-        const socket = new WebSocket(DEFAULT_WS_URL);
-        conn.socket = socket;
-        await new Promise((resolve, reject) => {
-            const cleanup = () => {
-                socket.off('open', handleOpen);
-                socket.off('error', handleError);
-            };
-            const handleOpen = () => {
-                cleanup();
-                resolve();
-            };
-            const handleError = (error) => {
-                cleanup();
-                reject(error);
-            };
-            socket.once('open', handleOpen);
-            socket.once('error', handleError);
-        });
-        socket.on('message', (data) => {
-            const payload = this.parsePayload(data.toString());
-            if (!payload)
-                return;
-            this.handlePayload(conn, payload);
-        });
-        socket.on('close', () => this.scheduleReconnect(conn));
-        socket.on('error', () => this.scheduleReconnect(conn));
-        const response = await this.sendRequest(conn, APP_CMD_SUBSCRIBE, {
-            bot_id: secret.botId,
-            secret: secret.secret,
-            device_id: conn.deviceId,
-        }, 20_000);
-        this.raiseForWeComError(response, 'subscribe');
-        conn.reconnectAttempt = 0;
-        if (conn.pingTimer)
-            clearInterval(conn.pingTimer);
-        conn.pingTimer = setInterval(() => {
-            void this.sendJson(conn, { cmd: APP_CMD_PING, headers: { req_id: this.newReqId('ping') }, body: {} }).catch(() => { });
-        }, 30_000);
-        conn.pingTimer.unref();
-    }
-    scheduleReconnect(conn) {
-        if (conn.stopped)
-            return;
-        if (conn.reconnectTimer || conn.connecting)
-            return;
-        const delay = RECONNECT_BACKOFF_MS[Math.min(conn.reconnectAttempt, RECONNECT_BACKOFF_MS.length - 1)];
-        conn.reconnectAttempt += 1;
-        conn.reconnectTimer = setTimeout(() => {
-            conn.reconnectTimer = null;
-            void this.connect(conn.config).catch(() => { });
-        }, delay);
-        conn.reconnectTimer.unref();
-    }
-    async sendJson(conn, payload) {
-        if (!conn.socket || conn.socket.readyState !== WebSocket.OPEN) {
-            throw new Error('WeCom websocket is not connected');
-        }
-        conn.socket.send(JSON.stringify(payload));
-    }
-    async sendRequest(conn, cmd, body, timeoutMs = 15_000) {
-        const reqId = this.newReqId(cmd);
-        return new Promise((resolve, reject) => {
-            const timer = setTimeout(() => {
-                conn.pending.delete(reqId);
-                reject(new Error(`WeCom request timed out: ${cmd}`));
-            }, timeoutMs);
-            timer.unref();
-            conn.pending.set(reqId, { resolve, reject, timer });
-            void this.sendJson(conn, { cmd, headers: { req_id: reqId }, body }).catch((error) => {
-                clearTimeout(timer);
-                conn.pending.delete(reqId);
-                reject(error instanceof Error ? error : new Error(String(error)));
-            });
-        });
-    }
-    handlePayload(conn, payload) {
-        const reqId = this.payloadReqId(payload);
-        const cmd = String(payload.cmd || '');
-        if (reqId && conn.pending.has(reqId) && !this.isCallbackCommand(cmd)) {
-            const pending = conn.pending.get(reqId);
-            clearTimeout(pending.timer);
-            conn.pending.delete(reqId);
-            pending.resolve(payload);
-            return;
-        }
-        if (!this.isCallbackCommand(cmd))
-            return;
-        const body = this.asRecord(payload.body);
-        if (!body)
-            return;
-        const messageId = String(body.msgid || reqId || randomUUID());
-        if (this.isDuplicate(conn, messageId))
-            return;
-        if (reqId)
-            conn.replyReqIds.set(messageId, reqId);
-        const sender = this.asRecord(body.from) ?? {};
-        const senderId = String(sender.userid || '');
-        const senderName = String(sender.name || sender.userid || '');
-        const conversationId = String(body.chatid || senderId).trim();
-        if (!conversationId)
-            return;
-        if (reqId)
-            conn.lastChatReqIds.set(conversationId, reqId);
-        const text = this.filterContextText(conn, conversationId, this.extractText(body).trim());
-        if (!text)
-            return;
-        this.onMessage?.({
-            managerSessionId: conn.config.managerSessionId,
-            channelId: conn.config.id,
-            channelType: 'wecom',
-            conversationId,
-            messageId,
-            sender: { id: senderId || conversationId, name: senderName || senderId || conversationId },
-            text,
-            attachments: [],
-            receivedAt: new Date().toISOString(),
-            replyTarget: '',
-        });
-    }
-    isDuplicate(conn, messageId) {
-        if (conn.dedup.has(messageId))
-            return true;
-        conn.dedup.add(messageId);
-        conn.dedupQueue.push(messageId);
-        if (conn.dedupQueue.length > 1_000) {
-            const removed = conn.dedupQueue.shift();
-            if (removed)
-                conn.dedup.delete(removed);
-        }
-        return false;
-    }
-    extractText(body) {
-        const msgtype = String(body.msgtype || '').toLowerCase();
-        if (msgtype === 'mixed') {
-            const mixed = this.asRecord(body.mixed);
-            const items = Array.isArray(mixed?.msg_item) ? mixed?.msg_item : [];
-            return items
-                .map((item) => {
-                const block = this.asRecord(item);
-                if (String(block?.msgtype || '').toLowerCase() !== 'text')
-                    return '';
-                return String(this.asRecord(block?.text)?.content || '').trim();
-            })
-                .filter(Boolean)
-                .join('\n')
-                .trim();
-        }
-        const text = String(this.asRecord(body.text)?.content || '').trim();
-        if (text)
-            return text;
-        if (msgtype === 'voice')
-            return String(this.asRecord(body.voice)?.content || '').trim();
-        if (msgtype === 'appmsg')
-            return String(this.asRecord(body.appmsg)?.title || '').trim();
-        return '';
-    }
-    filterContextText(conn, conversationId, text) {
-        if (!text.includes('外部企业微信群消息上下文'))
-            return text;
-        const lines = text.split(/\r?\n/);
-        const headerLines = [];
-        const newMessageLines = [];
-        let sawContextLine = false;
-        for (const line of lines) {
-            const normalized = this.normalizeContextLine(line);
-            if (!normalized) {
-                if (!sawContextLine)
-                    headerLines.push(line);
-                continue;
-            }
-            sawContextLine = true;
-            const key = `${conversationId}\n${normalized}`;
-            if (conn.seenContextLineKeys.has(key))
-                continue;
-            this.rememberContextLine(conn, key);
-            newMessageLines.push(line);
-        }
-        if (!sawContextLine)
-            return text;
-        if (!newMessageLines.length)
-            return '';
-        return [...headerLines, ...newMessageLines].filter((line) => line.trim()).join('\n').trim();
-    }
-    normalizeContextLine(line) {
-        const trimmed = line.trim().replace(/^=>\s*/, '');
-        return /^\d{2}-\d{2}\s+\d{2}:\d{2}:\d{2}\s+.+/.test(trimmed) ? trimmed : null;
-    }
-    rememberContextLine(conn, key) {
-        conn.seenContextLineKeys.add(key);
-        conn.seenContextLineQueue.push(key);
-        while (conn.seenContextLineQueue.length > 2_000) {
-            const removed = conn.seenContextLineQueue.shift();
-            if (removed)
-                conn.seenContextLineKeys.delete(removed);
-        }
-    }
-    raiseForWeComError(payload, operation) {
-        const errcode = typeof payload.errcode === 'number' ? payload.errcode : Number(payload.errcode || 0);
-        if (!Number.isFinite(errcode) || errcode === 0)
-            return;
-        const errmsg = String(payload.errmsg || 'unknown error');
-        throw new Error(`WeCom ${operation} failed: ${errmsg} (${errcode})`);
-    }
-    parsePayload(raw) {
-        try {
-            const parsed = JSON.parse(raw);
-            return parsed && typeof parsed === 'object' ? parsed : null;
-        }
-        catch {
-            return null;
-        }
-    }
-    payloadReqId(payload) {
-        const headers = this.asRecord(payload.headers);
-        const reqId = headers?.req_id;
-        return typeof reqId === 'string' && reqId ? reqId : null;
-    }
-    asRecord(value) {
-        return value && typeof value === 'object' && !Array.isArray(value) ? value : null;
-    }
-    isCallbackCommand(cmd) {
-        return cmd === APP_CMD_CALLBACK || cmd === APP_CMD_LEGACY_CALLBACK;
-    }
-    newReqId(prefix) {
-        return `${prefix}_${randomUUID().replace(/-/g, '')}`;
-    }
-}
+import{randomUUID as m}from"node:crypto";import u from"ws";import{ChannelSecretRegistry as f}from"./secret-registry.js";const C="wss://openws.work.weixin.qq.com",w="aibot_subscribe",_="aibot_msg_callback",y="aibot_callback",S="aibot_send_msg",R="aibot_respond_msg",I="ping",h=[2e3,5e3,1e4,3e4],g=4e3;class q{onMessage;type="wecom";secrets=new f;connections=new Map;constructor(e){this.onMessage=e}async connect(e){if(!e.enabled)return;const t=this.ensureConnection(e);return t.connecting||(t.stopped=!1,t.config=e,t.connecting=this.openConnection(t).finally(()=>{t.connecting=null})),t.connecting}async disconnect(e){const t=this.connections.get(e.id);if(!t)return;t.stopped=!0,t.reconnectTimer&&clearTimeout(t.reconnectTimer),t.pingTimer&&clearInterval(t.pingTimer),t.reconnectTimer=null,t.pingTimer=null;for(const r of t.pending.values())clearTimeout(r.timer),r.reject(new Error("WeCom adapter disconnected"));t.pending.clear();const n=t.socket;t.socket=null,n&&await new Promise(r=>{n.once("close",()=>r()),n.close(),setTimeout(r,500)}),this.connections.delete(e.id)}async send(e,t){const n=this.ensureConnection(e);await this.connect(e);const r=t.text.trim();if(!r)throw new Error("Reply text is required");const s=t.messageId?n.replyReqIds.get(t.messageId):void 0,i=n.lastChatReqIds.get(t.conversationId),o=s||i?{msg:{msgtype:"markdown",markdown:{content:r.slice(0,g)}},req_id:s||i}:{chatid:t.conversationId,msgtype:"markdown",markdown:{content:r.slice(0,g)}},c=s||i?R:S,d=await this.sendRequest(n,c,o);this.raiseForWeComError(d,"send message")}async health(e){const t=this.readSecret(e);return!t?.botId||!t.secret?{ok:!1,message:"WeCom adapter requires local credentials"}:this.connections.get(e.id)?.socket?.readyState===u.OPEN?{ok:!0}:{ok:!1,message:"WeCom websocket disconnected"}}ensureConnection(e){let t=this.connections.get(e.id);return t||(t={config:e,socket:null,connecting:null,deviceId:m().replace(/-/g,""),stopped:!1,reconnectAttempt:0,reconnectTimer:null,pingTimer:null,pending:new Map,replyReqIds:new Map,lastChatReqIds:new Map,dedup:new Set,dedupQueue:[],seenContextLineKeys:new Set,seenContextLineQueue:[]},this.connections.set(e.id,t)),t.config=e,t}readSecret(e){const t=this.secrets.get(e.secretRef);if(!t||t.type!=="wecom"||!t.botId||!t.secret)throw new Error("WeCom outbound adapter is not configured on this daemon");return t}async openConnection(e){const t=this.readSecret(e.config),n=new u(C);e.socket=n,await new Promise((s,i)=>{const o=()=>{n.off("open",c),n.off("error",d)},c=()=>{o(),s()},d=a=>{o(),i(a)};n.once("open",c),n.once("error",d)}),n.on("message",s=>{const i=this.parsePayload(s.toString());i&&this.handlePayload(e,i)}),n.on("close",()=>this.scheduleReconnect(e)),n.on("error",()=>this.scheduleReconnect(e));const r=await this.sendRequest(e,w,{bot_id:t.botId,secret:t.secret,device_id:e.deviceId},2e4);this.raiseForWeComError(r,"subscribe"),e.reconnectAttempt=0,e.pingTimer&&clearInterval(e.pingTimer),e.pingTimer=setInterval(()=>{this.sendJson(e,{cmd:I,headers:{req_id:this.newReqId("ping")},body:{}}).catch(()=>{})},3e4),e.pingTimer.unref()}scheduleReconnect(e){if(e.stopped||e.reconnectTimer||e.connecting)return;const t=h[Math.min(e.reconnectAttempt,h.length-1)];e.reconnectAttempt+=1,e.reconnectTimer=setTimeout(()=>{e.reconnectTimer=null,this.connect(e.config).catch(()=>{})},t),e.reconnectTimer.unref()}async sendJson(e,t){if(!e.socket||e.socket.readyState!==u.OPEN)throw new Error("WeCom websocket is not connected");e.socket.send(JSON.stringify(t))}async sendRequest(e,t,n,r=15e3){const s=this.newReqId(t);return new Promise((i,o)=>{const c=setTimeout(()=>{e.pending.delete(s),o(new Error(`WeCom request timed out: ${t}`))},r);c.unref(),e.pending.set(s,{resolve:i,reject:o,timer:c}),this.sendJson(e,{cmd:t,headers:{req_id:s},body:n}).catch(d=>{clearTimeout(c),e.pending.delete(s),o(d instanceof Error?d:new Error(String(d)))})})}handlePayload(e,t){const n=this.payloadReqId(t),r=String(t.cmd||"");if(n&&e.pending.has(n)&&!this.isCallbackCommand(r)){const p=e.pending.get(n);clearTimeout(p.timer),e.pending.delete(n),p.resolve(t);return}if(!this.isCallbackCommand(r))return;const s=this.asRecord(t.body);if(!s)return;const i=String(s.msgid||n||m());if(this.isDuplicate(e,i))return;n&&e.replyReqIds.set(i,n);const o=this.asRecord(s.from)??{},c=String(o.userid||""),d=String(o.name||o.userid||""),a=String(s.chatid||c).trim();if(!a)return;n&&e.lastChatReqIds.set(a,n);const l=this.filterContextText(e,a,this.extractText(s).trim());l&&this.onMessage?.({managerSessionId:e.config.managerSessionId,channelId:e.config.id,channelType:"wecom",conversationId:a,messageId:i,sender:{id:c||a,name:d||c||a},text:l,attachments:[],receivedAt:new Date().toISOString(),replyTarget:""})}isDuplicate(e,t){if(e.dedup.has(t))return!0;if(e.dedup.add(t),e.dedupQueue.push(t),e.dedupQueue.length>1e3){const n=e.dedupQueue.shift();n&&e.dedup.delete(n)}return!1}extractText(e){const t=String(e.msgtype||"").toLowerCase();if(t==="mixed"){const r=this.asRecord(e.mixed);return(Array.isArray(r?.msg_item)?r?.msg_item:[]).map(i=>{const o=this.asRecord(i);return String(o?.msgtype||"").toLowerCase()!=="text"?"":String(this.asRecord(o?.text)?.content||"").trim()}).filter(Boolean).join(`
+`).trim()}const n=String(this.asRecord(e.text)?.content||"").trim();return n||(t==="voice"?String(this.asRecord(e.voice)?.content||"").trim():t==="appmsg"?String(this.asRecord(e.appmsg)?.title||"").trim():"")}filterContextText(e,t,n){if(!n.includes("\u5916\u90E8\u4F01\u4E1A\u5FAE\u4FE1\u7FA4\u6D88\u606F\u4E0A\u4E0B\u6587"))return n;const r=n.split(/\r?\n/),s=[],i=[];let o=!1;for(const c of r){const d=this.normalizeContextLine(c);if(!d){o||s.push(c);continue}o=!0;const a=`${t}
+${d}`;e.seenContextLineKeys.has(a)||(this.rememberContextLine(e,a),i.push(c))}return o?i.length?[...s,...i].filter(c=>c.trim()).join(`
+`).trim():"":n}normalizeContextLine(e){const t=e.trim().replace(/^=>\s*/,"");return/^\d{2}-\d{2}\s+\d{2}:\d{2}:\d{2}\s+.+/.test(t)?t:null}rememberContextLine(e,t){for(e.seenContextLineKeys.add(t),e.seenContextLineQueue.push(t);e.seenContextLineQueue.length>2e3;){const n=e.seenContextLineQueue.shift();n&&e.seenContextLineKeys.delete(n)}}raiseForWeComError(e,t){const n=typeof e.errcode=="number"?e.errcode:Number(e.errcode||0);if(!Number.isFinite(n)||n===0)return;const r=String(e.errmsg||"unknown error");throw new Error(`WeCom ${t} failed: ${r} (${n})`)}parsePayload(e){try{const t=JSON.parse(e);return t&&typeof t=="object"?t:null}catch{return null}}payloadReqId(e){const n=this.asRecord(e.headers)?.req_id;return typeof n=="string"&&n?n:null}asRecord(e){return e&&typeof e=="object"&&!Array.isArray(e)?e:null}isCallbackCommand(e){return e===_||e===y}newReqId(e){return`${e}_${m().replace(/-/g,"")}`}}export{q as WeComChannelAdapter};

package/dist/src/commands/agent.js

@@ -1,131 +1,6 @@
-import chalk from 'chalk';
-import { loadConfig, saveConfig } from '../config/index.js';
-import { spawnCommandString } from '../agents/command-spec.js';
-function getCapsModels(caps) {
-    if (caps.models?.length)
-        return caps.models;
-    if (caps.model)
-        return [caps.model];
-    return [];
-}
-/**
- * Runs `<command> /caps` and returns the parsed caps JSON.
- */
-async function probeCaps(command) {
-    return new Promise((resolve, reject) => {
-        const proc = spawnCommandString(command, ['/caps'], { stdio: ['ignore', 'pipe', 'pipe'] });
-        let stdout = '';
-        let stderr = '';
-        proc.stdout?.on('data', (d) => { stdout += d.toString(); });
-        proc.stderr?.on('data', (d) => { stderr += d.toString(); });
-        proc.on('error', (err) => reject(new Error(`Failed to run command: ${err.message}`)));
-        proc.on('close', (_code) => {
-            const line = stdout.trim().split('\n').find((l) => l.trim().startsWith('{'));
-            if (!line) {
-                reject(new Error(`Command did not output valid caps JSON.\nstdout: ${stdout.trim()}\nstderr: ${stderr.trim()}`));
-                return;
-            }
-            try {
-                const caps = JSON.parse(line);
-                if (!caps.name || !caps.mode) {
-                    reject(new Error('Caps JSON must contain "name" and "mode" fields'));
-                    return;
-                }
-                if (getCapsModels(caps).length === 0) {
-                    reject(new Error('Caps JSON must contain either "model" or a non-empty "models" array'));
-                    return;
-                }
-                if (caps.mode !== 'spawn' && caps.mode !== 'stdio') {
-                    reject(new Error(`Caps "mode" must be "spawn" or "stdio", got: ${caps.mode}`));
-                    return;
-                }
-                resolve(caps);
-            }
-            catch {
-                reject(new Error(`Failed to parse caps JSON: ${line}`));
-            }
-        });
-        setTimeout(() => {
-            proc.kill('SIGTERM');
-            reject(new Error('Timed out waiting for caps response (5s)'));
-        }, 5000);
-    });
-}
-export function registerAgentCommand(program) {
-    const agent = program
-        .command('agent')
-        .description('Manage custom agents');
-    // ── add ──────────────────────────────────────────────────────────────────
-    agent
-        .command('add <name>')
-        .description('Register a custom agent')
-        .requiredOption('--command <cmd>', 'Command to run the agent (e.g. "python /path/agent.py")')
-        .action(async (name, opts) => {
-        if (!/^[a-zA-Z0-9_-]+$/.test(name)) {
-            console.error(chalk.red('✗ Agent name must contain only letters, numbers, hyphens, and underscores'));
-            process.exit(1);
-        }
-        console.log(chalk.gray(`Probing caps: ${opts.command} /caps ...`));
-        let caps;
-        try {
-            caps = await probeCaps(opts.command);
-        }
-        catch (err) {
-            console.error(chalk.red(`✗ ${err instanceof Error ? err.message : String(err)}`));
-            process.exit(1);
-        }
-        const config = loadConfig();
-        config.customAgents = config.customAgents ?? {};
-        config.customAgents[name] = { command: opts.command, caps };
-        saveConfig(config);
-        console.log(chalk.green(`✓ Agent "${name}" registered`));
-        console.log(chalk.gray(`  Name:    ${caps.name}`));
-        console.log(chalk.gray(`  Models:  ${getCapsModels(caps).join(', ')}`));
-        console.log(chalk.gray(`  Mode:    ${caps.mode}`));
-        if (caps.version)
-            console.log(chalk.gray(`  Version: ${caps.version}`));
-        console.log();
-        console.log(chalk.yellow(`  The agent will appear as "custom:${name}" in the Shennian app.`));
-        console.log(chalk.yellow('  Refresh Agents in the app to hot-reload this daemon, or restart the daemon as a fallback.'));
-    });
-    // ── list ─────────────────────────────────────────────────────────────────
-    agent
-        .command('list')
-        .description('List registered custom agents')
-        .action(() => {
-        const config = loadConfig();
-        const entries = Object.entries(config.customAgents ?? {});
-        if (entries.length === 0) {
-            console.log(chalk.gray('No custom agents registered.'));
-            console.log(chalk.gray('Add one with: shennian agent add <name> --command "<cmd>"'));
-            return;
-        }
-        console.log(chalk.bold('\nCustom agents:\n'));
-        for (const [name, entry] of entries) {
-            const models = entry.caps.models?.length
-                ? entry.caps.models.join(', ')
-                : entry.caps.model ?? '(none)';
-            console.log(`  ${chalk.cyan(name)}`);
-            console.log(`    ${chalk.gray('display:')}  ${entry.caps.name}`);
-            console.log(`    ${chalk.gray('models:')}   ${models}`);
-            console.log(`    ${chalk.gray('mode:')}     ${entry.caps.mode}`);
-            console.log(`    ${chalk.gray('command:')}  ${entry.command}`);
-            console.log();
-        }
-    });
-    // ── remove ────────────────────────────────────────────────────────────────
-    agent
-        .command('remove <name>')
-        .description('Unregister a custom agent')
-        .action((name) => {
-        const config = loadConfig();
-        if (!config.customAgents?.[name]) {
-            console.error(chalk.red(`✗ No agent named "${name}"`));
-            process.exit(1);
-        }
-        delete config.customAgents[name];
-        saveConfig(config);
-        console.log(chalk.green(`✓ Agent "${name}" removed`));
-        console.log(chalk.yellow('  Refresh Agents in the app to hot-reload this daemon, or restart the daemon as a fallback.'));
-    });
-}
+import o from"chalk";import{loadConfig as d,saveConfig as l}from"../config/index.js";import{spawnCommandString as p}from"../agents/command-spec.js";function g(m){return m.models?.length?m.models:m.model?[m.model]:[]}async function u(m){return new Promise((c,e)=>{const n=p(m,["/caps"],{stdio:["ignore","pipe","pipe"]});let s="",t="";n.stdout?.on("data",r=>{s+=r.toString()}),n.stderr?.on("data",r=>{t+=r.toString()}),n.on("error",r=>e(new Error(`Failed to run command: ${r.message}`))),n.on("close",r=>{const i=s.trim().split(`
+`).find(a=>a.trim().startsWith("{"));if(!i){e(new Error(`Command did not output valid caps JSON.
+stdout: ${s.trim()}
+stderr: ${t.trim()}`));return}try{const a=JSON.parse(i);if(!a.name||!a.mode){e(new Error('Caps JSON must contain "name" and "mode" fields'));return}if(g(a).length===0){e(new Error('Caps JSON must contain either "model" or a non-empty "models" array'));return}if(a.mode!=="spawn"&&a.mode!=="stdio"){e(new Error(`Caps "mode" must be "spawn" or "stdio", got: ${a.mode}`));return}c(a)}catch{e(new Error(`Failed to parse caps JSON: ${i}`))}}),setTimeout(()=>{n.kill("SIGTERM"),e(new Error("Timed out waiting for caps response (5s)"))},5e3)})}function $(m){const c=m.command("agent").description("Manage custom agents");c.command("add <name>").description("Register a custom agent").requiredOption("--command <cmd>",'Command to run the agent (e.g. "python /path/agent.py")').action(async(e,n)=>{/^[a-zA-Z0-9_-]+$/.test(e)||(console.error(o.red("\u2717 Agent name must contain only letters, numbers, hyphens, and underscores")),process.exit(1)),console.log(o.gray(`Probing caps: ${n.command} /caps ...`));let s;try{s=await u(n.command)}catch(r){console.error(o.red(`\u2717 ${r instanceof Error?r.message:String(r)}`)),process.exit(1)}const t=d();t.customAgents=t.customAgents??{},t.customAgents[e]={command:n.command,caps:s},l(t),console.log(o.green(`\u2713 Agent "${e}" registered`)),console.log(o.gray(`  Name:    ${s.name}`)),console.log(o.gray(`  Models:  ${g(s).join(", ")}`)),console.log(o.gray(`  Mode:    ${s.mode}`)),s.version&&console.log(o.gray(`  Version: ${s.version}`)),console.log(),console.log(o.yellow(`  The agent will appear as "custom:${e}" in the Shennian app.`)),console.log(o.yellow("  Refresh Agents in the app to hot-reload this daemon, or restart the daemon as a fallback."))}),c.command("list").description("List registered custom agents").action(()=>{const e=d(),n=Object.entries(e.customAgents??{});if(n.length===0){console.log(o.gray("No custom agents registered.")),console.log(o.gray('Add one with: shennian agent add <name> --command "<cmd>"'));return}console.log(o.bold(`
+Custom agents:
+`));for(const[s,t]of n){const r=t.caps.models?.length?t.caps.models.join(", "):t.caps.model??"(none)";console.log(`  ${o.cyan(s)}`),console.log(`    ${o.gray("display:")}  ${t.caps.name}`),console.log(`    ${o.gray("models:")}   ${r}`),console.log(`    ${o.gray("mode:")}     ${t.caps.mode}`),console.log(`    ${o.gray("command:")}  ${t.command}`),console.log()}}),c.command("remove <name>").description("Unregister a custom agent").action(e=>{const n=d();n.customAgents?.[e]||(console.error(o.red(`\u2717 No agent named "${e}"`)),process.exit(1)),delete n.customAgents[e],l(n),console.log(o.green(`\u2713 Agent "${e}" removed`)),console.log(o.yellow("  Refresh Agents in the app to hot-reload this daemon, or restart the daemon as a fallback."))})}export{$ as registerAgentCommand};

package/dist/src/commands/daemon-windows.js

@@ -1,45 +1,6 @@
-// @arch docs/architecture/cli/daemon.md#守护进程管理
-// @test src/__tests__/daemon-autostart.test.ts
-import path from 'node:path';
-export function escapeXml(text) {
-    return text
-        .replace(/&/g, '&')
-        .replace(/</g, '&lt;')
-        .replace(/>/g, '&gt;')
-        .replace(/"/g, '&quot;')
-        .replace(/'/g, '&apos;');
-}
-export function quoteSystemdArg(text) {
-    return `"${text.replace(/\\/g, '\\\\').replace(/"/g, '\\"')}"`;
-}
-export function quoteCmdArg(text) {
-    return `"${text.replace(/"/g, '""')}"`;
-}
-function escapeVbsString(text) {
-    return text.replace(/"/g, '""');
-}
-function escapeXmlValue(text) {
-    return escapeXml(text);
-}
-export function isWindowsCmdScript(filePath) {
-    const ext = path.extname(filePath).toLowerCase();
-    return ext === '.cmd' || ext === '.bat';
-}
-export function buildWindowsLauncherCommand(spec, logFile) {
-    const commandLine = [spec.command, ...spec.args].map(quoteCmdArg).join(' ');
-    const invocation = isWindowsCmdScript(spec.command) ? `call ${commandLine}` : commandLine;
-    return ['@echo off', 'setlocal', `${invocation} >> ${quoteCmdArg(logFile)} 2>&1`, ''].join('\r\n');
-}
-export function buildWindowsStartupVbs(cmdPath) {
-    return [
-        'Set shell = CreateObject("WScript.Shell")',
-        `shell.Run Chr(34) & "${escapeVbsString(cmdPath)}" & Chr(34), 0, False`,
-        '',
-    ].join('\r\n');
-}
-export function buildWindowsScheduledTaskXml(input) {
-    const joinedArgs = input.arguments.join(' ');
-    return `<?xml version="1.0" encoding="UTF-16"?>
+import i from"node:path";function a(e){return e.replace(/&/g,"&amp;").replace(/</g,"&lt;").replace(/>/g,"&gt;").replace(/"/g,"&quot;").replace(/'/g,"&apos;")}function d(e){return`"${e.replace(/\\/g,"\\\\").replace(/"/g,'\\"')}"`}function o(e){return`"${e.replace(/"/g,'""')}"`}function l(e){return e.replace(/"/g,'""')}function n(e){return a(e)}function s(e){const t=i.extname(e).toLowerCase();return t===".cmd"||t===".bat"}function m(e,t){const r=[e.command,...e.args].map(o).join(" ");return["@echo off","setlocal",`${s(e.command)?`call ${r}`:r} >> ${o(t)} 2>&1`,""].join(`\r
+`)}function g(e){return['Set shell = CreateObject("WScript.Shell")',`shell.Run Chr(34) & "${l(e)}" & Chr(34), 0, False`,""].join(`\r
+`)}function p(e){const t=e.arguments.join(" ");return`<?xml version="1.0" encoding="UTF-16"?>
 <Task version="1.4" xmlns="http://schemas.microsoft.com/windows/2004/02/mit/task">
   <RegistrationInfo>
     <Description>Shennian Agent Daemon</Description>
@@ -47,7 +8,7 @@ export function buildWindowsScheduledTaskXml(input) {
   <Triggers>
     <LogonTrigger>
       <Enabled>true</Enabled>
-      <UserId>${escapeXmlValue(input.userId)}</UserId>
+      <UserId>${n(e.userId)}</UserId>
     </LogonTrigger>
     <EventTrigger>
       <Enabled>true</Enabled>
@@ -61,7 +22,7 @@ export function buildWindowsScheduledTaskXml(input) {
   </Triggers>
   <Principals>
     <Principal id="Author">
-      <UserId>${escapeXmlValue(input.userId)}</UserId>
+      <UserId>${n(e.userId)}</UserId>
       <LogonType>InteractiveToken</LogonType>
       <RunLevel>LeastPrivilege</RunLevel>
     </Principal>
@@ -91,9 +52,8 @@ export function buildWindowsScheduledTaskXml(input) {
   </Settings>
   <Actions Context="Author">
     <Exec>
-      <Command>${escapeXmlValue(input.command)}</Command>
-      <Arguments>${escapeXmlValue(joinedArgs)}</Arguments>
+      <Command>${n(e.command)}</Command>
+      <Arguments>${n(t)}</Arguments>
     </Exec>
   </Actions>
-</Task>`;
-}
+</Task>`}export{m as buildWindowsLauncherCommand,p as buildWindowsScheduledTaskXml,g as buildWindowsStartupVbs,a as escapeXml,s as isWindowsCmdScript,o as quoteCmdArg,d as quoteSystemdArg};