team-anya 0.1.0

package/apps/server/dist/loid/context-builder.js

@@ -0,0 +1,413 @@
+import { getTask, getTasksByStatus, getRecentMessages, getMessageLogBySourceRef, } from "@team-anya/db";
+import { TaskStatus } from "@team-anya/core";
+import { memoryBrief } from "@team-anya/mcp-tools";
+export class LoidContextBuilder {
+    db;
+    workspacePath;
+    worktreeManager;
+    constructor(deps) {
+        this.db = deps.db;
+        this.workspacePath = deps.workspacePath;
+        this.worktreeManager = deps.worktreeManager;
+    }
+    /**
+     * 构建消息事件上下文（包含产品/项目注册表）
+     */
+    async buildMessageContext(message) {
+        const activeTasks = this.getActiveTasks();
+        const recentMessages = this.getRecentMessages(message.chatId);
+        const registry = await this.getRegistry();
+        if (registry.length > 0) {
+            const ids = registry.map((r) => r.id).join(", ");
+            console.log(`[ContextBuilder] 注入 registry: ${registry.length} 条 (${ids})`);
+        }
+        // 注入记忆摘要
+        let brief;
+        try {
+            const briefResult = await memoryBrief(this.workspacePath, {
+                sender_id: message.sender ?? undefined,
+                token_budget: 1500,
+            });
+            if (briefResult.brief) {
+                brief = briefResult.brief;
+            }
+        }
+        catch {
+            // 记忆摘要获取失败不阻塞主流程
+        }
+        return {
+            type: "new_message",
+            message,
+            activeTasks,
+            recentMessages,
+            registry,
+            memoryBrief: brief,
+        };
+    }
+    /**
+     * 构建交付验收上下文
+     */
+    buildDeliveryContext(taskId, exitCode, extras) {
+        const task = getTask(this.db, taskId);
+        const activeTasks = this.getActiveTasks();
+        return {
+            type: "delivery",
+            taskId,
+            briefContent: extras?.briefContent ?? null,
+            exitCode,
+            retryCount: task?.retry_count ?? 0,
+            maxRetries: task?.max_retries ?? 3,
+            prUrl: extras?.prUrl ?? task?.pr_url ?? null,
+            changedFiles: extras?.changedFiles ?? [],
+            testResults: extras?.testResults ?? null,
+            originalMessage: task?.context ?? null,
+            chatId: task?.source_ref
+                ? (getMessageLogBySourceRef(this.db, task.source_ref)?.chat_id ?? null)
+                : null,
+            conversationId: null,
+            activeTasks,
+            sourceChatId: task?.source_ref
+                ? (getMessageLogBySourceRef(this.db, task.source_ref)?.chat_id ??
+                    undefined)
+                : undefined,
+        };
+    }
+    /**
+     * 获取活跃任务列表
+     */
+    getActiveTasks() {
+        const statuses = [
+            TaskStatus.IN_PROGRESS,
+            TaskStatus.READY,
+            TaskStatus.NEED_CLARIFICATION,
+            TaskStatus.DELIVERING,
+            TaskStatus.BLOCKED,
+        ];
+        const tasks = [];
+        for (const status of statuses) {
+            const statusTasks = getTasksByStatus(this.db, status);
+            for (const t of statusTasks) {
+                tasks.push({
+                    task_id: t.task_id,
+                    title: t.title,
+                    status: t.status,
+                    assignee: t.assignee,
+                });
+            }
+        }
+        return tasks;
+    }
+    /**
+     * 从 WorktreeManager 加载项目注册表
+     */
+    async getRegistry() {
+        try {
+            const config = await this.worktreeManager.loadConfig();
+            return config.projects.map((p) => ({
+                id: p.projectId,
+                name: p.name,
+                description: p.description,
+                repos: p.repos.map((r) => r.name),
+                claudeMd: p.claudeMd,
+            }));
+        }
+        catch {
+            return [];
+        }
+    }
+    /**
+     * 获取最近消息历史
+     */
+    getRecentMessages(chatId) {
+        try {
+            if (!chatId)
+                return [];
+            const messages = getRecentMessages(this.db, {
+                limit: 30,
+                chat_id: chatId,
+            });
+            const maxContentLen = 500;
+            return messages.map((m) => {
+                const raw = m.content ?? "";
+                // 从 metadata JSON 中提取 media、回复关系、sender_open_id
+                let media = [];
+                let replyTo = null;
+                let senderOpenId = null;
+                if (m.metadata) {
+                    try {
+                        const parsed = JSON.parse(m.metadata);
+                        if (Array.isArray(parsed.media)) {
+                            media = parsed.media.map((item) => ({
+                                mediaType: String(item.mediaType ?? "file"),
+                                originalName: String(item.originalName ?? ""),
+                                localPath: String(item.localPath ?? ""),
+                            }));
+                        }
+                        if (parsed.parent_message_id) {
+                            replyTo = String(parsed.parent_message_id);
+                        }
+                        if (parsed.sender_open_id) {
+                            senderOpenId = String(parsed.sender_open_id);
+                        }
+                    }
+                    catch {
+                        // metadata 解析失败，忽略
+                    }
+                }
+                return {
+                    message_id: m.source_ref ?? null,
+                    sender: m.sender,
+                    sender_open_id: senderOpenId,
+                    content: raw.slice(0, maxContentLen),
+                    direction: m.direction,
+                    created_at: m.created_at,
+                    message_type: m.message_type ?? null,
+                    related_task_id: m.related_task_id ?? null,
+                    reply_to: replyTo,
+                    media,
+                    truncated: raw.length > maxContentLen,
+                };
+            });
+        }
+        catch {
+            return [];
+        }
+    }
+}
+// ── Prompt 格式化 ──
+export function formatMessageContextPrompt(ctx) {
+    const lines = [];
+    lines.push("强制执行步骤： 在生成任何回复之前，你必须先调用工具读取 CLAUDE.md 以及里面提到的所有的 md 文件。因为该文件中包含了决定你说话语气、你的灵魂，你的工作流程，以及你可以使用的工具（MCP）。即使你认为你知道答案，也严禁在未读取该文件的情况下直接回复。");
+    lines.push("--- 当前对话 ---");
+    const chatInfo = {
+        chat_id: ctx.message.chatId ?? null,
+        chat_type: ctx.message.isDirectMessage ? "p2p" : "group",
+    };
+    lines.push(JSON.stringify(chatInfo));
+    lines.push("");
+    lines.push("--- 当前事件 ---");
+    const event = {
+        type: "new_message",
+        message_id: ctx.message.metadata?.message_id ?? null,
+        sender_id: ctx.message.sender ?? "未知",
+        sender_name: ctx.message.senderName ?? null,
+        content: ctx.message.content,
+    };
+    if (ctx.message.mentionsAnya) {
+        event.mentions_anya = true;
+    }
+    if (ctx.message.metadata?.parent_message_id) {
+        event.reply_to = ctx.message.metadata.parent_message_id;
+    }
+    if (ctx.message.media.length > 0) {
+        event.media = ctx.message.media.map((m) => ({
+            type: m.mediaType,
+            name: m.originalName,
+            path: m.localPath,
+        }));
+    }
+    lines.push(JSON.stringify(event, null, 2));
+    if (ctx.message.media.length > 0) {
+        lines.push("");
+        lines.push("提示: 本条消息包含媒体附件，你可以通过 media.path 中的本地路径读取文件内容（图片可直接查看，文件可读取文本）。");
+    }
+    // 注入记忆摘要（在"当前状态"之前）
+    if (ctx.memoryBrief) {
+        lines.push("");
+        lines.push("--- 记忆摘要 ---");
+        lines.push(ctx.memoryBrief);
+    }
+    lines.push("");
+    lines.push("--- 当前状态 ---");
+    if (ctx.activeTasks.length > 0) {
+        lines.push("活跃任务:");
+        for (const t of ctx.activeTasks) {
+            lines.push(`  - [${t.status}] ${t.task_id}: ${t.title}${t.assignee ? ` (${t.assignee})` : ""}`);
+        }
+    }
+    else {
+        lines.push("活跃任务: 无");
+    }
+    if (ctx.recentMessages.length > 0) {
+        lines.push("");
+        lines.push("--- 对话历史 ---");
+        lines.push(`以下是当前对话最近 ${ctx.recentMessages.length} 条消息（时间倒序，最新在前）：`);
+        const history = ctx.recentMessages.map((m) => {
+            const entry = {
+                message_id: m.message_id,
+                time: m.created_at ?? "",
+                sender_id: m.direction === "outbound" ? "anya" : (m.sender_open_id ?? m.sender ?? "未知"),
+                sender_name: m.direction === "outbound" ? "Anya" : (m.sender ?? null),
+                direction: m.direction,
+                content: m.content,
+            };
+            if (m.reply_to) {
+                entry.reply_to = m.reply_to;
+            }
+            if (m.message_type && m.message_type !== "text") {
+                entry.message_type = m.message_type;
+            }
+            if (m.related_task_id) {
+                entry.related_task_id = m.related_task_id;
+            }
+            if (m.media.length > 0) {
+                entry.media = m.media.map((a) => ({
+                    type: a.mediaType,
+                    name: a.originalName,
+                    path: a.localPath,
+                }));
+            }
+            if (m.truncated) {
+                entry.truncated = true;
+            }
+            return entry;
+        });
+        lines.push(JSON.stringify(history, null, 2));
+    }
+    if (ctx.registry.length > 0) {
+        lines.push("");
+        lines.push("--- 可用项目 ---");
+        lines.push("派发任务时请通过 project_id 指定对应的项目:");
+        for (const entry of ctx.registry) {
+            const desc = entry.description ? ` — ${entry.description}` : "";
+            lines.push(`  - ${entry.id}: ${entry.name}${desc}`);
+            if (entry.repos.length > 0) {
+                lines.push(`    仓库: ${entry.repos.join(", ")}`);
+            }
+            if (entry.claudeMd) {
+                lines.push(`    技术栈:`);
+                for (const line of entry.claudeMd.split("\n")) {
+                    lines.push(`    ${line}`);
+                }
+            }
+        }
+    }
+    return lines.join("\n");
+}
+/**
+ * 合并多条缓冲消息为一个 prompt，供 Session drain 时投递给 CC
+ */
+export function formatBufferedMessages(messages) {
+    if (messages.length === 0)
+        return "";
+    const lines = [];
+    lines.push(`你在处理上一件事的过程中，该对话又收到了 ${messages.length} 条新消息：`);
+    lines.push("");
+    for (let i = 0; i < messages.length; i++) {
+        const msg = messages[i];
+        const time = msg.message.metadata?.timestamp ?? "";
+        const sender = msg.message.senderName ?? msg.message.sender ?? "未知";
+        lines.push(`--- 消息 ${i + 1} (${time} 来自 ${sender}) ---`);
+        lines.push(msg.message.content);
+        if (msg.message.media.length > 0) {
+            for (const m of msg.message.media) {
+                lines.push(`  附件: [${m.mediaType}] ${m.originalName} (${m.localPath})`);
+            }
+        }
+        lines.push("");
+    }
+    lines.push("请综合这些新消息，决定下一步行动。");
+    return lines.join("\n");
+}
+/**
+ * 崩溃恢复时从 DB 活跃任务状态生成上下文 prompt
+ */
+export function formatRecoveryPrompt(activeTasks) {
+    const lines = [];
+    lines.push("--- 崩溃恢复 ---");
+    lines.push("上一个会话异常终止，以下是从数据库恢复的活跃任务状态：");
+    lines.push("");
+    if (activeTasks.length > 0) {
+        for (const t of activeTasks) {
+            lines.push(`- [${t.status}] ${t.task_id}: ${t.title}${t.assignee ? ` (负责人: ${t.assignee})` : ""}`);
+        }
+    }
+    else {
+        lines.push("当前无活跃任务。");
+    }
+    lines.push("");
+    lines.push("请根据以上状态，决定需要优先处理的事项。");
+    return lines.join("\n");
+}
+/**
+ * 后续消息精简格式：已有 session 时不重复注入对话历史、项目注册表、强制读 CLAUDE.md 指令
+ */
+export function formatFollowUpPrompt(ctx) {
+    const lines = [];
+    // 当前事件（必须）
+    lines.push("--- 新消息 ---");
+    const event = {
+        message_id: ctx.message.metadata?.message_id ?? null,
+        sender_id: ctx.message.sender ?? "未知",
+        sender_name: ctx.message.senderName ?? null,
+        content: ctx.message.content,
+    };
+    if (ctx.message.mentionsAnya)
+        event.mentions_anya = true;
+    if (ctx.message.metadata?.parent_message_id)
+        event.reply_to = ctx.message.metadata.parent_message_id;
+    if (ctx.message.media.length > 0) {
+        event.media = ctx.message.media.map((m) => ({
+            type: m.mediaType,
+            name: m.originalName,
+            path: m.localPath,
+        }));
+    }
+    lines.push(JSON.stringify(event, null, 2));
+    if (ctx.message.media.length > 0) {
+        lines.push("");
+        lines.push("提示: 本条消息包含媒体附件，你可以通过 media.path 中的本地路径读取文件内容。");
+    }
+    // 简化版记忆摘要（只注入人的记忆 summary）
+    if (ctx.memoryBrief) {
+        lines.push("");
+        lines.push("--- 记忆提示 ---");
+        lines.push(ctx.memoryBrief);
+    }
+    // 活跃任务（简洁版）
+    if (ctx.activeTasks.length > 0) {
+        lines.push("");
+        lines.push("--- 活跃任务 ---");
+        for (const t of ctx.activeTasks) {
+            lines.push(`  - [${t.status}] ${t.task_id}: ${t.title}${t.assignee ? ` (${t.assignee})` : ""}`);
+        }
+    }
+    return lines.join("\n");
+}
+export function formatDeliveryContextPrompt(ctx) {
+    const lines = [];
+    lines.push("--- 当前事件 ---");
+    lines.push(`类型: Yor 交付`);
+    lines.push(`任务ID: ${ctx.taskId}`);
+    lines.push(`退出码: ${ctx.exitCode}`);
+    lines.push(`重试次数: ${ctx.retryCount}/${ctx.maxRetries}`);
+    if (ctx.prUrl) {
+        lines.push(`PR 链接: ${ctx.prUrl}`);
+    }
+    if (ctx.briefContent) {
+        lines.push("");
+        lines.push("原始 Brief:");
+        lines.push(ctx.briefContent);
+    }
+    if (ctx.changedFiles.length > 0) {
+        lines.push("");
+        lines.push("变更文件:");
+        for (const f of ctx.changedFiles) {
+            lines.push(`  - ${f}`);
+        }
+    }
+    if (ctx.testResults) {
+        lines.push("");
+        lines.push("测试结果:");
+        lines.push(ctx.testResults);
+    }
+    if (ctx.originalMessage) {
+        lines.push("");
+        lines.push(`原始用户指令: "${ctx.originalMessage}"`);
+    }
+    if (ctx.chatId) {
+        lines.push(`对话ID: ${ctx.chatId}（用于回复人类）`);
+    }
+    return lines.join("\n");
+}
+//# sourceMappingURL=context-builder.js.map