thatgfsj-code 3.0.4 → 3.0.6

package/src/mcp/client.ts

@@ -1,330 +0,0 @@
-/**
- * S07: MCP Client (Model Context Protocol)
- * Connect to MCP servers and expose their tools as agent tools
- * 
- * MCP turns the agent from a closed system into an open platform.
- * Any server can expose tools via the MCP JSON-RPC 2.0 protocol.
- */
-
-import { spawn, ChildProcess } from 'child_process';
-import type { Tool, ToolResult } from '../tools/types.js';
-
-// ==================== MCP Types ====================
-
-interface MCPJsonRPCRequest {
-  jsonrpc: '2.0';
-  id: number | string;
-  method: string;
-  params?: any;
-}
-
-interface MCPJsonRPCResponse {
-  jsonrpc: '2.0';
-  id: number | string;
-  result?: any;
-  error?: { code: number; message: string; data?: any };
-}
-
-interface MCPTool {
-  name: string;
-  description: string;
-  inputSchema: {
-    type: 'object';
-    properties: Record<string, any>;
-    required?: string[];
-  };
-}
-
-// ==================== MCP Client ====================
-
-export class MCPClient {
-  private process: ChildProcess | null = null;
-  private tools: Map<string, MCPTool> = new Map();
-  private requestId = 0;
-  private pendingRequests: Map<number | string, {
-    resolve: (value: any) => void;
-    reject: (reason: any) => void;
-  }> = new Map();
-  private name: string;
-  private connected = false;
-
-  constructor(name: string) {
-    this.name = name;
-  }
-
-  /**
-   * S07: Connect to an MCP server
-   * @param command Command to run (e.g., 'npx', 'node')
-   * @param args Arguments (e.g., ['-y', '@modelcontextprotocol/server-filesystem', '/path/to/dir'])
-   */
-  async connect(command: string, args: string[]): Promise<void> {
-    return new Promise((resolve, reject) => {
-      this.process = spawn(command, args, {
-        stdio: ['pipe', 'pipe', 'pipe']
-      });
-
-      let buffer = '';
-
-      this.process.stdout?.on('data', (data: Buffer) => {
-        buffer += data.toString();
-        const lines = buffer.split('\n');
-        buffer = lines.pop() || '';
-
-        for (const line of lines) {
-          if (line.trim()) {
-            this.handleMessage(line);
-          }
-        }
-      });
-
-      this.process.stderr?.on('data', (data: Buffer) => {
-        console.error(`[MCP ${this.name}] stderr:`, data.toString().trim());
-      });
-
-      this.process.on('error', (err) => {
-        reject(err);
-      });
-
-      this.process.on('close', (code) => {
-        this.connected = false;
-        console.log(`[MCP ${this.name}] exited with code ${code}`);
-      });
-
-      // Initialize MCP connection
-      this.sendRequest('initialize', {
-        protocolVersion: '2024-11-05',
-        capabilities: { tools: {} },
-        clientInfo: { name: 'thatgfsj-code', version: '1.0.0' }
-      }).then(() => {
-        this.connected = true;
-        // Send initialized notification
-        this.sendNotification('initialized', {});
-        // List available tools
-        return this.listTools();
-      }).then(() => {
-        resolve();
-      }).catch(reject);
-
-      // Timeout
-      setTimeout(() => reject(new Error('MCP connection timeout')), 30000);
-    });
-  }
-
-  /**
-   * S07: Handle incoming JSON-RPC message
-   */
-  private handleMessage(raw: string): void {
-    try {
-      const msg: MCPJsonRPCResponse = JSON.parse(raw);
-
-      // Handle response
-      const pending = this.pendingRequests.get(msg.id);
-      if (pending) {
-        this.pendingRequests.delete(msg.id);
-        if (msg.error) {
-          pending.reject(new Error(msg.error.message));
-        } else {
-          pending.resolve(msg.result);
-        }
-        return;
-      }
-    } catch {
-      // Not JSON, ignore
-    }
-  }
-
-  /**
-   * S07: Send a JSON-RPC request
-   */
-  private sendRequest(method: string, params: any): Promise<any> {
-    return new Promise((resolve, reject) => {
-      if (!this.process?.connected) {
-        reject(new Error('MCP process not connected'));
-        return;
-      }
-
-      const id = ++this.requestId;
-      this.pendingRequests.set(id, { resolve, reject });
-
-      const request: MCPJsonRPCRequest = {
-        jsonrpc: '2.0',
-        id,
-        method,
-        params
-      };
-
-      this.process.stdin?.write(JSON.stringify(request) + '\n');
-
-      // Timeout
-      setTimeout(() => {
-        if (this.pendingRequests.has(id)) {
-          this.pendingRequests.delete(id);
-          reject(new Error('MCP request timeout: ' + method));
-        }
-      }, 30000);
-    });
-  }
-
-  /**
-   * S07: Send a notification (no response expected)
-   */
-  private sendNotification(method: string, params: any): void {
-    if (!this.process?.connected) return;
-
-    const notification = {
-      jsonrpc: '2.0',
-      method,
-      params
-    };
-
-    this.process.stdin?.write(JSON.stringify(notification) + '\n');
-  }
-
-  /**
-   * S07: Call an MCP tool
-   */
-  async callTool(name: string, arguments_: Record<string, any>): Promise<ToolResult> {
-    try {
-      const result = await this.sendRequest('tools/call', {
-        name,
-        arguments: arguments_
-      });
-
-      return {
-        success: true,
-        output: typeof result === 'string' ? result : JSON.stringify(result)
-      };
-    } catch (error: any) {
-      return {
-        success: false,
-        error: error.message
-      };
-    }
-  }
-
-  /**
-   * S07: List tools from MCP server
-   */
-  private async listTools(): Promise<void> {
-    try {
-      const result = await this.sendRequest('tools/list', {});
-      const tools: MCPTool[] = result.tools || [];
-
-      this.tools.clear();
-      for (const tool of tools) {
-        this.tools.set(tool.name, tool);
-      }
-
-      console.log(`[MCP ${this.name}] Loaded ${tools.length} tools:`, [...this.tools.keys()].join(', '));
-    } catch (error: any) {
-      console.error(`[MCP ${this.name}] Failed to list tools:`, error.message);
-    }
-  }
-
-  /**
-   * S07: Get registered tools as agent Tool[]
-   */
-  getTools(): Tool[] {
-    return [...this.tools.values()].map(mcpTool => ({
-      name: this.name + ':' + mcpTool.name,
-      description: mcpTool.description,
-      parameters: Object.entries(mcpTool.inputSchema.properties || {}).map(([name, schema]: [string, any]) => ({
-        name,
-        type: schema.type || 'string',
-        description: schema.description || '',
-        required: (mcpTool.inputSchema.required || []).includes(name)
-      })),
-      execute: async (params: Record<string, any>): Promise<ToolResult> => {
-        return this.callTool(mcpTool.name, params);
-      }
-    }));
-  }
-
-  /**
-   * S07: Get tool names
-   */
-  getToolNames(): string[] {
-    return [...this.tools.keys()];
-  }
-
-  /**
-   * S07: Check if connected
-   */
-  isConnected(): boolean {
-    return this.connected;
-  }
-
-  /**
-   * S07: Disconnect from MCP server
-   */
-  disconnect(): void {
-    if (this.process) {
-      this.process.kill();
-      this.process = null;
-    }
-    this.connected = false;
-    this.tools.clear();
-    this.pendingRequests.clear();
-  }
-}
-
-// ==================== MCP Server Manager ====================
-
-export class MCPServerManager {
-  private clients: Map<string, MCPClient> = new Map();
-
-  /**
-   * S07: Add and connect an MCP server
-   */
-  async addServer(name: string, command: string, args: string[]): Promise<void> {
-    const client = new MCPClient(name);
-    await client.connect(command, args);
-    this.clients.set(name, client);
-  }
-
-  /**
-   * S07: Get all tools from all servers
-   */
-  getAllTools(): Tool[] {
-    const allTools: Tool[] = [];
-    for (const client of this.clients.values()) {
-      allTools.push(...client.getTools());
-    }
-    return allTools;
-  }
-
-  /**
-   * S07: Get tool names from all servers
-   */
-  getAllToolNames(): string[] {
-    const names: string[] = [];
-    for (const [serverName, client] of this.clients) {
-      for (const toolName of client.getToolNames()) {
-        names.push(serverName + ':' + toolName);
-      }
-    }
-    return names;
-  }
-
-  /**
-   * S07: Disconnect a server
-   */
-  removeServer(name: string): void {
-    const client = this.clients.get(name);
-    if (client) {
-      client.disconnect();
-      this.clients.delete(name);
-    }
-  }
-
-  /**
-   * S07: Get server status
-   */
-  getStatus(): Record<string, boolean> {
-    const status: Record<string, boolean> = {};
-    for (const [name, client] of this.clients) {
-      status[name] = client.isConnected();
-    }
-    return status;
-  }
-}

package/src/prompts/index.ts

@@ -1,260 +0,0 @@
-/**
- * System Prompt Builder - Dynamic prompt construction
- */
-
-import { readFileSync, existsSync, readdirSync } from 'fs';
-import { join } from 'path';
-import type { Tool } from '../tools/types.js';
-
-export interface SystemPromptConfig {
-  cwd?: string;
-  tools?: Tool[];
-  includeProjectMd?: boolean;
-  permissionMode?: 'accept' | 'deny' | 'ask';
-  date?: Date;
-  skillsPrompt?: string;
-}
-
-/**
- * v3.0.0: a single named fragment of the system prompt. The provider layer
- * attaches cache_control markers to volatile=false segments (Anthropic) and
- * the fingerprint module hashes only the non-volatile portion.
- */
-export interface SystemSegment {
-  /** Stable identifier, used in fingerprints and logging. */
-  name: string;
-  /** Rendered text content. May be empty string if the section was a no-op. */
-  content: string;
-  /**
-   * volatile=false: content never changes between requests within a session.
-   * volatile=true:  content may change (NWT events, current time).
-   */
-  volatile: boolean;
-}
-
-export class SystemPromptBuilder {
-  private config: Required<SystemPromptConfig>;
-
-  constructor(config: SystemPromptConfig = {}) {
-    this.config = {
-      cwd: config.cwd ?? process.cwd(),
-      tools: config.tools ?? [],
-      includeProjectMd: config.includeProjectMd ?? true,
-      permissionMode: config.permissionMode ?? 'ask',
-      date: config.date ?? new Date(),
-      skillsPrompt: config.skillsPrompt ?? '',
-    };
-  }
-
-  build(): string {
-    return this.buildSegments()
-      .map(s => s.content)
-      .filter(Boolean)
-      .join('\n\n');
-  }
-
-  /**
-   * v3.0.0: build the system prompt as named segments instead of a flat
-   * string. The provider layer (Anthropic) can attach cache_control markers
-   * to the segments it caches, and the fingerprint module can hash the
-   * static portion to detect when the immutable prefix changed.
-   *
-   * Ordering rule (Reasonix-style): immutable prefix first, volatile tail
-   * last. NWT history and Date move to the tail because both can change
-   * between rounds (Date: every build; NWT: when a new event is logged).
-   * Keeping them out of the prefix means the upstream cache keeps its
-   * prefix checkpoint stable across those changes.
-   *
-   * Segments with empty content are filtered out — they neither contribute
-   * to the string output nor to any cache fingerprint.
-   */
-  buildSegments(): SystemSegment[] {
-    return [
-      // Immutable prefix (cacheable as a single block on Anthropic)
-      { name: 'project-instructions', content: this.buildProjectInstructions(), volatile: false },
-      { name: 'identity',              content: this.buildIdentity(),              volatile: false },
-      { name: 'tool-instructions',     content: this.buildToolInstructions(),     volatile: false },
-      { name: 'environment',           content: this.buildEnvironment(),           volatile: false },
-      { name: 'permission-mode',       content: this.buildPermissionMode(),       volatile: false },
-      { name: 'skills',                content: this.buildSkills(),                volatile: false },
-      // Volatile tail — not part of the cache prefix
-      { name: 'nwt-history',           content: this.buildNwtHistory(),           volatile: true  },
-      { name: 'date-info',             content: this.buildDateInfo(),             volatile: true  },
-    ].filter(s => s.content);
-  }
-
-  private buildIdentity(): string {
-    return [
-      'You are Thatgfsj Code, an interactive coding agent.',
-      '',
-      'IMPORTANT: You MUST follow the user configuration above (CLAUDE.md, SKILLS.md, etc).',
-      'At the start of each task, read SKILLS.md to check for relevant skills.',
-      '',
-      'Tools available: file, shell, git, search, nwt.',
-      '',
-      'Rules:',
-      '- Follow the user\'s technical preferences from their config',
-      '- Be concise but thorough',
-      '- Verify changes after making them',
-      '',
-      '## NWT Auto-Logging (IMPORTANT)',
-      '',
-      'After completing any meaningful task, you MUST evaluate whether to log it to NWT.',
-      'Call `nwt log` automatically when ANY of these conditions are true:',
-      '',
-      '1. You created, modified, or deleted 2+ files',
-      '2. You made an architectural decision (chose a library, pattern, or approach)',
-      '3. You fixed a bug',
-      '4. You added a new feature or capability',
-      '5. You refactored existing code',
-      '6. You set up a new project or module',
-      '7. You resolved a security issue',
-      '8. You completed a multi-step task',
-      '',
-      'Do NOT log:',
-      '- Simple file reads or searches',
-      '- Trivial single-line changes',
-      '- Conversational questions with no code changes',
-      '',
-      'When logging, use:',
-      '  nwt log task="<imperative verb> <what>" summary="<what was done>" files="<files>" tags="<tags>"',
-      '',
-      'Tags: feature, fix, refactor, decision, milestone, test, docs, config, security, init',
-      '',
-      'IMPORTANT: Log silently without mentioning it to the user unless they ask.',
-    ].join('\n');
-  }
-
-  private buildToolInstructions(): string {
-    const tools = this.config.tools;
-    if (tools.length === 0) return '## Tools\n\nNo tools are currently registered.';
-
-    const toolDescs = tools.map(t => {
-      const params = t.parameters
-        .map(p => `  - ${p.name} (${p.type}${p.required ? ', required' : ''}): ${p.description}`)
-        .join('\n');
-      return `### ${t.name}\n${t.description}\n\nParameters:\n${params}`;
-    }).join('\n\n');
-
-    return `## Tools\n\nYou have access to the following tools:\n\n${toolDescs}\n\nTo use a tool, respond with a tool call.`;
-  }
-
-  private buildEnvironment(): string {
-    return `## Environment\n\nWorking directory: ${this.config.cwd}`;
-  }
-
-  private buildPermissionMode(): string {
-    const mode = this.config.permissionMode;
-    const explanations: Record<string, string> = {
-      accept: 'All tool calls are automatically allowed without confirmation.',
-      deny: 'All tool calls are blocked. You may only read and discuss.',
-      ask: 'Dangerous or destructive commands require user confirmation before execution.',
-    };
-    return `## Permission Mode\n\nCurrent mode: ${mode}\n\n${explanations[mode] || ''}`;
-  }
-
-  /**
-   * Read project instruction files (generic, works for any user)
-   */
-  private buildProjectInstructions(): string {
-    if (!this.config.includeProjectMd) return '';
-
-    const cwd = this.config.cwd;
-    const home = process.env.USERPROFILE || process.env.HOME || '';
-
-    // Project-level files (any project can have these)
-    const projectFiles = [
-      'CLAUDE.md', '.claude.md', 'AGENTS.md',
-      'Codex.md', '.Codex.md', 'CODEX.md', '.codex.md',
-      'CONVENTIONS.md', 'CONTRIBUTING.md',
-    ];
-
-    // User-level files (in home directory)
-    const userDirs = ['.claude', '.Codex', '.agents'];
-    const userFiles = ['CLAUDE.md', 'SKILLS.md', 'AGENTS.md', 'CONVENTIONS.md'];
-
-    const paths: string[] = [];
-
-    // Project-level
-    for (const f of projectFiles) {
-      paths.push(join(cwd, f));
-    }
-
-    // User-level
-    for (const dir of userDirs) {
-      for (const f of userFiles) {
-        paths.push(join(home, dir, f));
-      }
-    }
-
-    const sections: string[] = [];
-    const MAX_LEN = 3000;
-
-    for (const path of paths) {
-      if (existsSync(path)) {
-        try {
-          let content = readFileSync(path, 'utf-8').trim();
-          if (content) {
-            const filename = path.split(/[/\\]/).pop();
-            if (content.length > MAX_LEN) {
-              content = content.slice(0, MAX_LEN) + '\n... (truncated)';
-            }
-            sections.push(`[${filename}]\n${content}`);
-          }
-        } catch {}
-      }
-    }
-
-    if (sections.length === 0) return '';
-    return `## User Configuration\n\n${sections.join('\n\n')}`;
-  }
-
-  /**
-   * Auto-inject recent NWT history into system prompt
-   */
-  private buildNwtHistory(): string {
-    try {
-      const nwtDir = join(this.config.cwd, '.nwt', 'events');
-      if (!existsSync(nwtDir)) return '';
-
-      const files = readdirSync(nwtDir)
-        .filter(f => f.endsWith('.json'))
-        .sort()
-        .slice(-10); // Last 10 events
-
-      if (files.length === 0) return '';
-
-      const events = files.map(f => {
-        try {
-          const ev = JSON.parse(readFileSync(join(nwtDir, f), 'utf-8'));
-          const time = ev.timestamp?.split('T')[0] || '';
-          const files = ev.files?.length > 0 ? ` [${ev.files.join(', ')}]` : '';
-          return `- [${ev.id}] ${time} ${ev.task}: ${ev.summary}${files}`;
-        } catch {
-          return null;
-        }
-      }).filter(Boolean);
-
-      if (events.length === 0) return '';
-
-      return `## Project History (NWT)\n\nRecent evolution events:\n\n${events.join('\n')}`;
-    } catch {
-      return '';
-    }
-  }
-
-  private buildSkills(): string {
-    if (!this.config.skillsPrompt) return '';
-    return `## Active Skills\n\n${this.config.skillsPrompt}`;
-  }
-
-  private buildDateInfo(): string {
-    const now = this.config.date;
-    const iso = now.toISOString().replace('T', ' ').split('.')[0];
-    return `## Current Time\n\n${iso}`;
-  }
-
-  setTools(tools: Tool[]): this { this.config.tools = tools; return this; }
-  setCwd(cwd: string): this { this.config.cwd = cwd; return this; }
-  setPermissionMode(mode: 'accept' | 'deny' | 'ask'): this { this.config.permissionMode = mode; return this; }
-}

package/src/session/compactor.ts

@@ -1,103 +0,0 @@
-/**
- * Context Compactor - Progressive compression for conversation history
- * Migrated from old src/core/context-compactor.ts
- *
- * Strategy:
- * 1. Within limit: no compression
- * 2. Beyond limit: keep system + recent N turns + summarize middle
- */
-
-import type { ChatMessage } from '../types.js';
-
-export interface CompactorConfig {
-  /** Max messages before compression kicks in */
-  maxMessages?: number;
-  /** How many recent turns to always preserve */
-  preserveRecent?: number;
-}
-
-export interface CompressionResult {
-  originalCount: number;
-  compactedCount: number;
-  removedCount: number;
-}
-
-export class ContextCompactor {
-  private maxMessages: number;
-  private preserveRecent: number;
-
-  constructor(config: CompactorConfig = {}) {
-    this.maxMessages = config.maxMessages ?? 50;
-    this.preserveRecent = config.preserveRecent ?? 10;
-  }
-
-  /**
-   * Compact messages if needed
-   */
-  compact(messages: ChatMessage[]): { compacted: ChatMessage[]; result: CompressionResult } {
-    if (messages.length <= this.maxMessages) {
-      return {
-        compacted: messages,
-        result: { originalCount: messages.length, compactedCount: messages.length, removedCount: 0 },
-      };
-    }
-
-    const systemMsg = messages.find(m => m.role === 'system');
-    const others = messages.filter(m => m.role !== 'system');
-
-    if (others.length <= this.preserveRecent) {
-      return {
-        compacted: messages,
-        result: { originalCount: messages.length, compactedCount: messages.length, removedCount: 0 },
-      };
-    }
-
-    // Keep recent N turns
-    const recent = others.slice(-this.preserveRecent);
-
-    // Summarize the middle portion
-    const middle = others.slice(0, -this.preserveRecent);
-    const summary: ChatMessage = {
-      role: 'system',
-      content: `[Earlier conversation summary: ${middle.length} messages covering ${this.extractTopics(middle)}]`,
-    };
-
-    const compacted = systemMsg
-      ? [systemMsg, summary, ...recent]
-      : [summary, ...recent];
-
-    return {
-      compacted,
-      result: {
-        originalCount: messages.length,
-        compactedCount: compacted.length,
-        removedCount: messages.length - compacted.length,
-      },
-    };
-  }
-
-  /**
-   * Auto-compact if messages exceed limit
-   */
-  autoCompact(messages: ChatMessage[]): { compacted: ChatMessage[]; result: CompressionResult } {
-    return this.compact(messages);
-  }
-
-  /**
-   * Estimate token count (rough: ~4 chars per token)
-   */
-  estimateTokens(messages: ChatMessage[]): number {
-    return messages.reduce((sum, m) => sum + Math.ceil(m.content.length / 4) + 10, 0);
-  }
-
-  /**
-   * Extract topic summary from messages
-   */
-  private extractTopics(msgs: ChatMessage[]): string {
-    const topics = msgs
-      .filter(m => m.role === 'user')
-      .map(m => m.content.slice(0, 30))
-      .slice(0, 3);
-    return topics.join(', ') || 'various topics';
-  }
-}