@yeaft/webchat-agent 1.0.79 → 1.0.80

package/crew/control.js

@@ -1,444 +0,0 @@
-/**
- * Crew — 控制操作
- * pause, resume, stop, clear, abort, interrupt 等
- */
-import { join } from 'path';
-import { promises as fs } from 'fs';
-import { sendCrewMessage, sendCrewOutput, sendStatusUpdate, endRoleStreaming } from './ui-messages.js';
-import { saveRoleSessionId, clearRoleSessionId, createRoleQuery } from './role-query.js';
-import { saveSessionMeta, cleanupMessageShards } from './persistence.js';
-import { executeRoute, dispatchToRole } from './routing.js';
-import { saveRoleWorkSummary } from './task-files.js';
-import { getMessages } from '../crew-i18n.js';
-import { cleanupWorktrees } from './worktree.js';
-import { upsertCrewIndex } from './persistence.js';
-import { processHumanQueue } from './human-interaction.js';
-
-/**
- * 处理控制命令
- */
-export async function handleCrewControl(msg) {
-  // Lazy import to avoid circular dependency
-  const { crewSessions, resumeCrewSession } = await import('./session.js');
-
-  const { sessionId, action, targetRole } = msg;
-  let session = crewSessions.get(sessionId);
-  if (!session) {
-    // Auto-resume: try to restore from disk before giving up
-    console.log(`[Crew] Session ${sessionId} not in memory, attempting auto-resume...`);
-    try {
-      await resumeCrewSession({ sessionId });
-      session = crewSessions.get(sessionId);
-    } catch (e) {
-      console.warn(`[Crew] Auto-resume failed for ${sessionId}:`, e.message);
-    }
-    if (!session) {
-      console.warn(`[Crew] Session not found: ${sessionId} (even after auto-resume)`);
-      return;
-    }
-  }
-
-  switch (action) {
-    case 'pause':
-      await pauseAll(session);
-      break;
-    case 'resume':
-      await resumeSession(session);
-      break;
-    case 'stop_role':
-      if (targetRole) await stopRole(session, targetRole);
-      break;
-    case 'interrupt_role':
-      if (targetRole && msg.content) {
-        await interruptRole(session, targetRole, msg.content, 'human');
-      }
-      break;
-    case 'abort_role':
-      if (targetRole) await abortRole(session, targetRole);
-      break;
-    case 'clear_role':
-      if (targetRole) await clearSingleRole(session, targetRole);
-      break;
-    case 'stop_all':
-      await stopAll(session);
-      break;
-    case 'clear':
-      await clearSession(session);
-      break;
-    default:
-      console.warn(`[Crew] Unknown control action: ${action}`);
-  }
-}
-
-/**
- * 清空单个角色的对话
- */
-async function clearSingleRole(session, roleName) {
-  const roleState = session.roleStates.get(roleName);
-
-  // P0-1: 清除该角色在 humanMessageQueue 中的待处理消息，防止幽灵 ROUTE 执行
-  if (session.humanMessageQueue.length > 0) {
-    session.humanMessageQueue = session.humanMessageQueue.filter(m => m.target !== roleName);
-  }
-
-  // 保存 task 信息，clear 后恢复时传给 dispatchToRole（必须在 if 块外声明，恢复 dispatch 在块外引用）
-  const lastTask = roleState?.currentTask ? { ...roleState.currentTask } : null;
-
-  if (roleState) {
-    // 保存工作摘要到 task file（与 context_exceeded clear 一致）
-    if (roleState.accumulatedText) {
-      await saveRoleWorkSummary(session, roleName, roleState.accumulatedText).catch(e =>
-        console.warn(`[Crew] Failed to save work summary for ${roleName}:`, e.message));
-    }
-
-    // P1-3: abort 并等待 query iterator 退出，避免 abort+dispatch 竞态
-    if (roleState.abortController) {
-      roleState.abortController.abort();
-    }
-    // 等待 query 完成（iterator 会因 AbortError 退出）
-    if (roleState.query) {
-      try {
-        // Drain the iterator — it will throw AbortError and exit
-        // eslint-disable-next-line no-empty
-        for await (const _ of roleState.query) {}
-      } catch {
-        // Expected: AbortError or other cleanup errors
-      }
-    }
-
-    roleState.query = null;
-    roleState.inputStream = null;
-    roleState.turnActive = false;
-    roleState.claudeSessionId = null;
-    roleState.consecutiveErrors = 0;
-    roleState.accumulatedText = '';
-    roleState.lastDispatchContent = null;
-    roleState.lastDispatchFrom = null;
-    roleState.lastDispatchTaskId = null;
-    roleState.lastDispatchTaskTitle = null;
-    // P0-3: 重置 UI 相关状态，防止显示过时信息
-    roleState.currentTask = null;
-    roleState.currentTool = null;
-    roleState.lastTurnText = '';
-  }
-
-  await clearRoleSessionId(session.sharedDir, roleName);
-
-  sendCrewMessage({
-    type: 'crew_role_cleared',
-    sessionId: session.id,
-    role: roleName,
-    reason: 'manual'
-  });
-
-  sendCrewOutput(session, 'system', 'system', {
-    type: 'assistant',
-    message: { role: 'assistant', content: [{ type: 'text', text: `${roleName} 对话已清空` }] }
-  });
-  sendStatusUpdate(session);
-  console.log(`[Crew] Role ${roleName} cleared`);
-
-  // 从 messageHistory 提取该角色最近 5 条相关消息，构造记忆恢复 prompt
-  const roleMessages = session.messageHistory
-    .filter(m => m.from === roleName || m.to === roleName)
-    .slice(-5);
-  if (roleMessages.length > 0) {
-    const m = getMessages(session.language || 'zh-CN');
-    const summary = roleMessages
-      .map(msg => `[${msg.from} → ${msg.to}${msg.taskId ? ` (${msg.taskId})` : ''}] ${msg.content}`)
-      .join('\n');
-    const restorePrompt = `${m.memoryRestorePrompt}\n\n${summary}`;
-    await dispatchToRole(session, roleName, restorePrompt, 'system', lastTask?.taskId, lastTask?.taskTitle);
-  }
-}
-
-/**
- * 暂停所有角色
- */
-async function pauseAll(session) {
-  session.status = 'paused';
-
-  for (const [roleName, roleState] of session.roleStates) {
-    if (roleState.claudeSessionId) {
-      await saveRoleSessionId(session.sharedDir, roleName, roleState.claudeSessionId)
-        .catch(e => console.warn(`[Crew] Failed to save sessionId for ${roleName}:`, e.message));
-    }
-    if (roleState.abortController) {
-      roleState.abortController.abort();
-    }
-    roleState.wasActive = roleState.turnActive;
-    roleState.turnActive = false;
-    roleState.query = null;
-    roleState.inputStream = null;
-  }
-
-  console.log(`[Crew] Session ${session.id} paused, all active queries aborted`);
-
-  sendCrewOutput(session, 'system', 'system', {
-    type: 'assistant',
-    message: { role: 'assistant', content: [{ type: 'text', text: 'Session 已暂停' }] }
-  });
-  sendStatusUpdate(session);
-
-  await saveSessionMeta(session);
-}
-
-/**
- * 恢复 session
- */
-async function resumeSession(session) {
-  if (session.status !== 'paused') return;
-
-  session.status = 'running';
-  console.log(`[Crew] Session ${session.id} resumed`);
-
-  sendCrewOutput(session, 'system', 'system', {
-    type: 'assistant',
-    message: { role: 'assistant', content: [{ type: 'text', text: 'Session 已恢复' }] }
-  });
-  sendStatusUpdate(session);
-  // Status changed to running — persist
-  saveSessionMeta(session).catch(e => console.warn('[Crew] Failed to save after resume:', e.message));
-
-  if (session.pendingRoutes.length > 0) {
-    const pending = session.pendingRoutes.slice();
-    session.pendingRoutes = [];
-    console.log(`[Crew] Replaying ${pending.length} pending route(s)`);
-    // P1-5: 串行执行 pending routes，防止多个 route 指向同一角色时并发 dispatch
-    for (const { fromRole, route } of pending) {
-      try {
-        await executeRoute(session, fromRole, route);
-      } catch (err) {
-        console.warn(`[Crew] Pending route replay failed:`, err);
-      }
-    }
-    return;
-  }
-
-  await processHumanQueue(session);
-}
-
-/**
- * 中断角色当前 turn 并发送新消息
- */
-async function interruptRole(session, roleName, newContent, fromSource = 'human') {
-  const roleState = session.roleStates.get(roleName);
-  if (!roleState) {
-    console.warn(`[Crew] Cannot interrupt ${roleName}: no roleState`);
-    return;
-  }
-
-  console.log(`[Crew] Interrupting ${roleName}`);
-
-  endRoleStreaming(session, roleName);
-
-  if (roleState.claudeSessionId) {
-    await saveRoleSessionId(session.sharedDir, roleName, roleState.claudeSessionId)
-      .catch(e => console.warn(`[Crew] Failed to save sessionId for ${roleName}:`, e.message));
-  }
-
-  if (roleState.abortController) {
-    roleState.abortController.abort();
-  }
-
-  roleState.query = null;
-  roleState.inputStream = null;
-  roleState.turnActive = false;
-  roleState.accumulatedText = '';
-
-  // Mark pending tool messages as completed before notifying frontend
-  for (const m of session.uiMessages) {
-    if (m.role === roleName && m.type === 'tool' && !m.hasResult) {
-      m.hasResult = true;
-    }
-  }
-
-  sendCrewMessage({
-    type: 'crew_turn_completed',
-    sessionId: session.id,
-    role: roleName,
-    interrupted: true
-  });
-
-  sendStatusUpdate(session);
-
-  sendCrewOutput(session, 'system', 'system', {
-    type: 'assistant',
-    message: { role: 'assistant', content: [{ type: 'text', text: `${roleName} 被中断` }] }
-  });
-
-  await dispatchToRole(session, roleName, newContent, fromSource);
-}
-
-/**
- * 中止角色当前 turn
- */
-async function abortRole(session, roleName) {
-  const roleState = session.roleStates.get(roleName);
-  if (!roleState) {
-    console.warn(`[Crew] Cannot abort ${roleName}: no roleState`);
-    return;
-  }
-
-  if (!roleState.turnActive) {
-    console.log(`[Crew] ${roleName} is not active, nothing to abort`);
-    return;
-  }
-
-  console.log(`[Crew] Aborting ${roleName}`);
-
-  endRoleStreaming(session, roleName);
-
-  if (roleState.claudeSessionId) {
-    await saveRoleSessionId(session.sharedDir, roleName, roleState.claudeSessionId)
-      .catch(e => console.warn(`[Crew] Failed to save sessionId for ${roleName}:`, e.message));
-  }
-
-  if (roleState.abortController) {
-    roleState.abortController.abort();
-  }
-
-  roleState.query = null;
-  roleState.inputStream = null;
-  roleState.turnActive = false;
-  roleState.accumulatedText = '';
-
-  // Mark pending tool messages as completed before notifying frontend
-  for (const m of session.uiMessages) {
-    if (m.role === roleName && m.type === 'tool' && !m.hasResult) {
-      m.hasResult = true;
-    }
-  }
-
-  sendCrewMessage({
-    type: 'crew_turn_completed',
-    sessionId: session.id,
-    role: roleName,
-    interrupted: true
-  });
-
-  sendStatusUpdate(session);
-
-  sendCrewOutput(session, 'system', 'system', {
-    type: 'assistant',
-    message: { role: 'assistant', content: [{ type: 'text', text: `${roleName} 已中止` }] }
-  });
-}
-
-async function stopRole(session, roleName) {
-  const roleState = session.roleStates.get(roleName);
-  if (roleState) {
-    if (roleState.claudeSessionId) {
-      await saveRoleSessionId(session.sharedDir, roleName, roleState.claudeSessionId)
-        .catch(e => console.warn(`[Crew] Failed to save sessionId for ${roleName}:`, e.message));
-    }
-    if (roleState.abortController) {
-      roleState.abortController.abort();
-    }
-    roleState.query = null;
-    roleState.inputStream = null;
-    roleState.turnActive = false;
-    session.roleStates.delete(roleName);
-  }
-
-  sendCrewOutput(session, 'system', 'system', {
-    type: 'assistant',
-    message: { role: 'assistant', content: [{ type: 'text', text: `${roleName} 已停止` }] }
-  });
-  sendStatusUpdate(session);
-  console.log(`[Crew] Role ${roleName} stopped`);
-}
-
-/**
- * 终止整个 session
- */
-async function stopAll(session) {
-  const { crewSessions } = await import('./session.js');
-
-  session.status = 'stopped';
-
-  for (const [roleName, roleState] of session.roleStates) {
-    if (roleState.claudeSessionId) {
-      await saveRoleSessionId(session.sharedDir, roleName, roleState.claudeSessionId)
-        .catch(e => console.warn(`[Crew] Failed to save sessionId for ${roleName}:`, e.message));
-    }
-    if (roleState.abortController) {
-      roleState.abortController.abort();
-    }
-    console.log(`[Crew] Stopping role: ${roleName}`);
-  }
-  session.roleStates.clear();
-
-  sendCrewOutput(session, 'system', 'system', {
-    type: 'assistant',
-    message: { role: 'assistant', content: [{ type: 'text', text: 'Session 已终止' }] }
-  });
-  sendStatusUpdate(session);
-
-  await cleanupWorktrees(session.projectDir);
-
-  await saveSessionMeta(session);
-  await upsertCrewIndex(session);
-
-  crewSessions.delete(session.id);
-  console.log(`[Crew] Session ${session.id} stopped`);
-}
-
-/**
- * 清空 session
- */
-async function clearSession(session) {
-  // P1-2: 先重置 _processingHumanQueue，防止后续消息处理被阻塞
-  session._processingHumanQueue = false;
-
-  for (const [roleName, roleState] of session.roleStates) {
-    if (roleState.abortController) {
-      roleState.abortController.abort();
-    }
-    console.log(`[Crew] Clearing role: ${roleName}`);
-  }
-  session.roleStates.clear();
-
-  for (const [roleName] of session.roles) {
-    await clearRoleSessionId(session.sharedDir, roleName);
-  }
-
-  // P1-1: humanMessageQueue 在这里清空，确保不会有幽灵消息在 clear 后被处理
-  session.messageHistory = [];
-  session.uiMessages = [];
-  session.humanMessageQueue = [];
-  session.waitingHumanContext = null;
-  session.pendingRoutes = [];
-
-  // 清除 feature/task 数据，避免 UI 残留空 task 卡片
-  session.features.clear();
-  session._completedTaskIds = new Set();
-
-  session.round = 0;
-
-  // 重置计费统计（clearSession 清除了所有 claudeSessionId，后续 query 全新，费用从零开始）
-  session.costUsd = 0;
-  session.totalInputTokens = 0;
-  session.totalOutputTokens = 0;
-
-  const messagesPath = join(session.sharedDir, 'messages.json');
-  await fs.writeFile(messagesPath, '[]').catch(() => {});
-  await cleanupMessageShards(session.sharedDir);
-
-  session.status = 'running';
-
-  sendCrewMessage({
-    type: 'crew_session_cleared',
-    sessionId: session.id
-  });
-
-  sendCrewOutput(session, 'system', 'system', {
-    type: 'assistant',
-    message: { role: 'assistant', content: [{ type: 'text', text: '会话已清空，所有角色将使用全新对话' }] }
-  });
-  sendStatusUpdate(session);
-
-  await saveSessionMeta(session);
-
-  console.log(`[Crew] Session ${session.id} cleared`);
-}

package/crew/human-interaction.js

@@ -1,195 +0,0 @@
-/**
- * Crew — 人工交互
- * handleCrewHumanInput, processHumanQueue
- */
-import { dispatchToRole } from './routing.js';
-import { sendStatusUpdate } from './ui-messages.js';
-import { debouncedSaveSessionMeta } from './persistence.js';
-
-/**
- * Resolve @role mention from message content.
- * Returns { target, message } if a valid role is found, null otherwise.
- */
-export function resolveAtMention(content, session) {
-  const atMatch = content.match(/^@(\S+)\s*([\s\S]*)/);
-  if (!atMatch) return null;
-
-  const atTarget = atMatch[1];
-  const message = atMatch[2].trim() || content;
-
-  for (const [name, role] of session.roles) {
-    if (name === atTarget.toLowerCase()) {
-      return { target: name, message };
-    }
-    if (role.displayName === atTarget) {
-      return { target: name, message };
-    }
-    // Fuzzy: compound "name-displayName" pattern (e.g. "dev-1-托瓦兹" or partial displayName)
-    const compound = `${name}-${role.displayName}`;
-    if (compound.toLowerCase() === atTarget.toLowerCase()) {
-      return { target: name, message };
-    }
-  }
-  return null;
-}
-
-/**
- * 处理人的输入
- */
-export async function handleCrewHumanInput(msg) {
-  // Lazy import to avoid circular dependency
-  const { crewSessions, resumeCrewSession } = await import('./session.js');
-
-  const { sessionId, content, targetRole, files } = msg;
-  let session = crewSessions.get(sessionId);
-  if (!session) {
-    // Auto-resume: try to restore from disk before giving up
-    console.log(`[Crew] Session ${sessionId} not in memory, attempting auto-resume...`);
-    try {
-      await resumeCrewSession({ sessionId });
-      session = crewSessions.get(sessionId);
-    } catch (e) {
-      console.warn(`[Crew] Auto-resume failed for ${sessionId}:`, e.message);
-    }
-    if (!session) {
-      console.warn(`[Crew] Session not found: ${sessionId} (even after auto-resume)`);
-      return;
-    }
-  }
-
-  // Auto-resume: user sending a message = intent to continue → restore to running
-  if (session.status === 'paused' || session.status === 'stopped' || session.status === 'completed') {
-    console.log(`[Crew] Auto-resuming session from ${session.status} to running (user sent message)`);
-    session.status = 'running';
-    sendStatusUpdate(session);
-    debouncedSaveSessionMeta(session);
-  }
-
-  // Resolve @role mention from content (works regardless of targetRole from frontend)
-  const atMention = resolveAtMention(content, session);
-  // Effective target: explicit targetRole from frontend > @mention from content > null
-  const effectiveTargetRole = targetRole || (atMention ? atMention.target : null);
-  // If @mention found, use the stripped message (without @prefix); otherwise use original content
-  const effectiveContent = atMention ? atMention.message : content;
-
-  // Build dispatch content (supports image attachments)
-  function buildHumanContent(prefix, text) {
-    if (files && files.length > 0) {
-      const blocks = [];
-      for (const file of files) {
-        if (file.isImage || file.mimeType?.startsWith('image/')) {
-          blocks.push({
-            type: 'image',
-            source: { type: 'base64', media_type: file.mimeType, data: file.data }
-          });
-        }
-      }
-      blocks.push({ type: 'text', text: `${prefix}\n${text}` });
-      return blocks;
-    }
-    return `${prefix}\n${text}`;
-  }
-
-  // 记录到 uiMessages 用于恢复时重放
-  session.uiMessages.push({
-    role: 'human', roleIcon: '', roleName: '你',
-    type: 'text', content,
-    timestamp: Date.now()
-  });
-
-  // 如果在等待人工介入
-  if (session.status === 'waiting_human') {
-    const waitingContext = session.waitingHumanContext;
-    session.status = 'running';
-    session.waitingHumanContext = null;
-    sendStatusUpdate(session);
-    // Status changed + new human message — persist (debounced, dispatch will follow)
-    debouncedSaveSessionMeta(session);
-
-    const target = effectiveTargetRole || waitingContext?.fromRole || session.decisionMaker;
-    if (effectiveTargetRole) {
-      console.log(`[Crew] @mention override in waiting_human: routing to ${target} instead of ${waitingContext?.fromRole || session.decisionMaker}`);
-    }
-    await dispatchToRole(session, target, buildHumanContent('人工回复:', effectiveContent), 'human');
-    return;
-  }
-
-  // @role 指令 — effectiveTargetRole already resolved above
-  if (atMention && effectiveTargetRole) {
-    const target = effectiveTargetRole;
-    const message = effectiveContent;
-
-    // 检测纯 skill 命令（如 /context, /simplify），直接发送不加前缀
-    if (/^\/[a-zA-Z0-9_-]+(?:\s+.*)?$/s.test(message)) {
-      let roleState = session.roleStates.get(target);
-      if (!roleState || !roleState.query || !roleState.inputStream) {
-        const { createRoleQuery } = await import('./role-query.js');
-        roleState = await createRoleQuery(session, target);
-      }
-      // P1-4: 守卫 stream.enqueue
-      try {
-        if (roleState.inputStream && !roleState.inputStream.isDone) {
-          roleState.inputStream.enqueue({
-            type: 'user',
-            message: { role: 'user', content: message }
-          });
-        } else {
-          console.warn(`[Crew] Skill dispatch: stream closed for ${target}, recreating`);
-          const { createRoleQuery } = await import('./role-query.js');
-          roleState = await createRoleQuery(session, target);
-          roleState.inputStream.enqueue({
-            type: 'user',
-            message: { role: 'user', content: message }
-          });
-        }
-      } catch (enqueueErr) {
-        console.error(`[Crew] Skill dispatch enqueue failed for ${target}:`, enqueueErr.message);
-        const { createRoleQuery } = await import('./role-query.js');
-        roleState = await createRoleQuery(session, target);
-        roleState.inputStream.enqueue({
-          type: 'user',
-          message: { role: 'user', content: message }
-        });
-      }
-      sendStatusUpdate(session);
-      console.log(`[Crew] Skill command dispatched to ${target}: ${message}`);
-      return;
-    }
-    console.log(`[Crew] @mention routing to ${target}: ${message.substring(0, 50)}...`);
-    await dispatchToRole(session, target, buildHumanContent('人工消息:', message), 'human');
-    return;
-  }
-
-  // 默认发给决策者
-  const target = effectiveTargetRole || session.decisionMaker;
-  await dispatchToRole(session, target, buildHumanContent('人工消息:', effectiveContent), 'human');
-}
-
-/**
- * 处理排队的人的消息
- */
-export async function processHumanQueue(session) {
-  if (session.humanMessageQueue.length === 0) return;
-  if (session._processingHumanQueue) return;
-  session._processingHumanQueue = true;
-  try {
-    const msgs = session.humanMessageQueue.splice(0);
-    if (msgs.length === 1) {
-      const humanPrompt = `人工消息:\n${msgs[0].content}`;
-      await dispatchToRole(session, msgs[0].target, humanPrompt, 'human');
-    } else {
-      const byTarget = new Map();
-      for (const m of msgs) {
-        if (!byTarget.has(m.target)) byTarget.set(m.target, []);
-        byTarget.get(m.target).push(m.content);
-      }
-      for (const [target, contents] of byTarget) {
-        const combined = contents.join('\n\n---\n\n');
-        const humanPrompt = `人工消息:\n你有 ${contents.length} 条待处理消息，请一并分析并用多个 ROUTE 块并行分配：\n\n${combined}`;
-        await dispatchToRole(session, target, humanPrompt, 'human');
-      }
-    }
-  } finally {
-    session._processingHumanQueue = false;
-  }
-}