evolclaw 2.2.0 → 2.3.0

package/dist/utils/init-channel.js

@@ -0,0 +1,649 @@
+/**
+ * Channel-specific init flows: Feishu, WeChat, AUN
+ *
+ * Each channel provides:
+ *   - cmdInit<Channel>()  — standalone `evolclaw init <channel>` entry
+ *   - run<Channel>QrFlow() or setupAun*() — reusable primitives for the main init wizard
+ */
+import fs from 'fs';
+import readline from 'readline';
+import path from 'path';
+import os from 'os';
+import { resolvePaths } from '../paths.js';
+import { normalizeChannelInstances } from '../config.js';
+import { selectInstance } from './init.js';
+function ask(rl, question) {
+    return new Promise(resolve => rl.question(question, resolve));
+}
+// ==================== Feishu ====================
+const FEISHU_PROD_URL = 'https://accounts.feishu.cn';
+const LARK_PROD_URL = 'https://accounts.larksuite.com';
+const SKIP = Symbol('SKIP');
+const QUIT = Symbol('QUIT');
+class FeishuQrRegistrationClient {
+    baseUrl;
+    constructor(isLark = false) {
+        this.baseUrl = isLark ? LARK_PROD_URL : FEISHU_PROD_URL;
+    }
+    setDomain(isLark) {
+        this.baseUrl = isLark ? LARK_PROD_URL : FEISHU_PROD_URL;
+    }
+    async init() {
+        return this.postRegistration('init', {});
+    }
+    async begin() {
+        return this.postRegistration('begin', {
+            archetype: 'PersonalAgent',
+            auth_method: 'client_secret',
+            request_user_info: 'open_id',
+        });
+    }
+    async poll(deviceCode) {
+        return this.postRegistration('poll', { device_code: deviceCode });
+    }
+    async postRegistration(action, extraParams) {
+        const body = new URLSearchParams({ action, ...extraParams }).toString();
+        const res = await fetch(`${this.baseUrl}/oauth/v1/app/registration`, {
+            method: 'POST',
+            headers: { 'content-type': 'application/x-www-form-urlencoded' },
+            body,
+        });
+        const text = await res.text();
+        if (!text)
+            return {};
+        return JSON.parse(text);
+    }
+}
+async function runQrRegistrationFlow() {
+    const client = new FeishuQrRegistrationClient();
+    const initResult = await client.init();
+    const authMethods = Array.isArray(initResult.supported_auth_methods) ? initResult.supported_auth_methods : [];
+    if (!authMethods.includes('client_secret')) {
+        throw new Error('当前环境不支持 client_secret 注册');
+    }
+    const beginResult = await client.begin();
+    if (!beginResult.verification_uri_complete || !beginResult.device_code) {
+        throw new Error('服务端未返回扫码链接或 device_code');
+    }
+    // 显示二维码
+    try {
+        const qrterm = await import('qrcode-terminal');
+        await new Promise(resolve => {
+            qrterm.default.generate(beginResult.verification_uri_complete, { small: true }, (qr) => {
+                console.log(qr);
+                resolve();
+            });
+        });
+    }
+    catch {
+        console.log(`请在浏览器中打开此链接扫码: ${beginResult.verification_uri_complete}\n`);
+    }
+    console.log('请用飞书/Lark 扫描上方二维码...\n');
+    console.log('按 q 退出 | 按 s 跳过扫码手动输入 appId/appSecret\n');
+    let userAction = null;
+    const setupKeyListener = () => {
+        if (!process.stdin.isTTY)
+            return () => { };
+        process.stdin.setRawMode(true);
+        process.stdin.resume();
+        process.stdin.setEncoding('utf8');
+        const handler = (key) => {
+            if (key === 'q' || key === '\u0003')
+                userAction = QUIT;
+            if (key === 's')
+                userAction = SKIP;
+        };
+        process.stdin.on('data', handler);
+        return () => {
+            process.stdin.removeListener('data', handler);
+            process.stdin.setRawMode(false);
+            process.stdin.pause();
+        };
+    };
+    const cleanup = setupKeyListener();
+    const startedAt = Date.now();
+    let pollIntervalSeconds = Number(beginResult.interval ?? 5);
+    const expireInSeconds = Number(beginResult.expires_in ?? beginResult.expire_in ?? 600);
+    let domainResolved = false;
+    let currentDomain = 'feishu';
+    try {
+        while (Date.now() - startedAt < expireInSeconds * 1000) {
+            if (userAction === QUIT)
+                return QUIT;
+            if (userAction === SKIP)
+                return SKIP;
+            const pollResult = await client.poll(beginResult.device_code);
+            if (pollResult.user_info?.tenant_brand === 'lark' && !domainResolved) {
+                client.setDomain(true);
+                currentDomain = 'lark';
+                domainResolved = true;
+            }
+            if (pollResult.client_id && pollResult.client_secret) {
+                return {
+                    appId: pollResult.client_id,
+                    appSecret: pollResult.client_secret,
+                    domain: currentDomain,
+                    openId: pollResult.user_info?.open_id ?? '',
+                };
+            }
+            if (pollResult.error === 'authorization_pending') {
+                await new Promise(r => setTimeout(r, pollIntervalSeconds * 1000));
+                continue;
+            }
+            if (pollResult.error === 'slow_down') {
+                pollIntervalSeconds += 5;
+                await new Promise(r => setTimeout(r, pollIntervalSeconds * 1000));
+                continue;
+            }
+            if (pollResult.error === 'access_denied') {
+                throw new Error('用户拒绝了扫码授权');
+            }
+            if (pollResult.error === 'expired_token') {
+                throw new Error('扫码会话已过期');
+            }
+            if (pollResult.error) {
+                throw new Error(`扫码注册失败: ${pollResult.error}${pollResult.error_description ? ` - ${pollResult.error_description}` : ''}`);
+            }
+            await new Promise(r => setTimeout(r, pollIntervalSeconds * 1000));
+        }
+        throw new Error('等待扫码结果超时');
+    }
+    finally {
+        cleanup();
+    }
+}
+async function manualInput(rl) {
+    console.log('\n手动输入模式：\n');
+    let appId = '';
+    while (!appId) {
+        appId = (await ask(rl, '  飞书 App ID: ')).trim();
+        if (!appId)
+            console.log('  ⚠ 不能为空');
+    }
+    let appSecret = '';
+    while (!appSecret) {
+        appSecret = (await ask(rl, '  飞书 App Secret: ')).trim();
+        if (!appSecret)
+            console.log('  ⚠ 不能为空');
+    }
+    return { appId, appSecret, domain: 'unknown', openId: '' };
+}
+export async function runFeishuQrFlow() {
+    try {
+        const result = await runQrRegistrationFlow();
+        if (result === QUIT || result === SKIP)
+            return null;
+        return result;
+    }
+    catch (error) {
+        console.error(`\n登录失败: ${error instanceof Error ? error.message : error}`);
+        return null;
+    }
+}
+export async function cmdInitFeishu() {
+    const p = resolvePaths();
+    if (!fs.existsSync(p.config)) {
+        console.log(`❌ 配置文件不存在，请先运行 evolclaw init`);
+        return;
+    }
+    const config = JSON.parse(fs.readFileSync(p.config, 'utf-8'));
+    // Normalize existing instances and filter out placeholders
+    const allInstances = normalizeChannelInstances(config.channels?.feishu, 'feishu');
+    const validInstances = [];
+    for (let i = 0; i < allInstances.length; i++) {
+        const inst = allInstances[i];
+        if (!inst.appId || !inst.appSecret)
+            continue;
+        if (inst.appId.includes('your-') || inst.appId.includes('placeholder'))
+            continue;
+        if (inst.appSecret.includes('your-') || inst.appSecret.includes('placeholder'))
+            continue;
+        validInstances.push({ ...inst, originalIndex: i });
+    }
+    let choice = null;
+    if (validInstances.length > 0) {
+        const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+        try {
+            choice = await selectInstance(rl, 'feishu', validInstances);
+            if (choice === null)
+                return; // user cancelled
+        }
+        finally {
+            rl.close();
+        }
+    }
+    console.log('正在获取飞书登录二维码...\n');
+    let result;
+    try {
+        const flowResult = await runQrRegistrationFlow();
+        if (flowResult === QUIT) {
+            console.log('已退出');
+            return;
+        }
+        if (flowResult === SKIP) {
+            const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+            try {
+                result = await manualInput(rl);
+            }
+            finally {
+                rl.close();
+            }
+        }
+        else {
+            result = flowResult;
+        }
+    }
+    catch (error) {
+        console.error(`\n登录失败: ${error instanceof Error ? error.message : error}`);
+        process.exit(1);
+    }
+    // Write config to the correct slot
+    if (!config.channels)
+        config.channels = {};
+    if (choice && choice.action === 'overwrite' && Array.isArray(config.channels.feishu)) {
+        // Overwrite existing instance in array — use originalIndex to find the right slot
+        const idx = validInstances[choice.index]?.originalIndex ?? choice.index;
+        config.channels.feishu[idx].appId = result.appId;
+        config.channels.feishu[idx].appSecret = result.appSecret;
+        config.channels.feishu[idx].enabled = true;
+        if (result.openId)
+            config.channels.feishu[idx].owner = result.openId;
+    }
+    else if (choice && choice.action === 'overwrite' && !Array.isArray(config.channels.feishu)) {
+        // Overwrite single-object
+        config.channels.feishu = config.channels.feishu || {};
+        config.channels.feishu.appId = result.appId;
+        config.channels.feishu.appSecret = result.appSecret;
+        config.channels.feishu.enabled = true;
+        if (result.openId)
+            config.channels.feishu.owner = result.openId;
+        else
+            delete config.channels.feishu.owner;
+    }
+    else if (choice && choice.action === 'add') {
+        // Add new instance — upgrade to array if needed
+        const newInst = {
+            name: choice.name,
+            appId: result.appId,
+            appSecret: result.appSecret,
+            enabled: true,
+            ...(result.openId ? { owner: result.openId } : {}),
+        };
+        if (Array.isArray(config.channels.feishu)) {
+            config.channels.feishu.push(newInst);
+        }
+        else if (config.channels.feishu) {
+            // Upgrade single object to array
+            const oldInst = { ...config.channels.feishu, name: config.channels.feishu.name || 'feishu' };
+            config.channels.feishu = [oldInst, newInst];
+        }
+        else {
+            config.channels.feishu = [newInst];
+        }
+    }
+    else {
+        // First instance — single object format (backward compat)
+        config.channels.feishu = config.channels.feishu || {};
+        config.channels.feishu.appId = result.appId;
+        config.channels.feishu.appSecret = result.appSecret;
+        config.channels.feishu.enabled = true;
+        if (result.openId)
+            config.channels.feishu.owner = result.openId;
+        else
+            delete config.channels.feishu.owner;
+    }
+    if (!config.channels.defaultChannel)
+        config.channels.defaultChannel = 'feishu';
+    fs.writeFileSync(p.config, JSON.stringify(config, null, 2) + '\n');
+    console.log(`\n✅ 飞书连接成功！`);
+    console.log(`  App ID: ${result.appId}`);
+    if (result.openId) {
+        console.log(`  Owner: ${result.openId}`);
+    }
+    if (result.domain !== 'unknown') {
+        console.log(`  Domain: ${result.domain}`);
+    }
+    if (choice) {
+        console.log(`  实例: ${choice.name} (${choice.action === 'add' ? '新增' : '覆盖'})`);
+    }
+    console.log(`  配置已写入: ${p.config}`);
+    console.log(`\n现在可以启动服务: evolclaw restart`);
+}
+// ==================== WeChat ====================
+const DEFAULT_BASE_URL = 'https://ilinkai.weixin.qq.com';
+const BOT_TYPE = '3';
+const QR_POLL_TIMEOUT_MS = 35_000;
+const LOGIN_TIMEOUT_MS = 480_000;
+async function fetchQRCode(baseUrl) {
+    const base = baseUrl.endsWith('/') ? baseUrl : `${baseUrl}/`;
+    const url = `${base}ilink/bot/get_bot_qrcode?bot_type=${BOT_TYPE}`;
+    const res = await fetch(url);
+    if (!res.ok)
+        throw new Error(`QR fetch failed: ${res.status}`);
+    return (await res.json());
+}
+async function pollQRStatus(baseUrl, qrcode) {
+    const base = baseUrl.endsWith('/') ? baseUrl : `${baseUrl}/`;
+    const url = `${base}ilink/bot/get_qrcode_status?qrcode=${encodeURIComponent(qrcode)}`;
+    const controller = new AbortController();
+    const timer = setTimeout(() => controller.abort(), QR_POLL_TIMEOUT_MS);
+    try {
+        const res = await fetch(url, {
+            headers: { 'iLink-App-ClientVersion': '1' },
+            signal: controller.signal,
+        });
+        clearTimeout(timer);
+        if (!res.ok)
+            throw new Error(`QR status failed: ${res.status}`);
+        return (await res.json());
+    }
+    catch (err) {
+        clearTimeout(timer);
+        if (err instanceof Error && err.name === 'AbortError') {
+            return { status: 'wait' };
+        }
+        throw err;
+    }
+}
+export async function runWechatQrFlow() {
+    const qrResp = await fetchQRCode(DEFAULT_BASE_URL);
+    try {
+        const qrterm = await import('qrcode-terminal');
+        await new Promise(resolve => {
+            qrterm.default.generate(qrResp.qrcode_img_content, { small: true }, (qr) => {
+                console.log(qr);
+                resolve();
+            });
+        });
+    }
+    catch {
+        console.log(`请在浏览器中打开此链接扫码: ${qrResp.qrcode_img_content}\n`);
+    }
+    console.log('请用微信扫描上方二维码...\n');
+    const deadline = Date.now() + LOGIN_TIMEOUT_MS;
+    let scannedPrinted = false;
+    while (Date.now() < deadline) {
+        const status = await pollQRStatus(DEFAULT_BASE_URL, qrResp.qrcode);
+        switch (status.status) {
+            case 'wait':
+                process.stdout.write('.');
+                break;
+            case 'scaned':
+                if (!scannedPrinted) {
+                    console.log('\n👀 已扫码，请在微信中确认...');
+                    scannedPrinted = true;
+                }
+                break;
+            case 'expired':
+                console.error('\n二维码已过期');
+                return null;
+            case 'confirmed':
+                if (!status.ilink_bot_id || !status.bot_token) {
+                    console.error('\n登录失败：服务器未返回完整信息');
+                    return null;
+                }
+                return {
+                    baseUrl: status.baseurl || DEFAULT_BASE_URL,
+                    token: status.bot_token,
+                };
+        }
+        await new Promise(r => setTimeout(r, 1000));
+    }
+    console.log('\n登录超时');
+    return null;
+}
+export async function cmdInitWechat() {
+    const p = resolvePaths();
+    if (!fs.existsSync(p.config)) {
+        console.log(`❌ 配置文件不存在，请先运行 evolclaw init`);
+        return;
+    }
+    const config = JSON.parse(fs.readFileSync(p.config, 'utf-8'));
+    // Normalize existing instances and filter out placeholders
+    const allInstances = normalizeChannelInstances(config.channels?.wechat, 'wechat');
+    const validInstances = [];
+    for (let i = 0; i < allInstances.length; i++) {
+        const inst = allInstances[i];
+        if (!inst.token)
+            continue;
+        if (inst.token.includes('your-') || inst.token.includes('placeholder'))
+            continue;
+        validInstances.push({ ...inst, originalIndex: i });
+    }
+    let choice = null;
+    if (validInstances.length > 0) {
+        const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+        try {
+            choice = await selectInstance(rl, 'wechat', validInstances);
+            if (choice === null)
+                return; // user cancelled
+        }
+        finally {
+            rl.close();
+        }
+    }
+    console.log('正在获取微信登录二维码...\n');
+    const qrResp = await fetchQRCode(DEFAULT_BASE_URL);
+    // 终端显示二维码
+    try {
+        const qrterm = await import('qrcode-terminal');
+        await new Promise(resolve => {
+            qrterm.default.generate(qrResp.qrcode_img_content, { small: true }, (qr) => {
+                console.log(qr);
+                resolve();
+            });
+        });
+    }
+    catch {
+        console.log(`请在浏览器中打开此链接扫码: ${qrResp.qrcode_img_content}\n`);
+    }
+    console.log('请用微信扫描上方二维码...\n');
+    const deadline = Date.now() + LOGIN_TIMEOUT_MS;
+    let scannedPrinted = false;
+    while (Date.now() < deadline) {
+        const status = await pollQRStatus(DEFAULT_BASE_URL, qrResp.qrcode);
+        switch (status.status) {
+            case 'wait':
+                process.stdout.write('.');
+                break;
+            case 'scaned':
+                if (!scannedPrinted) {
+                    console.log('\n\ud83d\udc40 已扫码，请在微信中确认...');
+                    scannedPrinted = true;
+                }
+                break;
+            case 'expired':
+                console.log('\n二维码已过期，请重新运行 evolclaw init wechat');
+                process.exit(1);
+                break;
+            case 'confirmed': {
+                if (!status.ilink_bot_id || !status.bot_token) {
+                    console.error('\n登录失败：服务器未返回完整信息');
+                    process.exit(1);
+                }
+                const baseUrl = status.baseurl || DEFAULT_BASE_URL;
+                const token = status.bot_token;
+                // Write config to the correct slot
+                if (!config.channels)
+                    config.channels = {};
+                if (choice && choice.action === 'overwrite' && Array.isArray(config.channels.wechat)) {
+                    // Overwrite existing instance in array — use originalIndex to find the right slot
+                    const idx = validInstances[choice.index]?.originalIndex ?? choice.index;
+                    config.channels.wechat[idx].enabled = true;
+                    config.channels.wechat[idx].baseUrl = baseUrl;
+                    config.channels.wechat[idx].token = token;
+                }
+                else if (choice && choice.action === 'overwrite' && !Array.isArray(config.channels.wechat)) {
+                    // Overwrite single-object
+                    config.channels.wechat = config.channels.wechat || {};
+                    config.channels.wechat.enabled = true;
+                    config.channels.wechat.baseUrl = baseUrl;
+                    config.channels.wechat.token = token;
+                }
+                else if (choice && choice.action === 'add') {
+                    // Add new instance — upgrade to array if needed
+                    const newInst = {
+                        name: choice.name,
+                        enabled: true,
+                        baseUrl,
+                        token,
+                    };
+                    if (Array.isArray(config.channels.wechat)) {
+                        config.channels.wechat.push(newInst);
+                    }
+                    else if (config.channels.wechat) {
+                        // Upgrade single object to array
+                        const oldInst = { ...config.channels.wechat, name: config.channels.wechat.name || 'wechat' };
+                        config.channels.wechat = [oldInst, newInst];
+                    }
+                    else {
+                        config.channels.wechat = [newInst];
+                    }
+                }
+                else {
+                    // First instance — single object format (backward compat)
+                    config.channels.wechat = {
+                        enabled: true,
+                        baseUrl,
+                        token,
+                    };
+                }
+                if (!config.channels.defaultChannel)
+                    config.channels.defaultChannel = 'wechat';
+                fs.writeFileSync(p.config, JSON.stringify(config, null, 2) + '\n');
+                console.log(`\n✅ 微信连接成功！`);
+                console.log(`  Bot ID: ${status.ilink_bot_id}`);
+                console.log(`  User ID: ${status.ilink_user_id}`);
+                if (choice) {
+                    console.log(`  实例: ${choice.name} (${choice.action === 'add' ? '新增' : '覆盖'})`);
+                }
+                console.log(`  配置已写入: ${p.config}`);
+                console.log(`\n现在可以启动服务: evolclaw restart`);
+                return;
+            }
+        }
+        await new Promise(r => setTimeout(r, 1000));
+    }
+    console.log('\n登录超时，请重新运行');
+    process.exit(1);
+}
+// ==================== AUN ====================
+export async function checkAunEnvironment(rl) {
+    console.log('\n🔍 AUN 环境检查...\n');
+    // TS SDK (@eleans/aun-core-node) is bundled as npm dependency — no external deps needed
+    console.log('  ✓ @eleans/aun-core-node (TS SDK)');
+    console.log('');
+    return true;
+}
+function isValidAid(name) {
+    const labels = name.split('.');
+    return labels.length >= 3 && labels.every(l => /^[a-zA-Z0-9]([a-zA-Z0-9-]*[a-zA-Z0-9])?$/.test(l));
+}
+export async function setupAunAid(rl, _config) {
+    let aid = '';
+    let gatewayPort;
+    // Outer loop: allows retrying with a different AID
+    while (true) {
+        // Ask AID with format validation
+        aid = '';
+        while (!aid) {
+            aid = (await ask(rl, '  AUN Agent ID (例: mybot.agentid.pub): ')).trim();
+            if (!aid) {
+                console.log('  ⚠ 不能为空');
+                continue;
+            }
+            if (!isValidAid(aid)) {
+                console.log('  ⚠ 无效 AID 格式（需要合法域名，至少三级，如 alice.agentid.pub）');
+                aid = '';
+            }
+        }
+        const portStr = (await ask(rl, '  Gateway 端口 [留空使用默认 443]: ')).trim();
+        gatewayPort = portStr ? parseInt(portStr, 10) : undefined;
+        if (gatewayPort !== undefined && (isNaN(gatewayPort) || gatewayPort < 1 || gatewayPort > 65535)) {
+            console.log('  ⚠ 端口号无效，使用默认 443');
+            gatewayPort = undefined;
+        }
+        // Check if AID exists locally
+        const aunPath = path.join(os.homedir(), '.aun');
+        const aidDir = path.join(aunPath, 'AIDs', aid);
+        if (fs.existsSync(aidDir) && fs.existsSync(path.join(aidDir, 'private'))) {
+            console.log(`  ✓ AID ${aid} 已存在`);
+            break;
+        }
+        const answer = (await ask(rl, `  ⚠ AID ${aid} 本地不存在，是否创建？[Y/n] `)).trim().toLowerCase();
+        if (answer === 'n' || answer === 'no') {
+            console.log('  已跳过 AID 创建（启动时可能连接失败）');
+            break;
+        }
+        // Create AID using TS SDK directly
+        console.log('  正在创建 AID...');
+        let failed = false;
+        try {
+            const { AUNClient } = await import('@eleans/aun-core-node');
+            const client = new AUNClient({ aun_path: aunPath });
+            // Set gateway URL from AID domain + port
+            const domain = aid.split('.').slice(1).join('.');
+            const port = gatewayPort || 443;
+            client._gatewayUrl = `wss://gateway.${domain}:${port}/aun`;
+            const result = await client.auth.createAid({ aid });
+            console.log(`  ✓ AID ${result.aid} 创建成功`);
+            try {
+                await client.close();
+            }
+            catch { /* ignore */ }
+        }
+        catch (e) {
+            const msg = e.message || String(e);
+            console.log(`  ✗ AID 创建失败: ${msg.slice(0, 200)}`);
+            failed = true;
+        }
+        if (!failed)
+            break;
+        // Creation failed — retry or give up
+        const retry = (await ask(rl, '  → 重新输入 (r) / 跳过 (s) / 取消 (c)？[r/s/c] ')).trim().toLowerCase();
+        if (retry === 'c')
+            return null;
+        if (retry === 's')
+            break;
+        // default: retry with new AID
+    }
+    return { aid, gatewayPort };
+}
+export async function cmdInitAun() {
+    const p = resolvePaths();
+    if (!fs.existsSync(p.config)) {
+        console.log('❌ 配置文件不存在，请先运行 evolclaw init');
+        return;
+    }
+    const config = JSON.parse(fs.readFileSync(p.config, 'utf-8'));
+    const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+    try {
+        if (config.channels?.aun?.aid) {
+            const answer = (await ask(rl, '已有 AUN 配置，是否重新配置？[y/N] ')).trim().toLowerCase();
+            if (answer !== 'y' && answer !== 'yes') {
+                console.log('已取消');
+                return;
+            }
+        }
+        if (!await checkAunEnvironment(rl)) {
+            return;
+        }
+        const result = await setupAunAid(rl, config);
+        if (!result)
+            return;
+        if (!config.channels)
+            config.channels = {};
+        config.channels.aun = {
+            enabled: true,
+            aid: result.aid,
+            ...(result.gatewayPort && { gatewayPort: result.gatewayPort }),
+        };
+        if (!config.channels.defaultChannel)
+            config.channels.defaultChannel = 'aun';
+        fs.writeFileSync(p.config, JSON.stringify(config, null, 2) + '\n');
+        console.log('\n✓ AUN 配置已写入');
+    }
+    finally {
+        rl.close();
+    }
+}