hamster-wheel-cli 0.1.0

package/src/global-config.ts

@@ -0,0 +1,296 @@
+import os from 'node:os';
+import path from 'node:path';
+import fs from 'fs-extra';
+import type { Logger } from './logger';
+
+/**
+ * 全局快捷指令配置。
+ */
+export interface ShortcutConfig {
+  readonly name: string;
+  readonly command: string;
+}
+
+/**
+ * 全局配置结构。
+ */
+export interface GlobalConfig {
+  readonly shortcut?: ShortcutConfig;
+}
+
+/**
+ * 获取全局配置文件路径。
+ */
+export function getGlobalConfigPath(): string {
+  return path.join(os.homedir(), '.wheel-ai', 'config.toml');
+}
+
+function stripTomlComment(line: string): string {
+  let quote: '"' | '\'' | null = null;
+  let escaped = false;
+  for (let i = 0; i < line.length; i += 1) {
+    const char = line[i];
+    if (escaped) {
+      escaped = false;
+      continue;
+    }
+    if (quote) {
+      if (quote === '"' && char === '\\') {
+        escaped = true;
+        continue;
+      }
+      if (char === quote) {
+        quote = null;
+      }
+      continue;
+    }
+    if (char === '"' || char === '\'') {
+      quote = char;
+      continue;
+    }
+    if (char === '#' || char === ';') {
+      return line.slice(0, i);
+    }
+  }
+  return line;
+}
+
+function findUnquotedIndex(text: string, target: string): number {
+  let quote: '"' | '\'' | null = null;
+  let escaped = false;
+  for (let i = 0; i < text.length; i += 1) {
+    const char = text[i];
+    if (escaped) {
+      escaped = false;
+      continue;
+    }
+    if (quote) {
+      if (quote === '"' && char === '\\') {
+        escaped = true;
+        continue;
+      }
+      if (char === quote) {
+        quote = null;
+      }
+      continue;
+    }
+    if (char === '"' || char === '\'') {
+      quote = char;
+      continue;
+    }
+    if (char === target) {
+      return i;
+    }
+  }
+  return -1;
+}
+
+function parseTomlString(raw: string): string | null {
+  const value = raw.trim();
+  if (value.length < 2) return null;
+  const quote = value[0];
+  if (quote !== '"' && quote !== '\'') return null;
+  let result = '';
+  let escaped = false;
+  for (let i = 1; i < value.length; i += 1) {
+    const char = value[i];
+    if (quote === '"') {
+      if (escaped) {
+        switch (char) {
+          case 'n':
+            result += '\n';
+            break;
+          case 't':
+            result += '\t';
+            break;
+          case 'r':
+            result += '\r';
+            break;
+          case '"':
+          case '\\':
+            result += char;
+            break;
+          default:
+            result += char;
+            break;
+        }
+        escaped = false;
+        continue;
+      }
+      if (char === '\\') {
+        escaped = true;
+        continue;
+      }
+      if (char === quote) {
+        const rest = value.slice(i + 1).trim();
+        if (rest.length > 0) return null;
+        return result;
+      }
+      result += char;
+      continue;
+    }
+
+    if (char === quote) {
+      const rest = value.slice(i + 1).trim();
+      if (rest.length > 0) return null;
+      return result;
+    }
+    result += char;
+  }
+  return null;
+}
+
+function normalizeShortcutName(name: string): string | null {
+  const trimmed = name.trim();
+  if (!trimmed) return null;
+  if (/\s/.test(trimmed)) return null;
+  return trimmed;
+}
+
+/**
+ * 解析全局 TOML 配置文本。
+ */
+export function parseGlobalConfig(content: string): GlobalConfig {
+  const lines = content.split(/\r?\n/);
+  let currentSection: string | null = null;
+  const shortcut: Record<string, string> = {};
+
+  for (const rawLine of lines) {
+    const line = stripTomlComment(rawLine).trim();
+    if (!line) continue;
+
+    const sectionMatch = /^\[(.+)\]$/.exec(line);
+    if (sectionMatch) {
+      currentSection = sectionMatch[1].trim();
+      continue;
+    }
+
+    if (currentSection !== 'shortcut') continue;
+
+    const equalIndex = findUnquotedIndex(line, '=');
+    if (equalIndex <= 0) continue;
+
+    const key = line.slice(0, equalIndex).trim();
+    const valuePart = line.slice(equalIndex + 1).trim();
+    if (!key || !valuePart) continue;
+
+    const parsedValue = parseTomlString(valuePart);
+    if (parsedValue === null) continue;
+
+    shortcut[key] = parsedValue;
+  }
+
+  const name = normalizeShortcutName(shortcut.name ?? '');
+  const command = (shortcut.command ?? '').trim();
+  if (!name || !command) {
+    return {};
+  }
+
+  return {
+    shortcut: {
+      name,
+      command
+    }
+  };
+}
+
+/**
+ * 读取用户目录下的全局配置。
+ */
+export async function loadGlobalConfig(logger?: Logger): Promise<GlobalConfig | null> {
+  const filePath = getGlobalConfigPath();
+  const exists = await fs.pathExists(filePath);
+  if (!exists) return null;
+
+  try {
+    const content = await fs.readFile(filePath, 'utf8');
+    return parseGlobalConfig(content);
+  } catch (error) {
+    const message = error instanceof Error ? error.message : String(error);
+    logger?.warn(`读取全局配置失败，已忽略：${message}`);
+    return null;
+  }
+}
+
+/**
+ * 将命令行字符串拆解为参数数组（支持引号与转义）。
+ */
+export function splitCommandArgs(command: string): string[] {
+  const args: string[] = [];
+  let current = '';
+  let quote: '"' | '\'' | null = null;
+  let escaped = false;
+
+  for (let i = 0; i < command.length; i += 1) {
+    const char = command[i];
+    if (escaped) {
+      current += char;
+      escaped = false;
+      continue;
+    }
+
+    if (quote) {
+      if (quote === '"' && char === '\\') {
+        escaped = true;
+        continue;
+      }
+      if (char === quote) {
+        quote = null;
+        continue;
+      }
+      current += char;
+      continue;
+    }
+
+    if (char === '"' || char === '\'') {
+      quote = char;
+      continue;
+    }
+
+    if (/\s/.test(char)) {
+      if (current.length > 0) {
+        args.push(current);
+        current = '';
+      }
+      continue;
+    }
+
+    if (char === '\\') {
+      escaped = true;
+      continue;
+    }
+
+    current += char;
+  }
+
+  if (current.length > 0) {
+    args.push(current);
+  }
+
+  return args;
+}
+
+function normalizeShortcutArgs(args: string[]): string[] {
+  if (args.length > 0 && args[0] === 'run') {
+    return args.slice(1);
+  }
+  return args;
+}
+
+/**
+ * 应用全局快捷指令，将别名替换为 run 子命令参数。
+ */
+export function applyShortcutArgv(argv: string[], config: GlobalConfig | null): string[] {
+  if (!config?.shortcut) return argv;
+  if (argv.length < 3) return argv;
+  const commandIndex = 2;
+  if (argv[commandIndex] !== config.shortcut.name) return argv;
+
+  const shortcutArgs = normalizeShortcutArgs(splitCommandArgs(config.shortcut.command));
+  return [
+    ...argv.slice(0, commandIndex),
+    'run',
+    ...shortcutArgs,
+    ...argv.slice(commandIndex + 1)
+  ];
+}

package/src/index.ts

@@ -0,0 +1,3 @@
+export { runCli } from './cli';
+export { runLoop } from './loop';
+export type { LoopConfig, AiCliConfig, WorktreeConfig, PrConfig, TestConfig, WebhookConfig } from './types';

package/src/logger.ts

@@ -0,0 +1,122 @@
+import fs from 'fs-extra';
+import { pad2 } from './utils';
+
+type Colorizer = (value: string) => string;
+type ConsoleMethodName = 'log' | 'warn' | 'error';
+
+const wrap = (code: string): Colorizer => (value: string) => `\u001b[${code}m${value}\u001b[0m`;
+
+const colors = {
+  blue: wrap('34'),
+  green: wrap('32'),
+  yellow: wrap('33'),
+  red: wrap('31'),
+  magenta: wrap('35'),
+  gray: wrap('90')
+} as const;
+
+/**
+ * 日志器配置项。
+ */
+export interface LoggerOptions {
+  readonly verbose?: boolean;
+  readonly logFile?: string;
+}
+
+/**
+ * 带颜色的日志输出器，可选写入日志文件。
+ */
+export class Logger {
+  private readonly verbose: boolean;
+  private readonly logFile?: string;
+  private logFileEnabled: boolean;
+  private logFileErrored: boolean;
+
+  constructor(options: LoggerOptions = {}) {
+    this.verbose = options.verbose ?? false;
+    const trimmedPath = options.logFile?.trim();
+    this.logFile = trimmedPath && trimmedPath.length > 0 ? trimmedPath : undefined;
+    this.logFileEnabled = Boolean(this.logFile);
+    this.logFileErrored = false;
+
+    if (this.logFile) {
+      try {
+        fs.ensureFileSync(this.logFile);
+      } catch (error) {
+        this.disableFileWithError(error);
+      }
+    }
+  }
+
+  info(message: string): void {
+    this.emit('log', colors.blue, 'info', '  ', message);
+  }
+
+  success(message: string): void {
+    this.emit('log', colors.green, 'ok', '    ', message);
+  }
+
+  warn(message: string): void {
+    this.emit('warn', colors.yellow, 'warn', '  ', message);
+  }
+
+  error(message: string): void {
+    this.emit('error', colors.red, 'err', '   ', message);
+  }
+
+  debug(message: string): void {
+    if (!this.verbose) return;
+    this.emit('log', colors.magenta, 'dbg', '   ', message);
+  }
+
+  private emit(method: ConsoleMethodName, colorizer: Colorizer, label: string, padding: string, message: string): void {
+    const now = new Date();
+    const consoleLine = this.formatConsoleLine(now, colorizer(label), padding, message);
+    const fileLine = this.formatFileLine(now, label, padding, message);
+    console[method](consoleLine);
+    this.writeFileLine(fileLine);
+  }
+
+  private formatConsoleLine(date: Date, label: string, padding: string, message: string): string {
+    const timestamp = this.formatTimestamp(date);
+    return `${colors.gray(timestamp)} ${label}${padding}${message}`;
+  }
+
+  private formatFileLine(date: Date, label: string, padding: string, message: string): string {
+    const timestamp = this.formatTimestamp(date);
+    return `${timestamp} ${label}${padding}${message}`;
+  }
+
+  private writeFileLine(line: string): void {
+    if (!this.logFileEnabled || !this.logFile) return;
+    try {
+      fs.appendFileSync(this.logFile, `${line}\n`, 'utf8');
+    } catch (error) {
+      this.disableFileWithError(error);
+    }
+  }
+
+  private disableFileWithError(error: unknown): void {
+    this.logFileEnabled = false;
+    if (this.logFileErrored) return;
+    this.logFileErrored = true;
+    const message = error instanceof Error ? error.message : String(error);
+    const target = this.logFile ? ` (${this.logFile})` : '';
+    console.warn(`日志文件写入失败${target}，已停止写入：${message}`);
+  }
+
+  private formatTimestamp(date: Date): string {
+    const year = date.getFullYear();
+    const month = pad2(date.getMonth() + 1);
+    const day = pad2(date.getDate());
+    const hours = pad2(date.getHours());
+    const minutes = pad2(date.getMinutes());
+    const seconds = pad2(date.getSeconds());
+    return `${year}-${month}-${day} ${hours}:${minutes}:${seconds}`;
+  }
+}
+
+/**
+ * 默认日志器实例。
+ */
+export const defaultLogger = new Logger();