workplace-pua-cli 0.4.0

package/src/utils/formatter.ts

@@ -0,0 +1,104 @@
+import chalk from 'chalk';
+
+export type OutputFormat = 'text' | 'markdown' | 'json';
+
+export interface FormattedOutput {
+  format: OutputFormat;
+  content: string;
+  metadata?: {
+    role?: string;
+    severity?: string;
+    provider?: string;
+    model?: string;
+    tokens?: number;
+    timestamp?: string;
+  };
+}
+
+export class OutputFormatter {
+  private outputFormat: OutputFormat;
+
+  constructor(format: OutputFormat = 'text') {
+    this.outputFormat = format;
+  }
+
+  format(data: FormattedOutput): string {
+    switch (this.outputFormat) {
+      case 'json':
+        return this.formatJson(data);
+      case 'markdown':
+        return this.formatMarkdown(data);
+      case 'text':
+      default:
+        return this.formatText(data);
+    }
+  }
+
+  private formatJson(data: FormattedOutput): string {
+    return JSON.stringify({
+      output: data.content,
+      metadata: data.metadata || {},
+      timestamp: new Date().toISOString()
+    }, null, 2);
+  }
+
+  private formatMarkdown(data: FormattedOutput): string {
+    let result = '';
+
+    if (data.metadata?.role) {
+      result += `## ${data.metadata.role}\n\n`;
+    }
+
+    result += data.content;
+    result += '\n';
+
+    if (data.metadata) {
+      result += '\n---\n\n';
+      result += '### Metadata\n\n';
+      result += '| Key | Value |\n';
+      result += '|-----|-------|\n';
+
+      for (const [key, value] of Object.entries(data.metadata)) {
+        result += `| ${key} | ${value} |\n`;
+      }
+    }
+
+    return result;
+  }
+
+  private formatText(data: FormattedOutput): string {
+    return data.content;
+  }
+
+  print(data: FormattedOutput): void {
+    const formatted = this.format(data);
+
+    switch (this.outputFormat) {
+      case 'json':
+        console.log(formatted);
+        break;
+      case 'markdown':
+        console.log(formatted);
+        break;
+      case 'text':
+      default:
+        // 对于文本格式，处理颜色
+        if (data.metadata?.role === 'boss') {
+          console.log(chalk.red(formatted));
+        } else if (data.metadata?.role === 'employee') {
+          console.log(chalk.yellow(formatted));
+        } else {
+          console.log(formatted);
+        }
+        break;
+    }
+  }
+
+  setFormat(format: OutputFormat): void {
+    this.outputFormat = format;
+  }
+
+  getFormat(): OutputFormat {
+    return this.outputFormat;
+  }
+}

package/src/utils/logger.ts

@@ -0,0 +1,31 @@
+import chalk from 'chalk';
+
+export class Logger {
+  info(message: string): void {
+    console.log(chalk.blue('ℹ'), message);
+  }
+
+  success(message: string): void {
+    console.log(chalk.green('✓'), message);
+  }
+
+  warning(message: string): void {
+    console.log(chalk.yellow('⚠'), message);
+  }
+
+  error(message: string): void {
+    console.error(chalk.red('✗'), message);
+  }
+
+  debug(message: string): void {
+    if (process.env.DEBUG) {
+      console.log(chalk.gray('…'), message);
+    }
+  }
+
+  system(message: string): void {
+    console.log(chalk.cyan('►'), message);
+  }
+}
+
+export const logger = new Logger();

package/src/utils/stream.ts

@@ -0,0 +1,76 @@
+import chalk from 'chalk';
+import { StreamChunk } from '../llm/base';
+
+export class StreamPrinter {
+  private buffer: string = '';
+  private currentLine: string = '';
+
+  constructor(private readonly roleColor: (text: string) => string = chalk.green) {}
+
+  /**
+   * Print a streaming chunk to the terminal
+   */
+  printChunk(chunk: StreamChunk): void {
+    if (chunk.content) {
+      // Accumulate content for smoother display
+      this.buffer += chunk.content;
+
+      // Print directly for real-time effect
+      process.stdout.write(chunk.content);
+    }
+
+    if (chunk.done) {
+      // Add newline when done
+      process.stdout.write('\n');
+    }
+  }
+
+  /**
+   * Print a complete message (non-streaming)
+   */
+  printMessage(message: string): void {
+    console.log(this.roleColor(message));
+  }
+
+  /**
+   * Print user input
+   */
+  printUserInput(input: string): void {
+    console.log(chalk.gray('┌─────────────────────────────────────'));
+    console.log(chalk.gray('│ 你:'), input);
+    console.log(chalk.gray('└─────────────────────────────────────'));
+  }
+
+  /**
+   * Print assistant response header
+   */
+  printResponseHeader(role: string): void {
+    const roleLabel = role === 'boss' ? '老板' : '员工';
+    const color = role === 'boss' ? chalk.red.bold : chalk.yellow.bold;
+    console.log();
+    console.log(color(`┌─ ${roleLabel} ─────────────────────────────`));
+  }
+
+  /**
+   * Print assistant response footer
+   */
+  printResponseFooter(): void {
+    console.log(chalk.gray('└─────────────────────────────────────'));
+    console.log();
+  }
+
+  /**
+   * Print error message
+   */
+  printError(message: string): void {
+    console.error(chalk.red('✗ 错误:'), message);
+  }
+
+  /**
+   * Clear current line
+   */
+  clearLine(): void {
+    process.stdout.clearLine(0);
+    process.stdout.cursorTo(0);
+  }
+}

package/tsconfig.json

@@ -0,0 +1,20 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "CommonJS",
+    "moduleResolution": "node",
+    "lib": ["ES2022"],
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "strict": false,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "resolveJsonModule": true,
+    "declaration": false,
+    "sourceMap": false,
+    "types": ["node"]
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist", "**/*.test.ts"]
+}

package/vitest.config.ts

@@ -0,0 +1,18 @@
+import { defineConfig } from 'vitest/config';
+
+export default defineConfig({
+  test: {
+    globals: {
+      describe: 'readonly',
+      it: 'readonly',
+      expect: 'readonly'
+    },
+    environment: 'node',
+    include: ['src/**/*.test.ts', 'tests/**/*.test.ts'],
+    coverage: {
+      provider: 'v8',
+      reporter: ['text', 'json', 'html'],
+      exclude: ['**/*.test.ts', '**/dist/**']
+    }
+  }
+});