@yvhitxcel/opencode-remote 0.16.3 → 0.18.0

package/dist/weixin/api.js

@@ -53,7 +53,14 @@ async function apiFetch(params) {
     }
     catch (err) {
         clearTimeout(t);
-        console.error(`[apiFetch] ${params.label} error:`, err.message);
+        const details = {
+            url: url.toString(),
+            label: params.label,
+            message: err.message,
+            cause: err.cause ? (err.cause.message || String(err.cause)) : undefined,
+            code: err.code,
+        };
+        console.error(`[apiFetch] ${params.label} error:`, JSON.stringify(details));
         throw err;
     }
 }
@@ -61,7 +68,7 @@ async function apiFetch(params) {
 // Login APIs
 // ---------------------------------------------------------------------------
 /**
- * Fetch QR code for login
+ * Fetch QR code for login (admin auth)
  */
 export async function fetchQRCode(baseUrl = DEFAULT_BASE_URL, botType = '3') {
     const base = ensureTrailingSlash(baseUrl);
@@ -122,26 +129,47 @@ export async function getUpdates(params) {
     }
 }
 /**
- * Send a message
+ * Send a message with rate limit retry
  */
 export async function sendMessage(params) {
-    const rawText = await apiFetch({
-        baseUrl: params.baseUrl,
-        endpoint: 'ilink/bot/sendmessage',
-        body: JSON.stringify({
-            ...params.body,
-            base_info: { channel_version: '1.0.0' },
-        }),
-        token: params.token,
-        timeoutMs: params.timeoutMs ?? DEFAULT_API_TIMEOUT_MS,
-        label: 'sendMessage',
-    });
-    try {
-        JSON.parse(rawText);
-    }
-    catch (e) {
-        console.debug('[api] Non-JSON response:', e.message);
+    const MAX_ATTEMPTS = 3;
+    let lastErr;
+    for (let attempt = 1; attempt <= MAX_ATTEMPTS; attempt++) {
+        try {
+            const rawText = await apiFetch({
+                baseUrl: params.baseUrl,
+                endpoint: 'ilink/bot/sendmessage',
+                body: JSON.stringify({
+                    ...params.body,
+                    base_info: { channel_version: '1.0.0' },
+                }),
+                token: params.token,
+                timeoutMs: params.timeoutMs ?? DEFAULT_API_TIMEOUT_MS,
+                label: 'sendMessage',
+            });
+            // 微信返回限流错误码时，rawText 含 errcode，重试
+            try {
+                const j = JSON.parse(rawText);
+                if (j && typeof j.errcode === 'number' && (j.errcode === -1001 || j.errcode === -1002 || j.errcode === 45009 || j.errcode === 45047)) {
+                    const backoff = 1000 * attempt;
+                    console.warn(`[sendMessage] rate-limited (errcode=${j.errcode}), retry in ${backoff}ms (${attempt}/${MAX_ATTEMPTS})`);
+                    await new Promise(r => setTimeout(r, backoff));
+                    continue;
+                }
+            } catch (e) { console.debug('[sendMessage] parse error:', e.message); }
+            return;
+        } catch (err) {
+            lastErr = err;
+            if (attempt < MAX_ATTEMPTS && /AbortError|fetch failed|ECONNRESET/i.test(err.message || '')) {
+                const backoff = 1000 * attempt;
+                console.warn(`[sendMessage] transient error, retry in ${backoff}ms: ${err.message}`);
+                await new Promise(r => setTimeout(r, backoff));
+                continue;
+            }
+            throw err;
+        }
     }
+    throw lastErr;
 }
 /**
  * Get bot config (includes typing_ticket)

package/dist/weixin/bot.js

@@ -1,7 +1,6 @@
-import { existsSync, mkdirSync, readFileSync, writeFileSync, statSync, chmodSync, unlinkSync } from 'fs';
+import { existsSync, mkdirSync, readFileSync, writeFileSync, statSync, chmodSync, unlinkSync, readdirSync } from 'fs';
 import { homedir } from 'os';
 import { join } from 'path';
-import { initSessionManager, loadSessionMapping, saveSessionMapping } from '../core/session.js';
 import { initOpenCode, initFetchConfig } from '../opencode/client.js';
 import { getAuthStatus } from '../core/auth.js';
 import { registry } from '../core/registry.js';
@@ -9,15 +8,32 @@ import { fetchQRCode, pollQRStatus, getUpdates } from './api.js';
 import { DEFAULT_BASE_URL } from './types.js';
 import { createWeixinAdapter } from './adapter.js';
 import { handleMessage } from './handler.js';
+import { userAdapterMap } from './user-adapter-map.js';
+import { initState } from '../core/state.js';
+import { initLogger, cleanOldLogs, logger } from '../core/log.js';
+import { LRUSessionMap } from '../core/lru.js';
+import { encryptCredential, decryptCredential } from '../core/crypto.js';
 export { COMMAND_ALIASES, detectCommand } from '../core/router.js';
 
+let _initialized = false;
+function initBot() {
+    if (_initialized) return;
+    _initialized = true;
+    initLogger();
+    cleanOldLogs();
+    initState();
+    logger.info('Bot starting', { ts: new Date().toISOString() });
+}
+
 const CONFIG_DIR = join(homedir(), '.opencode-remote');
 const WEIXIN_DIR = join(CONFIG_DIR, 'weixin');
+const CREDENTIALS_DIR = join(WEIXIN_DIR, 'credentials');
 const INSTANCE_ID = process.env.OPENCODE_INSTANCE_ID || 'default';
 const CREDENTIALS_FILE = INSTANCE_ID === 'default'
     ? join(WEIXIN_DIR, 'credentials.json')
     : join(WEIXIN_DIR, `credentials-${INSTANCE_ID}.json`);
-const RESTART_NOTIFY_FILE = join(WEIXIN_DIR, 'restart-notify.json');
+
+const botInstances = [];
 
 export async function loginWithQR(baseUrl = DEFAULT_BASE_URL, onQRCode) {
     console.log('Starting Weixin login...');
@@ -48,21 +64,112 @@ export async function loginWithQR(baseUrl = DEFAULT_BASE_URL, onQRCode) {
     } catch (e) { console.error('Login error:', e); return null; }
 }
 
-function ensureDirs() { if (!existsSync(WEIXIN_DIR)) mkdirSync(WEIXIN_DIR, { recursive: true }); }
-export function loadWeixinCredentials() {
+function ensureDirs() {
+    if (!existsSync(WEIXIN_DIR)) mkdirSync(WEIXIN_DIR, { recursive: true });
+    if (!existsSync(CREDENTIALS_DIR)) mkdirSync(CREDENTIALS_DIR, { recursive: true });
+}
+
+export function loadAllCredentials() {
     ensureDirs();
-    if (!existsSync(CREDENTIALS_FILE)) return null;
-    try { return JSON.parse(readFileSync(CREDENTIALS_FILE, 'utf-8')); } catch (e) { console.debug('[credentials] Failed to parse:', e.message); return null; }
+    if (existsSync(CREDENTIALS_DIR)) {
+        const files = readdirSync(CREDENTIALS_DIR).filter(f => f.endsWith('.json'));
+        if (files.length > 0) {
+            return files.map(f => {
+                try {
+                    const raw = readFileSync(join(CREDENTIALS_DIR, f), 'utf-8');
+                    const obj = JSON.parse(raw);
+                    // 检测是否加密信封 → 解密
+                    if (obj && obj.v === 1 && obj.enc) {
+                        const decrypted = decryptCredential(obj.enc);
+                        if (decrypted) return JSON.parse(decrypted);
+                    }
+                    return obj;
+                }
+                catch (e) { console.debug('[credentials] Failed to parse:', f, e.message); return null; }
+            }).filter(Boolean);
+        }
+    }
+    if (existsSync(CREDENTIALS_FILE)) {
+        try { return [JSON.parse(readFileSync(CREDENTIALS_FILE, 'utf-8'))]; } catch (e) { console.debug('[credentials] Failed to parse legacy:', e.message); }
+    }
+    return [];
 }
+
+export function loadWeixinCredentials() {
+    const all = loadAllCredentials();
+    return all.length > 0 ? all[0] : null;
+}
+
 export function saveWeixinCredentials(creds) {
     ensureDirs();
-    writeFileSync(CREDENTIALS_FILE, JSON.stringify({ ...creds, savedAt: new Date().toISOString() }, null, 2), 'utf-8');
-    try { const s = statSync(CREDENTIALS_FILE); chmodSync(CREDENTIALS_FILE, (s.mode & 0o777) | 0o600); } catch (e) { console.warn('[credentials] chmod failed:', e.message); }
+    const filePath = join(CREDENTIALS_DIR, `credentials-${creds.accountId}.json`);
+    const plain = JSON.stringify({ ...creds, savedAt: new Date().toISOString() }, null, 2);
+    const enc = encryptCredential(plain);
+    const envelope = { v: 1, enc, savedAt: new Date().toISOString() };
+    writeFileSync(filePath, JSON.stringify(envelope, null, 2), 'utf-8');
+    try { const s = statSync(filePath); chmodSync(filePath, (s.mode & 0o777) | 0o600); } catch (e) { console.warn('[credentials] chmod failed:', e.message); }
+}
+
+export function saveCredential(creds) {
+    saveWeixinCredentials(creds);
 }
 
 let _restartCallback = null;
 function setRestartCallback(fn) { _restartCallback = fn; }
+
+async function runPollingLoop(adapter, baseUrl, token, openCodeSessions, signal) {
+    let buf = '';
+    let retryCount = 0;
+    while (!signal.aborted) {
+        try {
+            const resp = await getUpdates({ baseUrl, token, get_updates_buf: buf });
+            if (signal.aborted) break;
+            if (resp.get_updates_buf) buf = resp.get_updates_buf;
+            for (const msg of (resp.msgs || [])) {
+                if (msg.message_type !== 1) continue;
+                const textItem = msg.item_list?.find((i) => i.type === 1);
+                const text = textItem?.text_item?.text;
+                const fromUserId = msg.from_user_id;
+                if (!fromUserId || !text) continue;
+                userAdapterMap.set(fromUserId, adapter);
+                const messageId = msg.message_id?.toString();
+                if (adapter.isDuplicate(messageId, `${fromUserId}:${text}`)) continue;
+                if (msg.context_token) adapter.contextTokens.set(fromUserId, { value: msg.context_token, _ts: Date.now() });
+                try { await handleMessage(adapter, { platform: 'weixin', threadId: fromUserId, userId: fromUserId, messageId }, text, openCodeSessions); } catch (e) { console.error('Handle error:', e); }
+            }
+        } catch (e) {
+            if (signal.aborted) break;
+            const errMsg = e.message || '';
+            const isConnReset = errMsg.includes('ECONNRESET') || errMsg.includes('fetch failed');
+            if (isConnReset) {
+                retryCount++;
+                const delay = Math.min(2000 * retryCount, 15000);
+                console.error(`[bot] Connection error (${retryCount}), retry in ${delay}ms...`);
+                await new Promise(r => setTimeout(r, delay));
+            } else {
+                console.error('Polling error:', e);
+                await new Promise(r => setTimeout(r, 2000));
+            }
+        }
+    }
+}
+
+export function addBotInstance(creds, openCodeSessions) {
+    const baseUrl = creds.baseUrl || DEFAULT_BASE_URL;
+    const token = creds.token;
+    const botId = creds.accountId;
+    const adapter = createWeixinAdapter(baseUrl, token, botId);
+    const abortController = new AbortController();
+    const instance = { adapter, abortController, creds };
+    botInstances.push(instance);
+    runPollingLoop(adapter, baseUrl, token, openCodeSessions, abortController.signal).catch(e => console.error('[bot] Polling loop ended:', e));
+    return instance;
+}
+
 export async function startWeixinBot(botConfig, restartFn) {
+    // 仅在子进程启动时初始化日志和状态 (不在父进程 import 时)
+    initBot();
+    process.on('unhandledRejection', (reason) => { console.error('[bot] Unhandled Rejection:', reason); });
     if (restartFn) _restartCallback = restartFn;
     console.log('');
     console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
@@ -72,78 +179,78 @@ export async function startWeixinBot(botConfig, restartFn) {
 
     await registry.loadBuiltInPlugins();
 
-    try { await initFetchConfig(); } catch (e) { console.warn('⚠️ Fetch config failed:', e); }
-    let credentials = loadWeixinCredentials();
-    if (!credentials) {
+    let credentialsList = loadAllCredentials();
+    if (credentialsList.length === 0) {
         console.log('No saved credentials. Starting login...');
-        credentials = await loginWithQR(botConfig.weixinBaseUrl || DEFAULT_BASE_URL);
-        if (!credentials) { console.error('Login failed'); process.exit(1); }
+        const creds = await loginWithQR(botConfig.weixinBaseUrl || DEFAULT_BASE_URL);
+        if (!creds) { console.error('Login failed'); process.exit(1); }
+        credentialsList = [creds];
     }
-    console.log(`Using account: ${credentials.accountId}`);
-    const baseUrl = credentials.baseUrl || DEFAULT_BASE_URL;
-    const token = credentials.token;
-    const botId = credentials.accountId;
-    initSessionManager(botConfig);
-    const openCodeSessions = new Map();
-    const adapter = createWeixinAdapter(baseUrl, token, botId);
-    try { await initOpenCode(); console.log('OpenCode ready'); } catch (e) { console.error('Failed to init OpenCode:', e); }
-    
+    const firstCreds = credentialsList[0];
+    console.log(`Using account: ${firstCreds.accountId}${credentialsList.length > 1 ? ` (+${credentialsList.length - 1} more)` : ''}`);
+    const openCodeSessions = new LRUSessionMap({ maxSize: 100, ttlMs: 30 * 60 * 1000, name: 'opencode-sessions' });
+    // 定期清理过期 session (每 5 分钟)
+    setInterval(() => openCodeSessions.cleanup(), 5 * 60 * 1000).unref();
+
+    let opencodeServer = null;
     try {
         const opencode = await initOpenCode();
         if (opencode) {
+            opencodeServer = opencode.server;
+            globalThis.__opencodeServer = opencode.server;
+            console.log('OpenCode ready');
             const result = await opencode.client.session.list();
             if (!result.error && result.data && result.data.length > 0) {
                 const sorted = result.data.sort((a, b) => (b.updated_at || 0) - (a.updated_at || 0));
                 const latest = sorted[0];
                 console.log(`Latest OpenCode session: ${latest.title || 'Untitled'} (${latest.id.slice(0, 8)}...)`);
-                globalThis.__latestOpenCodeSession = { id: latest.id, directory: latest.directory };
                 if (latest.directory) {
-                    console.log(`Project directory: ${latest.directory}`);
-                    globalThis.__autoProjectDir = latest.directory;
+                    // resume 的目录可能指向 bot 自身而非 openchat 项目
+                    // 如果 resume 目录没 lab.mjs，尝试 fallback 到 openchat 项目
+                    const projectDir = existsSync(`${latest.directory}/bridge/bin/lab.mjs`) ? latest.directory : (existsSync('F:\\openchat\\bridge\\bin\\lab.mjs') ? 'F:\\openchat' : latest.directory);
+                    console.log(`Project directory: ${projectDir}`);
+                    globalThis.__autoProjectDir = projectDir;
                 }
             }
         }
-    } catch (e) { console.warn('⚠️ Auto-resume failed:', e.message); }
-    
+    } catch (e) { console.error('Failed to init OpenCode:', e); }
+
     if (!getAuthStatus().weixin) {
         console.log('\n🔒 Bot not secured! First user to send /start becomes owner.\n');
     }
-    
-    try {
-        if (existsSync(RESTART_NOTIFY_FILE)) {
-            const data = JSON.parse(readFileSync(RESTART_NOTIFY_FILE, 'utf8'));
-            if (data.threadId && Date.now() - data.time < 60000) {
-                await adapter.reply(data.threadId, '✅ Bot 重启完成！');
-            }
-            unlinkSync(RESTART_NOTIFY_FILE);
-        }
-    } catch (e) { console.warn('[restart-notify] Failed to read restart file:', e.message); }
-    
-    let running = true;
+
+    for (const creds of credentialsList) {
+        console.log(`Starting bot for account: ${creds.accountId}`);
+        addBotInstance(creds, openCodeSessions);
+    }
+
+    // IPC 心跳：每 30s 通知父进程还活着
+    const hbTimer = setInterval(() => { try { process.send?.({ type: 'heartbeat', ts: Date.now() }); } catch {} }, 30_000);
+    if (hbTimer.unref) hbTimer.unref();
+
     let shouldRestart = false;
     const shutdown = (restart = false) => {
         console.log(restart ? '\nRestarting...' : '\nShutting down...');
-        saveSessionMapping();
-        running = false;
         shouldRestart = restart;
-        for (const [, s] of openCodeSessions.entries()) { try { s.server?.shutdown?.(); } catch (e) { console.warn('[shutdown] Server shutdown error:', e.message); } }
+        for (const instance of botInstances) {
+            try { instance.abortController.abort(); } catch (e) { }
+        }
+        // 关掉 opencode server 进程，防重启后端口冲突
+        try { globalThis.__opencodeServer?.kill?.(); } catch (e) { console.warn('[shutdown] Server kill error:', e.message); }
         openCodeSessions.clear();
     };
-    
-    globalThis.__weixinBotShutdown = (restart = false) => shutdown(restart);
-    globalThis.__weixinBotRunning = () => running;
 
-    let buf = '';
-    let retryCount = 0;
-    console.log('Polling for messages...');
+    globalThis.__weixinBotShutdown = (restart = false) => shutdown(restart);
+    globalThis.__weixinBotRunning = () => botInstances.some(i => !i.abortController.signal.aborted);
 
     if (process.env.OPENCODE_RESTART === '1') {
         try {
+            const firstAdapter = botInstances[0]?.adapter;
             const restartInfoPath = join(process.env.HOME || process.cwd(), '.opencode-remote', '.restart_user.json');
             if (existsSync(restartInfoPath)) {
                 const restartInfo = JSON.parse(readFileSync(restartInfoPath, 'utf8'));
-                if (Date.now() - restartInfo.time < 60000) {
-                    await adapter.reply(restartInfo.threadId, '✅ Bot 重启完成！');
+                if (Date.now() - restartInfo.time < 60000 && firstAdapter) {
+                    await firstAdapter.reply(restartInfo.threadId, '✅ Bot 重启完成！');
                     console.log('Sent restart notification to user');
                 }
                 unlinkSync(restartInfoPath);
@@ -153,38 +260,20 @@ export async function startWeixinBot(botConfig, restartFn) {
         }
     }
 
-    while (running) {
-        try {
-            const resp = await getUpdates({ baseUrl, token, get_updates_buf: buf });
-            if (!running) break;
-            if (resp.get_updates_buf) buf = resp.get_updates_buf;
-            for (const msg of (resp.msgs || [])) {
-                if (msg.message_type !== 1) continue;
-                const textItem = msg.item_list?.find((i) => i.type === 1);
-                const text = textItem?.text_item?.text;
-                const fromUserId = msg.from_user_id;
-                if (!fromUserId || !text) continue;
-                const messageId = msg.message_id?.toString();
-                if (adapter.isDuplicate(messageId)) continue;
-                if (msg.context_token) adapter.contextTokens.set(fromUserId, msg.context_token);
-                handleMessage(adapter, { platform: 'weixin', threadId: fromUserId, userId: fromUserId, messageId }, text, openCodeSessions).catch(e => console.error('Handle error:', e));
-            }
-        } catch (e) {
-            if (!running) break;
-            const errMsg = e.message || '';
-            const isConnReset = errMsg.includes('ECONNRESET') || errMsg.includes('fetch failed');
-            if (isConnReset) {
-                retryCount++;
-                const delay = Math.min(2000 * retryCount, 15000);
-                console.error(`[bot] Connection error (${retryCount}), retry in ${delay}ms...`);
-                await new Promise(r => setTimeout(r, delay));
-            } else {
-                console.error('Polling error:', e);
-                await new Promise(r => setTimeout(r, 2000));
-            }
-        }
-    }
-    
+    console.log(`✅ ${botInstances.length} bot instance(s) running`);
+    console.log('📡 Listening for WeChat messages...');
+
+    await new Promise(resolve => {
+        globalThis.__weixinBotShutdownAndExit = (restart) => {
+            shutdown(restart);
+            resolve();
+        };
+        // 收到信号时清理 opencode server 后再退出
+        const handleSignal = () => { shutdown(false); resolve(); };
+        process.on('SIGINT', handleSignal);
+        process.on('SIGTERM', handleSignal);
+    });
+
     if (shouldRestart) {
         console.log('✅ Bot shutdown complete, exiting for restart...');
         process.exit(0);