workplace-pua-cli 0.4.0

package/src/__tests__/config/settings.test.ts

@@ -0,0 +1,48 @@
+import { describe, it, expect, beforeEach } from 'vitest';
+import { loadConfig, DEFAULTS, type ProviderType } from '../config/settings';
+
+describe('config/settings', () => {
+  beforeEach(() => {
+    // 清除环境变量
+    delete process.env.ZHIPUAI_API_KEY;
+    delete process.env.OPENAI_API_KEY;
+  });
+
+  describe('loadConfig', () => {
+    it('should return defaults when no config exists', () => {
+      const config = loadConfig({});
+
+      expect(config.provider).toBe(DEFAULTS.provider);
+      expect(config.model).toBe(DEFAULTS.model);
+      expect(config.role).toBe(DEFAULTS.role);
+    });
+
+    it('should prioritize CLI options over env vars', () => {
+      process.env.ZHIPUAI_API_KEY = 'env-key';
+
+      const config = loadConfig({
+        apiKey: 'cli-key',
+        provider: 'zhipu' as ProviderType
+      } as any);
+
+      expect(config.apiKey).toBe('cli-key');
+    });
+
+    it('should load from environment variables', () => {
+      process.env.ZHIPUAI_API_KEY = 'test-key';
+
+      const config = loadConfig({});
+
+      expect(config.apiKey).toBe('test-key');
+    });
+  });
+
+  describe('DEFAULTS', () => {
+    it('should have correct default values', () => {
+      expect(DEFAULTS.provider).toBe('zhipu');
+      expect(DEFAULTS.model).toBe('glm-4.7');
+      expect(DEFAULTS.role).toBe('boss');
+      expect(DEFAULTS.severity).toBe('medium');
+    });
+  });
+});

package/src/__tests__/prompts/boss.test.ts

@@ -0,0 +1,35 @@
+import { describe, it, expect } from 'vitest';
+import { getBossSystemMessage } from '../prompts/boss';
+
+describe('prompts/boss', () => {
+  describe('getBossSystemMessage', () => {
+    it('should return mild template', () => {
+      const prompt = getBossSystemMessage('mild');
+
+      expect(prompt).toContain('喜欢说教但不严厉');
+      expect(prompt).toContain('为你好');
+    });
+
+    it('should return medium template', () => {
+      const prompt = getBossSystemMessage('medium');
+
+      expect(prompt).toContain('典型职场 PUA 老板');
+      expect(prompt).toContain('要对齐');
+      expect(prompt).toContain('闭环');
+    });
+
+    it('should return extreme template', () => {
+      const prompt = getBossSystemMessage('extreme');
+
+      expect(prompt).toContain('极度挑剔');
+      expect(prompt).toContain('刻薄');
+      expect(prompt).toContain('巨婴');
+    });
+
+    it('should contain role-specific keywords', () => {
+      const prompt = getBossSystemMessage('medium');
+
+      expect(prompt).toMatch(/年轻人|锻炼|学习|提升/);
+    });
+  });
+});

package/src/commands/chat.ts

@@ -0,0 +1,328 @@
+import readline from 'readline';
+import chalk from 'chalk';
+import ora from 'ora';
+import type { Ora } from 'ora';
+import { Message } from '../llm/base';
+import { getBossSystemMessage, getEmployeeSystemMessage } from '../prompts';
+import { createLLM } from '../llm/factory';
+import { getProviderBaseUrl } from '../config/settings';
+import { StreamPrinter } from '../utils/stream';
+import { sessionManager } from '../history/session';
+import { SessionStorage, type SessionData } from '../config/session-storage';
+import { logger } from '../utils/logger';
+import { type ProviderType } from '../config/providers';
+
+export interface ChatOptions {
+  apiKey: string;
+  provider: ProviderType;
+  model: string;
+  role: 'boss' | 'employee';
+  severity: 'mild' | 'medium' | 'extreme';
+}
+
+// 全局会话存储实例
+const sessionStorage = new SessionStorage();
+
+// 扩展 readline 接口以支持动态属性
+declare module 'readline' {
+  interface Interface {
+    [key: string]: any;
+  }
+}
+
+export async function chatCommand(options: ChatOptions): Promise<void> {
+  // Create session
+  const sessionId = `session-${Date.now()}`;
+  sessionManager.createSession(sessionId);
+
+  // Create readline interface first
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+    prompt: chalk.green('❯ ')
+  });
+
+  // 保存元数据到 rl 以便在命令处理中使用
+  rl['role'] = options.role;
+  rl['severity'] = options.severity;
+  rl['provider'] = options.provider;
+  rl['model'] = options.model;
+
+  // Set up system message
+  const systemMessage =
+    options.role === 'boss'
+      ? getBossSystemMessage(options.severity)
+      : getEmployeeSystemMessage(options.severity);
+
+  sessionManager.addMessage({ role: 'system', content: systemMessage });
+
+  // Create LLM instance
+  const llm = createLLM(options.provider, {
+    apiKey: options.apiKey,
+    model: options.model,
+    baseUrl: getProviderBaseUrl(options.provider),
+  });
+
+  const printer = new StreamPrinter(
+    options.role === 'boss' ? chalk.red : chalk.yellow
+  );
+
+  // Print welcome message
+  const roleLabel = options.role === 'boss' ? '老板模式' : '员工模式';
+  const roleEmoji = options.role === 'boss' ? '👔' : '👤';
+  const severityLabel = {
+    mild: '温和',
+    medium: '标准',
+    extreme: '极端'
+  }[options.severity];
+
+  console.log();
+  console.log(chalk.cyan('╔═══════════════════════════════════════════════════════════╗'));
+  console.log(chalk.cyan('║') + chalk.bold.white(`            ${roleEmoji} PUA CLI - ${roleLabel}                 `) + chalk.cyan('║'));
+  console.log(chalk.cyan('║') + `   Provider: ${chalk.gray(options.provider)}   强度: ${chalk.gray(severityLabel)}   ` + chalk.cyan('║'));
+  console.log(chalk.cyan('║') + `   模型: ${chalk.gray(options.model)}                                     ` + chalk.cyan('║'));
+  console.log(chalk.cyan('╚═══════════════════════════════════════════════════════════╝'));
+  console.log();
+  console.log(chalk.gray('输入 /help 查看可用命令，输入 /exit 退出'));
+  console.log();
+
+  rl.prompt();
+
+  rl.on('line', async (input) => {
+    const trimmedInput = input.trim();
+
+    // Handle commands
+    if (trimmedInput.startsWith('/')) {
+      await handleCommand(trimmedInput, rl, printer);
+      rl.prompt();
+      return;
+    }
+
+    if (!trimmedInput) {
+      rl.prompt();
+      return;
+    }
+
+    // Add user message to session
+    sessionManager.addMessage({ role: 'user', content: trimmedInput });
+
+    // Print user input
+    printer.printUserInput(trimmedInput);
+
+    // Call LLM
+    let spinner: any = null;
+    let hasStarted = false;
+
+    try {
+      printer.printResponseHeader(options.role);
+
+      const messages = sessionManager.getMessages();
+
+      await llm.chatStream(messages, (chunk) => {
+        if (!hasStarted) {
+          hasStarted = true;
+          if (spinner) {
+            spinner.stop();
+            spinner = null;
+          }
+        }
+        printer.printChunk(chunk);
+      });
+
+      printer.printResponseFooter();
+
+    } catch (error) {
+      if (spinner) spinner.stop();
+      printer.printError(error instanceof Error ? error.message : String(error));
+    }
+
+    rl.prompt();
+  });
+
+  rl.on('close', () => {
+    console.log();
+    logger.info('再见！');
+    process.exit(0);
+  });
+}
+
+async function handleCommand(
+  command: string,
+  rl: readline.Interface,
+  printer: StreamPrinter
+): Promise<void> {
+  const [cmd, ...args] = command.split(' ');
+
+  switch (cmd) {
+    case '/help':
+      printHelp();
+      break;
+
+    case '/exit':
+    case '/quit':
+    case '/q':
+      console.log();
+      logger.info('再见！');
+      rl.close();
+      process.exit(0);
+      break;
+
+    case '/clear':
+      sessionManager.clearCurrentSession();
+      logger.success('会话历史已清空');
+      break;
+
+    case '/history':
+      console.log(sessionManager.getFormattedHistory());
+      break;
+
+    case '/info':
+      console.log(sessionManager.getSessionInfo());
+      break;
+
+    case '/save':
+      await handleSaveCommand(args, rl);
+      break;
+
+    case '/sessions':
+      await handleSessionsCommand(rl);
+      break;
+
+    case '/load':
+      await handleLoadCommand(args, rl);
+      break;
+
+    default:
+      logger.warning(`未知命令: ${cmd}`);
+      console.log(chalk.gray('输入 /help 查看可用命令'));
+  }
+}
+
+function printHelp(): void {
+  console.log();
+  console.log(chalk.bold('可用命令:'));
+  console.log(chalk.gray('─').repeat(50));
+  console.log('  /help          显示此帮助信息');
+  console.log('  /clear         清空当前会话历史');
+  console.log('  /history       显示会话历史记录');
+  console.log('  /info          显示会话统计信息');
+  console.log('  /save [名称]  保存当前会话');
+  console.log('  /sessions      列出所有已保存会话');
+  console.log('  /load <ID>     加载指定会话');
+  console.log('  /exit          退出程序');
+  console.log();
+}
+
+async function handleSaveCommand(args: string[], rl: readline.Interface): Promise<void> {
+  const sessionName = args.join(' ') || '未命名会话';
+  const currentMessages = sessionManager.getMessages();
+
+  const spinner = ora('保存会话中...').start();
+  try {
+    const savedSession = sessionStorage.saveSession({
+      name: sessionName,
+      description: `包含 ${currentMessages.length} 条消息`,
+      messages: currentMessages,
+      metadata: {
+        role: rl['role'] || 'boss',
+        severity: rl['severity'] || 'medium',
+        provider: rl['provider'] || 'zhipu',
+        model: rl['model'] || 'glm-4.7'
+      }
+    });
+
+    spinner.stop();
+    logger.success(`会话已保存: ${savedSession.name} (ID: ${savedSession.id})`);
+  } catch (error) {
+    spinner.stop();
+    logger.error(`保存失败: ${error instanceof Error ? error.message : String(error)}`);
+  }
+
+  rl.prompt();
+}
+
+async function handleSessionsCommand(rl: readline.Interface): Promise<void> {
+  const spinner = ora('加载会话列表...').start();
+  try {
+    const sessions = sessionStorage.listSessions();
+    spinner.stop();
+
+    if (sessions.length === 0) {
+      logger.info('暂无已保存的会话');
+      rl.prompt();
+      return;
+    }
+
+    console.log();
+    console.log(chalk.bold('已保存的会话:'));
+    console.log(chalk.gray('─').repeat(60));
+
+    for (const session of sessions) {
+      const messageCount = session.messages?.length || 0;
+      const timeAgo = getTimeAgo(session.updatedAt);
+
+      console.log(`  ${chalk.cyan(session.id.padEnd(12))}  ${chalk.white(session.name.padEnd(20))}  ${chalk.gray(`(${messageCount} 条消息, ${timeAgo})`)}`);
+    }
+
+    console.log();
+    logger.info('使用 /load <ID> 加载会话');
+    rl.prompt();
+  } catch (error) {
+    spinner.stop();
+    logger.error(`加载失败: ${error instanceof Error ? error.message : String(error)}`);
+    rl.prompt();
+  }
+}
+
+async function handleLoadCommand(args: string[], rl: readline.Interface): Promise<void> {
+  const sessionId = args[0];
+
+  if (!sessionId) {
+    logger.error('请指定会话 ID');
+    console.log(chalk.gray('使用 /sessions 查看所有会话'));
+    rl.prompt();
+    return;
+  }
+
+  const spinner = ora('加载会话中...').start();
+  try {
+    const session = sessionStorage.loadSession(sessionId);
+
+    if (!session) {
+      spinner.stop();
+      logger.error(`未找到会话: ${sessionId}`);
+      rl.prompt();
+      return;
+    }
+
+    spinner.stop();
+
+    // 加载会话消息
+    sessionManager.clearCurrentSession();
+    for (const msg of session.messages || []) {
+      sessionManager.addMessage({ role: msg.role as any, content: msg.content });
+    }
+
+    logger.success(`已加载会话: ${session.name}`);
+    console.log();
+    console.log(chalk.gray(`会话包含 ${session.messages?.length || 0} 条消息`));
+    console.log();
+
+    rl.prompt();
+  } catch (error) {
+    spinner.stop();
+    logger.error(`加载失败: ${error instanceof Error ? error.message : String(error)}`);
+    rl.prompt();
+  }
+}
+
+function getTimeAgo(timestamp: string): string {
+  const now = Date.now();
+  const past = new Date(timestamp).getTime();
+  const diff = Math.floor((now - past) / 1000);
+
+  if (diff < 60) return `${diff} 秒前`;
+  if (diff < 3600) return `${Math.floor(diff / 60)} 分钟前`;
+  if (diff < 86400) return `${Math.floor(diff / 3600)} 小时前`;
+  return `${Math.floor(diff / 86400)} 天前`;
+}

package/src/commands/config.ts

@@ -0,0 +1,283 @@
+import chalk from 'chalk';
+import input from '@inquirer/input';
+import select from '@inquirer/select';
+import confirm from '@inquirer/confirm';
+import { saveGlobalConfig, loadGlobalConfig, ensureConfigDir, type GlobalConfig } from '../config/storage';
+import { PROVIDERS, getProvider, validateApiKey, validateBaseUrl, type ProviderType } from '../config/providers';
+import { logger } from '../utils/logger';
+
+export interface ConfigWizardOptions {
+  autoMode?: boolean;
+  initialProvider?: ProviderType;
+}
+
+/**
+ * Run the configuration wizard
+ */
+export async function configWizard(options: ConfigWizardOptions = {}): Promise<GlobalConfig> {
+  const { autoMode = false, initialProvider } = options;
+
+  // Show welcome message
+  if (!autoMode) {
+    console.log();
+    console.log(chalk.cyan.bold('╔═══════════════════════════════════════════════════════════╗'));
+    console.log(chalk.cyan.bold('║') + '              ' + chalk.white.bold('PUA CLI 配置向导') + '                    ' + chalk.cyan.bold('║'));
+    console.log(chalk.cyan.bold('╚═══════════════════════════════════════════════════════════╝'));
+    console.log();
+    console.log(chalk.gray('欢迎使用 PUA CLI！让我们先配置一些基本信息。'));
+    console.log();
+  }
+
+  // Load existing config if available
+  let existingConfig = loadGlobalConfig();
+
+  // Step 1: Select Provider
+  const providerAnswer = await select({
+    message: '选择 AI 服务提供商:',
+    choices: [
+      {
+        name: `${chalk.green('1.')} 智谱 AI ${chalk.gray('(国产，推荐)')}`,
+        value: 'zhipu',
+        description: '国产大模型，稳定可靠，响应快速',
+      },
+      {
+        name: `${chalk.blue('2.')} OpenAI ${chalk.gray('(GPT-4o)')}`,
+        value: 'openai',
+        description: '国际通用，支持 GPT-4o、GPT-4o-mini 等模型',
+      },
+    ],
+    default: existingConfig?.currentProvider || initialProvider || 'zhipu',
+  });
+
+  const provider = providerAnswer as ProviderType;
+  const providerDef = getProvider(provider)!;
+
+  console.log();
+  console.log(chalk.gray(`已选择: ${chalk.white.bold(providerDef.name)}`));
+  console.log();
+
+  // Step 2: Input API Key
+  let apiKey = '';
+  let apiKeyValid = false;
+
+  while (!apiKeyValid) {
+    apiKey = await input({
+      message: `请输入 ${providerDef.name} API Key:`,
+      default: existingConfig?.providers[provider]?.apiKey || '',
+      validate: (value: string) => {
+        if (!value || value.trim().length === 0) {
+          return 'API Key 不能为空';
+        }
+        const validation = validateApiKey(provider, value);
+        if (!validation.valid) {
+          return validation.error || 'API Key 无效';
+        }
+        return true;
+      },
+    });
+
+    const validation = validateApiKey(provider, apiKey);
+    apiKeyValid = validation.valid;
+
+    if (!apiKeyValid) {
+      console.log(chalk.red(validation.error || 'API Key 无效，请重新输入'));
+    }
+  }
+
+  console.log();
+  console.log(chalk.green('✓ API Key 已设置'));
+  console.log();
+
+  // Step 3: Input Base URL (optional)
+  const useCustomUrl = await confirm({
+    message: '是否使用自定义 Base URL（用于代理）？',
+    default: false,
+  });
+
+  let baseUrl = providerDef.defaultBaseUrl;
+
+  if (useCustomUrl) {
+    baseUrl = await input({
+      message: '请输入 Base URL:',
+      default: existingConfig?.providers[provider]?.baseUrl || providerDef.defaultBaseUrl,
+      validate: (value: string) => {
+        const validation = validateBaseUrl(value);
+        if (!validation.valid) {
+          return validation.error || 'Base URL 无效';
+        }
+        return true;
+      },
+    });
+  }
+
+  console.log();
+
+  // Step 4: Select Default Model
+  const model = await select({
+    message: '选择默认模型:',
+    choices: providerDef.defaultModels.map((m) => ({
+      name: m,
+      value: m,
+    })),
+    default: existingConfig?.defaults.model || providerDef.defaultModels[0],
+  });
+
+  console.log();
+  console.log(chalk.gray(`已选择: ${chalk.white(model)}`));
+  console.log();
+
+  // Step 5: Select Default Role
+  const role = await select({
+    message: '选择默认角色:',
+    choices: [
+      { name: '老板模式 (PUA 别人)', value: 'boss' },
+      { name: '员工模式 (被 PUA)', value: 'employee' },
+    ],
+    default: existingConfig?.defaults.role || 'boss',
+  });
+
+  console.log();
+  console.log(chalk.gray(`已选择: ${chalk.white(role === 'boss' ? '老板模式' : '员工模式')}`));
+  console.log();
+
+  // Step 6: Select Default Severity
+  const severity = await select({
+    message: '选择 PUA 强度:',
+    choices: [
+      { name: '温和 (mild)', value: 'mild' },
+      { name: '标准 (medium)', value: 'medium' },
+      { name: '极端 (extreme)', value: 'extreme' },
+    ],
+    default: existingConfig?.defaults.severity || 'medium',
+  });
+
+  console.log();
+  console.log(chalk.gray(`已选择: ${chalk.white(severity)}`));
+  console.log();
+
+  // Step 7: Confirm and Save
+  const shouldSave = await confirm({
+    message: '确认保存配置？',
+    default: true,
+  });
+
+  if (!shouldSave) {
+    console.log();
+    logger.warning('配置已取消');
+    throw new Error('配置已取消');
+  }
+
+  // Build config object
+  const newConfig: GlobalConfig = {
+    currentProvider: provider,
+    providers: {
+      ...existingConfig?.providers,
+      [provider]: {
+        apiKey,
+        baseUrl,
+        models: providerDef.defaultModels,
+      },
+    },
+    defaults: {
+      provider,
+      model,
+      role: role as 'boss' | 'employee',
+      severity: severity as 'mild' | 'medium' | 'extreme',
+    },
+    onboardingCompleted: true,
+  };
+
+  // Save config
+  ensureConfigDir();
+  saveGlobalConfig(newConfig);
+
+  console.log();
+  console.log(chalk.green('✓ 配置已保存'));
+  console.log();
+  console.log(chalk.gray('配置文件位置:'), chalk.white(getConfigPath()));
+  console.log();
+
+  if (!autoMode) {
+    console.log(chalk.cyan('─────────────────────────────────────────────────────────────'));
+    console.log();
+    console.log(chalk.white.bold('配置完成！现在你可以开始使用 PUA CLI 了：'));
+    console.log();
+    console.log('  ' + chalk.white('pua chat --role boss') + chalk.gray('           # 启动老板模式'));
+    console.log('  ' + chalk.white('pua prompt --role boss "你好"') + chalk.gray('     # 单次提示'));
+    console.log();
+    console.log(chalk.gray('更多选项请运行: pua --help'));
+    console.log();
+  }
+
+  return newConfig;
+}
+
+/**
+ * Get the config file path for display
+ */
+function getConfigPath(): string {
+  const platform = process.platform;
+
+  if (platform === 'win32') {
+    return '%APPDATA%\\pua-cli\\config.json';
+  } else {
+    return '~/.config/pua-cli/config.json';
+  }
+}
+
+/**
+ * Show current configuration
+ */
+export async function showConfig(): Promise<void> {
+  const config = loadGlobalConfig();
+
+  if (!config) {
+    console.log();
+    logger.warning('未找到配置文件');
+    console.log(chalk.gray('运行 ') + chalk.white('pua config') + chalk.gray(' 来配置 PUA CLI'));
+    console.log();
+    return;
+  }
+
+  console.log();
+  console.log(chalk.cyan.bold('当前配置:'));
+  console.log(chalk.gray('─').repeat(50));
+  console.log();
+
+  const providerDef = getProvider(config.currentProvider);
+  console.log(chalk.bold('当前 Provider:'), chalk.white(providerDef?.name || config.currentProvider));
+  console.log();
+
+  console.log(chalk.bold('已配置的 Providers:'));
+  for (const [id, providerConfig] of Object.entries(config.providers)) {
+    const p = getProvider(id as any);
+    const isCurrent = id === config.currentProvider;
+    const prefix = isCurrent ? chalk.green('→') : ' ';
+
+    console.log(`  ${prefix} ${chalk.white(p?.name || id)}`);
+    console.log(`    API Key: ${chalk.gray(maskApiKey(providerConfig.apiKey))}`);
+    if (providerConfig.baseUrl) {
+      console.log(`    Base URL: ${chalk.gray(providerConfig.baseUrl)}`);
+    }
+    console.log();
+  }
+
+  console.log(chalk.bold('默认设置:'));
+  console.log(`  模型: ${chalk.white(config.defaults.model)}`);
+  console.log(`  角色: ${chalk.white(config.defaults.role)}`);
+  console.log(`  强度: ${chalk.white(config.defaults.severity)}`);
+  console.log();
+
+  console.log(chalk.gray('配置文件:'), chalk.white(getConfigPath()));
+  console.log();
+}
+
+/**
+ * Mask API key for display
+ */
+function maskApiKey(apiKey: string): string {
+  if (apiKey.length <= 8) {
+    return '*'.repeat(apiKey.length);
+  }
+  return apiKey.slice(0, 4) + '*'.repeat(apiKey.length - 8) + apiKey.slice(-4);
+}