ultra-dex 2.2.0 → 3.1.0

package/lib/commands/memory.js

@@ -0,0 +1,80 @@
+/**
+ * ultra-dex memory command
+ * Manage persistent memory for AI agents
+ */
+
+import chalk from 'chalk';
+import { ultraMemory } from '../mcp/memory.js';
+
+export function registerMemoryCommand(program) {
+  const memory = program
+    .command('memory')
+    .description('Manage persistent agent memory');
+
+  memory
+    .command('list')
+    .description('List all remembered facts')
+    .option('--json', 'Output as JSON')
+    .action(async (options) => {
+      const items = await ultraMemory.getAll();
+      
+      if (options.json) {
+        console.log(JSON.stringify(items, null, 2));
+        return;
+      }
+
+      console.log(chalk.cyan.bold('\n🧠 Ultra-Dex Persistent Memory\n'));
+      
+      if (items.length === 0) {
+        console.log(chalk.gray('  Memory is empty.'));
+        return;
+      }
+
+      items.forEach((item, i) => {
+        console.log(chalk.white(`${i + 1}. [${new Date(item.timestamp).toLocaleDateString()}] (${item.source})`));
+        console.log(chalk.gray(`   ${item.text}`));
+        if (item.tags && item.tags.length > 0) {
+          console.log(chalk.blue(`   Tags: ${item.tags.join(', ')}`));
+        }
+        console.log();
+      });
+    });
+
+  memory
+    .command('add <text>')
+    .description('Add a fact to memory')
+    .option('-t, --tags <tags>', 'Comma-separated tags')
+    .action(async (text, options) => {
+      const tags = options.tags ? options.tags.split(',').map(t => t.trim()) : [];
+      await ultraMemory.remember(text, tags, 'manual');
+      console.log(chalk.green('✅ Fact remembered.'));
+    });
+
+  memory
+    .command('search <query>')
+    .description('Search memory')
+    .action(async (query) => {
+      const results = await ultraMemory.search(query);
+      console.log(chalk.cyan.bold(`\n🔍 Search Results for "${query}":\n`));
+      
+      if (results.length === 0) {
+        console.log(chalk.gray('  No matches found.'));
+        return;
+      }
+
+      results.forEach((item, i) => {
+        console.log(chalk.white(`${i + 1}. [${new Date(item.timestamp).toLocaleDateString()}]`));
+        console.log(chalk.gray(`   ${item.text}`));
+        console.log();
+      });
+    });
+
+  memory
+    .command('clear')
+    .description('Clear all memory')
+    .option('--before <date>', 'Clear before date (ISO)')
+    .action(async (options) => {
+      await ultraMemory.clear(options.before);
+      console.log(chalk.green('✅ Memory cleared.'));
+    });
+}

package/lib/commands/plan.js

@@ -0,0 +1,82 @@
+import fs from 'fs/promises';
+import path from 'path';
+import chalk from 'chalk';
+
+export async function loadState() {
+  try {
+    const content = await fs.readFile(path.resolve(process.cwd(), '.ultra/state.json'), 'utf8');
+    return JSON.parse(content);
+  } catch (error) {
+    console.error(chalk.red('Failed to load .ultra/state.json. Is the project initialized?'));
+    return null;
+  }
+}
+
+export async function saveState(state) {
+  try {
+    await fs.mkdir(path.resolve(process.cwd(), '.ultra'), { recursive: true });
+    await fs.writeFile(
+      path.resolve(process.cwd(), '.ultra/state.json'), 
+      JSON.stringify(state, null, 2)
+    );
+    return true;
+  } catch (error) {
+    console.error(chalk.red('Failed to save state:'), error);
+    return false;
+  }
+}
+
+export function generateMarkdown(state) {
+  const { project, phases } = state;
+  let md = `# ${project.name} - Implementation Plan\n\n`;
+  md += `> **Generated by Ultra-Dex Core**\n`;
+  md += `> Version: ${project.version}\n`;
+  md += `> Mode: ${project.mode}\n\n`;
+
+  md += `## 🚀 Execution Phases\n\n`;
+
+  phases.forEach(phase => {
+    const statusIcon = phase.status === 'completed' ? '✅' : phase.status === 'in_progress' ? '🔄' : '⏳';
+    md += `### ${statusIcon} ${phase.name}\n\n`;
+    
+    if (phase.steps && phase.steps.length > 0) {
+      phase.steps.forEach(step => {
+        const stepIcon = step.status === 'completed' ? '- [x]' : '- [ ]';
+        md += `${stepIcon} **${step.id}**: ${step.task}\n`;
+      });
+    } else {
+      md += `_No steps defined for this phase._\n`;
+    }
+    md += `\n`;
+  });
+
+  md += `## 🤖 Agent Registry\n\n`;
+  if (state.agents && state.agents.registry) {
+    state.agents.registry.forEach(agent => {
+      const active = state.agents.active.includes(agent) ? '(Active)' : '';
+      md += `- ${agent} ${active}\n`;
+    });
+  }
+
+  md += `\n---\n`;
+  md += `*This file is strictly read-only. Edit .ultra/state.json to update.*`;
+
+  return md;
+}
+
+export function registerPlanCommand(program) {
+  program
+    .command('plan')
+    .description('Generate IMPLEMENTATION-PLAN.md from state.json')
+    .action(async () => {
+      const state = await loadState();
+      if (!state) return;
+
+      console.log(chalk.blue(`Generating plan for ${state.project.name}...`));
+      
+      const markdown = generateMarkdown(state);
+      await fs.writeFile(path.resolve(process.cwd(), 'IMPLEMENTATION-PLAN.md'), markdown);
+      
+      console.log(chalk.green(`✅ IMPLEMENTATION-PLAN.md generated successfully.`));
+    });
+}

package/lib/commands/review.js

@@ -10,6 +10,8 @@ import fs from 'fs/promises';
 import path from 'path';
 import { createProvider, getDefaultProvider, checkConfiguredProviders } from '../providers/index.js';
 import { SYSTEM_PROMPT, generateReviewPrompt } from '../templates/prompts/review-code.js';
+import { validateSafePath } from '../utils/validation.js';
+import { buildGraph, queryGraph, getImpactAnalysis } from '../utils/graph.js'; // Import CPG utils
 
 // File patterns to scan
 const CODE_PATTERNS = {
@@ -106,6 +108,12 @@ export function registerReviewCommand(program) {
     .action(async (options) => {
       console.log(chalk.cyan('\n🔍 Ultra-Dex Code Review\n'));
 
+      const dirValidation = validateSafePath(options.dir, 'Review directory');
+      if (dirValidation !== true) {
+        console.log(chalk.red(dirValidation));
+        process.exit(1);
+      }
+
       const reviewDir = path.resolve(options.dir);
 
       // Check for plan
@@ -120,11 +128,29 @@ export function registerReviewCommand(program) {
         return;
       }
 
-      // Get directory structure
-      const spinner = ora('Scanning codebase...').start();
+      // Get directory structure & Build Graph
+      const spinner = ora('Scanning codebase & Building Graph...').start();
       const structure = await getDirectoryStructure(reviewDir);
       const keyFiles = await findKeyFiles(reviewDir);
-      spinner.succeed('Codebase scanned');
+      
+      // GOD MODE: Build CPG
+      let graphSummary = "Graph Not Available";
+      try {
+        const graph = await buildGraph();
+        graphSummary = `
+Code Property Graph Stats:
+- Files: ${graph.nodes.filter(n => n.type === 'file').length}
+- Functions: ${graph.nodes.filter(n => n.type === 'function').length}
+- Dependencies (Edges): ${graph.edges.length}
+
+Top Dependencies:
+${graph.edges.slice(0, 10).map(e => `- ${e.source} -> ${e.target}`).join('\n')}
+        `;
+      } catch (e) {
+        // Fallback if graph fails
+      }
+
+      spinner.succeed('Codebase scanned & Graph built');
 
       if (options.quick) {
         // Quick review - just check structure
@@ -181,13 +207,16 @@ export function registerReviewCommand(program) {
 
       // Build file summary
       const filesSummary = keyFiles.map(f => `### ${f.path}\n\`\`\`\n${f.content}\n\`\`\``).join('\n\n');
+      
+      // Inject Graph Summary into context
+      const contextWithGraph = `${structure}\n\n## ARCHITECTURAL GRAPH (TRUTH)\n${graphSummary}`;
 
-      spinner.start('Analyzing code against plan...');
+      spinner.start('Analyzing code & graph against plan...');
 
       try {
         const result = await provider.generate(
           SYSTEM_PROMPT,
-          generateReviewPrompt(plan.slice(0, 15000), structure, filesSummary)
+          generateReviewPrompt(plan.slice(0, 15000), contextWithGraph, filesSummary)
         );
 
         spinner.succeed('Analysis complete');

package/lib/commands/run.js

@@ -0,0 +1,233 @@
+/**
+ * ultra-dex run command
+ * Execute agent tasks automatically (the "swarm" approach)
+ */
+
+import chalk from 'chalk';
+import ora from 'ora';
+import inquirer from 'inquirer';
+import fs from 'fs/promises';
+import path from 'path';
+import { createProvider, getDefaultProvider, checkConfiguredProviders } from '../providers/index.js';
+import { validateSafePath } from '../utils/validation.js';
+import { projectGraph } from '../mcp/graph.js';
+
+const AGENTS = {
+  planner: {
+    name: '@Planner',
+    role: 'Task Breakdown Specialist',
+    systemPrompt: `You are @Planner. Break down features into atomic tasks.
+Use >> SEARCH_CODE: "query" to find existing patterns.
+Output format:
+## Task Breakdown
+### Task 1: [Name]
+- Agent: @Backend | @Frontend | ...
+- Description: ...
+`,
+  },
+  cto: {
+    name: '@CTO',
+    role: 'Technical Architecture Lead',
+    systemPrompt: `You are @CTO. Make tech decisions, design architecture, set standards.
+Use >> READ_CODE: "path/to/file" to review existing architecture.`,
+  },
+  backend: {
+    name: '@Backend',
+    role: 'API & Business Logic Developer',
+    systemPrompt: `You are @Backend. Write API/Service code.
+Available commands:
+>> READ_CODE: "path" - Read a file
+>> WRITE_CODE: "path" "content" - Create/Update a file
+>> SEARCH_CODE: "query" - Search codebase
+>> DELEGATE: @AgentName "Task" - Delegate work`,
+  },
+  frontend: {
+    name: '@Frontend',
+    role: 'UI/UX Developer',
+    systemPrompt: `You are @Frontend. Build React/Next.js components.
+Available commands:
+>> READ_CODE: "path"
+>> WRITE_CODE: "path" "content"
+>> SEARCH_CODE: "query"`,
+  },
+  database: {
+    name: '@Database',
+    role: 'Database Architect',
+    systemPrompt: `You are @Database. Design schemas.
+Available commands:
+>> READ_CODE: "path"
+>> WRITE_CODE: "path" "content"`,
+  },
+  testing: {
+    name: '@Testing',
+    role: 'QA Engineer',
+    systemPrompt: `You are @Testing. Write tests.
+Available commands:
+>> READ_CODE: "path"
+>> WRITE_CODE: "path" "content"`,
+  },
+  reviewer: {
+    name: '@Reviewer',
+    role: 'Code Review Specialist',
+    systemPrompt: `You are @Reviewer. Audit code.
+Available commands:
+>> READ_CODE: "path"
+>> SEARCH_CODE: "query"`,
+  },
+};
+
+async function readProjectContext() {
+  const context = {};
+  try { context.plan = await fs.readFile('IMPLEMENTATION-PLAN.md', 'utf8'); } catch { context.plan = null; }
+  try { context.context = await fs.readFile('CONTEXT.md', 'utf8'); } catch { context.context = null; }
+  try { 
+    context.state = JSON.parse(await fs.readFile('.ultra/state.json', 'utf8')); 
+  } catch { 
+    try {
+        context.state = JSON.parse(await fs.readFile('.ultra-dex/state.json', 'utf8'));
+    } catch {
+        context.state = null; 
+    }
+  }
+  
+  // Graph Scan (God Mode)
+  try {
+    await projectGraph.scan();
+    context.graph = projectGraph.getSummary();
+  } catch (e) {
+    context.graph = null;
+  }
+  
+  return context;
+}
+
+export async function runAgentLoop(agentName, task, provider, projectContext, depth = 0) {
+  if (depth > 5) return `[System]: Max delegation depth reached.`;
+
+  const agent = AGENTS[agentName.toLowerCase()];
+  if (!agent) return `[System]: Unknown agent @${agentName}`;
+
+  const spinner = ora(`${agent.name} is working...`).start();
+  
+  const graphInfo = projectContext.graph 
+    ? `## Codebase Graph\n- Files: ${projectContext.graph.nodeCount}\n- Dependencies: ${projectContext.graph.edgeCount}\n` 
+    : '';
+
+  const historySection = projectContext.history ? `## Execution History\n${projectContext.history}\n\n` : '';
+  const contextSection = projectContext.context ? `## Context\n${projectContext.context.slice(0, 3000)}\n\n${graphInfo}${historySection}` : '';
+  const prompt = `${contextSection}## Task\n${task}\n\nYou can use tools by outputting:
+>> READ_CODE: "filePath"
+>> WRITE_CODE: "filePath" "fullContent"
+>> SEARCH_CODE: "query"
+>> DELEGATE: @AgentName "Task"`;
+
+  try {
+    const result = await provider.generate(agent.systemPrompt, prompt);
+    spinner.succeed(`${agent.name} completed.`);
+    
+    let content = result.content;
+
+    // Tool Execution Logic (God Mode)
+    const readMatch = content.match(/>>\s*READ_CODE:\s*["'](.+?)["']/);
+    const writeMatch = content.match(/>>\s*WRITE_CODE:\s*["'](.+?)["']\s*["']([\s\S]+?)["']/);
+    const searchMatch = content.match(/>>\s*SEARCH_CODE:\s*["'](.+?)["']/);
+    const delegateMatch = content.match(/>>\s*DELEGATE:\s*@(\w+)\s*["'](.+?)["']/);
+
+    if (readMatch) {
+      const filePath = readMatch[1];
+      console.log(chalk.cyan(`\n🔍 ${agent.name} is reading ${filePath}...`));
+      try {
+        const fileContent = await fs.readFile(path.resolve(process.cwd(), filePath), 'utf8');
+        const nextPrompt = `Output of READ_CODE "${filePath}":\n\`\`\`\n${fileContent}\n\`\`\`\n\nPlease proceed with your task.`;
+        return await runAgentLoop(agentName, `${task}\n\n${nextPrompt}`, provider, projectContext, depth + 1);
+      } catch (e) {
+        return await runAgentLoop(agentName, `${task}\n\nError reading ${filePath}: ${e.message}`, provider, projectContext, depth + 1);
+      }
+    }
+
+    if (writeMatch) {
+      const filePath = writeMatch[1];
+      const newContent = writeMatch[2];
+      console.log(chalk.green(`\n💾 ${agent.name} is writing to ${filePath}...`));
+      try {
+        await fs.mkdir(path.dirname(path.resolve(process.cwd(), filePath)), { recursive: true });
+        await fs.writeFile(path.resolve(process.cwd(), filePath), newContent, 'utf8');
+        const nextPrompt = `Successfully wrote ${filePath}. Please proceed or delegate verification.`;
+        return await runAgentLoop(agentName, `${task}\n\n${nextPrompt}`, provider, projectContext, depth + 1);
+      } catch (e) {
+        return await runAgentLoop(agentName, `${task}\n\nError writing ${filePath}: ${e.message}`, provider, projectContext, depth + 1);
+      }
+    }
+
+    if (delegateMatch) {
+      const nextAgent = delegateMatch[1];
+      const nextTask = delegateMatch[2];
+      console.log(chalk.cyan(`\n↪️  ${agent.name} is delegating to @${nextAgent}: "${nextTask}"`));
+      const subResult = await runAgentLoop(nextAgent, nextTask, provider, projectContext, depth + 1);
+      return `${content}\n\n---\n\n## Delegated Result from @${nextAgent}\n${subResult}`;
+    }
+
+    return content;
+  } catch (err) {
+    spinner.fail(`${agent.name} failed: ${err.message}`);
+    return `[Error]: ${err.message}`;
+  }
+}
+
+export function registerRunCommand(program) {
+  program.command('run <agent>')
+    .description('Execute an agent task automatically')
+    .option('-t, --task <task>', 'Task to execute')
+    .option('-p, --provider <provider>', 'AI provider')
+    .option('-k, --key <apiKey>', 'API key')
+    .option('-o, --output <file>', 'Output file')
+    .action(async (agentName, options) => {
+      const configured = checkConfiguredProviders();
+      const hasProvider = configured.some(p => p.configured) || options.key;
+
+      if (!hasProvider) {
+        console.log(chalk.yellow('⚠️  No AI provider configured.'));
+        return;
+      }
+
+      let task = options.task;
+      if (!task) {
+        const { taskInput } = await inquirer.prompt([{ 
+          type: 'input', name: 'taskInput', message: `Task for ${agentName}?`
+        }]);
+        task = taskInput;
+      }
+
+      const context = await readProjectContext();
+      const providerId = options.provider || getDefaultProvider();
+      const provider = createProvider(providerId, { apiKey: options.key, maxTokens: 8000 });
+
+      const finalOutput = await runAgentLoop(agentName, task, provider, context);
+      
+      if (options.output) {
+        await fs.writeFile(options.output, finalOutput);
+        console.log(chalk.green(`\n✅ Saved to ${options.output}`));
+      }
+    });
+}
+
+export function registerSwarmCommand(program) {
+  program.command('swarm <feature>')
+    .description('Run a full agent swarm for a feature')
+    .option('-p, --provider <provider>', 'AI provider')
+    .option('-k, --key <apiKey>', 'API key')
+    .action(async (feature, options) => {
+      console.log(chalk.cyan('\n🐝 Ultra-Dex Agent Swarm\n'));
+      const context = await readProjectContext();
+      const providerId = options.provider || getDefaultProvider();
+      const provider = createProvider(providerId, { apiKey: options.key, maxTokens: 8000 });
+
+      console.log(chalk.bold('Step 1: 📋 @Planner breaking down feature...'));
+      const plan = await runAgentLoop('planner', feature, provider, context);
+      
+      console.log(chalk.bold('\nStep 2: 🏗️  @CTO reviewing architecture...'));
+      await runAgentLoop('cto', `Review plan:\n${plan}`, provider, context);
+    });
+}
+
+export default { registerRunCommand, registerSwarmCommand };