thatgfsj-code 0.3.0 → 0.4.0

package/src/tui/welcome.ts

@@ -1,6 +1,6 @@
 /**
  * Welcome / Setup Wizard
- * Migrated and simplified from old src/repl/welcome.ts
+ * Supports all providers + custom relay stations (中转站)
  */
 
 import chalk from 'chalk';
@@ -8,7 +8,7 @@ import readline from 'readline';
 import { existsSync, mkdirSync, writeFileSync } from 'fs';
 import { join } from 'path';
 import { homedir } from 'os';
-import { PROVIDERS, getModelsForProvider, listProviders } from '../config/providers.js';
+import { PROVIDERS, getModelsForProvider, listProviders, isCustomProvider } from '../config/providers.js';
 import type { ProviderName } from '../config/types.js';
 
 export class WelcomeScreen {
@@ -23,7 +23,7 @@ export class WelcomeScreen {
     console.log(chalk.cyan('|') + chalk.yellow(' 快速开始指南').padEnd(w) + chalk.cyan('|'));
     console.log(chalk.cyan('|') + ' '.repeat(w) + chalk.cyan('|'));
     console.log(chalk.cyan('|') + chalk.gray(' 运行 gfcode init 配置你的 API Key').padEnd(w) + chalk.cyan('|'));
-    console.log(chalk.cyan('|') + chalk.gray(' 或设置环境变量: OPENAI_API_KEY, SILICONFLOW_API_KEY 等').padEnd(w) + chalk.cyan('|'));
+    console.log(chalk.cyan('|') + chalk.gray(' 支持: OpenAI / Claude / DeepSeek / Kimi / 智谱 / 中转站').padEnd(w) + chalk.cyan('|'));
     console.log(chalk.cyan('+') + '-'.repeat(w) + '+');
     console.log();
   }
@@ -48,7 +48,8 @@ export class WelcomeScreen {
       console.log(chalk.white('  步骤 1/3: 选择 AI 提供商\n'));
       const providers = listProviders();
       providers.forEach((p, i) => {
-        console.log(chalk.green(`  ${i + 1}. ${p.name}`));
+        const num = (i + 1).toString().padStart(2);
+        console.log(chalk.green(`  ${num}. ${p.name}`));
       });
       console.log();
 
@@ -56,46 +57,99 @@ export class WelcomeScreen {
       const selected = providers[parseInt(choice) - 1] || providers[0];
       const providerName = selected.key;
 
-      // Step 2: Enter API Key
-      console.log(chalk.white('\n  步骤 2/3: 输入 API Key\n'));
-      console.log(chalk.gray(`  获取地址: ${PROVIDERS[providerName].baseUrl.replace('/v1', '').replace('/v1beta', '')}`));
+      // Step 2: Enter API Key (and base URL for custom providers)
+      console.log(chalk.white('\n  步骤 2/3: 配置连接信息\n'));
+
+      let baseUrl: string | undefined;
+
+      if (isCustomProvider(providerName)) {
+        // Custom provider needs base URL
+        console.log(chalk.yellow('  自定义 Provider (中转站) 模式'));
+        console.log(chalk.gray('  支持 OpenAI 兼容 或 Anthropic 兼容 的中转站'));
+        console.log();
+        baseUrl = await question(chalk.green('  请输入中转站 Base URL: '));
+        if (!baseUrl) {
+          console.log(chalk.red('  ❌ 中转站必须提供 Base URL'));
+          rl.close();
+          return;
+        }
+        // Remove trailing slash
+        baseUrl = baseUrl.replace(/\/+$/, '');
+        // Remove /v1 suffix if present (we'll add it in the provider)
+        // Actually keep it as-is since the user knows their endpoint
+      }
+
       const apiKey = await question(chalk.green('  请输入 API Key: '));
 
       // Step 3: Choose model
       console.log(chalk.white('\n  步骤 3/3: 选择模型\n'));
-      const models = getModelsForProvider(providerName);
-      models.forEach((m, i) => {
-        console.log(chalk.green(`  ${i + 1}. ${m.name}`) + chalk.gray(` - ${m.desc}`));
-      });
-      console.log();
-
-      const modelChoice = await question(chalk.green('  请选择模型 (1-' + models.length + '): '));
-      const selectedModel = models[parseInt(modelChoice) - 1] || models[0];
 
-      // Save config
-      const configDir = join(homedir(), '.thatgfsj');
-      const configPath = join(configDir, 'config.json');
-
-      if (!existsSync(configDir)) {
-        mkdirSync(configDir, { recursive: true });
+      if (isCustomProvider(providerName)) {
+        // Custom provider: let user type model name
+        console.log(chalk.gray('  输入中转站支持的模型名称'));
+        console.log(chalk.gray('  例如: gpt-4o-mini, claude-sonnet-4-20250514, deepseek-chat'));
+        const model = await question(chalk.green('  模型名称: '));
+        if (!model) {
+          console.log(chalk.red('  ❌ 必须指定模型'));
+          rl.close();
+          return;
+        }
+
+        // Save config
+        this.saveConfig(providerName, model, apiKey, baseUrl);
+      } else {
+        // Standard provider: show model list
+        const models = getModelsForProvider(providerName);
+        models.forEach((m, i) => {
+          const num = (i + 1).toString().padStart(2);
+          console.log(chalk.green(`  ${num}. ${m.name}`) + chalk.gray(` - ${m.desc}`));
+        });
+        console.log();
+
+        const modelChoice = await question(chalk.green('  请选择模型 (1-' + models.length + '): '));
+        const selectedModel = models[parseInt(modelChoice) - 1] || models[0];
+
+        this.saveConfig(providerName, selectedModel.id, apiKey, baseUrl);
       }
 
-      writeFileSync(configPath, JSON.stringify({
-        provider: providerName,
-        model: selectedModel.id,
-        apiKey,
-        temperature: 0.7,
-        maxTokens: 4096,
-      }, null, 2));
-
       console.log(chalk.green('\n  ✅ 配置已保存!'));
-      console.log(chalk.gray(`  提供商: ${providerName}`));
-      console.log(chalk.gray(`  模型: ${selectedModel.name}`));
-      console.log(chalk.gray(`  配置文件: ${configPath}`));
+      console.log(chalk.gray(`  配置文件: ${join(homedir(), '.thatgfsj', 'config.json')}`));
       console.log(chalk.cyan('\n  运行 gfcode 开始使用\n'));
 
     } finally {
       rl.close();
     }
   }
+
+  /**
+   * Save config to file
+   */
+  private static saveConfig(provider: ProviderName, model: string, apiKey: string, baseUrl?: string): void {
+    const configDir = join(homedir(), '.thatgfsj');
+    const configPath = join(configDir, 'config.json');
+
+    if (!existsSync(configDir)) {
+      mkdirSync(configDir, { recursive: true });
+    }
+
+    const config: Record<string, any> = {
+      provider,
+      model,
+      apiKey,
+      temperature: 0.7,
+      maxTokens: 4096,
+    };
+
+    if (baseUrl) {
+      config.baseUrl = baseUrl;
+    }
+
+    writeFileSync(configPath, JSON.stringify(config, null, 2));
+
+    console.log(chalk.gray(`  提供商: ${provider}`));
+    console.log(chalk.gray(`  模型: ${model}`));
+    if (baseUrl) {
+      console.log(chalk.gray(`  Base URL: ${baseUrl}`));
+    }
+  }
 }