clawt 2.18.0 → 2.19.0

package/src/utils/fs.ts

@@ -1,4 +1,5 @@
-import { existsSync, mkdirSync, readdirSync, rmdirSync } from 'node:fs';
+import { existsSync, mkdirSync, readdirSync, rmdirSync, statSync } from 'node:fs';
+import { join } from 'node:path';
 
 /**
  * 确保目录存在，不存在则递归创建
@@ -26,3 +27,33 @@ export function removeEmptyDir(dirPath: string): boolean {
   }
   return false;
 }
+
+/**
+ * 递归计算目录占用的磁盘大小（字节）
+ * @param {string} dirPath - 目录路径
+ * @returns {number} 目录总大小（字节）
+ */
+export function calculateDirSize(dirPath: string): number {
+  let totalSize = 0;
+
+  try {
+    const entries = readdirSync(dirPath, { withFileTypes: true });
+
+    for (const entry of entries) {
+      const fullPath = join(dirPath, entry.name);
+      try {
+        if (entry.isDirectory()) {
+          totalSize += calculateDirSize(fullPath);
+        } else if (entry.isFile()) {
+          totalSize += statSync(fullPath).size;
+        }
+      } catch {
+        // 忽略无法访问的文件
+      }
+    }
+  } catch {
+    // 忽略无法读取的目录
+  }
+
+  return totalSize;
+}

package/src/utils/index.ts

@@ -51,8 +51,8 @@ export { sanitizeBranchName, generateBranchNames, validateBranchesNotExist } fro
 export { validateMainWorktree, validateGitInstalled, validateClaudeCodeInstalled } from './validation.js';
 export { createWorktrees, getProjectWorktrees, getProjectWorktreeDir, cleanupWorktrees, getWorktreeStatus, createWorktreesByBranches } from './worktree.js';
 export { loadConfig, writeDefaultConfig, writeConfig, saveConfig, getConfigValue, ensureClawtDirs, parseConcurrency } from './config.js';
-export { printSuccess, printError, printWarning, printInfo, printHint, printSeparator, printDoubleSeparator, confirmAction, confirmDestructiveAction, formatWorktreeStatus, isWorktreeIdle, formatDuration, formatRelativeTime } from './formatter.js';
-export { ensureDir, removeEmptyDir } from './fs.js';
+export { printSuccess, printError, printWarning, printInfo, printHint, printSeparator, printDoubleSeparator, confirmAction, confirmDestructiveAction, formatWorktreeStatus, isWorktreeIdle, formatDuration, formatRelativeTime, formatDiskSize, formatLocalISOString } from './formatter.js';
+export { ensureDir, removeEmptyDir, calculateDirSize } from './fs.js';
 export { multilineInput } from './prompt.js';
 export { launchInteractiveClaude, hasClaudeSessionHistory, launchInteractiveClaudeInNewTerminal } from './claude.js';
 export { getSnapshotPath, hasSnapshot, getSnapshotModifiedTime, readSnapshotTreeHash, readSnapshot, writeSnapshot, removeSnapshot, removeProjectSnapshots, getProjectSnapshotBranches } from './validate-snapshot.js';

package/tests/unit/utils/formatter.test.ts

@@ -1,5 +1,5 @@
 import { describe, it, expect, vi } from 'vitest';
-import { formatWorktreeStatus, printSuccess, printError, printWarning, printInfo, printSeparator, printDoubleSeparator, isWorktreeIdle, formatDuration, formatRelativeTime } from '../../../src/utils/formatter.js';
+import { formatWorktreeStatus, printSuccess, printError, printWarning, printInfo, printSeparator, printDoubleSeparator, isWorktreeIdle, formatDuration, formatRelativeTime, formatDiskSize, formatLocalISOString } from '../../../src/utils/formatter.js';
 import { createWorktreeStatus } from '../../helpers/fixtures.js';
 
 describe('formatWorktreeStatus', () => {
@@ -177,3 +177,93 @@ describe('formatRelativeTime', () => {
     expect(formatRelativeTime(future.toISOString())).toBe('刚刚');
   });
 });
+
+describe('formatDiskSize', () => {
+  it('0 字节时返回 "0 B"', () => {
+    expect(formatDiskSize(0)).toBe('0 B');
+  });
+
+  it('小于 1 KB 时以 B 为单位', () => {
+    expect(formatDiskSize(1)).toBe('1 B');
+    expect(formatDiskSize(512)).toBe('512 B');
+    expect(formatDiskSize(1023)).toBe('1023 B');
+  });
+
+  it('恰好 1 KB 时返回 "1.0 KB"', () => {
+    expect(formatDiskSize(1024)).toBe('1.0 KB');
+  });
+
+  it('KB 范围内保留一位小数', () => {
+    expect(formatDiskSize(1536)).toBe('1.5 KB');
+    expect(formatDiskSize(10240)).toBe('10.0 KB');
+    expect(formatDiskSize(1024 * 1024 - 1)).toBe('1024.0 KB');
+  });
+
+  it('恰好 1 MB 时返回 "1.0 MB"', () => {
+    expect(formatDiskSize(1024 * 1024)).toBe('1.0 MB');
+  });
+
+  it('MB 范围内保留一位小数', () => {
+    expect(formatDiskSize(1.5 * 1024 * 1024)).toBe('1.5 MB');
+    expect(formatDiskSize(256 * 1024 * 1024)).toBe('256.0 MB');
+    expect(formatDiskSize(1024 * 1024 * 1024 - 1)).toBe('1024.0 MB');
+  });
+
+  it('恰好 1 GB 时返回 "1.0 GB"', () => {
+    expect(formatDiskSize(1024 * 1024 * 1024)).toBe('1.0 GB');
+  });
+
+  it('GB 范围内保留一位小数', () => {
+    expect(formatDiskSize(1.5 * 1024 * 1024 * 1024)).toBe('1.5 GB');
+    expect(formatDiskSize(10 * 1024 * 1024 * 1024)).toBe('10.0 GB');
+  });
+});
+
+describe('formatLocalISOString', () => {
+  it('返回值包含日期和时间部分', () => {
+    const date = new Date('2025-06-15T10:30:00Z');
+    const result = formatLocalISOString(date);
+    // 结果应符合 ISO 8601 带时区偏移的格式
+    expect(result).toMatch(/^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\.\d{3}[+-]\d{2}:\d{2}$/);
+  });
+
+  it('返回值不以 Z 结尾（应包含时区偏移）', () => {
+    const date = new Date();
+    const result = formatLocalISOString(date);
+    expect(result).not.toMatch(/Z$/);
+  });
+
+  it('时区偏移格式为 +HH:MM 或 -HH:MM', () => {
+    const date = new Date();
+    const result = formatLocalISOString(date);
+    const tzPart = result.slice(-6);
+    expect(tzPart).toMatch(/^[+-]\d{2}:\d{2}$/);
+  });
+
+  it('时区偏移量与本机 getTimezoneOffset 一致', () => {
+    const date = new Date('2025-01-01T12:00:00Z');
+    const result = formatLocalISOString(date);
+    const tzPart = result.slice(-6);
+    const sign = tzPart[0];
+    const hours = parseInt(tzPart.slice(1, 3), 10);
+    const minutes = parseInt(tzPart.slice(4, 6), 10);
+
+    // 从偏移字符串反推总分钟数
+    const totalMinutesFromResult = (sign === '+' ? 1 : -1) * (hours * 60 + minutes);
+    const expectedMinutes = -date.getTimezoneOffset();
+    expect(totalMinutesFromResult).toBe(expectedMinutes);
+  });
+
+  it('不同日期对象产生不同的输出', () => {
+    const date1 = new Date('2025-01-01T00:00:00Z');
+    const date2 = new Date('2025-06-15T12:30:00Z');
+    expect(formatLocalISOString(date1)).not.toBe(formatLocalISOString(date2));
+  });
+
+  it('毫秒部分被保留', () => {
+    const date = new Date('2025-03-20T08:15:30.123Z');
+    const result = formatLocalISOString(date);
+    // 毫秒值应出现在结果中
+    expect(result).toContain('.123');
+  });
+});

package/tests/unit/utils/fs.test.ts

@@ -6,15 +6,17 @@ vi.mock('node:fs', () => ({
   mkdirSync: vi.fn(),
   readdirSync: vi.fn(),
   rmdirSync: vi.fn(),
+  statSync: vi.fn(),
 }));
 
-import { existsSync, mkdirSync, readdirSync, rmdirSync } from 'node:fs';
-import { ensureDir, removeEmptyDir } from '../../../src/utils/fs.js';
+import { existsSync, mkdirSync, readdirSync, rmdirSync, statSync } from 'node:fs';
+import { ensureDir, removeEmptyDir, calculateDirSize } from '../../../src/utils/fs.js';
 
 const mockedExistsSync = vi.mocked(existsSync);
 const mockedMkdirSync = vi.mocked(mkdirSync);
 const mockedReaddirSync = vi.mocked(readdirSync);
 const mockedRmdirSync = vi.mocked(rmdirSync);
+const mockedStatSync = vi.mocked(statSync);
 
 describe('ensureDir', () => {
   it('目录不存在时创建', () => {
@@ -52,3 +54,124 @@ describe('removeEmptyDir', () => {
     expect(mockedReaddirSync).not.toHaveBeenCalled();
   });
 });
+
+/**
+ * 创建模拟的 Dirent 对象
+ * @param {string} name - 文件/目录名
+ * @param {'file' | 'directory'} type - 类型
+ * @returns {import('node:fs').Dirent} 模拟的 Dirent 对象
+ */
+function createMockDirent(name: string, type: 'file' | 'directory') {
+  return {
+    name,
+    isFile: () => type === 'file',
+    isDirectory: () => type === 'directory',
+    isBlockDevice: () => false,
+    isCharacterDevice: () => false,
+    isSymbolicLink: () => false,
+    isFIFO: () => false,
+    isSocket: () => false,
+    parentPath: '',
+    path: '',
+  } as import('node:fs').Dirent;
+}
+
+describe('calculateDirSize', () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it('空目录返回 0', () => {
+    mockedReaddirSync.mockReturnValue([]);
+    expect(calculateDirSize('/tmp/empty')).toBe(0);
+  });
+
+  it('只有文件时累加所有文件大小', () => {
+    mockedReaddirSync.mockReturnValue([
+      createMockDirent('a.txt', 'file'),
+      createMockDirent('b.txt', 'file'),
+    ]);
+    mockedStatSync
+      .mockReturnValueOnce({ size: 100 } as import('node:fs').Stats)
+      .mockReturnValueOnce({ size: 200 } as import('node:fs').Stats);
+
+    expect(calculateDirSize('/tmp/dir')).toBe(300);
+  });
+
+  it('递归计算子目录中的文件大小', () => {
+    // 第一次调用：顶层目录包含一个文件和一个子目录
+    mockedReaddirSync
+      .mockReturnValueOnce([
+        createMockDirent('file.txt', 'file'),
+        createMockDirent('subdir', 'directory'),
+      ])
+      // 第二次调用：子目录包含一个文件
+      .mockReturnValueOnce([
+        createMockDirent('nested.txt', 'file'),
+      ]);
+
+    mockedStatSync
+      .mockReturnValueOnce({ size: 500 } as import('node:fs').Stats)
+      .mockReturnValueOnce({ size: 300 } as import('node:fs').Stats);
+
+    expect(calculateDirSize('/tmp/dir')).toBe(800);
+  });
+
+  it('readdirSync 抛出异常时返回 0', () => {
+    mockedReaddirSync.mockImplementation(() => {
+      throw new Error('权限不足');
+    });
+    expect(calculateDirSize('/tmp/no-access')).toBe(0);
+  });
+
+  it('statSync 对个别文件抛出异常时跳过该文件继续计算', () => {
+    mockedReaddirSync.mockReturnValue([
+      createMockDirent('good.txt', 'file'),
+      createMockDirent('bad.txt', 'file'),
+      createMockDirent('ok.txt', 'file'),
+    ]);
+    mockedStatSync
+      .mockReturnValueOnce({ size: 100 } as import('node:fs').Stats)
+      .mockImplementationOnce(() => { throw new Error('无法访问'); })
+      .mockReturnValueOnce({ size: 200 } as import('node:fs').Stats);
+
+    expect(calculateDirSize('/tmp/dir')).toBe(300);
+  });
+
+  it('多层嵌套目录正确累加', () => {
+    // 顶层：子目录 a
+    mockedReaddirSync
+      .mockReturnValueOnce([
+        createMockDirent('a', 'directory'),
+      ])
+      // a 目录：文件 + 子目录 b
+      .mockReturnValueOnce([
+        createMockDirent('x.txt', 'file'),
+        createMockDirent('b', 'directory'),
+      ])
+      // b 目录：一个文件
+      .mockReturnValueOnce([
+        createMockDirent('y.txt', 'file'),
+      ]);
+
+    mockedStatSync
+      .mockReturnValueOnce({ size: 1000 } as import('node:fs').Stats)
+      .mockReturnValueOnce({ size: 2000 } as import('node:fs').Stats);
+
+    expect(calculateDirSize('/tmp/root')).toBe(3000);
+  });
+
+  it('只包含子目录（无直接文件）时正确递归', () => {
+    mockedReaddirSync
+      .mockReturnValueOnce([
+        createMockDirent('sub', 'directory'),
+      ])
+      .mockReturnValueOnce([
+        createMockDirent('inner.txt', 'file'),
+      ]);
+
+    mockedStatSync.mockReturnValueOnce({ size: 42 } as import('node:fs').Stats);
+
+    expect(calculateDirSize('/tmp/dir')).toBe(42);
+  });
+});