@stacksjs/ai 0.70.55 → 0.70.57

package/src/buddy.ts

@@ -1,619 +0,0 @@
-/**
- * Buddy - Voice AI Code Assistant
- *
- * This module contains the shared state and utilities for Buddy,
- * an AI-powered code assistant that helps users modify codebases
- * through voice commands.
- */
-
-import type {
-  AIDriver,
-  AIMessage,
-  BuddyApiKeys,
-  BuddyConfig,
-  BuddyState,
-  GitHubCredentials,
-  RepoState,
-  StreamingResult,
-} from './types'
-import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs'
-import { homedir } from 'node:os'
-import { dirname, join } from 'node:path'
-import { claudeAgent } from './agents'
-import { createAnthropicDriver, createClaudeAgentSDKDriver, createOllamaDriver, createOpenAIDriver } from './drivers'
-
-// =============================================================================
-// Configuration
-// =============================================================================
-
-export const CONFIG: BuddyConfig = {
-  workDir: join(homedir(), 'Code', '.buddy-repos'),
-  commitMessage: 'chore: wip',
-  ollamaHost: process.env.OLLAMA_HOST || 'http://localhost:11434',
-  ollamaModel: process.env.OLLAMA_MODEL || 'llama3.2',
-}
-
-// API Keys state (can be set at runtime via settings endpoint)
-export const apiKeys: BuddyApiKeys = {
-  anthropic: process.env.ANTHROPIC_API_KEY,
-  openai: process.env.OPENAI_API_KEY,
-  claudeCliHost: process.env.BUDDY_EC2_HOST,
-}
-
-// =============================================================================
-// State Management
-// =============================================================================
-
-// State singleton
-const state: BuddyState = {
-  repo: null,
-  conversationHistory: [],
-  currentDriver: 'claude-cli-local',
-  github: null,
-}
-
-// Ensure work directory exists
-if (!existsSync(CONFIG.workDir)) {
-  mkdirSync(CONFIG.workDir, { recursive: true })
-}
-
-// Buddy State Manager
-export interface BuddyStateManager {
-  getState: () => BuddyState
-  setRepo: (repo: RepoState | null) => void
-  setCurrentDriver: (driver: string) => void
-  setGitHub: (github: GitHubCredentials | null) => void
-  addToHistory: (message: AIMessage) => void
-  clearHistory: () => void
-}
-
-export const buddyState: BuddyStateManager = {
-  getState: (): BuddyState => state,
-  setRepo: (repo: RepoState | null): void => { state.repo = repo },
-  setCurrentDriver: (driver: string): void => { state.currentDriver = driver },
-  setGitHub: (github: GitHubCredentials | null): void => { state.github = github },
-  addToHistory: (message: AIMessage): void => { state.conversationHistory.push(message) },
-  clearHistory: (): void => { state.conversationHistory = [] },
-}
-
-// =============================================================================
-// AI Driver Utilities
-// =============================================================================
-
-/**
- * Build system prompt for AI with repository context
- */
-export function buildSystemPrompt(context: string): string {
-  return `You are Buddy, an AI code assistant that helps users modify codebases through voice commands.
-
-${state.repo
-    ? `You are working on the repository: ${state.repo.name}
-Branch: ${state.repo.branch}
-Path: ${state.repo.path}
-
-${context}`
-    : 'No repository is currently open.'}
-
-When the user gives you a command:
-1. Analyze what they want to do
-2. Identify the files that need to be modified
-3. Generate the exact code changes needed
-4. Respond with a structured format showing:
-  - Summary of changes
-  - Files to modify/create with full content
-  - Any additional notes
-
-Format file changes as:
-FILE: path/to/file.ts
-\`\`\`typescript
-// Full file content
-\`\`\`
-
-Be concise but thorough. The user will review and commit your changes.`
-}
-
-/**
- * Create a mock driver for testing
- */
-function createMockDriver(): AIDriver {
-  return {
-    name: 'Mock',
-    async process(command: string): Promise<string> {
-      await new Promise(resolve => setTimeout(resolve, 1000))
-
-      const lowerCommand = command.toLowerCase()
-
-      if (lowerCommand.includes('readme') || lowerCommand.includes('documentation')) {
-        return `I'll update the README.md file for you.
-
-Analyzing the repository structure...
-
-FILE: README.md
-\`\`\`markdown
-# Project Name
-
-## Installation
-
-\`\`\`bash
-npm install
-# or
-bun install
-\`\`\`
-
-## Usage
-
-\`\`\`bash
-npm run start
-\`\`\`
-\`\`\`
-
-File modified: README.md
-Lines added: 12`
-      }
-
-      if (lowerCommand.includes('fix') || lowerCommand.includes('bug')) {
-        return `I'll analyze and fix the issue.
-
-Scanning for potential bugs...
-
-FILE: src/utils.ts
-\`\`\`typescript
-export function getData(data: { value?: string }) {
-  return data?.value ?? 'default';
-}
-\`\`\`
-
-Files modified: src/utils.ts
-Lines changed: 4`
-      }
-
-      return `I understand you want to: "${command}"
-
-I'll analyze the repository and implement this change.
-
-FILE: src/main.ts
-\`\`\`typescript
-// Updated based on your request
-export function main() {
-  console.log('Changes applied');
-}
-\`\`\`
-
-Files modified: 1`
-    },
-  }
-}
-
-/**
- * Get driver instance by name
- */
-export function getDriver(driverName: string): AIDriver {
-  const currentState = buddyState.getState()
-
-  switch (driverName) {
-    case 'claude-cli-local':
-      return claudeAgent.createLocal({ cwd: currentState.repo?.path })
-
-    case 'claude-cli-ec2':
-      return claudeAgent.createEC2({
-        cwd: currentState.repo?.path,
-        ec2Host: apiKeys.claudeCliHost,
-      })
-
-    case 'claude':
-    case 'anthropic':
-      if (!apiKeys.anthropic) {
-        throw new Error('Anthropic API key not set. Configure your API key in settings.')
-      }
-      return createAnthropicDriver({ apiKey: apiKeys.anthropic })
-
-    case 'openai':
-      if (!apiKeys.openai) {
-        throw new Error('OpenAI API key not set. Configure your API key in settings.')
-      }
-      return createOpenAIDriver({ apiKey: apiKeys.openai })
-
-    case 'ollama':
-      return createOllamaDriver({
-        host: CONFIG.ollamaHost,
-        model: CONFIG.ollamaModel,
-      })
-
-    case 'claude-sdk':
-    case 'claude-agent-sdk':
-      return createClaudeAgentSDKDriver({
-        cwd: currentState.repo?.path,
-        maxTurns: 25,
-        permissionMode: 'bypassPermissions',
-      })
-
-    case 'mock':
-      return createMockDriver()
-
-    default:
-      throw new Error(`Unknown driver: ${driverName}. Available: claude-cli-local, claude-cli-ec2, claude, claude-sdk, openai, ollama, mock`)
-  }
-}
-
-/**
- * Get list of available AI drivers
- */
-export function getAvailableDrivers(): string[] {
-  return ['claude-cli-local', 'claude-cli-ec2', 'claude', 'claude-sdk', 'openai', 'ollama', 'mock']
-}
-
-// =============================================================================
-// Repository Operations
-// =============================================================================
-
-/**
- * Get repository structure for context
- */
-export async function getRepoContext(repoPath: string): Promise<string> {
-  const { $: _$ } = await import('bun')
-  const treeResult = await _$`cd ${repoPath} && find . -type f -not -path "*/node_modules/*" -not -path "*/.git/*" -not -name "*.lock" | head -50`.quiet()
-  const files = treeResult.text().trim()
-
-  let readme = ''
-  const readmePath = join(repoPath, 'README.md')
-  if (existsSync(readmePath)) {
-    readme = readFileSync(readmePath, 'utf-8').slice(0, 2000)
-  }
-
-  let packageJson = ''
-  const packagePath = join(repoPath, 'package.json')
-  if (existsSync(packagePath)) {
-    packageJson = readFileSync(packagePath, 'utf-8')
-  }
-
-  return `
-Repository Structure:
-${files}
-
-${readme ? `README.md (excerpt):\n${readme}\n` : ''}
-${packageJson ? `package.json:\n${packageJson}\n` : ''}
-`.trim()
-}
-
-/**
- * Clone or open a repository
- */
-export async function openRepository(input: string): Promise<RepoState> {
-  const { $: _$ } = await import('bun')
-  let repoPath: string
-  let repoName: string
-
-  if (input.includes('github.com') || input.startsWith('git@')) {
-    repoName = input.split('/').pop()?.replace('.git', '') || 'repo'
-    repoPath = join(CONFIG.workDir, repoName)
-
-    if (existsSync(repoPath)) {
-      // Repository already exists, pull latest
-      await _$`cd ${repoPath} && git pull --rebase`.quiet()
-    }
-    else {
-      // Clone the repository
-      await _$`git clone ${input} ${repoPath}`.quiet()
-    }
-  }
-  else {
-    repoPath = input.startsWith('~') ? input.replace('~', homedir()) : input
-
-    if (!existsSync(repoPath)) {
-      throw new Error(`Local path does not exist: ${repoPath}`)
-    }
-
-    if (!existsSync(join(repoPath, '.git'))) {
-      throw new Error(`Not a git repository: ${repoPath}`)
-    }
-
-    repoName = repoPath.split('/').pop() || 'repo'
-  }
-
-  const branchResult = await _$`cd ${repoPath} && git branch --show-current`.quiet()
-  const branch = branchResult.text().trim()
-
-  const statusResult = await _$`cd ${repoPath} && git status --porcelain`.quiet()
-  const hasChanges = statusResult.text().trim().length > 0
-
-  const lastCommitResult = await _$`cd ${repoPath} && git log -1 --format="%h %s"`.quiet()
-  const lastCommit = lastCommitResult.text().trim()
-
-  const repoState: RepoState = {
-    path: repoPath,
-    name: repoName,
-    branch,
-    hasChanges,
-    lastCommit,
-  }
-
-  buddyState.setRepo(repoState)
-  buddyState.clearHistory()
-  return repoState
-}
-
-/**
- * Apply file changes from AI response
- */
-export async function applyChanges(aiResponse: string): Promise<string[]> {
-  const currentState = buddyState.getState()
-
-  if (!currentState.repo) {
-    throw new Error('No repository opened')
-  }
-
-  const modifiedFiles: string[] = []
-  const filePattern = /FILE:\s*([^\n]+)\n```\w*\n([\s\S]*?)```/g
-
-  for (const match of aiResponse.matchAll(filePattern)) {
-    const filePath = match[1].trim()
-    const content = match[2]
-
-    const fullPath = join(currentState.repo.path, filePath)
-
-    const dir = dirname(fullPath)
-    if (!existsSync(dir)) {
-      mkdirSync(dir, { recursive: true })
-    }
-
-    writeFileSync(fullPath, content)
-    modifiedFiles.push(filePath)
-  }
-
-  if (modifiedFiles.length > 0 && currentState.repo) {
-    currentState.repo.hasChanges = true
-  }
-
-  return modifiedFiles
-}
-
-/**
- * Configure git user for commits
- */
-export async function configureGitUser(): Promise<void> {
-  const currentState = buddyState.getState()
-  if (!currentState.repo || !currentState.github) return
-
-  const { $: _$ } = await import('bun')
-  const { name, email } = currentState.github
-
-  await _$`cd ${currentState.repo.path} && git config user.name ${name}`.quiet()
-  await _$`cd ${currentState.repo.path} && git config user.email ${email}`.quiet()
-}
-
-/**
- * Stage and commit changes
- */
-export async function commitChanges(): Promise<string> {
-  const currentState = buddyState.getState()
-
-  if (!currentState.repo) {
-    throw new Error('No repository opened')
-  }
-
-  const { $: _$ } = await import('bun')
-
-  if (currentState.github) {
-    await configureGitUser()
-  }
-
-  await _$`cd ${currentState.repo.path} && git add -A`.quiet()
-  await _$`cd ${currentState.repo.path} && git commit -m ${CONFIG.commitMessage}`.quiet()
-
-  const hashResult = await _$`cd ${currentState.repo.path} && git rev-parse --short HEAD`.quiet()
-  const commitHash = hashResult.text().trim()
-
-  currentState.repo.hasChanges = false
-  currentState.repo.lastCommit = commitHash
-
-  return commitHash
-}
-
-/**
- * Push changes to remote
- */
-export async function pushChanges(): Promise<void> {
-  const currentState = buddyState.getState()
-
-  if (!currentState.repo) {
-    throw new Error('No repository opened')
-  }
-
-  const { $: _$ } = await import('bun')
-  await _$`cd ${currentState.repo.path} && git push`.quiet()
-}
-
-// =============================================================================
-// Command Processing
-// =============================================================================
-
-/**
- * Process command with selected AI driver
- */
-export async function processCommand(command: string, driverName?: string): Promise<string> {
-  const currentState = buddyState.getState()
-
-  if (!currentState.repo) {
-    throw new Error('No repository opened')
-  }
-
-  const normalizedDriver = driverName || currentState.currentDriver
-  const driver = getDriver(normalizedDriver)
-
-  if (driverName) {
-    buddyState.setCurrentDriver(driverName)
-  }
-
-  const context = await getRepoContext(currentState.repo.path)
-  const systemPrompt = buildSystemPrompt(context)
-  const response = await driver.process(command, systemPrompt, currentState.conversationHistory)
-
-  buddyState.addToHistory({ role: 'user', content: command })
-  buddyState.addToHistory({ role: 'assistant', content: response })
-
-  return response
-}
-
-/**
- * Process command with streaming output using Claude CLI
- */
-export async function buddyProcessStreaming(
-  command: string,
-  driverName?: string,
-  history?: Array<{role: string; content: string}>,
-): Promise<StreamingResult> {
-  const currentState = buddyState.getState()
-
-  if (!currentState.repo) {
-    throw new Error('No repository opened')
-  }
-
-  const normalizedDriver = driverName || currentState.currentDriver
-  const streamingDrivers = ['claude-cli-local', 'claude-sdk']
-  if (!streamingDrivers.includes(normalizedDriver)) {
-    throw new Error(`Streaming only supported for ${streamingDrivers.join(', ')} drivers. Current: ${normalizedDriver}`)
-  }
-
-  if (driverName) {
-    buddyState.setCurrentDriver(driverName)
-  }
-
-  // Build command with conversation history for context
-  let contextualCommand = command
-  if (history && history.length > 0) {
-    let conversationContext = '## Previous Conversation\nHere is our conversation so far:\n\n'
-    for (const msg of history) {
-      const role = msg.role === 'user' ? 'User' : 'Assistant'
-      conversationContext += `**${role}:** ${msg.content}\n\n`
-    }
-    conversationContext += '---\n\n## Current Request\n'
-    contextualCommand = conversationContext + command
-  }
-
-  const result = await claudeAgent.processStreaming(contextualCommand, currentState.repo.path)
-
-  // Update history when streaming completes
-  result.fullResponse.then((response) => {
-    buddyState.addToHistory({ role: 'user', content: command })
-    buddyState.addToHistory({ role: 'assistant', content: response })
-  })
-
-  return result
-}
-
-/**
- * Stream a simple Q&A response using the Anthropic API directly.
- * This provides true token-by-token streaming like ChatGPT/Claude web.
- * Use this for questions/explanations that don't require agentic tool use.
- */
-export async function buddyStreamSimple(
-  command: string,
-  history?: Array<{ role: string; content: string }>,
-): Promise<StreamingResult> {
-
-  if (!apiKeys.anthropic) {
-    throw new Error('Anthropic API key not set. Configure your API key in settings.')
-  }
-
-  // Build conversation messages
-  const messages: AIMessage[] = []
-  if (history && history.length > 0) {
-    for (const msg of history) {
-      messages.push({
-        role: msg.role as 'user' | 'assistant',
-        content: msg.content,
-      })
-    }
-  }
-  messages.push({ role: 'user', content: command })
-
-  // For simple Q&A mode, use a generic helpful assistant prompt
-  // Don't include repo context - this keeps answers general and not code-focused
-  const systemPrompt = `You are a helpful AI assistant. Answer questions naturally and conversationally.
-You can discuss any topic - technology, science, philosophy, everyday questions, or anything else the user asks about.
-Be concise but thorough. If the user asks about coding or their project specifically, help with that too.`
-
-  const encoder = new TextEncoder()
-  let fullResponse = ''
-  let resolveFullResponse: (value: string) => void
-  const fullResponsePromise = new Promise<string>((resolve) => {
-    resolveFullResponse = resolve
-  })
-
-  // Create streaming response using Anthropic API
-  const stream = new ReadableStream<Uint8Array>({
-    async start(controller) {
-      try {
-        const response = await fetch('https://api.anthropic.com/v1/messages', {
-          method: 'POST',
-          headers: {
-            'Content-Type': 'application/json',
-            'x-api-key': apiKeys.anthropic!,
-            'anthropic-version': '2023-06-01',
-          },
-          body: JSON.stringify({
-            model: 'claude-sonnet-4-20250514',
-            max_tokens: 4096,
-            system: systemPrompt,
-            stream: true,
-            messages,
-          }),
-        })
-
-        if (!response.ok) {
-          const error = await response.text()
-          throw new Error(`Claude API error: ${error}`)
-        }
-
-        const reader = response.body?.getReader()
-        if (!reader) throw new Error('No response body')
-
-        const decoder = new TextDecoder()
-        let buffer = ''
-
-        while (true) {
-          const { done, value } = await reader.read()
-          if (done) break
-
-          buffer += decoder.decode(value, { stream: true })
-          const lines = buffer.split('\n')
-          buffer = lines.pop() || ''
-
-          for (const line of lines) {
-            if (line.startsWith('data: ')) {
-              const data = line.slice(6)
-              if (data === '[DONE]') continue
-
-              try {
-                const event = JSON.parse(data) as { type?: string; delta?: { text?: string } }
-                if (event.type === 'content_block_delta' && event.delta?.text) {
-                  const text = event.delta.text
-                  fullResponse += text
-                  controller.enqueue(encoder.encode(text))
-                }
-              }
-              catch {
-                // Skip invalid JSON
-              }
-            }
-          }
-        }
-
-        // Update history and resolve the full response
-        buddyState.addToHistory({ role: 'user', content: command })
-        buddyState.addToHistory({ role: 'assistant', content: fullResponse })
-        resolveFullResponse(fullResponse)
-        controller.close()
-      }
-      catch (error) {
-        resolveFullResponse(fullResponse) // Resolve with what we have
-        controller.error(error)
-      }
-    },
-  })
-
-  return {
-    stream,
-    fullResponse: fullResponsePromise,
-  }
-}