@steerable/agent-shell-web 0.6.48 → 0.6.49

package/src/components/chat/turn-files.ts

@@ -13,6 +13,14 @@ export interface TurnFile {
   kind: 'created' | 'modified';
   /** 字节数（展示用，可缺省）。 */
   size?: number;
+  /** 新增行数（展示用，如 +110）。 */
+  additions?: number;
+  /** 删除行数（展示用，如 -0）。 */
+  deletions?: number;
+  /** 类别：deliverable = present_files 声明的最终交付文件；intermediate = 其余本轮写过的文件。 */
+  category?: 'deliverable' | 'intermediate';
+  /** present_files 给出的一行说明（交付卡片副标题）。 */
+  description?: string;
 }
 
 /** 解析 SSE 事件 / 持久化元数据里的文件列表；形状不符返回 null（与 parseTurnBlocks 同约定）。 */
@@ -28,6 +36,14 @@ export function parseTurnFiles(raw: unknown): TurnFile[] | null {
       path: rec.path,
       kind: rec.kind,
       ...(typeof rec.size === 'number' ? { size: rec.size } : {}),
+      ...(typeof rec.additions === 'number' ? { additions: rec.additions } : {}),
+      ...(typeof rec.deletions === 'number' ? { deletions: rec.deletions } : {}),
+      ...(rec.category === 'deliverable' || rec.category === 'intermediate'
+        ? { category: rec.category }
+        : {}),
+      ...(typeof rec.description === 'string' && rec.description
+        ? { description: rec.description }
+        : {}),
     });
   }
   return files;
@@ -41,6 +57,17 @@ export function splitTurnFilePath(filePath: string): { dir: string; name: string
   return { dir: normalized.slice(0, idx + 1), name: normalized.slice(idx + 1) };
 }
 
+/** 获取小写扩展名（含前导点）。 */
+export function getFileExtension(filePath: string): string {
+  const normalized = filePath.replace(/[/\\]+$/, '');
+  const lastDot = normalized.lastIndexOf('.');
+  const lastSlash = Math.max(normalized.lastIndexOf('/'), normalized.lastIndexOf('\\'));
+  if (lastDot > lastSlash && lastDot >= 0) {
+    return normalized.slice(lastDot).toLowerCase();
+  }
+  return '';
+}
+
 /** 紧凑的字节数展示（不足 1 KB 显示 B）。 */
 export function formatFileSize(size: number | undefined): string | null {
   if (size == null || !Number.isFinite(size) || size < 0) return null;
@@ -48,3 +75,249 @@ export function formatFileSize(size: number | undefined): string | null {
   if (size < 1024 * 1024) return `${(size / 1024).toFixed(1)} KB`;
   return `${(size / (1024 * 1024)).toFixed(1)} MB`;
 }
+
+/** 是否为临时文件 / Office 锁定文件（如 ~$ 开头的文件名），产物列表中一律排除。 */
+export function isIgnoredTurnFile(filePath: string): boolean {
+  const { name } = splitTurnFilePath(filePath);
+  if (!name) return true;
+  if (name === '.DS_Store' || name === 'Thumbs.db') return true;
+  // Office 临时锁定文件
+  if (name.startsWith('~$')) return true;
+  // 临时文件与编辑器交换文件
+  if (name.endsWith('.tmp') || name.endsWith('.swp') || name.endsWith('~')) return true;
+  return false;
+}
+
+/** 交付物扩展名（最终产物，如电子表格、幻灯片、文档、图片、音视频、独立页面、压缩包）。 */
+export const DELIVERABLE_EXTENSIONS = new Set([
+  // 表格 / Spreadsheets
+  '.xlsx', '.xls', '.csv', '.tsv', '.numbers',
+  // 幻灯片 / Presentations
+  '.pptx', '.ppt', '.key', // shell-neutral:allow — Office 幻灯片扩展名，不是产品品牌
+  // 文档 / Documents
+  '.docx', '.doc', '.pdf', '.pages', '.epub', '.rtf',
+  // 图像与富媒体 / Images & Media
+  '.png', '.jpg', '.jpeg', '.gif', '.svg', '.webp', '.mp4', '.mov', '.mp3',
+  // 独立文档输出与压缩包 / Standalone HTML & Archives
+  '.html', '.htm', '.zip', '.tar.gz', '.tar', '.7z',
+]);
+
+/**
+ * 后端按 present_files 声明给出 category；没有 category 的是声明机制上线前
+ * 落库的消息，退回扩展名规则。
+ */
+export function getTurnFileCategory(file: TurnFile): 'deliverable' | 'intermediate' {
+  if (file.category) return file.category;
+  const ext = getFileExtension(file.path);
+  if (DELIVERABLE_EXTENSIONS.has(ext)) return 'deliverable';
+  return 'intermediate';
+}
+
+export type DeliverableFileKind =
+  | 'spreadsheet'
+  | 'presentation'
+  | 'document'
+  | 'pdf'
+  | 'image'
+  | 'archive'
+  | 'code'
+  | 'file';
+
+export interface DeliverableFileMeta {
+  label: string;
+  extBadge: string;
+  kind: DeliverableFileKind;
+  themeColor: 'emerald' | 'amber' | 'blue' | 'rose' | 'purple' | 'slate';
+}
+
+export function getDeliverableMeta(filePath: string): DeliverableFileMeta {
+  const ext = getFileExtension(filePath);
+  const extClean = ext.replace(/^\./, '').toUpperCase() || 'FILE';
+
+  switch (ext) {
+    case '.xlsx':
+    case '.xls':
+    case '.csv':
+    case '.tsv':
+    case '.numbers':
+      return {
+        label: '电子表格',
+        extBadge: extClean,
+        kind: 'spreadsheet',
+        themeColor: 'emerald',
+      };
+    case '.pptx': // shell-neutral:allow — Office 幻灯片扩展名，不是产品品牌
+    case '.ppt': // shell-neutral:allow — Office 幻灯片扩展名，不是产品品牌
+    case '.key':
+      return {
+        label: '演示文稿',
+        extBadge: extClean,
+        kind: 'presentation',
+        themeColor: 'amber',
+      };
+    case '.docx':
+    case '.doc':
+    case '.pages':
+    case '.rtf':
+      return {
+        label: '文档',
+        extBadge: extClean,
+        kind: 'document',
+        themeColor: 'blue',
+      };
+    case '.pdf':
+      return {
+        label: '文档',
+        extBadge: 'PDF',
+        kind: 'pdf',
+        themeColor: 'rose',
+      };
+    case '.png':
+    case '.jpg':
+    case '.jpeg':
+    case '.gif':
+    case '.webp':
+    case '.svg':
+      return {
+        label: '图像',
+        extBadge: extClean,
+        kind: 'image',
+        themeColor: 'purple',
+      };
+    case '.zip':
+    case '.tar':
+    case '.gz':
+    case '.7z':
+      return {
+        label: '压缩包',
+        extBadge: extClean,
+        kind: 'archive',
+        themeColor: 'amber',
+      };
+    case '.html':
+    case '.htm':
+      return {
+        label: '网页',
+        extBadge: 'HTML',
+        kind: 'document',
+        themeColor: 'blue',
+      };
+    default:
+      return {
+        label: '文件',
+        extBadge: extClean,
+        kind: 'file',
+        themeColor: 'slate',
+      };
+  }
+}
+
+/** 规范化展示路径（若在特定项目/脚本目录下，提取便于阅读的相对路径）。 */
+export function formatIntermediateDisplayPath(filePath: string): string {
+  const normalized = filePath.replace(/\\/g, '/');
+  const markers = ['/review_work/', '/src/', '/scripts/', '/packages/', '/work/', '/tests/'];
+  for (const marker of markers) {
+    const idx = normalized.indexOf(marker);
+    if (idx >= 0) {
+      return normalized.slice(idx + 1);
+    }
+  }
+  const { dir, name } = splitTurnFilePath(filePath);
+  if (!dir) return name;
+  const parts = dir.replace(/\/$/, '').split(/[/\\]/);
+  const lastFolder = parts[parts.length - 1];
+  return lastFolder ? `${lastFolder}/${name}` : name;
+}
+
+export interface GroupedTurnFiles {
+  deliverables: TurnFile[];
+  intermediates: TurnFile[];
+  totalAdditions: number;
+  totalDeletions: number;
+}
+
+/** 从 unified diff 解析增删行数。 */
+function parseDiffStats(diffText: string): { additions: number; deletions: number } {
+  let additions = 0;
+  let deletions = 0;
+  for (const line of diffText.split('\n')) {
+    if (line.startsWith('+') && !line.startsWith('+++')) additions++;
+    else if (line.startsWith('-') && !line.startsWith('---')) deletions++;
+  }
+  return { additions, deletions };
+}
+
+/** 把本轮文件按最终交付物（最终文件）与中间修改文件分类，并合并行数统计。 */
+export function groupTurnFiles(
+  files: TurnFile[],
+  executedActions?: Array<{ tool?: string; arguments?: unknown; result?: unknown }>,
+): GroupedTurnFiles {
+  // 1. 过滤垃圾/临时锁定文件
+  const validFiles = files.filter((f) => !isIgnoredTurnFile(f.path));
+
+  // 2. 从 executedActions 中补充 diff 统计
+  const statsByPath = new Map<string, { additions: number; deletions: number }>();
+  if (Array.isArray(executedActions)) {
+    for (const act of executedActions) {
+      const args = act.arguments && typeof act.arguments === 'object' ? (act.arguments as Record<string, unknown>) : null;
+      const rawPath = typeof args?.path === 'string' ? args.path : null;
+      if (rawPath) {
+        if (act.tool === 'local_edit_file') {
+          const res = act.result && typeof act.result === 'object' ? (act.result as Record<string, unknown>) : null;
+          if (typeof res?.diff === 'string') {
+            const diffStats = parseDiffStats(res.diff);
+            const prev = statsByPath.get(rawPath) ?? { additions: 0, deletions: 0 };
+            statsByPath.set(rawPath, {
+              additions: prev.additions + diffStats.additions,
+              deletions: prev.deletions + diffStats.deletions,
+            });
+          }
+        } else if (act.tool === 'local_write_file') {
+          if (typeof args?.content === 'string') {
+            const lines = args.content ? args.content.split('\n').length : 0;
+            statsByPath.set(rawPath, { additions: lines, deletions: 0 });
+          }
+        }
+      }
+    }
+  }
+
+  const deliverables: TurnFile[] = [];
+  const intermediates: TurnFile[] = [];
+  let totalAdditions = 0;
+  let totalDeletions = 0;
+
+  for (const file of validFiles) {
+    const cat = getTurnFileCategory(file);
+    // 匹配统计：尝试直接匹配或 basename 匹配
+    let stats = statsByPath.get(file.path);
+    if (!stats) {
+      for (const [p, s] of statsByPath.entries()) {
+        if (p.endsWith(file.path) || file.path.endsWith(p)) {
+          stats = s;
+          break;
+        }
+      }
+    }
+
+    const additions = file.additions ?? stats?.additions;
+    const deletions = file.deletions ?? stats?.deletions;
+
+    const enriched: TurnFile = {
+      ...file,
+      category: cat,
+      ...(additions !== undefined ? { additions } : {}),
+      ...(deletions !== undefined ? { deletions } : {}),
+    };
+
+    if (cat === 'deliverable') {
+      deliverables.push(enriched);
+    } else {
+      intermediates.push(enriched);
+      if (typeof additions === 'number') totalAdditions += additions;
+      if (typeof deletions === 'number') totalDeletions += deletions;
+    }
+  }
+
+  return { deliverables, intermediates, totalAdditions, totalDeletions };
+}