hamster-wheel-cli 0.1.0

package/tests/logs.test.ts

@@ -0,0 +1,33 @@
+import assert from 'node:assert/strict';
+import path from 'node:path';
+import { test } from 'node:test';
+import { buildAutoLogFilePath, formatCommandLine, formatTimeString, getLogMetaPath, getLogsDir, sanitizeBranchName } from '../src/logs';
+
+test('formatTimeString 输出 14 位时间串', () => {
+  const date = new Date(2025, 0, 1, 14, 1, 1);
+  assert.equal(formatTimeString(date), '20250101140101');
+});
+
+test('sanitizeBranchName 会替换非法字符', () => {
+  assert.equal(sanitizeBranchName('feature/foo/bar'), 'feature-foo-bar');
+  assert.equal(sanitizeBranchName(' fix:bug? '), 'fix-bug');
+});
+
+test('buildAutoLogFilePath 会拼接日志目录与分支名', () => {
+  const date = new Date(2025, 0, 1, 8, 0, 0);
+  const filePath = buildAutoLogFilePath('feat/alpha', date);
+  const expected = path.join(getLogsDir(), 'wheel-ai-auto-log-20250101080000-feat-alpha.log');
+  assert.equal(filePath, expected);
+});
+
+test('getLogMetaPath 使用同名 json', () => {
+  const logFile = path.join('/tmp', 'wheel-ai-auto-log-20250101140101-main.log');
+  const metaPath = getLogMetaPath(logFile);
+  const expected = path.join(getLogsDir(), 'wheel-ai-auto-log-20250101140101-main.json');
+  assert.equal(metaPath, expected);
+});
+
+test('formatCommandLine 会为包含空格的参数加引号', () => {
+  const commandLine = formatCommandLine(['node', 'cli.js', 'run', '--log-file', 'a b']);
+  assert.equal(commandLine, 'node cli.js run --log-file "a b"');
+});

package/tests/prompt.test.ts

@@ -0,0 +1,20 @@
+import assert from 'node:assert/strict';
+import { test } from 'node:test';
+import { buildPrompt } from '../src/ai';
+
+test('buildPrompt 输出包含核心段落与迭代信息', () => {
+  const prompt = buildPrompt({
+    task: 'demo task',
+    workflowGuide: 'workflow guide',
+    plan: 'current plan',
+    notes: 'historical notes',
+    iteration: 2
+  });
+
+  assert.ok(prompt.includes('# 背景任务'));
+  assert.ok(prompt.includes('demo task'));
+  assert.ok(prompt.includes('workflow guide'));
+  assert.ok(prompt.includes('current plan'));
+  assert.ok(prompt.includes('historical notes'));
+  assert.ok(prompt.includes('迭代'));
+});

package/tests/run-command-stream.test.ts

@@ -0,0 +1,60 @@
+import assert from 'node:assert/strict';
+import { afterEach, beforeEach, test } from 'node:test';
+import { Logger } from '../src/logger';
+import { runCommand } from '../src/utils';
+
+type ConsoleMethodName = 'log';
+
+const ansiPattern = /\u001b\[[0-9;]*m/g;
+const timestampPattern = /^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2} /;
+
+const originalConsole: Record<ConsoleMethodName, Console['log']> = {
+  log: console.log
+};
+
+const outputs: string[] = [];
+
+const stripAnsi = (value: string): string => value.replace(ansiPattern, '');
+
+const capture = (method: ConsoleMethodName, sink: string[]): void => {
+  const writer = (...data: unknown[]): void => {
+    sink.push(data.map(item => String(item)).join(' '));
+  };
+  console[method] = writer as Console['log'];
+};
+
+const hasLine = (lines: string[], message: string): boolean => {
+  return lines.some(line => {
+    if (!timestampPattern.test(line)) return false;
+    const withoutTimestamp = line.replace(timestampPattern, '');
+    return withoutTimestamp === `info  ${message}`;
+  });
+};
+
+beforeEach(() => {
+  outputs.length = 0;
+  capture('log', outputs);
+});
+
+afterEach(() => {
+  console.log = originalConsole.log;
+});
+
+test('runCommand 流式输出会写入日志', async () => {
+  const logger = new Logger({ verbose: false });
+  const script = "process.stdout.write('alpha\\nbeta\\n'); process.stderr.write('gamma\\n');";
+
+  await runCommand(process.execPath, ['-e', script], {
+    logger,
+    stream: {
+      enabled: true,
+      stdoutPrefix: '[AI] ',
+      stderrPrefix: '[AI-err] '
+    }
+  });
+
+  const cleaned = outputs.map(stripAnsi);
+  assert.ok(hasLine(cleaned, '[AI] alpha'));
+  assert.ok(hasLine(cleaned, '[AI] beta'));
+  assert.ok(hasLine(cleaned, '[AI-err] gamma'));
+});

package/tests/summary.test.ts

@@ -0,0 +1,58 @@
+import assert from 'node:assert/strict';
+import { test } from 'node:test';
+import { buildFallbackSummary, ensurePrBodySections, parseDeliverySummary } from '../src/summary';
+import type { TestRunResult } from '../src/types';
+
+test('parseDeliverySummary 解析 JSON 输出', () => {
+  const output = JSON.stringify({
+    commitTitle: 'chore: 更新 PR 文案生成',
+    commitBody: '- 优化 PR 标题\n- 补充提交摘要',
+    prTitle: 'chore: 更新 PR 文案生成',
+    prBody: '# 变更摘要\n- 优化 PR 标题\n\n# 测试结果\n- 未运行\n\n# 风险与回滚\n- 风险：低\n- 回滚：git revert 对应提交'
+  });
+
+  const parsed = parseDeliverySummary(output);
+
+  assert.ok(parsed);
+  assert.equal(parsed?.commitTitle, 'chore: 更新 PR 文案生成');
+  assert.equal(parsed?.prTitle, 'chore: 更新 PR 文案生成');
+});
+
+test('parseDeliverySummary 支持 code fence 输出', () => {
+  const output = [
+    '```json',
+    '{"commitTitle":"chore: 文案优化","prTitle":"chore: 文案优化","prBody":"# 变更摘要\\n- 更新\\n\\n# 测试结果\\n- 未运行\\n\\n# 风险与回滚\\n- 风险：待评估","commitBody":"- 更新"}',
+    '```'
+  ].join('\n');
+
+  const parsed = parseDeliverySummary(output);
+
+  assert.ok(parsed);
+  assert.equal(parsed?.commitBody, '- 更新');
+});
+
+test('ensurePrBodySections 在缺少标题时使用兜底模板', () => {
+  const testResults: TestRunResult[] = [];
+  const fallback = {
+    commitTitle: 'chore: 补充提交摘要',
+    commitBody: '- 调整 PR 文案输出',
+    testResults
+  };
+
+  const prBody = ensurePrBodySections('简短描述', fallback);
+
+  assert.ok(prBody.includes('# 变更摘要'));
+  assert.ok(prBody.includes('# 测试结果'));
+  assert.ok(prBody.includes('# 风险与回滚'));
+  assert.ok(prBody.includes('- 调整 PR 文案输出'));
+});
+
+test('buildFallbackSummary 使用任务生成标题', () => {
+  const summary = buildFallbackSummary({
+    task: '优化 PR 标题与提交信息生成',
+    testResults: null
+  });
+
+  assert.ok(summary.prTitle.includes('优化 PR 标题与提交信息生成'));
+  assert.ok(summary.prBody.includes('# 变更摘要'));
+});

package/tests/token-usage.test.ts

@@ -0,0 +1,33 @@
+import assert from 'node:assert/strict';
+import { test } from 'node:test';
+import { mergeTokenUsage, parseTokenUsage } from '../src/ai';
+
+test('parseTokenUsage 可解析 total_tokens 行', () => {
+  const logs = 'model: demo\nTotal_tokens: 321\n';
+  const usage = parseTokenUsage(logs);
+
+  assert.ok(usage);
+  assert.equal(usage?.totalTokens, 321);
+});
+
+test('parseTokenUsage 可累加 prompt/output tokens', () => {
+  const logs = 'prompt_tokens: 120\ncompletion_tokens: 30\n';
+  const usage = parseTokenUsage(logs);
+
+  assert.ok(usage);
+  assert.equal(usage?.totalTokens, 150);
+  assert.equal(usage?.inputTokens, 120);
+  assert.equal(usage?.outputTokens, 30);
+});
+
+test('mergeTokenUsage 会累计各轮数据', () => {
+  const merged = mergeTokenUsage(
+    { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
+    { inputTokens: 5, outputTokens: 10, totalTokens: 15 }
+  );
+
+  assert.ok(merged);
+  assert.equal(merged?.totalTokens, 45);
+  assert.equal(merged?.inputTokens, 15);
+  assert.equal(merged?.outputTokens, 30);
+});

package/tests/utils.test.ts

@@ -0,0 +1,8 @@
+import assert from 'node:assert/strict';
+import { test } from 'node:test';
+import { pad2 } from '../src/utils';
+
+test('pad2 会补齐两位数字', () => {
+  assert.equal(pad2(1), '01');
+  assert.equal(pad2(12), '12');
+});

package/tests/webhook.test.ts

@@ -0,0 +1,89 @@
+import assert from 'node:assert/strict';
+import { test } from 'node:test';
+import { Logger } from '../src/logger';
+import { buildWebhookPayload, normalizeWebhookUrls, sendWebhookNotifications, type FetchLike } from '../src/webhook';
+
+test('normalizeWebhookUrls 会去除空值并裁剪空白', () => {
+  const urls = normalizeWebhookUrls(['  https://example.com  ', '', '   ', 'http://a.local']);
+  assert.deepEqual(urls, ['https://example.com', 'http://a.local']);
+});
+
+test('buildWebhookPayload 会补齐时间戳', () => {
+  const payload = buildWebhookPayload({
+    event: 'task_start',
+    task: 'demo',
+    iteration: 0,
+    stage: '任务开始'
+  });
+  assert.equal(payload.event, 'task_start');
+  assert.equal(payload.task, 'demo');
+  assert.equal(payload.iteration, 0);
+  assert.ok(payload.timestamp.length > 0);
+});
+
+test('sendWebhookNotifications 在无配置时直接跳过', async () => {
+  const logger = new Logger({ verbose: false });
+  let called = false;
+  const fetcher: FetchLike = async () => {
+    called = true;
+    return { ok: true, status: 200, statusText: 'OK' };
+  };
+
+  const payload = buildWebhookPayload({
+    event: 'task_start',
+    task: 'demo',
+    iteration: 0,
+    stage: '任务开始',
+    timestamp: '2024-01-01T00:00:00.000Z'
+  });
+
+  await sendWebhookNotifications(null, payload, logger, fetcher);
+  assert.equal(called, false);
+});
+
+test('sendWebhookNotifications 会向多个地址发送', async () => {
+  const logger = new Logger({ verbose: false });
+  const calls: Array<{ url: string; body: string; contentType: string }> = [];
+  const fetcher: FetchLike = async (url, init) => {
+    calls.push({
+      url,
+      body: init.body,
+      contentType: init.headers['content-type']
+    });
+    return { ok: true, status: 204, statusText: 'No Content' };
+  };
+
+  const payload = buildWebhookPayload({
+    event: 'iteration_start',
+    task: 'demo',
+    branch: 'feat/demo',
+    iteration: 2,
+    stage: '开始第 2 轮迭代',
+    timestamp: '2024-01-01T00:00:00.000Z'
+  });
+
+  await sendWebhookNotifications({ urls: ['https://a.test', 'https://b.test'], timeoutMs: 1000 }, payload, logger, fetcher);
+
+  assert.equal(calls.length, 2);
+  assert.equal(calls[0].contentType, 'application/json');
+  const decoded = JSON.parse(calls[0].body) as typeof payload;
+  assert.equal(decoded.task, 'demo');
+  assert.equal(decoded.branch, 'feat/demo');
+});
+
+test('sendWebhookNotifications 捕获异常并不中断', async () => {
+  const logger = new Logger({ verbose: false });
+  const payload = buildWebhookPayload({
+    event: 'task_end',
+    task: 'demo',
+    iteration: 1,
+    stage: '任务结束',
+    timestamp: '2024-01-01T00:00:00.000Z'
+  });
+
+  await assert.doesNotReject(async () => {
+    await sendWebhookNotifications({ urls: ['https://fail.test'] }, payload, logger, async () => {
+      throw new Error('boom');
+    });
+  });
+});

package/tsconfig.json

@@ -0,0 +1,18 @@
+{
+  "compilerOptions": {
+    "target": "ES2020",
+    "module": "CommonJS",
+    "moduleResolution": "node",
+    "outDir": "dist",
+    "rootDir": "src",
+    "esModuleInterop": true,
+    "forceConsistentCasingInFileNames": true,
+    "strict": true,
+    "skipLibCheck": true,
+    "resolveJsonModule": true,
+    "noImplicitOverride": true,
+    "noUnusedLocals": false,
+    "noUnusedParameters": false
+  },
+  "include": ["src/**/*"]
+}

package/tsup.config.ts

@@ -0,0 +1,18 @@
+import { defineConfig } from 'tsup';
+
+export default defineConfig({
+  entry: {
+    cli: 'src/cli.ts',
+    index: 'src/index.ts'
+  },
+  format: ['cjs'],
+  target: 'node18',
+  dts: true,
+  sourcemap: true,
+  clean: true,
+  minify: false,
+  shims: false,
+  banner: {
+    js: '#!/usr/bin/env node'
+  }
+});