thatgfsj-code 0.3.0 → 0.3.1

package/src/tools/shell.ts

@@ -1,136 +0,0 @@
-/**
- * Shell Tool - Execute shell commands with security checks
- * Migrated from old src/tools/shell.ts (permission logic inlined)
- */
-
-import type { Tool, ToolResult, ToolContext } from './types.js';
-import { exec } from 'child_process';
-import { promisify } from 'util';
-
-const execAsync = promisify(exec);
-
-// Dangerous command patterns - blocked immediately
-const DANGEROUS_PATTERNS = [
-  /^rm\s+-rf\s+\//i,
-  /^del\s+\/f\s+\/s\s+\/q/i,
-  /^format\s+[a-z]:/i,
-  /^mkfs/i,
-  /^dd\s+if=/i,
-  /^shred/i,
-  /^cat\s+\/dev\/null\s*>/i,
-  />\s*\/dev\/sda/i,
-  /^rm\s+-rf\s+\$HOME/i,
-  /^rm\s+-rf\s+%USERPROFILE%/i,
-  /^curl\s+.*\|.*sh/i,
-  /^wget\s+.*\|.*sh/i,
-  /^eval\s+/i,
-  /base64\s+-d\s+.*\|/i,
-];
-
-// Commands that need user confirmation
-const CONFIRM_REQUIRED = [
-  'rm -rf',
-  'rmdir',
-  'del /s /q',
-  'format',
-  'mkfs',
-  'dd',
-  'shutdown',
-  'reboot',
-  'pkill',
-  'killall',
-  'git push --force',
-  'git push -f',
-];
-
-export class ShellTool implements Tool {
-  name = 'shell';
-  description = 'Execute shell commands and scripts. Some commands require user confirmation.';
-
-  inputSchema = {
-    type: 'object' as const,
-    properties: {
-      command: { type: 'string', description: 'Shell command to execute' },
-      cwd: { type: 'string', description: 'Working directory' },
-      timeout: { type: 'number', description: 'Timeout in seconds', default: 30 },
-    },
-    required: ['command'],
-  };
-
-  metadata = {
-    permissions: ['execute', 'write', 'network'] as ('read' | 'write' | 'execute' | 'network')[],
-    tags: ['shell', 'system', 'dangerous'],
-    maxDuration: 120000,
-    version: '1.0.0',
-  };
-
-  parameters = [
-    { name: 'command', type: 'string', description: 'Shell command to execute', required: true },
-    { name: 'cwd', type: 'string', description: 'Working directory', required: false },
-    { name: 'timeout', type: 'number', description: 'Timeout in seconds', required: false },
-  ];
-
-  /**
-   * Check if command matches dangerous patterns
-   */
-  private isDangerous(command: string): boolean {
-    return DANGEROUS_PATTERNS.some(pattern => pattern.test(command.trim()));
-  }
-
-  /**
-   * Check if command needs user confirmation
-   */
-  private needsConfirmation(command: string): boolean {
-    const lower = command.toLowerCase();
-    return CONFIRM_REQUIRED.some(cmd => lower.includes(cmd.toLowerCase()));
-  }
-
-  async execute(params: Record<string, any>, ctx?: ToolContext): Promise<ToolResult> {
-    const { command, cwd, timeout = 30 } = params;
-
-    if (!command || typeof command !== 'string') {
-      return { success: false, error: 'Command is required and must be a string' };
-    }
-
-    const trimmed = command.trim();
-
-    // Block dangerous commands
-    if (this.isDangerous(trimmed)) {
-      return { success: false, error: `Blocked: dangerous command "${trimmed.substring(0, 50)}..."` };
-    }
-
-    if (trimmed.length > 10000) {
-      return { success: false, error: 'Command too long (max 10000 characters)' };
-    }
-
-    // Ask for confirmation on risky commands
-    if (ctx?.confirmAction && this.needsConfirmation(trimmed)) {
-      const confirmed = await ctx.confirmAction(
-        `⚠️  Confirm command:\n  ${command}\n\n[y] Yes  [n] No`
-      );
-      if (!confirmed) {
-        return { success: false, error: 'Command cancelled by user' };
-      }
-    }
-
-    try {
-      const options: any = {
-        timeout: timeout * 1000,
-        maxBuffer: 10 * 1024 * 1024, // 10MB
-      };
-      if (cwd) options.cwd = cwd;
-
-      const { stdout, stderr } = await execAsync(command, options);
-      const stdoutStr = stdout?.toString() || '';
-      const stderrStr = stderr?.toString() || '';
-      const output = stdoutStr + (stderrStr ? `\n[stderr]: ${stderrStr}` : '');
-
-      return { success: true, output: output.trim() || '(command executed successfully with no output)' };
-    } catch (error: any) {
-      if (error.killed) {
-        return { success: false, error: 'Command timed out' };
-      }
-      return { success: false, error: error.message || 'Command execution failed' };
-    }
-  }
-}

package/src/tools/types.ts

@@ -1,122 +0,0 @@
-/**
- * Tool interface definitions
- * Extracted from old src/core/types.ts
- */
-
-// ==================== Tool Parameter & Schema ====================
-
-export interface ToolParameter {
-  name: string;
-  type: string;
-  description: string;
-  required: boolean;
-  default?: any;
-  enum?: string[];
-}
-
-export interface ToolInputSchema {
-  type: 'object';
-  properties: Record<string, {
-    type: string;
-    description: string;
-    default?: any;
-    enum?: string[];
-  }>;
-  required?: string[];
-}
-
-export interface ToolMetadata {
-  permissions?: ('read' | 'write' | 'execute' | 'network')[];
-  maxDuration?: number;
-  tags?: string[];
-  version?: string;
-  deprecated?: string;
-}
-
-// ==================== Tool Interface ====================
-
-export interface ToolContext {
-  sessionId?: string;
-  workingDirectory?: string;
-  confirmAction?: (msg: string) => Promise<boolean>;
-  signal?: AbortSignal;
-  toolCallId?: string;
-}
-
-export interface ToolResult {
-  success: boolean;
-  output?: string;
-  error?: string;
-  data?: any;
-}
-
-export interface Tool {
-  name: string;
-  description: string;
-  parameters: ToolParameter[];
-  inputSchema?: ToolInputSchema;
-  metadata?: ToolMetadata;
-  execute(params: Record<string, any>, ctx?: ToolContext): Promise<ToolResult>;
-}
-
-// ==================== Tool Builder ====================
-
-export class ToolBuilder {
-  private _name = '';
-  private _description = '';
-  private _parameters: ToolParameter[] = [];
-  private _inputSchema: ToolInputSchema = { type: 'object', properties: {} };
-  private _metadata: ToolMetadata = {};
-  private _fn?: (params: Record<string, any>, ctx: ToolContext) => Promise<ToolResult>;
-
-  name(n: string): this { this._name = n; return this; }
-  description(d: string): this { this._description = d; return this; }
-
-  param(p: ToolParameter): this {
-    this._parameters.push(p);
-    this._inputSchema.properties[p.name] = {
-      type: p.type,
-      description: p.description,
-      default: p.default,
-      enum: p.enum,
-    };
-    return this;
-  }
-
-  stringParam(name: string, description: string, required = true, defaultVal?: string): this {
-    return this.param({ name, type: 'string', description, required, default: defaultVal });
-  }
-
-  booleanParam(name: string, description: string, required = false, defaultVal?: boolean): this {
-    return this.param({ name, type: 'boolean', description, required, default: defaultVal });
-  }
-
-  permissions(...perms: ('read' | 'write' | 'execute' | 'network')[]): this {
-    this._metadata.permissions = perms;
-    return this;
-  }
-
-  tags(...t: string[]): this {
-    this._metadata.tags = t;
-    return this;
-  }
-
-  handle(fn: (params: Record<string, any>, ctx: ToolContext) => Promise<ToolResult>): this {
-    this._fn = fn;
-    return this;
-  }
-
-  build(): Tool {
-    if (!this._name || !this._fn) {
-      throw new Error('Tool name and handler are required');
-    }
-    return {
-      name: this._name,
-      description: this._description,
-      parameters: this._parameters,
-      inputSchema: this._inputSchema,
-      metadata: this._metadata,
-      execute: this._fn,
-    };
-  }
-}

package/src/tui/index.ts

@@ -1,8 +0,0 @@
-/**
- * TUI Module - Terminal User Interface
- */
-
-export { REPL } from './repl.js';
-export { REPLInput } from './input.js';
-export { REPLOutput } from './output.js';
-export { WelcomeScreen } from './welcome.js';

package/src/tui/input.ts

@@ -1,56 +0,0 @@
-/**
- * REPL Input Handler
- * Migrated from old src/repl/input.ts
- */
-
-import readline from 'readline';
-import chalk from 'chalk';
-
-export class REPLInput {
-  private rl: readline.Interface;
-  private history: string[] = [];
-  private historyIndex: number = -1;
-
-  constructor() {
-    this.rl = readline.createInterface({
-      input: process.stdin,
-      output: process.stdout,
-      completer: this.completer.bind(this),
-      historySize: 100,
-      tabSize: 4,
-    });
-
-    // Handle Ctrl+C
-    this.rl.on('SIGINT', () => {
-      process.stdout.write(chalk.gray('\n\n👋 Exiting...\n'));
-      process.exit(0);
-    });
-
-    // Track history
-    this.rl.on('line', (line) => {
-      if (line.trim()) {
-        this.history.push(line);
-        if (this.history.length > 100) this.history.shift();
-      }
-      this.historyIndex = this.history.length;
-    });
-  }
-
-  async prompt(prefix: string = chalk.green('\n> ')): Promise<string> {
-    return new Promise((resolve) => {
-      this.rl.question(prefix, (answer) => {
-        resolve(answer);
-      });
-    });
-  }
-
-  close(): void {
-    this.rl.close();
-  }
-
-  private completer(line: string): [string[], string] {
-    const commands = ['exit', 'quit', 'clear', 'help', 'tools', 'model'];
-    const hits = commands.filter(c => c.startsWith(line.toLowerCase()));
-    return [hits.length ? hits : [], line];
-  }
-}

package/src/tui/output.ts

@@ -1,83 +0,0 @@
-/**
- * REPL Output Handler
- * Migrated from old src/repl/output.ts
- */
-
-import chalk from 'chalk';
-import ora, { Ora } from 'ora';
-
-export class REPLOutput {
-  private spinner: Ora | null = null;
-
-  printError(error: string): void {
-    console.error(chalk.red(`\n❌ Error: ${error}`));
-  }
-
-  printWarning(warning: string): void {
-    console.warn(chalk.yellow(`\n⚠️  ${warning}`));
-  }
-
-  printSuccess(message: string): void {
-    console.log(chalk.green(`\n✅ ${message}`));
-  }
-
-  printInfo(message: string): void {
-    console.log(chalk.gray(message));
-  }
-
-  printDivider(char = '─', length = 40): void {
-    console.log(chalk.gray(char.repeat(length)));
-  }
-
-  printHeader(text: string): void {
-    this.printDivider();
-    console.log(chalk.cyan(text));
-    this.printDivider();
-  }
-
-  startSpinner(text: string): void {
-    this.spinner = ora({ text, color: 'cyan', spinner: 'dots' }).start();
-  }
-
-  stopSpinner(): void {
-    if (this.spinner) {
-      this.spinner.stop();
-      this.spinner = null;
-    }
-  }
-
-  stopSpinnerFail(text?: string): void {
-    if (this.spinner) {
-      this.spinner.fail(text || this.spinner.text);
-      this.spinner = null;
-    }
-  }
-
-  clear(): void {
-    console.clear();
-  }
-
-  printBanner(): void {
-    console.log(chalk.cyan(`
-╔═══════════════════════════════════════╗
-║     🤖 Thatgfsj Code v0.3.0        ║
-║     AI Coding Assistant              ║
-╚═══════════════════════════════════════╝
-    `));
-  }
-
-  printHelp(): void {
-    console.log(`
-${chalk.cyan('Commands:')}
-  ${chalk.gray('exit, Ctrl+C')}  - Exit
-  ${chalk.gray('clear')}           - Clear screen
-  ${chalk.gray('help')}            - Show this help
-  ${chalk.gray('tools')}            - List available tools
-  ${chalk.gray('model')}            - Show current model
-
-${chalk.cyan('Tips:')}
-  • Use ${chalk.gray('↑/↓')} for command history
-  • Use ${chalk.gray('Tab')} for auto-complete
-    `);
-  }
-}

package/src/tui/repl.ts

@@ -1,131 +0,0 @@
-/**
- * REPL - Main interactive loop
- * Migrated from old src/repl/loop.ts + src/index.ts interactive mode
- */
-
-import chalk from 'chalk';
-import { REPLInput } from './input.js';
-import { REPLOutput } from './output.js';
-import { App } from '../app/index.js';
-import { ProjectContext } from '../utils/project.js';
-
-export class REPL {
-  private input: REPLInput;
-  private output: REPLOutput;
-  private app: App;
-  private running = false;
-
-  constructor(app: App) {
-    this.input = new REPLInput();
-    this.output = new REPLOutput();
-    this.app = app;
-  }
-
-  /**
-   * Start the REPL
-   */
-  async start(): Promise<void> {
-    this.running = true;
-
-    // Show banner
-    this.output.clear();
-    this.output.printBanner();
-    this.output.printInfo('Type "help" for available commands\n');
-    this.output.printDivider();
-
-    // Show project context
-    console.log(chalk.gray(ProjectContext.getSummary()));
-    this.output.printDivider();
-    console.log();
-
-    // Main loop
-    while (this.running) {
-      try {
-        const userInput = await this.input.prompt();
-
-        if (!userInput.trim()) continue;
-
-        // Handle commands
-        const handled = await this.handleCommand(userInput.trim());
-        if (handled) continue;
-
-        // Process with AI
-        await this.processInput(userInput.trim());
-      } catch (error: any) {
-        if (error.message === 'SIGINT') continue;
-        this.output.printError(error.message);
-      }
-    }
-  }
-
-  /**
-   * Handle built-in commands
-   */
-  private async handleCommand(input: string): Promise<boolean> {
-    const cmd = input.toLowerCase();
-
-    switch (cmd) {
-      case 'exit':
-      case 'quit':
-        this.output.printInfo('\n👋 Goodbye!');
-        this.running = false;
-        this.input.close();
-        return true;
-
-      case 'clear':
-        this.output.clear();
-        this.output.printBanner();
-        return true;
-
-      case 'help':
-        this.output.printHelp();
-        return true;
-
-      case 'tools':
-        this.output.printHeader('🔧 Available Tools');
-        for (const tool of this.app.tools.list()) {
-          this.output.printInfo(`  ${tool.name} - ${tool.description}`);
-        }
-        console.log();
-        return true;
-
-      case 'model':
-        this.output.printHeader('🤖 Current Model');
-        const cfg = this.app.config.get();
-        this.output.printInfo(`  Provider: ${cfg.provider}`);
-        this.output.printInfo(`  Model: ${cfg.model}`);
-        this.output.printInfo(`  Base URL: ${cfg.baseUrl || 'default'}`);
-        console.log();
-        return true;
-
-      default:
-        return false;
-    }
-  }
-
-  /**
-   * Process user input with AI
-   */
-  private async processInput(input: string): Promise<void> {
-    this.app.session.addMessage('user', input);
-
-    this.output.startSpinner('Thinking...');
-    let fullResponse = '';
-
-    try {
-      const stream = this.app.getAgent().run(this.app.session.getMessages());
-      for await (const chunk of stream) {
-        this.output.stopSpinner();
-        process.stdout.write(chunk);
-        fullResponse += chunk;
-      }
-
-      console.log(); // newline
-
-      this.app.session.addMessage('assistant', fullResponse);
-      this.app.session.truncate();
-    } catch (error: any) {
-      this.output.stopSpinnerFail(chalk.red(`Error: ${error.message}`));
-    }
-  }
-}

package/src/tui/welcome.ts

@@ -1,101 +0,0 @@
-/**
- * Welcome / Setup Wizard
- * Migrated and simplified from old src/repl/welcome.ts
- */
-
-import chalk from 'chalk';
-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 type { ProviderName } from '../config/types.js';
-
-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();
-  }
-
-  /**
-   * Interactive setup wizard
-   */
-  static async interactiveSetup(): Promise<void> {
-    console.clear();
-    console.log(chalk.cyan('\n  🤖 Thatgfsj Code 配置向导\n'));
-
-    const rl = readline.createInterface({
-      input: process.stdin,
-      output: process.stdout,
-    });
-
-    const question = (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'));
-      const providers = listProviders();
-      providers.forEach((p, i) => {
-        console.log(chalk.green(`  ${i + 1}. ${p.name}`));
-      });
-      console.log();
-
-      const choice = await question(chalk.green('  请选择 (1-' + providers.length + '): '));
-      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}`));
-      });
-      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 });
-      }
-
-      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.cyan('\n  运行 gfcode 开始使用\n'));
-
-    } finally {
-      rl.close();
-    }
-  }
-}

package/src/types.ts

@@ -1,42 +0,0 @@
-/**
- * Global shared types for Thatgfsj Code
- * Only the most fundamental types used across modules
- */
-
-export type Role = 'system' | 'user' | 'assistant' | 'tool';
-
-export interface ChatMessage {
-  role: Role;
-  content: string;
-  name?: string;
-  tool_calls?: ToolCall[];
-  tool_call_id?: string;
-  reasoning_content?: string;
-}
-
-export interface ToolCall {
-  id: string;
-  type: 'function';
-  function: {
-    name: string;
-    arguments: string;
-  };
-}
-
-export interface ChatResponse {
-  content: string;
-  role: 'assistant';
-  usage?: {
-    prompt_tokens: number;
-    completion_tokens: number;
-    total_tokens: number;
-  };
-  tool_calls?: ToolCall[];
-  reasoning_content?: string;
-}
-
-export interface ChatOptions {
-  temperature?: number;
-  maxTokens?: number;
-  stream?: boolean;
-}