agent-muster 0.1.1 → 0.1.2

package/dist/feishu.js

@@ -0,0 +1,232 @@
+import { randomUUID } from 'node:crypto';
+import { EventEmitter } from 'node:events';
+import { mkdir, readFile, rename, writeFile } from 'node:fs/promises';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import { Client, Domain, EventDispatcher, LoggerLevel, WSClient, registerApp } from '@larksuiteoapi/node-sdk';
+import QRCode from 'qrcode';
+const directory = join(homedir(), '.agent-muster');
+const dataFile = join(directory, 'feishu-apps.json');
+export class FeishuRegistry extends EventEmitter {
+    apps = [];
+    clients = new Map();
+    registrations = new Map();
+    writeQueue = Promise.resolve();
+    async load() {
+        try {
+            this.apps = JSON.parse(await readFile(dataFile, 'utf8'));
+        }
+        catch (error) {
+            if (error.code !== 'ENOENT')
+                throw error;
+        }
+    }
+    list() {
+        return this.apps.map(({ clientSecret, conversations, ...app }) => ({
+            ...app, connectionStatus: this.clients.get(app.id)?.ws.getConnectionStatus().state,
+        }));
+    }
+    getBinding(id) {
+        const app = this.apps.find(item => item.id === id);
+        return app ? { project: app.project, agent: app.agent } : undefined;
+    }
+    getSession(id, chatId) {
+        return this.apps.find(item => item.id === id)?.conversations[chatId];
+    }
+    sessionBindings() {
+        return this.apps.flatMap(app => Object.values(app.conversations).map(sessionId => ({ sessionId, appId: app.id })));
+    }
+    async setSession(id, chatId, sessionId) {
+        this.getApp(id).conversations[chatId] = sessionId;
+        await this.persist();
+    }
+    async start() {
+        const id = randomUUID();
+        const registration = { controller: new AbortController(), result: { status: 'WAITING' } };
+        this.registrations.set(id, registration);
+        return new Promise((resolve, reject) => {
+            void registerApp({
+                source: 'agent-muster', createOnly: true, signal: registration.controller.signal,
+                appPreset: { name: 'Agent Muster', desc: '通过飞书与本地编码 Agent 对话' },
+                addons: {
+                    scopes: { tenant: ['im:message:send_as_bot', 'im:message.p2p_msg:readonly', 'im:message.group_at_msg:readonly'] },
+                    events: { items: { tenant: ['im.message.receive_v1'] } },
+                },
+                onQRCodeReady: ({ url, expireIn }) => {
+                    registration.expiresAt = Date.now() + expireIn * 1000;
+                    void QRCode.toString(url, { type: 'svg', margin: 1, width: 220 }).then(svg => {
+                        resolve({ id, verificationUrl: url, qrCode: `data:image/svg+xml;base64,${Buffer.from(svg).toString('base64')}`, expiresAt: registration.expiresAt, interval: 2 });
+                    }).catch(error => { registration.controller.abort(); reject(error); });
+                },
+            }).then(async (result) => {
+                if (result.user_info?.tenant_brand === 'lark')
+                    throw new Error('请使用飞书账号创建应用');
+                const app = {
+                    id: randomUUID(), clientId: result.client_id, clientSecret: result.client_secret,
+                    name: 'Agent Muster', icon: null, project: null, agent: null, conversations: {},
+                };
+                this.apps.push(app);
+                try {
+                    await this.persist();
+                }
+                catch (error) {
+                    this.apps = this.apps.filter(item => item.id !== app.id);
+                    throw error;
+                }
+                this.subscribe(app);
+                registration.result = { status: 'SUCCESS', appId: app.id };
+            }).catch((error) => {
+                registration.result = { status: error.code === 'expired_token' ? 'EXPIRED' : 'FAIL', reason: error.description || error.message };
+                if (!registration.expiresAt)
+                    this.registrations.delete(id);
+                reject(error);
+            });
+        });
+    }
+    poll(id) {
+        const registration = this.registrations.get(id);
+        if (!registration)
+            throw new Error('扫码会话不存在');
+        if (registration.result.status === 'WAITING' && registration.expiresAt && Date.now() >= registration.expiresAt) {
+            registration.result = { status: 'EXPIRED' };
+            registration.controller.abort();
+        }
+        return registration.result;
+    }
+    async cancelRegistration(id) {
+        const registration = this.registrations.get(id);
+        if (!registration)
+            return;
+        registration.controller.abort();
+        this.registrations.delete(id);
+    }
+    startSubscriptions() { for (const app of this.apps)
+        this.subscribe(app); }
+    shutdown() {
+        for (const registration of this.registrations.values())
+            registration.controller.abort();
+        for (const client of this.clients.values())
+            client.ws.close({ force: true });
+        this.clients.clear();
+    }
+    subscribe(app) {
+        if (this.clients.has(app.id))
+            return;
+        const config = { appId: app.clientId, appSecret: app.clientSecret, domain: Domain.Feishu, loggerLevel: LoggerLevel.warn };
+        const api = new Client(config);
+        const ws = new WSClient(config);
+        const eventDispatcher = new EventDispatcher({}).register({
+            'im.message.receive_v1': data => { this.emit('message', { appId: app.id, message: data }); },
+        });
+        this.clients.set(app.id, { api, ws });
+        void ws.start({ eventDispatcher }).catch(error => console.error(`飞书应用 ${app.clientId} 消息监听失败：`, error));
+    }
+    async send(id, chatId, content) {
+        const client = this.clients.get(id);
+        if (!client)
+            throw new Error('飞书应用消息监听未启动');
+        // 按 UTF-8 字节分段，避免超过飞书文本消息大小限制。
+        let chunk = '';
+        let size = 0;
+        const chunks = [];
+        for (const character of content) {
+            const bytes = Buffer.byteLength(character);
+            if (size + bytes > 20000) {
+                chunks.push(chunk);
+                chunk = '';
+                size = 0;
+            }
+            chunk += character;
+            size += bytes;
+        }
+        if (chunk)
+            chunks.push(chunk);
+        for (const text of chunks) {
+            const result = await client.api.im.message.create({
+                params: { receive_id_type: 'chat_id' },
+                data: { receive_id: chatId, msg_type: 'text', content: JSON.stringify({ text }) },
+            });
+            if (result.code !== 0)
+                throw new Error(`飞书消息发送失败：${result.msg} (${result.code})`);
+        }
+    }
+    async sendToSession(id, sessionId, content) {
+        const chatId = Object.entries(this.getApp(id).conversations).find(([, value]) => value === sessionId)?.[0];
+        if (!chatId)
+            throw new Error('此对话尚无飞书发送目标');
+        await this.send(id, chatId, content);
+    }
+    async bind(id, binding, projects, agents) {
+        if (binding.project !== undefined && binding.project !== null && !projects.includes(binding.project))
+            throw new Error('项目不存在');
+        if (binding.agent !== undefined && binding.agent !== null && !agents.includes(binding.agent))
+            throw new Error('Agent 不可用');
+        const app = this.getApp(id);
+        if ((binding.project !== undefined && binding.project !== app.project) || (binding.agent !== undefined && binding.agent !== app.agent))
+            app.conversations = {};
+        if (binding.project !== undefined)
+            app.project = binding.project;
+        if (binding.agent !== undefined)
+            app.agent = binding.agent;
+        await this.persist();
+        return this.list().find(item => item.id === id);
+    }
+    async syncMetadata(id) {
+        const app = this.getApp(id);
+        const client = this.clients.get(id);
+        if (!client)
+            throw new Error('飞书应用消息监听未启动');
+        const result = await client.api.request({ url: '/open-apis/bot/v3/info', method: 'GET' });
+        if (result.code !== 0 || !result.bot)
+            throw new Error(`飞书应用信息获取失败：${result.msg}`);
+        app.name = result.bot.app_name;
+        app.icon = result.bot.avatar_url;
+        await this.persist();
+        return this.list().find(item => item.id === id);
+    }
+    async remove(id) {
+        this.getApp(id);
+        this.clients.get(id)?.ws.close({ force: true });
+        this.clients.delete(id);
+        this.apps = this.apps.filter(item => item.id !== id);
+        await this.persist();
+    }
+    async unbindSession(sessionId) {
+        let changed = false;
+        for (const app of this.apps) {
+            for (const [chatId, id] of Object.entries(app.conversations)) {
+                if (id !== sessionId)
+                    continue;
+                delete app.conversations[chatId];
+                changed = true;
+            }
+        }
+        if (changed)
+            await this.persist();
+    }
+    async unbindProject(project) {
+        const apps = this.apps.filter(app => app.project === project);
+        for (const app of apps) {
+            app.project = null;
+            app.conversations = {};
+        }
+        if (apps.length)
+            await this.persist();
+    }
+    getApp(id) {
+        const app = this.apps.find(item => item.id === id);
+        if (!app)
+            throw new Error('飞书应用不存在');
+        return app;
+    }
+    persist() {
+        const snapshot = JSON.stringify(this.apps);
+        const write = this.writeQueue.then(async () => {
+            await mkdir(directory, { recursive: true, mode: 0o700 });
+            await writeFile(`${dataFile}.tmp`, snapshot, { mode: 0o600 });
+            await rename(`${dataFile}.tmp`, dataFile);
+        });
+        this.writeQueue = write.catch(() => { });
+        return write;
+    }
+}

package/dist/server.js

@@ -26,7 +26,7 @@ async function chooseProjectDirectory() {
         throw error;
     }
 }
-export async function startServer(port, registry, projects, sessions, dingtalk) {
+export async function startServer(port, registry, projects, sessions, dingtalk, feishu, dev = false) {
     const app = express();
     app.disable('x-powered-by');
     app.use((request, response, next) => {
@@ -75,8 +75,10 @@ export async function startServer(port, registry, projects, sessions, dingtalk)
                 return;
             }
             await dingtalk.unbindProject(path);
+            await feishu.unbindProject(path);
             for (const session of projectSessions) {
                 await dingtalk.unbindSession(session.id);
+                await feishu.unbindSession(session.id);
                 await sessions.remove(session.id);
             }
             if (registered)
@@ -151,6 +153,61 @@ export async function startServer(port, registry, projects, sessions, dingtalk)
             response.status(500).json({ error: error.message });
         }
     });
+    app.get('/api/feishu/apps', (_request, response) => response.json(feishu.list()));
+    app.post('/api/feishu/apps/:id/sync', async (request, response) => {
+        try {
+            response.json(await feishu.syncMetadata(request.params.id));
+        }
+        catch (error) {
+            response.status(502).json({ error: error.message });
+        }
+    });
+    app.post('/api/feishu/registration', async (_request, response) => {
+        try {
+            response.status(201).json(await feishu.start());
+        }
+        catch (error) {
+            response.status(502).json({ error: error.message });
+        }
+    });
+    app.get('/api/feishu/registration/:id', async (request, response) => {
+        try {
+            response.json(await feishu.poll(request.params.id));
+        }
+        catch (error) {
+            response.status(502).json({ error: error.message });
+        }
+    });
+    app.patch('/api/feishu/apps/:id', async (request, response) => {
+        try {
+            const { project, agent } = request.body || {};
+            if (project !== undefined && project !== null && typeof project !== 'string')
+                throw new Error('项目无效');
+            if (agent !== undefined && agent !== null && typeof agent !== 'string')
+                throw new Error('Agent 无效');
+            response.json(await feishu.bind(request.params.id, { project, agent }, projects.list(), registry.list().filter(item => item.installed).map(item => item.id)));
+        }
+        catch (error) {
+            response.status(400).json({ error: error.message });
+        }
+    });
+    app.delete('/api/feishu/apps/:id', async (request, response) => {
+        try {
+            if (!feishu.getBinding(request.params.id)) {
+                response.status(404).json({ error: '飞书应用不存在' });
+                return;
+            }
+            await feishu.remove(request.params.id);
+            response.json({ id: request.params.id });
+        }
+        catch (error) {
+            response.status(500).json({ error: error.message });
+        }
+    });
+    app.delete('/api/feishu/registration/:id', async (request, response) => {
+        await feishu.cancelRegistration(request.params.id);
+        response.status(204).end();
+    });
     app.get('/api/sessions', (_request, response) => response.json(sessions.list()));
     app.get('/api/sessions/:id', (request, response) => {
         const session = sessions.get(request.params.id);
@@ -230,6 +287,24 @@ export async function startServer(port, registry, projects, sessions, dingtalk)
             response.status(400).json({ error: error.message });
         }
     });
+    app.post('/api/sessions/:id/feishu/messages', async (request, response) => {
+        try {
+            const { content } = request.body || {};
+            if (typeof content !== 'string' || !content.trim())
+                throw new Error('请输入消息内容');
+            const session = sessions.get(request.params.id);
+            if (!session)
+                throw new Error('会话不存在');
+            if (!session.feishuAppId)
+                throw new Error('此对话未关联飞书应用');
+            await feishu.sendToSession(session.feishuAppId, session.id, content.trim());
+            response.json(sessions.recordFeishuMessage(session.id, content.trim()));
+        }
+        catch (error) {
+            console.error(`飞书会话 ${request.params.id} 消息发送失败：`, error);
+            response.status(400).json({ error: error.message });
+        }
+    });
     app.post('/api/sessions/:id/stop', (request, response) => {
         try {
             response.json(sessions.stop(request.params.id));
@@ -245,6 +320,7 @@ export async function startServer(port, registry, projects, sessions, dingtalk)
                 return;
             }
             await dingtalk.unbindSession(request.params.id);
+            await feishu.unbindSession(request.params.id);
             await sessions.remove(request.params.id);
             response.json({ id: request.params.id });
         }
@@ -272,9 +348,21 @@ export async function startServer(port, registry, projects, sessions, dingtalk)
         const heartbeat = setInterval(() => response.write(': keepalive\n\n'), 15000);
         request.on('close', () => { clearInterval(heartbeat); sessions.off(session.id, listener); });
     });
-    app.use(express.static(webRoot));
-    app.get('/{*path}', (_request, response) => response.sendFile(join(webRoot, 'index.html')));
     const server = createServer(app);
+    if (dev) {
+        const { createServer: createViteServer } = await import('vite');
+        const vite = await createViteServer({
+            configFile: resolve(webRoot, '../vite.config.ts'),
+            root: resolve(webRoot, '..'),
+            server: { middlewareMode: true, hmr: { server } },
+        });
+        app.use(vite.middlewares);
+        server.once('close', () => { void vite.close(); });
+    }
+    else {
+        app.use(express.static(webRoot));
+        app.get('/{*path}', (_request, response) => response.sendFile(join(webRoot, 'index.html')));
+    }
     await new Promise((done, reject) => {
         server.once('error', reject);
         server.listen(port, '127.0.0.1', done);

package/dist/sessions.js

@@ -91,6 +91,13 @@ export class SessionManager extends EventEmitter {
         session.dingtalkAppId = appId;
         this.save(session);
     }
+    linkFeishuSession(id, appId) {
+        const session = this.sessions.get(id);
+        if (!session || session.feishuAppId === appId)
+            return;
+        session.feishuAppId = appId;
+        this.save(session);
+    }
     isRunning(id) {
         return this.processes.has(id);
     }
@@ -99,7 +106,7 @@ export class SessionManager extends EventEmitter {
         this.writeQueue = this.writeQueue.then(() => writeSession(session.id, snapshot))
             .catch(error => console.error('Failed to save session:', error));
     }
-    append(session, type, text, detail, kind, comparison, dingtalkConversation) {
+    append(session, type, text, detail, kind, comparison, dingtalkConversation, feishuConversation) {
         const event = {
             id: (session.events.at(-1)?.id || 0) + 1,
             type,
@@ -109,6 +116,7 @@ export class SessionManager extends EventEmitter {
             ...(comparison === undefined ? {} : { comparison }),
             timestamp: Date.now(),
             ...(dingtalkConversation ? { dingtalkConversation } : {}),
+            ...(feishuConversation ? { feishuConversation } : {}),
         };
         session.events.push(event);
         if (session.events.length > 500)
@@ -128,7 +136,7 @@ export class SessionManager extends EventEmitter {
         };
         this.append(session, 'usage', JSON.stringify(session.usage));
     }
-    async create(agent, cwd, prompt, model, dingtalkAppId, dingtalkConversation, images) {
+    async create(agent, cwd, prompt, model, dingtalkAppId, dingtalkConversation, images, feishu) {
         if (!adapters[agent])
             throw new Error('Unsupported agent');
         if (!prompt.trim())
@@ -141,6 +149,7 @@ export class SessionManager extends EventEmitter {
             id: `ses_${randomUUID()}`,
             agent,
             ...(dingtalkAppId ? { dingtalkAppId } : {}),
+            ...(feishu ? { feishuAppId: feishu.appId, feishuConversation: feishu.conversation } : {}),
             ...(dingtalkConversation ? { dingtalkConversation } : {}),
             model,
             cwd: resolve(cwd),
@@ -151,11 +160,11 @@ export class SessionManager extends EventEmitter {
             events: [],
         };
         this.sessions.set(session.id, session);
-        this.append(session, 'message', session.prompt, undefined, undefined, undefined, dingtalkConversation);
+        this.append(session, 'message', session.prompt, undefined, undefined, undefined, dingtalkConversation, feishu?.conversation);
         this.start(session, session.prompt, images);
         return session;
     }
-    send(id, message, model, dingtalkConversation, images) {
+    send(id, message, model, dingtalkConversation, images, feishuConversation) {
         const session = this.sessions.get(id);
         if (!session)
             throw new Error('Session not found');
@@ -169,7 +178,9 @@ export class SessionManager extends EventEmitter {
             session.model = model || undefined;
         if (dingtalkConversation)
             session.dingtalkConversation = dingtalkConversation;
-        this.append(session, 'message', message.trim(), undefined, undefined, undefined, dingtalkConversation);
+        if (feishuConversation)
+            session.feishuConversation = feishuConversation;
+        this.append(session, 'message', message.trim(), undefined, undefined, undefined, dingtalkConversation, feishuConversation);
         this.start(session, message.trim(), images);
         return session;
     }
@@ -180,6 +191,13 @@ export class SessionManager extends EventEmitter {
         this.append(session, 'dingtalk_message', content);
         return session;
     }
+    recordFeishuMessage(id, content) {
+        const session = this.sessions.get(id);
+        if (!session)
+            throw new Error('会话不存在');
+        this.append(session, 'feishu_message', content);
+        return session;
+    }
     stop(id) {
         const session = this.sessions.get(id);
         if (!session)

package/dist/updates.js

@@ -0,0 +1,40 @@
+import { execFile, spawn } from 'node:child_process';
+import { readFile } from 'node:fs/promises';
+import { promisify } from 'node:util';
+import { gt, valid } from 'semver';
+const execute = promisify(execFile);
+const npmCommand = process.platform === 'win32' ? 'npm.cmd' : 'npm';
+const shell = process.platform === 'win32';
+async function getUpdate() {
+    const { name, version } = JSON.parse(await readFile(new URL('../package.json', import.meta.url), 'utf8'));
+    const { stdout } = await execute(npmCommand, ['view', `${name}@latest`, 'version', '--json', '--fetch-timeout=5000', '--fetch-retries=0'], { shell, timeout: 10000 });
+    const latest = JSON.parse(stdout);
+    if (typeof latest !== 'string' || !valid(latest))
+        throw new Error('npm 返回的版本号无效');
+    return { name, current: version, latest };
+}
+export async function checkForUpdates() {
+    const { name, current, latest } = await getUpdate();
+    if (!gt(latest, current))
+        return;
+    console.log(`\n发现新版本：${current} → ${latest}\n运行 ${name} update 更新全局安装，或运行 npx ${name}@latest 启动最新版本。\n`);
+}
+export async function updatePackage() {
+    const { name, current, latest } = await getUpdate();
+    if (!gt(latest, current)) {
+        console.log(`当前版本 ${current} 已是最新版本。`);
+        return;
+    }
+    console.log(`正在更新 ${name}：${current} → ${latest}`);
+    await new Promise((resolve, reject) => {
+        const child = spawn(npmCommand, ['install', '--global', `${name}@${latest}`], { shell, stdio: 'inherit' });
+        child.once('error', reject);
+        child.once('close', (code, signal) => {
+            if (code === 0)
+                resolve();
+            else
+                reject(new Error(`npm 更新失败（${signal ? `信号：${signal}` : `退出码：${code}`}）`));
+        });
+    });
+    console.log(`已更新至 ${latest}，请重新启动 Agent Muster。`);
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "agent-muster",
-  "version": "0.1.1",
-  "description": "A local web interface for managing coding agents, projects, sessions, and DingTalk integrations.",
+  "version": "0.1.2",
+  "description": "A local web interface for managing coding agents, projects, sessions, and DingTalk/Feishu integrations.",
   "author": "sexyAmole",
   "license": "UNLICENSED",
   "homepage": "https://github.com/sexyAmole/agent-muster",
@@ -17,7 +17,8 @@
     "coding-agent",
     "codex",
     "claude-code",
-    "dingtalk"
+    "dingtalk",
+    "feishu"
   ],
   "type": "module",
   "packageManager": "pnpm@10.33.4",
@@ -25,7 +26,7 @@
     "agent-muster": "./bin/agent-muster.js"
   },
   "scripts": {
-    "dev": "pnpm run build && tsx watch src/cli.ts web --no-open",
+    "dev": "tsx watch src/cli.ts web --dev --no-open",
     "build": "tsc -p tsconfig.json && vite build --config web/vite.config.ts",
     "prepack": "pnpm run build",
     "typecheck": "tsc -p tsconfig.json --noEmit && tsc -p web/tsconfig.json --noEmit",
@@ -34,9 +35,11 @@
   "files": [
     "bin",
     "dist",
-    "web/dist"
+    "web/dist",
+    "README.zh-CN.md"
   ],
   "dependencies": {
+    "@larksuiteoapi/node-sdk": "^1.74.0",
     "dingtalk-stream": "2.1.6-beta.1",
     "express": "^5.1.0",
     "qrcode": "^1.5.4",
@@ -44,6 +47,7 @@
     "react-dom": "^19.1.0",
     "react-markdown": "^10.1.0",
     "remark-gfm": "^4.0.1",
+    "semver": "^7.8.5",
     "undici": "^8.11.2"
   },
   "devDependencies": {
@@ -52,6 +56,7 @@
     "@types/qrcode": "^1.5.6",
     "@types/react": "^19.1.0",
     "@types/react-dom": "^19.1.0",
+    "@types/semver": "^7.8.0",
     "@vitejs/plugin-react": "^4.7.0",
     "tsx": "^4.20.0",
     "typescript": "^5.8.0",