team-anya-cli 0.1.0

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

@@ -0,0 +1,271 @@
+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 进程并建立连接
+     */
+    async connect() {
+        // 注册 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);
+        await this.broker.spawn({
+            id: this.instanceId,
+            role: 'loid',
+            backendConfig: this.config.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;
+        this.removeBrokerListeners();
+        const orphans = this.pendingMessages.splice(0);
+        this.callbacks.onCrash(orphans);
+    }
+    /**
+     * 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

package/apps/server/dist/loid/worktree-manager.js

@@ -0,0 +1,191 @@
+import { execFile as execFileCb } from 'node:child_process';
+import { mkdir, access, readFile } from 'node:fs/promises';
+import { join } from 'node:path';
+import { promisify } from 'node:util';
+import { getAllProjects, upsertProject, getProjectRepos, syncProjectRepos } from '@team-anya/db';
+const execFile = promisify(execFileCb);
+export class WorktreeManager {
+    workspacePath;
+    projectsConfigPath;
+    reposPath;
+    db;
+    logger;
+    config = null;
+    constructor(deps) {
+        this.workspacePath = deps.workspacePath;
+        this.projectsConfigPath = deps.projectsConfigPath;
+        this.reposPath = deps.reposPath;
+        this.db = deps.db;
+        this.logger = deps.logger ?? { info: console.log, error: console.error };
+    }
+    /**
+     * 加载项目配置
+     *
+     * 优先从 DB 读取。如果 DB 为空且 JSON 文件存在，则从 JSON 导入到 DB。
+     */
+    async loadConfig() {
+        if (this.config !== null)
+            return this.config;
+        // 先从 DB 读取
+        const dbProjects = getAllProjects(this.db);
+        if (dbProjects.length > 0) {
+            this.config = {
+                projects: dbProjects.map(p => {
+                    const repos = getProjectRepos(this.db, p.project_id);
+                    return {
+                        projectId: p.project_id,
+                        name: p.name,
+                        description: p.description ?? undefined,
+                        repos: repos.map(r => ({
+                            name: r.name,
+                            gitUrl: r.git_url ?? undefined,
+                            repoPath: r.repo_path ?? undefined,
+                            defaultBranch: r.default_branch ?? undefined,
+                        })),
+                        claudeMd: p.claude_md ?? undefined,
+                        platform: p.platform ?? 'github',
+                    };
+                }),
+            };
+            return this.config;
+        }
+        // DB 为空，尝试从 JSON 文件导入
+        try {
+            await access(this.projectsConfigPath);
+            const content = await readFile(this.projectsConfigPath, 'utf-8');
+            const parsed = JSON.parse(content);
+            const jsonProjects = (parsed.projects ?? []);
+            // 导入到 DB（项目 + repos 关联表）
+            for (const p of jsonProjects) {
+                upsertProject(this.db, {
+                    project_id: p.projectId,
+                    name: p.name,
+                    description: p.description,
+                    platform: p.platform ?? 'github',
+                    claude_md: p.claudeMd,
+                });
+                syncProjectRepos(this.db, p.projectId, p.repos.map(r => ({
+                    name: r.name,
+                    git_url: r.gitUrl,
+                    repo_path: r.repoPath,
+                    default_branch: r.defaultBranch,
+                })));
+            }
+            this.logger.info(`[WorktreeManager] 从 JSON 导入 ${jsonProjects.length} 个项目到 DB`);
+            this.config = { projects: jsonProjects };
+            this.validateUniqueIds(this.config);
+        }
+        catch (err) {
+            if (err instanceof Error && err.message.startsWith('ID 冲突')) {
+                throw err;
+            }
+            this.logger.info('[WorktreeManager] 项目注册表为空，使用空列表');
+            this.config = { projects: [] };
+        }
+        return this.config;
+    }
+    /**
+     * 向后兼容：返回 projects 列表
+     */
+    async loadProjects() {
+        const config = await this.loadConfig();
+        return config.projects;
+    }
+    /**
+     * 重新从 DB 加载（清除缓存）
+     */
+    invalidateCache() {
+        this.config = null;
+    }
+    validateUniqueIds(config) {
+        const ids = new Set();
+        for (const p of config.projects) {
+            if (ids.has(p.projectId)) {
+                throw new Error(`ID 冲突: "${p.projectId}" 在配置中重复出现`);
+            }
+            ids.add(p.projectId);
+        }
+    }
+    /**
+     * 解析 repoPath：如果配置中未指定，使用 {reposPath}/{name} 作为默认值
+     */
+    resolveRepoPath(repoPath, name) {
+        return repoPath ?? join(this.reposPath, name);
+    }
+    findProject(id) {
+        return this.config?.projects.find(p => p.projectId === id);
+    }
+    /**
+     * 自动初始化：遍历所有配置的仓库，缺失的自动 clone
+     * 适合在服务启动时或手动初始化时调用
+     */
+    async ensureRepos() {
+        await this.loadConfig();
+        await mkdir(this.reposPath, { recursive: true });
+        const result = { cloned: [], existed: [], failed: [] };
+        // 收集所有需要检查的仓库
+        for (const project of this.config.projects) {
+            for (const repo of project.repos) {
+                const repoPath = this.resolveRepoPath(repo.repoPath, repo.name);
+                // 检查是否已存在
+                try {
+                    await access(join(repoPath, '.git'));
+                    result.existed.push(repo.name);
+                    continue;
+                }
+                catch {
+                    // 不存在，需要 clone
+                }
+                if (!repo.gitUrl) {
+                    result.failed.push({
+                        name: repo.name,
+                        error: `仓库不存在且未配置 gitUrl: ${repoPath}`,
+                    });
+                    continue;
+                }
+                try {
+                    this.logger.info(`[WorktreeManager] 克隆 ${repo.name}: ${repo.gitUrl}`);
+                    const cloneArgs = ['clone'];
+                    if (repo.defaultBranch) {
+                        cloneArgs.push('--branch', repo.defaultBranch);
+                    }
+                    cloneArgs.push(repo.gitUrl, repoPath);
+                    await execFile('git', cloneArgs);
+                    result.cloned.push(repo.name);
+                    this.logger.info(`[WorktreeManager] 克隆完成: ${repo.name}`);
+                }
+                catch (err) {
+                    const msg = err instanceof Error ? err.message : String(err);
+                    result.failed.push({ name: repo.name, error: msg });
+                    this.logger.error(`[WorktreeManager] 克隆失败 (${repo.name}):`, err);
+                }
+            }
+        }
+        return result;
+    }
+    /**
+     * 获取项目配置（不执行任何 git 命令）
+     * 找到项目 → 返回 project 模式（repos 数组）
+     * 未找到 → 返回 adhoc 模式
+     */
+    async getProjectConfig(projectId) {
+        await this.loadConfig();
+        if (projectId) {
+            const project = this.findProject(projectId);
+            if (project) {
+                return {
+                    mode: 'project',
+                    platform: project.platform ?? 'github',
+                    repos: project.repos.map(repo => ({
+                        name: repo.name,
+                        repo_path: this.resolveRepoPath(repo.repoPath, repo.name),
+                        default_branch: repo.defaultBranch ?? 'main',
+                    })),
+                    claudeMd: project.claudeMd,
+                };
+            }
+        }
+        return { mode: 'adhoc', platform: 'local' };
+    }
+}
+//# sourceMappingURL=worktree-manager.js.map