@yeaft/webchat-agent 0.0.233 → 0.0.235

package/crew/routing.js

@@ -0,0 +1,194 @@
+/**
+ * Crew — 路由解析与执行
+ * parseRoutes, executeRoute, buildRoutePrompt, dispatchToRole
+ */
+import { join } from 'path';
+import { sendCrewMessage, sendCrewOutput, sendStatusUpdate } from './ui-messages.js';
+import { ensureTaskFile, appendTaskRecord, readTaskFile, updateKanban, readKanban } from './task-files.js';
+import { createRoleQuery } from './role-query.js';
+
+/** Format role label */
+function roleLabel(r) {
+  return r.icon ? `${r.icon} ${r.displayName}` : r.displayName;
+}
+
+/**
+ * 从累积文本中解析所有 ROUTE 块（支持多 ROUTE + task 字段）
+ * @returns {Array<{ to, summary, taskId, taskTitle }>}
+ */
+export function parseRoutes(text) {
+  const routes = [];
+  const regex = /---ROUTE---\s*\n([\s\S]*?)---END_ROUTE---/g;
+  let match;
+
+  while ((match = regex.exec(text)) !== null) {
+    const block = match[1];
+    const toMatch = block.match(/to:\s*(.+)/i);
+    if (!toMatch) continue;
+
+    const summaryMatch = block.match(/summary:\s*([\s\S]+)/i);
+    const taskMatch = block.match(/^task:\s*(.+)/im);
+    const taskTitleMatch = block.match(/^taskTitle:\s*(.+)/im);
+
+    routes.push({
+      to: toMatch[1].trim().toLowerCase(),
+      summary: summaryMatch ? summaryMatch[1].trim() : '',
+      taskId: taskMatch ? taskMatch[1].trim() : null,
+      taskTitle: taskTitleMatch ? taskTitleMatch[1].trim() : null
+    });
+  }
+
+  return routes;
+}
+
+/**
+ * 执行路由
+ */
+export async function executeRoute(session, fromRole, route) {
+  const { to, summary, taskId, taskTitle } = route;
+
+  // 如果 session 已暂停或停止，保存为 pendingRoutes
+  if (session.status === 'paused' || session.status === 'stopped') {
+    session.pendingRoutes.push({ fromRole, route });
+    console.log(`[Crew] Session ${session.status}, route saved as pending: ${fromRole} -> ${to}`);
+    return;
+  }
+
+  // Task 文件自动管理（fire-and-forget）
+  if (taskId && summary) {
+    const fromRoleConfig = session.roles.get(fromRole);
+    if (fromRoleConfig?.isDecisionMaker && taskTitle && to !== 'human') {
+      ensureTaskFile(session, taskId, taskTitle, to, summary)
+        .catch(e => console.warn(`[Crew] Failed to create task file ${taskId}:`, e.message));
+    }
+    appendTaskRecord(session, taskId, fromRole, summary)
+      .catch(e => console.warn(`[Crew] Failed to append task record ${taskId}:`, e.message));
+
+    // 更新工作看板：推断状态
+    const { getMessages } = await import('../crew-i18n.js');
+    const m = getMessages(session.language || 'zh-CN');
+    const toRoleConfig = session.roles.get(to);
+    let status = m.kanbanStatusDev;
+    if (toRoleConfig) {
+      switch (toRoleConfig.roleType) {
+        case 'reviewer': status = m.kanbanStatusReview; break;
+        case 'tester': status = m.kanbanStatusTest; break;
+        default:
+          if (toRoleConfig.isDecisionMaker) status = m.kanbanStatusDecision;
+      }
+    }
+    updateKanban(session, {
+      taskId, taskTitle, assignee: to,
+      status, summary: summary.substring(0, 100)
+    }).catch(e => console.warn(`[Crew] Failed to update kanban:`, e.message));
+  }
+
+  // 发送路由消息（UI 显示）
+  sendCrewOutput(session, fromRole, 'route', null, { routeTo: to, routeSummary: summary });
+
+  // 路由到 human
+  if (to === 'human') {
+    session.status = 'waiting_human';
+    session.waitingHumanContext = {
+      fromRole,
+      reason: 'requested',
+      message: summary
+    };
+    sendCrewMessage({
+      type: 'crew_human_needed',
+      sessionId: session.id,
+      fromRole,
+      reason: 'requested',
+      message: summary
+    });
+    sendStatusUpdate(session);
+    return;
+  }
+
+  // 路由到指定角色
+  if (session.roles.has(to)) {
+    if (session.humanMessageQueue.length > 0) {
+      const { processHumanQueue } = await import('./human-interaction.js');
+      await processHumanQueue(session);
+    } else {
+      const taskPrompt = buildRoutePrompt(fromRole, summary, session);
+      await dispatchToRole(session, to, taskPrompt, fromRole, taskId, taskTitle);
+    }
+  } else {
+    console.warn(`[Crew] Unknown route target: ${to}`);
+    const errorMsg = `路由目标 "${to}" 不存在。来自 ${fromRole} 的消息: ${summary}`;
+    await dispatchToRole(session, session.decisionMaker, errorMsg, 'system');
+  }
+}
+
+/**
+ * 构建路由转发的 prompt
+ */
+export function buildRoutePrompt(fromRole, summary, session) {
+  const fromRoleConfig = session.roles.get(fromRole);
+  const fromName = fromRoleConfig ? roleLabel(fromRoleConfig) : fromRole;
+  return `来自 ${fromName} 的消息:\n${summary}\n\n请开始你的工作。完成后通过 ROUTE 块传递给下一个角色。`;
+}
+
+/**
+ * 向角色发送消息
+ */
+export async function dispatchToRole(session, roleName, content, fromSource, taskId, taskTitle) {
+  if (session.status === 'paused' || session.status === 'stopped' || session.status === 'initializing') {
+    console.log(`[Crew] Session ${session.status}, skipping dispatch to ${roleName}`);
+    return;
+  }
+
+  let roleState = session.roleStates.get(roleName);
+
+  // 如果角色没有 query 实例，创建一个（支持 resume）
+  if (!roleState || !roleState.query || !roleState.inputStream) {
+    roleState = await createRoleQuery(session, roleName);
+  }
+
+  // 设置 task
+  if (taskId) {
+    roleState.currentTask = { taskId, taskTitle };
+  }
+
+  // Task 上下文注入
+  const effectiveTaskId = taskId || roleState.currentTask?.taskId;
+  if (effectiveTaskId && typeof content === 'string') {
+    const taskContent = await readTaskFile(session, effectiveTaskId);
+    if (taskContent) {
+      content = `${content}\n\n---\n<task-context file="context/features/${effectiveTaskId}.md">\n${taskContent}\n</task-context>`;
+    }
+  }
+
+  // 看板上下文注入（角色重启后知道全局状态）
+  if (typeof content === 'string') {
+    const kanbanContent = await readKanban(session);
+    if (kanbanContent) {
+      content = `${content}\n\n---\n<kanban file="context/kanban.md">\n${kanbanContent}\n</kanban>`;
+    }
+  }
+
+  // 记录消息历史
+  session.messageHistory.push({
+    from: fromSource,
+    to: roleName,
+    content: typeof content === 'string' ? content.substring(0, 200) : '...',
+    taskId: taskId || roleState.currentTask?.taskId || null,
+    timestamp: Date.now()
+  });
+
+  // 发送
+  roleState.lastDispatchContent = content;
+  roleState.lastDispatchFrom = fromSource;
+  roleState.lastDispatchTaskId = taskId || null;
+  roleState.lastDispatchTaskTitle = taskTitle || null;
+  roleState.turnActive = true;
+  roleState.accumulatedText = '';
+  roleState.inputStream.enqueue({
+    type: 'user',
+    message: { role: 'user', content }
+  });
+
+  sendStatusUpdate(session);
+  console.log(`[Crew] Dispatched to ${roleName} from ${fromSource}${taskId ? ` (task: ${taskId})` : ''}`);
+}

package/crew/session.js

@@ -0,0 +1,474 @@
+/**
+ * Crew Session — 核心数据结构、角色展开和 Session 生命周期管理
+ */
+import { promises as fs } from 'fs';
+import { join, isAbsolute } from 'path';
+import ctx from '../context.js';
+import { getMessages } from '../crew-i18n.js';
+import { initWorktrees } from './worktree.js';
+import { initSharedDir, writeRoleClaudeMd, updateSharedClaudeMd } from './shared-dir.js';
+import {
+  loadCrewIndex, upsertCrewIndex, removeFromCrewIndex,
+  loadSessionMeta, saveSessionMeta, loadSessionMessages, getMaxShardIndex
+} from './persistence.js';
+import { sendCrewMessage, sendCrewOutput, sendStatusUpdate } from './ui-messages.js';
+
+// =====================================================================
+// Data Structures
+// =====================================================================
+
+/** @type {Map<string, CrewSession>} */
+export const crewSessions = new Map();
+
+// =====================================================================
+// Role Multi-Instance Expansion
+// =====================================================================
+
+const SHORT_PREFIX = {
+  developer: 'dev',
+  tester: 'test',
+  reviewer: 'rev'
+};
+
+const EXPANDABLE_ROLES = new Set(['developer', 'tester', 'reviewer']);
+
+/**
+ * 展开角色列表：count > 1 的执行者角色展开为多个实例
+ */
+export function expandRoles(roles) {
+  const devRole = roles.find(r => r.name === 'developer');
+  const devCount = devRole?.count > 1 ? devRole.count : 1;
+
+  const expanded = [];
+  for (const role of roles) {
+    const isExpandable = EXPANDABLE_ROLES.has(role.name);
+    const count = isExpandable ? devCount : 1;
+
+    if (count <= 1) {
+      expanded.push({
+        ...role,
+        roleType: role.name,
+        groupIndex: isExpandable ? 1 : 0
+      });
+    } else {
+      const prefix = SHORT_PREFIX[role.name] || role.name;
+      for (let i = 1; i <= count; i++) {
+        expanded.push({
+          ...role,
+          name: `${prefix}-${i}`,
+          displayName: `${role.displayName}-${i}`,
+          roleType: role.name,
+          groupIndex: i,
+          count: undefined
+        });
+      }
+    }
+  }
+  return expanded;
+}
+
+/** Format role label */
+export function roleLabel(r) {
+  return r.icon ? `${r.icon} ${r.displayName}` : r.displayName;
+}
+
+// =====================================================================
+// Path Validation
+// =====================================================================
+
+function isValidProjectDir(dir) {
+  if (!dir || typeof dir !== 'string') return false;
+  if (!isAbsolute(dir)) return false;
+  if (/(?:^|[\\/])\.\.(?:[\\/]|$)/.test(dir)) return false;
+  return true;
+}
+
+// =====================================================================
+// Session Lifecycle
+// =====================================================================
+
+/**
+ * 查找指定 projectDir 的已有 crew session
+ */
+async function findExistingSessionByProjectDir(projectDir) {
+  const normalizedDir = projectDir.replace(/\/+$/, '');
+
+  for (const [, session] of crewSessions) {
+    if (session.projectDir.replace(/\/+$/, '') === normalizedDir
+        && session.status !== 'completed') {
+      return { sessionId: session.id, source: 'active' };
+    }
+  }
+
+  const index = await loadCrewIndex();
+  const agentId = ctx.CONFIG?.agentName || null;
+  const match = index.find(e =>
+    e.projectDir.replace(/\/+$/, '') === normalizedDir
+    && (!agentId || !e.agentId || e.agentId === agentId)
+    && e.status !== 'completed'
+  );
+
+  if (match) {
+    const meta = await loadSessionMeta(match.sharedDir);
+    if (meta) return { sessionId: match.sessionId, source: 'index' };
+    await removeFromCrewIndex(match.sessionId);
+  }
+
+  return null;
+}
+
+/**
+ * 创建 Crew Session
+ */
+export async function createCrewSession(msg) {
+  const {
+    sessionId,
+    projectDir,
+    sharedDir: sharedDirRel,
+    name,
+    roles: rawRoles = [],
+    teamType = 'dev',
+    language = 'zh-CN',
+    userId,
+    username
+  } = msg;
+
+  // 同目录检查
+  const existingSession = await findExistingSessionByProjectDir(projectDir);
+  if (existingSession) {
+    console.log(`[Crew] Found existing session for ${projectDir}: ${existingSession.sessionId}, auto-resuming`);
+    await resumeCrewSession({ sessionId: existingSession.sessionId, userId, username });
+    return;
+  }
+
+  const roles = expandRoles(rawRoles);
+  const sharedDir = sharedDirRel?.startsWith('/')
+    ? sharedDirRel
+    : join(projectDir, sharedDirRel || '.crew');
+  const decisionMaker = roles.find(r => r.isDecisionMaker)?.name || roles[0]?.name || null;
+
+  const session = {
+    id: sessionId,
+    projectDir,
+    sharedDir,
+    name: name || '',
+    roles: new Map(roles.map(r => [r.name, r])),
+    roleStates: new Map(),
+    decisionMaker,
+    status: 'initializing',
+    round: 0,
+    costUsd: 0,
+    totalInputTokens: 0,
+    totalOutputTokens: 0,
+    messageHistory: [],
+    uiMessages: [],
+    humanMessageQueue: [],
+    waitingHumanContext: null,
+    pendingRoutes: [],
+    features: new Map(),
+    _completedTaskIds: new Set(),
+    initProgress: null,
+    userId,
+    username,
+    agentId: ctx.CONFIG?.agentName || null,
+    teamType,
+    language,
+    createdAt: Date.now()
+  };
+
+  crewSessions.set(sessionId, session);
+
+  sendCrewMessage({
+    type: 'crew_session_created',
+    sessionId,
+    projectDir,
+    sharedDir,
+    name: name || '',
+    roles: roles.map(r => ({
+      name: r.name,
+      displayName: r.displayName,
+      icon: r.icon,
+      description: r.description,
+      isDecisionMaker: r.isDecisionMaker || false,
+      model: r.model,
+      roleType: r.roleType,
+      groupIndex: r.groupIndex
+    })),
+    decisionMaker,
+    userId,
+    username
+  });
+
+  sendStatusUpdate(session);
+
+  try {
+    session.initProgress = 'roles';
+    sendStatusUpdate(session);
+    await initSharedDir(sharedDir, roles, projectDir, language);
+
+    const groupIndices = [...new Set(roles.filter(r => r.groupIndex > 0).map(r => r.groupIndex))];
+    if (groupIndices.length > 0) {
+      session.initProgress = 'worktrees';
+      sendStatusUpdate(session);
+    }
+    const worktreeMap = await initWorktrees(projectDir, roles);
+
+    for (const role of roles) {
+      if (role.groupIndex > 0 && worktreeMap.has(role.groupIndex)) {
+        role.workDir = worktreeMap.get(role.groupIndex);
+        await writeRoleClaudeMd(sharedDir, role, language);
+      }
+    }
+
+    await upsertCrewIndex(session);
+    await saveSessionMeta(session);
+
+    if (session.status === 'initializing') {
+      session.status = 'running';
+    }
+    session.initProgress = null;
+    sendStatusUpdate(session);
+  } catch (e) {
+    console.error('[Crew] Session initialization failed:', e);
+    if (session.status === 'initializing') {
+      session.status = 'running';
+    }
+    session.initProgress = null;
+    sendStatusUpdate(session);
+    sendCrewMessage({
+      type: 'crew_output',
+      sessionId,
+      roleName: 'system',
+      roleIcon: 'S',
+      roleDisplayName: '系统',
+      content: `工作环境初始化失败: ${e.message}`,
+      isTurnEnd: true
+    });
+  }
+
+  return session;
+}
+
+// =====================================================================
+// List & Resume Sessions
+// =====================================================================
+
+/**
+ * 列出所有 crew sessions
+ */
+export async function handleListCrewSessions(msg) {
+  const { requestId, _requestClientId } = msg;
+  const index = await loadCrewIndex();
+
+  const agentId = ctx.CONFIG?.agentName || null;
+  const filtered = agentId
+    ? index.filter(e => !e.agentId || e.agentId === agentId)
+    : index;
+
+  for (const entry of filtered) {
+    const active = crewSessions.get(entry.sessionId);
+    if (active) {
+      entry.status = active.status;
+    }
+  }
+
+  ctx.sendToServer({
+    type: 'crew_sessions_list',
+    requestId,
+    _requestClientId,
+    sessions: filtered
+  });
+}
+
+/**
+ * 检查工作目录下是否存在 .crew 目录
+ */
+export async function handleCheckCrewExists(msg) {
+  const { projectDir, requestId, _requestClientId } = msg;
+  if (!projectDir || !isValidProjectDir(projectDir)) {
+    ctx.sendToServer({
+      type: 'crew_exists_result',
+      requestId,
+      _requestClientId,
+      exists: false,
+      error: 'projectDir is required'
+    });
+    return;
+  }
+
+  const crewDir = join(projectDir, '.crew');
+  try {
+    const stat = await fs.stat(crewDir);
+    if (stat.isDirectory()) {
+      let sessionInfo = null;
+      try {
+        const sessionPath = join(crewDir, 'session.json');
+        const data = await fs.readFile(sessionPath, 'utf-8');
+        sessionInfo = JSON.parse(data);
+      } catch {}
+      ctx.sendToServer({
+        type: 'crew_exists_result',
+        requestId,
+        _requestClientId,
+        exists: true,
+        projectDir,
+        sessionInfo
+      });
+    } else {
+      ctx.sendToServer({
+        type: 'crew_exists_result',
+        requestId,
+        _requestClientId,
+        exists: false,
+        projectDir
+      });
+    }
+  } catch {
+    ctx.sendToServer({
+      type: 'crew_exists_result',
+      requestId,
+      _requestClientId,
+      exists: false,
+      projectDir
+    });
+  }
+}
+
+/**
+ * 删除工作目录下的 .crew 目录
+ */
+export async function handleDeleteCrewDir(msg) {
+  const { projectDir } = msg;
+  if (!isValidProjectDir(projectDir)) return;
+  const crewDir = join(projectDir, '.crew');
+  try {
+    await fs.rm(crewDir, { recursive: true, force: true });
+  } catch {}
+}
+
+/**
+ * 恢复已停止的 crew session
+ */
+export async function resumeCrewSession(msg) {
+  const { sessionId, userId, username } = msg;
+
+  if (crewSessions.has(sessionId)) {
+    const session = crewSessions.get(sessionId);
+    const roles = Array.from(session.roles.values());
+    if ((!session.uiMessages || session.uiMessages.length === 0) && session.sharedDir) {
+      const loaded = await loadSessionMessages(session.sharedDir);
+      session.uiMessages = loaded.messages;
+    }
+    const cleanedMessages = (session.uiMessages || []).map(m => {
+      const { _streaming, ...rest } = m;
+      return rest;
+    });
+    const hasOlderMessages = await getMaxShardIndex(session.sharedDir) > 0;
+
+    sendCrewMessage({
+      type: 'crew_session_restored',
+      sessionId,
+      projectDir: session.projectDir,
+      sharedDir: session.sharedDir,
+      name: session.name || '',
+      roles: roles.map(r => ({
+        name: r.name, displayName: r.displayName, icon: r.icon,
+        description: r.description, isDecisionMaker: r.isDecisionMaker || false,
+        groupIndex: r.groupIndex, roleType: r.roleType, model: r.model
+      })),
+      decisionMaker: session.decisionMaker,
+      userId: session.userId,
+      username: session.username,
+      uiMessages: cleanedMessages,
+      hasOlderMessages
+    });
+    sendStatusUpdate(session);
+    return;
+  }
+
+  const index = await loadCrewIndex();
+  const indexEntry = index.find(e => e.sessionId === sessionId);
+  if (!indexEntry) {
+    sendCrewMessage({ type: 'error', sessionId, message: 'Crew session not found in index' });
+    return;
+  }
+
+  const meta = await loadSessionMeta(indexEntry.sharedDir);
+  if (!meta) {
+    sendCrewMessage({ type: 'error', sessionId, message: 'Crew session metadata not found' });
+    return;
+  }
+
+  const roles = meta.roles || [];
+  const decisionMaker = meta.decisionMaker || roles[0]?.name || null;
+  const session = {
+    id: sessionId,
+    projectDir: meta.projectDir,
+    sharedDir: meta.sharedDir || indexEntry.sharedDir,
+    name: meta.name || '',
+    roles: new Map(roles.map(r => [r.name, r])),
+    roleStates: new Map(),
+    decisionMaker,
+    status: 'waiting_human',
+    round: meta.round || 0,
+    costUsd: meta.costUsd || 0,
+    totalInputTokens: meta.totalInputTokens || 0,
+    totalOutputTokens: meta.totalOutputTokens || 0,
+    messageHistory: [],
+    uiMessages: [],
+    humanMessageQueue: [],
+    waitingHumanContext: null,
+    pendingRoutes: [],
+    features: new Map((meta.features || []).map(f => [f.taskId, f])),
+    _completedTaskIds: new Set(meta._completedTaskIds || []),
+    userId: userId || meta.userId,
+    username: username || meta.username,
+    agentId: meta.agentId || ctx.CONFIG?.agentName || null,
+    teamType: meta.teamType || 'dev',
+    language: meta.language || 'zh-CN',
+    createdAt: meta.createdAt || Date.now()
+  };
+  crewSessions.set(sessionId, session);
+
+  const loaded = await loadSessionMessages(session.sharedDir);
+  session.uiMessages = loaded.messages;
+
+  sendCrewMessage({
+    type: 'crew_session_restored',
+    sessionId,
+    projectDir: session.projectDir,
+    sharedDir: session.sharedDir,
+    name: session.name || '',
+    roles: roles.map(r => ({
+      name: r.name, displayName: r.displayName, icon: r.icon,
+      description: r.description, isDecisionMaker: r.isDecisionMaker || false,
+      groupIndex: r.groupIndex, roleType: r.roleType, model: r.model
+    })),
+    decisionMaker,
+    userId: session.userId,
+    username: session.username,
+    uiMessages: session.uiMessages,
+    hasOlderMessages: loaded.hasOlderMessages
+  });
+  sendStatusUpdate(session);
+
+  await upsertCrewIndex(session);
+  await saveSessionMeta(session);
+
+  console.log(`[Crew] Session ${sessionId} resumed, waiting for human input`);
+}
+
+/**
+ * 更新 crew session 的 name
+ */
+export async function handleUpdateCrewSession(msg) {
+  const { sessionId, name } = msg;
+  const session = crewSessions.get(sessionId);
+  if (!session) {
+    console.warn(`[Crew] Session not found for update: ${sessionId}`);
+    return;
+  }
+  if (name !== undefined) session.name = name;
+  await saveSessionMeta(session);
+  await upsertCrewIndex(session);
+}