@yeaft/webchat-agent 0.0.234 → 0.0.236

package/crew/human-interaction.js

@@ -0,0 +1,115 @@
+/**
+ * Crew — 人工交互
+ * handleCrewHumanInput, processHumanQueue
+ */
+import { dispatchToRole } from './routing.js';
+import { sendStatusUpdate } from './ui-messages.js';
+
+/**
+ * 处理人的输入
+ */
+export async function handleCrewHumanInput(msg) {
+  // Lazy import to avoid circular dependency
+  const { crewSessions } = await import('./session.js');
+
+  const { sessionId, content, targetRole, files } = msg;
+  const session = crewSessions.get(sessionId);
+  if (!session) {
+    console.warn(`[Crew] Session not found: ${sessionId}`);
+    return;
+  }
+
+  // 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);
+
+    const target = targetRole || waitingContext?.fromRole || session.decisionMaker;
+    await dispatchToRole(session, target, buildHumanContent('人工回复:', content), 'human');
+    return;
+  }
+
+  // 解析 @role 指令
+  const atMatch = content.match(/^@(\S+)\s*([\s\S]*)/);
+  if (atMatch) {
+    const atTarget = atMatch[1];
+    const message = atMatch[2].trim() || content;
+
+    let target = null;
+    for (const [name, role] of session.roles) {
+      if (name === atTarget.toLowerCase()) {
+        target = name;
+        break;
+      }
+      if (role.displayName === atTarget) {
+        target = name;
+        break;
+      }
+    }
+
+    if (target) {
+      await dispatchToRole(session, target, buildHumanContent('人工消息:', message), 'human');
+      return;
+    }
+  }
+
+  // 默认发给决策者
+  const target = targetRole || session.decisionMaker;
+  await dispatchToRole(session, target, buildHumanContent('人工消息:', content), '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;
+  }
+}

package/crew/persistence.js

@@ -0,0 +1,287 @@
+/**
+ * Crew — 持久化管理
+ * Session 索引 (~/.claude/crew-sessions.json)、session 元数据、消息分片
+ */
+import { promises as fs } from 'fs';
+import { join } from 'path';
+import { homedir } from 'os';
+
+// =====================================================================
+// Crew Session Index (~/.claude/crew-sessions.json)
+// =====================================================================
+
+const CREW_INDEX_PATH = join(homedir(), '.claude', 'crew-sessions.json');
+
+// 写入锁：防止并发写入导致文件损坏
+let _indexWriteLock = Promise.resolve();
+
+export async function loadCrewIndex() {
+  try { return JSON.parse(await fs.readFile(CREW_INDEX_PATH, 'utf-8')); }
+  catch { return []; }
+}
+
+async function saveCrewIndex(index) {
+  const doWrite = async () => {
+    await fs.mkdir(join(homedir(), '.claude'), { recursive: true });
+    const data = JSON.stringify(index, null, 2);
+    // 先写临时文件再 rename，保证原子性
+    const tmpPath = CREW_INDEX_PATH + '.tmp';
+    await fs.writeFile(tmpPath, data);
+    await fs.rename(tmpPath, CREW_INDEX_PATH);
+  };
+  // 串行化写入
+  _indexWriteLock = _indexWriteLock.then(doWrite, doWrite);
+  return _indexWriteLock;
+}
+
+function sessionToIndexEntry(session) {
+  return {
+    sessionId: session.id,
+    projectDir: session.projectDir,
+    sharedDir: session.sharedDir,
+    status: session.status,
+    name: session.name || '',
+    userId: session.userId,
+    username: session.username,
+    agentId: session.agentId || null,
+    createdAt: session.createdAt,
+    updatedAt: Date.now()
+  };
+}
+
+export async function upsertCrewIndex(session) {
+  const index = await loadCrewIndex();
+  const entry = sessionToIndexEntry(session);
+  const idx = index.findIndex(e => e.sessionId === session.id);
+  if (idx >= 0) index[idx] = entry; else index.push(entry);
+  await saveCrewIndex(index);
+}
+
+export async function removeFromCrewIndex(sessionId) {
+  // Lazy import to avoid circular dependency
+  const { crewSessions } = await import('./session.js');
+
+  const index = await loadCrewIndex();
+  const entry = index.find(e => e.sessionId === sessionId);
+  const filtered = index.filter(e => e.sessionId !== sessionId);
+  if (filtered.length !== index.length) {
+    await saveCrewIndex(filtered);
+    console.log(`[Crew] Removed session ${sessionId} from index`);
+  }
+  // 从内存中也移除（防止 sendConversationList 重新加入）
+  if (crewSessions.has(sessionId)) {
+    crewSessions.delete(sessionId);
+    console.log(`[Crew] Removed session ${sessionId} from active sessions`);
+  }
+  // 删除磁盘上的 session 数据文件
+  const sharedDir = entry?.sharedDir;
+  if (sharedDir) {
+    try {
+      for (const file of ['session.json', 'messages.json']) {
+        await fs.unlink(join(sharedDir, file)).catch(() => {});
+      }
+      // Clean up message shard files
+      await cleanupMessageShards(sharedDir);
+      console.log(`[Crew] Cleaned session files in ${sharedDir}`);
+    } catch (e) {
+      console.warn(`[Crew] Failed to clean session files:`, e.message);
+    }
+  }
+}
+
+// =====================================================================
+// Session Metadata (.crew/session.json)
+// =====================================================================
+
+const MESSAGE_SHARD_SIZE = 256 * 1024; // 256KB per shard
+
+export async function saveSessionMeta(session) {
+  const meta = {
+    sessionId: session.id,
+    projectDir: session.projectDir,
+    sharedDir: session.sharedDir,
+    name: session.name || '',
+    status: session.status,
+    roles: Array.from(session.roles.values()).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,
+    round: session.round,
+    createdAt: session.createdAt,
+    updatedAt: Date.now(),
+    userId: session.userId,
+    username: session.username,
+    agentId: session.agentId || null,
+    teamType: session.teamType || 'dev',
+    language: session.language || 'zh-CN',
+    costUsd: session.costUsd,
+    totalInputTokens: session.totalInputTokens,
+    totalOutputTokens: session.totalOutputTokens,
+    features: Array.from(session.features.values()),
+    _completedTaskIds: Array.from(session._completedTaskIds || [])
+  };
+  await fs.writeFile(join(session.sharedDir, 'session.json'), JSON.stringify(meta, null, 2));
+  // 保存 UI 消息历史（用于恢复时重放）
+  if (session.uiMessages && session.uiMessages.length > 0) {
+    // 清理 _streaming 标记后保存
+    const cleaned = session.uiMessages.map(m => {
+      const { _streaming, ...rest } = m;
+      return rest;
+    });
+    const json = JSON.stringify(cleaned);
+    // 超过阈值时直接归档（rotateMessages 内部写两个文件，避免双写）
+    if (json.length > MESSAGE_SHARD_SIZE && !session._rotating) {
+      await rotateMessages(session, cleaned);
+    } else {
+      await fs.writeFile(join(session.sharedDir, 'messages.json'), json);
+    }
+  }
+}
+
+/**
+ * 归档旧消息到分片文件（logrotate 风格）
+ */
+async function rotateMessages(session, cleaned) {
+  session._rotating = true;
+  try {
+    const halfLen = Math.floor(cleaned.length / 2);
+    let splitIdx = halfLen;
+    for (let i = halfLen; i > Math.max(0, halfLen - 20); i--) {
+      if (cleaned[i].type === 'route' || cleaned[i].type === 'system') {
+        splitIdx = i + 1;
+        break;
+      }
+    }
+    if (splitIdx === halfLen) {
+      for (let i = halfLen + 1; i < Math.min(cleaned.length - 1, halfLen + 20); i++) {
+        if (cleaned[i].type === 'route' || cleaned[i].type === 'system') {
+          splitIdx = i + 1;
+          break;
+        }
+      }
+    }
+    splitIdx = Math.max(1, Math.min(splitIdx, cleaned.length - 1));
+
+    const archivePart = cleaned.slice(0, splitIdx);
+    const remainPart = cleaned.slice(splitIdx);
+
+    const maxShard = await getMaxShardIndex(session.sharedDir);
+    for (let i = maxShard; i >= 1; i--) {
+      const src = join(session.sharedDir, `messages.${i}.json`);
+      const dst = join(session.sharedDir, `messages.${i + 1}.json`);
+      await fs.rename(src, dst).catch(() => {});
+    }
+
+    await fs.writeFile(join(session.sharedDir, 'messages.1.json'), JSON.stringify(archivePart));
+    await fs.writeFile(join(session.sharedDir, 'messages.json'), JSON.stringify(remainPart));
+    session.uiMessages = remainPart.map(m => ({ ...m }));
+
+    console.log(`[Crew] Rotated messages: archived ${archivePart.length} msgs to shard 1, kept ${remainPart.length} in active`);
+  } finally {
+    session._rotating = false;
+  }
+}
+
+/**
+ * 获取当前最大分片编号
+ */
+export async function getMaxShardIndex(sharedDir) {
+  let max = 0;
+  try {
+    const files = await fs.readdir(sharedDir);
+    for (const f of files) {
+      const match = f.match(/^messages\.(\d+)\.json$/);
+      if (match) {
+        const idx = parseInt(match[1], 10);
+        if (idx > max) max = idx;
+      }
+    }
+  } catch { /* dir may not exist */ }
+  return max;
+}
+
+/**
+ * 删除所有消息分片文件
+ */
+export async function cleanupMessageShards(sharedDir) {
+  try {
+    const files = await fs.readdir(sharedDir);
+    for (const f of files) {
+      if (/^messages\.\d+\.json$/.test(f)) {
+        await fs.unlink(join(sharedDir, f)).catch(() => {});
+      }
+    }
+  } catch { /* dir may not exist */ }
+}
+
+export async function loadSessionMeta(sharedDir) {
+  try { return JSON.parse(await fs.readFile(join(sharedDir, 'session.json'), 'utf-8')); }
+  catch { return null; }
+}
+
+export async function loadSessionMessages(sharedDir) {
+  let messages = [];
+  try { messages = JSON.parse(await fs.readFile(join(sharedDir, 'messages.json'), 'utf-8')); }
+  catch { /* file may not exist */ }
+  let hasOlderMessages = false;
+  try {
+    await fs.access(join(sharedDir, 'messages.1.json'));
+    hasOlderMessages = true;
+  } catch { /* no older shards */ }
+  return { messages, hasOlderMessages };
+}
+
+/**
+ * 加载历史消息分片（前端上滑到顶部时按需请求）
+ */
+export async function handleLoadCrewHistory(msg) {
+  const { sessionId, requestId } = msg;
+  const shardIndex = parseInt(msg.shardIndex, 10);
+
+  // Lazy import to avoid circular dependency
+  const { crewSessions } = await import('./session.js');
+  const { sendCrewMessage } = await import('./ui-messages.js');
+
+  if (!Number.isFinite(shardIndex) || shardIndex < 1) {
+    sendCrewMessage({
+      type: 'crew_history_loaded',
+      sessionId,
+      shardIndex: msg.shardIndex,
+      requestId,
+      messages: [],
+      hasMore: false
+    });
+    return;
+  }
+  const session = crewSessions.get(sessionId);
+  if (!session) {
+    sendCrewMessage({
+      type: 'crew_history_loaded',
+      sessionId,
+      shardIndex,
+      requestId,
+      messages: [],
+      hasMore: false
+    });
+    return;
+  }
+
+  const shardPath = join(session.sharedDir, `messages.${shardIndex}.json`);
+  let messages = [];
+  try {
+    messages = JSON.parse(await fs.readFile(shardPath, 'utf-8'));
+  } catch { /* shard file doesn't exist */ }
+
+  const hasMore = shardIndex < await getMaxShardIndex(session.sharedDir);
+
+  sendCrewMessage({
+    type: 'crew_history_loaded',
+    sessionId,
+    shardIndex,
+    requestId,
+    messages,
+    hasMore
+  });
+}

package/crew/role-management.js

@@ -0,0 +1,131 @@
+/**
+ * Crew — 动态角色管理
+ * addRoleToSession, removeRoleFromSession
+ */
+import { initRoleDir, updateSharedClaudeMd } from './shared-dir.js';
+import { saveRoleSessionId } from './role-query.js';
+import { sendCrewMessage, sendCrewOutput, sendStatusUpdate } from './ui-messages.js';
+
+/** Format role label */
+function roleLabel(r) {
+  return r.icon ? `${r.icon} ${r.displayName}` : r.displayName;
+}
+
+/**
+ * 向现有 session 动态添加角色
+ */
+export async function addRoleToSession(msg) {
+  // Lazy import to avoid circular dependency
+  const { crewSessions, expandRoles } = await import('./session.js');
+
+  const { sessionId, role } = msg;
+  const session = crewSessions.get(sessionId);
+  if (!session) {
+    console.warn(`[Crew] Session not found: ${sessionId}`);
+    return;
+  }
+
+  const rolesToAdd = expandRoles([role]);
+
+  for (const r of rolesToAdd) {
+    if (session.roles.has(r.name)) {
+      console.warn(`[Crew] Role already exists: ${r.name}`);
+      continue;
+    }
+
+    session.roles.set(r.name, r);
+
+    if (r.isDecisionMaker) {
+      session.decisionMaker = r.name;
+    }
+    if (!session.decisionMaker) {
+      session.decisionMaker = r.name;
+    }
+
+    await initRoleDir(session.sharedDir, r, session.language || 'zh-CN');
+
+    console.log(`[Crew] Role added: ${r.name} (${r.displayName}) to session ${sessionId}`);
+
+    sendCrewMessage({
+      type: 'crew_role_added',
+      sessionId,
+      role: {
+        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: session.decisionMaker
+    });
+
+    sendCrewOutput(session, 'system', 'system', {
+      type: 'assistant',
+      message: { role: 'assistant', content: [{ type: 'text', text: `${roleLabel(r)} 加入了群聊` }] }
+    });
+  }
+
+  await updateSharedClaudeMd(session);
+  sendStatusUpdate(session);
+}
+
+/**
+ * 从 session 移除角色
+ */
+export async function removeRoleFromSession(msg) {
+  const { crewSessions } = await import('./session.js');
+
+  const { sessionId, roleName } = msg;
+  const session = crewSessions.get(sessionId);
+  if (!session) {
+    console.warn(`[Crew] Session not found: ${sessionId}`);
+    return;
+  }
+
+  const role = session.roles.get(roleName);
+  if (!role) {
+    console.warn(`[Crew] Role not found: ${roleName}`);
+    return;
+  }
+
+  // 停止角色的 query
+  const roleState = session.roleStates.get(roleName);
+  if (roleState) {
+    if (roleState.claudeSessionId) {
+      await saveRoleSessionId(session.sharedDir, roleName, roleState.claudeSessionId);
+    }
+    if (roleState.abortController) {
+      roleState.abortController.abort();
+    }
+    session.roleStates.delete(roleName);
+  }
+
+  session.roles.delete(roleName);
+
+  if (session.decisionMaker === roleName) {
+    const remaining = Array.from(session.roles.values());
+    const newDM = remaining.find(r => r.isDecisionMaker) || remaining[0];
+    session.decisionMaker = newDM?.name || null;
+  }
+
+  await updateSharedClaudeMd(session);
+
+  console.log(`[Crew] Role removed: ${roleName} from session ${sessionId}`);
+
+  sendCrewMessage({
+    type: 'crew_role_removed',
+    sessionId,
+    roleName,
+    decisionMaker: session.decisionMaker
+  });
+
+  sendCrewOutput(session, 'system', 'system', {
+    type: 'assistant',
+    message: { role: 'assistant', content: [{ type: 'text', text: `${roleLabel(role)} 离开了群聊` }] }
+  });
+
+  sendStatusUpdate(session);
+}