thatgfsj-code 0.3.0 → 0.4.1

package/src/tui/welcome.ts

@@ -1,6 +1,5 @@
 /**
- * Welcome / Setup Wizard
- * Migrated and simplified from old src/repl/welcome.ts
+ * Welcome / Setup Wizard - Clean UI
  */
 
 import chalk from 'chalk';
@@ -8,94 +7,152 @@ 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';
 
+const line = chalk.gray('─'.repeat(52));
+
 export class WelcomeScreen {
-  /**
-   * Show welcome if no API key is configured
-   */
+
   static show(hasApiKey: boolean): void {
     if (hasApiKey) return;
 
-    const w = 62;
-    console.log(chalk.cyan('+') + chalk.white.bold(' Thatgfsj Code ') + chalk.cyan('-'.repeat(w - 16)) + '+');
-    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('+') + '-'.repeat(w) + '+');
+    console.log();
+    console.log(chalk.cyan.bold('  ⚡ Thatgfsj Code') + chalk.gray(' v0.4.1'));
+    console.log(chalk.gray('  AI Coding Assistant'));
+    console.log(line);
+    console.log();
+    console.log(chalk.yellow('  ⚠  No API key configured'));
+    console.log();
+    console.log(chalk.gray('  Run ') + chalk.cyan.bold('gfcode init') + chalk.gray(' to set up your provider.'));
+    console.log();
+    console.log(chalk.gray('  Supported providers:'));
+    console.log(chalk.gray('  OpenAI · Claude · DeepSeek · Kimi · GLM · Gemini · 中转站'));
+    console.log(line);
     console.log();
   }
 
-  /**
-   * Interactive setup wizard
-   */
   static async interactiveSetup(): Promise<void> {
     console.clear();
-    console.log(chalk.cyan('\n  🤖 Thatgfsj Code 配置向导\n'));
+    console.log();
+    console.log(chalk.cyan.bold('  ⚡ Thatgfsj Code') + chalk.gray(' - Setup'));
+    console.log(line);
+    console.log();
 
     const rl = readline.createInterface({
       input: process.stdin,
       output: process.stdout,
     });
 
-    const question = (prompt: string): Promise<string> =>
+    const ask = (prompt: string): Promise<string> =>
       new Promise(resolve => rl.question(prompt, answer => resolve(answer.trim())));
 
     try {
-      // Step 1: Choose provider
-      console.log(chalk.white('  步骤 1/3: 选择 AI 提供商\n'));
+      // ── Step 1: Provider ────────────────────────────────
+      console.log(chalk.bold('  1. Choose Provider'));
+      console.log();
       const providers = listProviders();
       providers.forEach((p, i) => {
-        console.log(chalk.green(`  ${i + 1}. ${p.name}`));
+        const num = chalk.cyan((i + 1).toString().padStart(2));
+        console.log(`    ${num}  ${p.name}`);
       });
       console.log();
 
-      const choice = await question(chalk.green('  请选择 (1-' + providers.length + '): '));
+      const choice = await ask(chalk.cyan('    ❯ '));
       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', '')}`));
-      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}`));
-      });
+      // ── Step 2: API Key ─────────────────────────────────
+      console.log();
+      console.log(chalk.bold('  2. API Key'));
       console.log();
 
-      const modelChoice = await question(chalk.green('  请选择模型 (1-' + models.length + '): '));
-      const selectedModel = models[parseInt(modelChoice) - 1] || models[0];
+      let baseUrl: string | undefined;
+
+      if (isCustomProvider(providerName)) {
+        console.log(chalk.yellow('    Custom Provider (Relay Station)'));
+        console.log(chalk.gray('    OpenAI-compatible or Anthropic-compatible'));
+        console.log();
+        baseUrl = await ask(chalk.cyan('    Base URL ❯ '));
+        if (!baseUrl) {
+          console.log(chalk.red('    ❌ Base URL required'));
+          rl.close();
+          return;
+        }
+        baseUrl = baseUrl.replace(/\/+$/, '');
+      }
 
-      // Save config
-      const configDir = join(homedir(), '.thatgfsj');
-      const configPath = join(configDir, 'config.json');
+      const apiKey = await ask(chalk.cyan('    API Key ❯ '));
 
-      if (!existsSync(configDir)) {
-        mkdirSync(configDir, { recursive: true });
-      }
+      // ── Step 3: Model ───────────────────────────────────
+      console.log();
+      console.log(chalk.bold('  3. Choose Model'));
+      console.log();
 
-      writeFileSync(configPath, JSON.stringify({
-        provider: providerName,
-        model: selectedModel.id,
-        apiKey,
-        temperature: 0.7,
-        maxTokens: 4096,
-      }, null, 2));
+      if (isCustomProvider(providerName)) {
+        console.log(chalk.gray('    Enter model name for your relay station'));
+        console.log(chalk.gray('    e.g. gpt-4o-mini, claude-sonnet-4-20250514'));
+        console.log();
+        const model = await ask(chalk.cyan('    Model ❯ '));
+        if (!model) {
+          console.log(chalk.red('    ❌ Model name required'));
+          rl.close();
+          return;
+        }
+        this.saveConfig(providerName, model, apiKey, baseUrl);
+      } else {
+        const models = getModelsForProvider(providerName);
+        models.forEach((m, i) => {
+          const num = chalk.cyan((i + 1).toString().padStart(2));
+          console.log(`    ${num}  ${chalk.white(m.name)}  ${chalk.gray(m.desc)}`);
+        });
+        console.log();
+
+        const modelChoice = await ask(chalk.cyan('    ❯ '));
+        const selectedModel = models[parseInt(modelChoice) - 1] || models[0];
+        this.saveConfig(providerName, selectedModel.id, apiKey, baseUrl);
+      }
 
-      console.log(chalk.green('\n  ✅ 配置已保存!'));
-      console.log(chalk.gray(`  提供商: ${providerName}`));
-      console.log(chalk.gray(`  模型: ${selectedModel.name}`));
-      console.log(chalk.gray(`  配置文件: ${configPath}`));
-      console.log(chalk.cyan('\n  运行 gfcode 开始使用\n'));
+      // ── Done ────────────────────────────────────────────
+      console.log();
+      console.log(line);
+      console.log(chalk.green.bold('  ✅ Configuration saved!'));
+      console.log();
+      console.log(chalk.gray('  Config: ') + chalk.white(join(homedir(), '.thatgfsj', 'config.json')));
+      console.log(chalk.gray('  Run ') + chalk.cyan.bold('gfcode') + chalk.gray(' to start.'));
+      console.log(line);
+      console.log();
 
     } finally {
       rl.close();
     }
   }
+
+  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();
+    console.log(chalk.gray('    Provider: ') + chalk.white(provider));
+    console.log(chalk.gray('    Model:    ') + chalk.white(model));
+    if (baseUrl) {
+      console.log(chalk.gray('    URL:      ') + chalk.white(baseUrl));
+    }
+  }
 }