team-anya 0.2.0

package/apps/server/dist/loid/session.js

@@ -0,0 +1,276 @@
+import { insertCCSession, updateCCSessionEnded } from '@team-anya/db';
+// ── 常量 ──
+const DEFAULT_IDLE_TIMEOUT_MS = 2 * 60 * 60 * 1000; // 2 小时
+const DEFAULT_MAX_TURNS = 50;
+// ── LoidSession ──
+export class LoidSession {
+    state = 'idle';
+    broker;
+    instanceId;
+    config;
+    callbacks;
+    pendingMessages = [];
+    turnCount = 0;
+    maxTurns;
+    idleTimer = null;
+    lastActivityAt = new Date();
+    connected = false;
+    cliSessionId = null;
+    /** Broker 事件监听器引用（用于 dispose 时清理） */
+    boundResultListener = null;
+    boundCrashListener = null;
+    boundExitedListener = null;
+    constructor(config, callbacks) {
+        this.config = config;
+        this.callbacks = callbacks;
+        this.maxTurns = config.maxTurns ?? DEFAULT_MAX_TURNS;
+        this.broker = config.broker;
+        this.instanceId = config.instanceId;
+    }
+    // ── 公开方法 ──
+    /**
+     * 通过 CCBroker 启动 CC 进程并建立连接
+     * @param resumeSessionId 传入则尝试 resume 到之前的 CC session
+     */
+    async connect(resumeSessionId) {
+        // 注册 broker 事件监听（过滤当前 instanceId）
+        this.boundResultListener = (id, _role, result) => {
+            if (id === this.instanceId) {
+                this.handleResult(result);
+            }
+        };
+        this.boundCrashListener = (id, _role, _error) => {
+            if (id === this.instanceId) {
+                this.handleCrash();
+            }
+        };
+        this.boundExitedListener = (id) => {
+            if (id === this.instanceId && this.connected) {
+                if (this.state !== 'disposed') {
+                    this.config.logger?.error('[LoidSession] CC 进程异常退出，触发崩溃恢复');
+                    this.handleCrash();
+                }
+            }
+        };
+        this.broker.on('instance.result', this.boundResultListener);
+        this.broker.on('instance.crash', this.boundCrashListener);
+        this.broker.on('instance.exited', this.boundExitedListener);
+        // 监听 CC session ID
+        const sessionIdListener = (id, cliSessionId) => {
+            if (id === this.instanceId && this.config.db) {
+                this.cliSessionId = cliSessionId;
+                insertCCSession(this.config.db, {
+                    session_id: cliSessionId,
+                    role: 'loid',
+                    instance_id: this.instanceId,
+                    chat_id: this.config.chatId,
+                    project_path: this.config.backendConfig.workingDir,
+                });
+                this.config.logger?.info(`[LoidSession] 记录 CC session: ${cliSessionId}`);
+                // 一次性监听，获取后移除
+                this.broker.removeListener('instance.sessionId', sessionIdListener);
+            }
+        };
+        this.broker.on('instance.sessionId', sessionIdListener);
+        const backendConfig = resumeSessionId
+            ? { ...this.config.backendConfig, resumeSessionId }
+            : this.config.backendConfig;
+        await this.broker.spawn({
+            id: this.instanceId,
+            role: 'loid',
+            backendConfig,
+            logFile: this.config.logFile,
+        });
+        this.connected = true;
+        this.resetIdleTimer();
+    }
+    /**
+     * 投递消息：IDLE 时直接发送，EXECUTING 时缓冲
+     * 返回 false 表示 session 不接受消息（DISPOSED）
+     */
+    enqueue(msg) {
+        if (this.state === 'disposed') {
+            return false;
+        }
+        this.lastActivityAt = new Date();
+        if (this.state === 'idle') {
+            this.state = 'executing';
+            this.clearIdleTimer();
+            try {
+                this.broker.sendPrompt(this.instanceId, msg.prompt);
+            }
+            catch (err) {
+                this.config.logger?.error('[LoidSession] sendPrompt 失败:', err);
+                this.handleCrash();
+            }
+        }
+        else {
+            // EXECUTING → 缓冲
+            this.pendingMessages.push(msg);
+        }
+        return true;
+    }
+    /**
+     * 获取当前状态
+     */
+    getState() {
+        return this.state;
+    }
+    /**
+     * 获取当前轮次数
+     */
+    getTurnCount() {
+        return this.turnCount;
+    }
+    /**
+     * 获取缓冲区消息数
+     */
+    getPendingCount() {
+        return this.pendingMessages.length;
+    }
+    /**
+     * 获取最后活跃时间（供 LRU 淘汰使用）
+     */
+    getLastActivityAt() {
+        return this.lastActivityAt;
+    }
+    /**
+     * 销毁 session，释放资源
+     */
+    async dispose() {
+        if (this.state === 'disposed')
+            return;
+        this.state = 'disposed';
+        this.clearIdleTimer();
+        this.connected = false;
+        // 更新 cc_sessions.ended_at
+        if (this.cliSessionId && this.config.db) {
+            updateCCSessionEnded(this.config.db, this.cliSessionId);
+        }
+        this.removeBrokerListeners();
+        await this.broker.dispose(this.instanceId).catch(() => { });
+    }
+    // ── 私有方法 ──
+    /**
+     * CC 执行完一轮后的 drain loop
+     */
+    handleResult(result) {
+        this.callbacks.onResult?.(result);
+        this.turnCount++;
+        this.lastActivityAt = new Date();
+        if (this.state === 'disposed')
+            return;
+        // drain: 缓冲区有消息 → 合并发送
+        if (this.pendingMessages.length > 0) {
+            const buffered = this.pendingMessages.splice(0);
+            const prompt = this.formatBufferedMessages(buffered);
+            try {
+                this.broker.sendPrompt(this.instanceId, prompt);
+            }
+            catch (err) {
+                this.config.logger?.error('[LoidSession] drain sendPrompt 失败:', err);
+                this.handleCrash();
+            }
+            return;
+        }
+        // 达到硬上限 → 直接 dispose（不生成摘要）
+        if (this.turnCount >= this.maxTurns) {
+            this.config.logger?.info(`[LoidSession] 达到硬上限 (${this.maxTurns} 轮)，销毁 session`);
+            this.disposeAndNotify();
+            return;
+        }
+        // 无待处理消息 → 回 IDLE，启动空闲计时器
+        this.state = 'idle';
+        this.resetIdleTimer();
+    }
+    /**
+     * 崩溃处理：交还未处理消息，标记 disposed
+     */
+    handleCrash() {
+        if (this.state === 'disposed')
+            return;
+        this.state = 'disposed';
+        this.clearIdleTimer();
+        this.connected = false;
+        const savedSessionId = this.cliSessionId;
+        this.removeBrokerListeners();
+        const orphans = this.pendingMessages.splice(0);
+        this.callbacks.onCrash(orphans, savedSessionId);
+    }
+    /**
+     * dispose 并通知回调
+     */
+    async disposeAndNotify() {
+        await this.dispose();
+        this.callbacks.onDisposed();
+    }
+    /**
+     * 重置空闲计时器
+     */
+    resetIdleTimer() {
+        this.clearIdleTimer();
+        const timeout = this.config.idleTimeoutMs ?? DEFAULT_IDLE_TIMEOUT_MS;
+        this.idleTimer = setTimeout(() => {
+            if (this.state === 'idle') {
+                this.config.logger?.info(`[LoidSession] 空闲超时 (${timeout}ms)，销毁 session`);
+                this.disposeAndNotify();
+            }
+        }, timeout);
+    }
+    /**
+     * 清除空闲计时器
+     */
+    clearIdleTimer() {
+        if (this.idleTimer) {
+            clearTimeout(this.idleTimer);
+            this.idleTimer = null;
+        }
+    }
+    /**
+     * 移除 broker 事件监听器
+     */
+    removeBrokerListeners() {
+        if (this.boundResultListener) {
+            this.broker.removeListener('instance.result', this.boundResultListener);
+            this.boundResultListener = null;
+        }
+        if (this.boundCrashListener) {
+            this.broker.removeListener('instance.crash', this.boundCrashListener);
+            this.boundCrashListener = null;
+        }
+        if (this.boundExitedListener) {
+            this.broker.removeListener('instance.exited', this.boundExitedListener);
+            this.boundExitedListener = null;
+        }
+    }
+    /**
+     * 合并缓冲消息为单个 prompt
+     * 临时内联实现（正式版本由 context-builder 的 formatBufferedMessages 提供）
+     */
+    formatBufferedMessages(messages) {
+        if (messages.length === 1) {
+            return messages[0].prompt;
+        }
+        const lines = [];
+        lines.push(`你在处理上一件事的过程中，该对话又收到了 ${messages.length} 条新消息：`);
+        lines.push('');
+        for (let i = 0; i < messages.length; i++) {
+            const msg = messages[i];
+            const time = msg.enqueuedAt.toLocaleTimeString('zh-CN', {
+                hour: '2-digit',
+                minute: '2-digit',
+                second: '2-digit',
+                hour12: false,
+            });
+            const typeLabel = msg.type === 'delivery' ? ' [交付]'
+                : msg.type === 'recovery' ? ' [恢复]'
+                    : '';
+            lines.push(`--- 消息 ${i + 1} (${time}${typeLabel}) ---`);
+            lines.push(msg.prompt);
+            lines.push('');
+        }
+        lines.push('请综合这些新消息，决定下一步行动。');
+        return lines.join('\n');
+    }
+}
+//# sourceMappingURL=session.js.map