@xagent-ai/cli 1.2.2 → 1.3.0

package/src/memory.ts

@@ -1,344 +1,344 @@
-import fs from 'fs/promises';
-import fsSync from 'fs';
-import path from 'path';
-import os from 'os';
-import { glob } from 'glob';
-import crypto from 'crypto';
-
-export interface MemoryFile {
-  path: string;
-  content: string;
-  level: 'global' | 'project' | 'subdirectory';
-  projectRoot?: string;
-}
-
-export class MemoryManager {
-  private memoriesDir: string;
-  private globalMemoryPath: string;
-  private projectMemoryPath: string;
-  private projectRoot: string | null = null;
-  private memoryFiles: MemoryFile[] = [];
-  private contextFileNames: string | string[];
-
-  constructor(projectRoot?: string, contextFileName: string | string[] = 'XAGENT.md') {
-    this.memoriesDir = path.join(os.homedir(), '.xagent', 'memories');
-    this.globalMemoryPath = path.join(this.memoriesDir, 'global.md');
-
-    if (projectRoot) {
-      this.projectRoot = path.resolve(projectRoot);
-      this.projectMemoryPath = this.getProjectMemoryPath(this.projectRoot);
-    } else {
-      this.projectMemoryPath = '';
-    }
-    this.contextFileNames = contextFileName;
-  }
-
-  private getProjectMemoryPath(projectRoot: string): string {
-    const hash = crypto.createHash('md5').update(projectRoot).digest('hex').substring(0, 16);
-    const sanitizedName = projectRoot.replace(/[:\\\/]/g, '_').replace(/[^a-zA-Z0-9_\-]/g, '');
-    const name = sanitizedName.length > 50 ? sanitizedName.substring(0, 50) : sanitizedName;
-    return path.join(this.memoriesDir, `project_${name}_${hash}.md`);
-  }
-
-  async loadMemory(): Promise<string> {
-    this.memoryFiles = [];
-    let combinedMemory = '';
-
-    const globalMemory = await this.loadMemoryFile(this.globalMemoryPath, 'global');
-    if (globalMemory) {
-      this.memoryFiles.push(globalMemory);
-      combinedMemory += globalMemory.content + '\n\n';
-    } else {
-      // Global memory not found, create default content
-      const defaultContent = '# Global Context\n\nGlobal preferences and settings will be added here.';
-      await this.saveMemory(defaultContent, 'global');
-      const newMemory = await this.loadMemoryFile(this.globalMemoryPath, 'global');
-      if (newMemory) {
-        this.memoryFiles.push(newMemory);
-        combinedMemory += newMemory.content + '\n\n';
-      }
-    }
-
-    if (this.projectMemoryPath) {
-      const projectMemory = await this.loadMemoryFile(this.projectMemoryPath, 'project');
-      if (projectMemory) {
-        this.memoryFiles.push(projectMemory);
-        combinedMemory += projectMemory.content + '\n\n';
-      }
-      // else {
-      //   // Project memory not found, create default content
-      //   const defaultContent = '# Project Context\n\nProject-specific context will be added here.';
-      //   await this.saveMemory(defaultContent, 'project');
-      //   const newMemory = await this.loadMemoryFile(this.projectMemoryPath, 'project');
-      //   if (newMemory) {
-      //     this.memoryFiles.push(newMemory);
-      //     combinedMemory += newMemory.content + '\n\n';
-      //   }
-      // }
-    }
-
-    return this.processImports(combinedMemory);
-  }
-
-  private async loadMemoryFile(filePath: string, level: 'global' | 'project' | 'subdirectory'): Promise<MemoryFile | null> {
-    try {
-      const content = await fs.readFile(filePath, 'utf-8');
-      return { path: filePath, content, level };
-    } catch (error) {
-      if ((error as NodeJS.ErrnoException).code !== 'ENOENT') {
-        console.error(`Failed to load memory file ${filePath}:`, error);
-      }
-      return null;
-    }
-  }
-
-  // private async loadSubdirectoryMemories(): Promise<MemoryFile[]> {
-  //   const memories: MemoryFile[] = [];
-    
-  //   if (!this.projectMemoryPath) {
-  //     return memories;
-  //   }
-
-  //   const projectRoot = path.dirname(this.projectMemoryPath);
-    
-  //   try {
-  //     const files = await glob('**/XAGENT.md', {
-  //       cwd: projectRoot,
-  //       ignore: ['node_modules/**', '.git/**', 'dist/**', 'build/**']
-  //     });
-
-  //     for (const file of files) {
-  //       if (file !== 'XAGENT.md') {
-  //         const filePath = path.join(projectRoot, file);
-  //         const memory = await this.loadMemoryFile(filePath, 'subdirectory');
-  //         if (memory) {
-  //           memories.push(memory);
-  //         }
-  //       }
-  //     }
-  //   } catch (error) {
-  //     console.error('Failed to load subdirectory memories:', error);
-  //   }
-
-  //   return memories;
-  // }
-
-  private processImports(content: string): string {
-    // Only match actual import statements: import ... from '@xxx' or import '@xxx'
-    const importRegex = /import\s+(?:(?:\{[^}]*\}|\* as \w+|\w+)\s+from\s+)?['"](@[^'"]+)['"]/g;
-    let processedContent = content;
-    const visitedFiles = new Set<string>();
-
-    const processImport = (match: string, importPath: string): string => {
-      const absolutePath = path.isAbsolute(importPath)
-        ? importPath
-        : path.resolve(process.cwd(), importPath);
-
-      if (visitedFiles.has(absolutePath)) {
-        return '';
-      }
-
-      visitedFiles.add(absolutePath);
-
-      try {
-        const importedContent = fsSync.readFileSync(absolutePath, 'utf-8');
-        return `\n\n${importedContent}\n\n`;
-      } catch (error) {
-        console.warn(`Failed to import ${importPath}:`, error);
-        return '';
-      }
-    };
-
-    let maxIterations = 10;
-    let iterations = 0;
-
-    while (importRegex.test(processedContent) && iterations < maxIterations) {
-      processedContent = processedContent.replace(importRegex, processImport);
-      iterations++;
-    }
-
-    return processedContent;
-  }
-
-  async saveMemory(content: string, scope: 'global' | 'project' = 'global'): Promise<void> {
-    const filePath = scope === 'global' ? this.globalMemoryPath : this.projectMemoryPath;
-
-    if (!filePath) {
-      throw new Error('Project memory path not set');
-    }
-
-    const dir = path.dirname(filePath);
-    await fs.mkdir(dir, { recursive: true });
-
-    await fs.writeFile(filePath, content, 'utf-8');
-
-    if (scope === 'global') {
-      const existingMemory = this.memoryFiles.find(m => m.level === 'global');
-      if (existingMemory) {
-        existingMemory.content = content;
-      } else {
-        this.memoryFiles.push({ path: filePath, content, level: 'global' });
-      }
-    } else {
-      const existingMemory = this.memoryFiles.find(m => m.level === 'project');
-      if (existingMemory) {
-        existingMemory.content = content;
-      } else {
-        this.memoryFiles.push({ path: filePath, content, level: 'project', projectRoot: this.projectRoot || undefined });
-      }
-    }
-  }
-
-  async addMemoryEntry(entry: string, scope: 'global' | 'project' = 'global'): Promise<void> {
-    const filePath = scope === 'global' ? this.globalMemoryPath : this.projectMemoryPath;
-
-    if (!filePath) {
-      throw new Error('Project memory path not set');
-    }
-
-    let existingContent = '';
-    try {
-      existingContent = await fs.readFile(filePath, 'utf-8');
-    } catch (error) {
-      if ((error as NodeJS.ErrnoException).code !== 'ENOENT') {
-        throw error;
-      }
-    }
-
-    const newContent = existingContent + `\n\n## Memory Entry (${new Date().toISOString()})\n\n${entry}\n`;
-    await this.saveMemory(newContent, scope);
-  }
-
-  getMemoryFiles(): MemoryFile[] {
-    return [...this.memoryFiles];
-  }
-
-  getMemoriesDir(): string {
-    return this.memoriesDir;
-  }
-
-  async initializeProject(projectRoot: string): Promise<void> {
-    this.projectRoot = path.resolve(projectRoot);
-    this.projectMemoryPath = this.getProjectMemoryPath(this.projectRoot);
-
-    const existingMemory = await this.loadMemoryFile(this.projectMemoryPath, 'project');
-    if (existingMemory) {
-      console.log('XAGENT.md already exists. Skipping initialization.');
-      return;
-    }
-
-    console.log('Creating XAGENT.md...');
-    await this.saveMemory('# Project Context\n\nProject-specific context will be added here.', 'project');
-    
-    console.log('Analyzing project structure...');
-    const analysis = await this.analyzeProject(projectRoot);
-    
-    console.log('Generating project-specific context...');
-    await this.saveMemory(analysis, 'project');
-    
-    console.log('✅ XAGENT.md has been successfully populated with project-specific information.');
-  }
-
-  private async analyzeProject(projectRoot: string): Promise<string> {
-    let analysis = '# Project Context\n\n';
-
-    try {
-      const packageJsonPath = path.join(projectRoot, 'package.json');
-      const packageJson = await this.loadJsonFile(packageJsonPath);
-      
-      if (packageJson) {
-        analysis += '## Project Overview\n\n';
-        analysis += `Name: ${packageJson.name || 'Unknown'}\n`;
-        analysis += `Version: ${packageJson.version || 'Unknown'}\n`;
-        analysis += `Description: ${packageJson.description || 'No description'}\n\n`;
-
-        if (packageJson.dependencies) {
-          analysis += '## Dependencies\n\n';
-          analysis += '```json\n' + JSON.stringify(packageJson.dependencies, null, 2) + '\n```\n\n';
-        }
-
-        if (packageJson.scripts) {
-          analysis += '## Scripts\n\n';
-          Object.entries(packageJson.scripts).forEach(([name, script]) => {
-            analysis += `- ${name}: ${script}\n`;
-          });
-          analysis += '\n';
-        }
-      }
-
-      const tsconfigPath = path.join(projectRoot, 'tsconfig.json');
-      const tsconfig = await this.loadJsonFile(tsconfigPath);
-      
-      if (tsconfig) {
-        analysis += '## TypeScript Configuration\n\n';
-        analysis += '```json\n' + JSON.stringify(tsconfig, null, 2) + '\n```\n\n';
-      }
-
-      analysis += '## Project Structure\n\n';
-      analysis += '```\n';
-      const structure = await this.getProjectStructure(projectRoot);
-      analysis += structure;
-      analysis += '```\n\n';
-
-      analysis += '## Development Guidelines\n\n';
-      analysis += '- Follow existing code style and conventions\n';
-      analysis += '- Write clear, descriptive commit messages\n';
-      analysis += '- Add tests for new features\n';
-      analysis += '- Update documentation as needed\n';
-    } catch (error) {
-      console.error('Failed to analyze project:', error);
-      analysis += '## Project Analysis Failed\n\n';
-      analysis += 'Unable to analyze project structure automatically.\n';
-    }
-
-    return analysis;
-  }
-
-  private async loadJsonFile(filePath: string): Promise<any> {
-    try {
-      const content = await fs.readFile(filePath, 'utf-8');
-      return JSON.parse(content);
-    } catch (error) {
-      return null;
-    }
-  }
-
-  private async getProjectStructure(rootPath: string, maxDepth: number = 3, currentDepth: number = 0): Promise<string> {
-    if (currentDepth >= maxDepth) {
-      return '';
-    }
-
-    let structure = '';
-    
-    try {
-      const entries = await fs.readdir(rootPath, { withFileTypes: true });
-      
-      for (const entry of entries) {
-        if (entry.name.startsWith('.') || entry.name === 'node_modules' || entry.name === 'dist') {
-          continue;
-        }
-
-        if (entry.isDirectory()) {
-          structure += '  '.repeat(currentDepth) + entry.name + '/\n';
-          const subPath = path.join(rootPath, entry.name);
-          structure += await this.getProjectStructure(subPath, maxDepth, currentDepth + 1);
-        } else {
-          structure += '  '.repeat(currentDepth) + entry.name + '\n';
-        }
-      }
-    } catch (error) {
-      console.error(`Failed to read directory ${rootPath}:`, error);
-    }
-
-    return structure;
-  }
-}
-
-let memoryManagerInstance: MemoryManager | null = null;
-
-export function getMemoryManager(projectRoot?: string, contextFileName?: string | string[]): MemoryManager {
-  if (!memoryManagerInstance || projectRoot) {
-    memoryManagerInstance = new MemoryManager(projectRoot, contextFileName);
-  }
-  return memoryManagerInstance;
-}
+import fs from 'fs/promises';
+import fsSync from 'fs';
+import path from 'path';
+import os from 'os';
+import { glob } from 'glob';
+import crypto from 'crypto';
+
+export interface MemoryFile {
+  path: string;
+  content: string;
+  level: 'global' | 'project' | 'subdirectory';
+  projectRoot?: string;
+}
+
+export class MemoryManager {
+  private memoriesDir: string;
+  private globalMemoryPath: string;
+  private projectMemoryPath: string;
+  private projectRoot: string | null = null;
+  private memoryFiles: MemoryFile[] = [];
+  private contextFileNames: string | string[];
+
+  constructor(projectRoot?: string, contextFileName: string | string[] = 'XAGENT.md') {
+    this.memoriesDir = path.join(os.homedir(), '.xagent', 'memories');
+    this.globalMemoryPath = path.join(this.memoriesDir, 'global.md');
+
+    if (projectRoot) {
+      this.projectRoot = path.resolve(projectRoot);
+      this.projectMemoryPath = this.getProjectMemoryPath(this.projectRoot);
+    } else {
+      this.projectMemoryPath = '';
+    }
+    this.contextFileNames = contextFileName;
+  }
+
+  private getProjectMemoryPath(projectRoot: string): string {
+    const hash = crypto.createHash('md5').update(projectRoot).digest('hex').substring(0, 16);
+    const sanitizedName = projectRoot.replace(/[:\\\/]/g, '_').replace(/[^a-zA-Z0-9_\-]/g, '');
+    const name = sanitizedName.length > 50 ? sanitizedName.substring(0, 50) : sanitizedName;
+    return path.join(this.memoriesDir, `project_${name}_${hash}.md`);
+  }
+
+  async loadMemory(): Promise<string> {
+    this.memoryFiles = [];
+    let combinedMemory = '';
+
+    const globalMemory = await this.loadMemoryFile(this.globalMemoryPath, 'global');
+    if (globalMemory) {
+      this.memoryFiles.push(globalMemory);
+      combinedMemory += globalMemory.content + '\n\n';
+    } else {
+      // Global memory not found, create default content
+      const defaultContent = '# Global Context\n\nGlobal preferences and settings will be added here.';
+      await this.saveMemory(defaultContent, 'global');
+      const newMemory = await this.loadMemoryFile(this.globalMemoryPath, 'global');
+      if (newMemory) {
+        this.memoryFiles.push(newMemory);
+        combinedMemory += newMemory.content + '\n\n';
+      }
+    }
+
+    if (this.projectMemoryPath) {
+      const projectMemory = await this.loadMemoryFile(this.projectMemoryPath, 'project');
+      if (projectMemory) {
+        this.memoryFiles.push(projectMemory);
+        combinedMemory += projectMemory.content + '\n\n';
+      }
+      // else {
+      //   // Project memory not found, create default content
+      //   const defaultContent = '# Project Context\n\nProject-specific context will be added here.';
+      //   await this.saveMemory(defaultContent, 'project');
+      //   const newMemory = await this.loadMemoryFile(this.projectMemoryPath, 'project');
+      //   if (newMemory) {
+      //     this.memoryFiles.push(newMemory);
+      //     combinedMemory += newMemory.content + '\n\n';
+      //   }
+      // }
+    }
+
+    return this.processImports(combinedMemory);
+  }
+
+  private async loadMemoryFile(filePath: string, level: 'global' | 'project' | 'subdirectory'): Promise<MemoryFile | null> {
+    try {
+      const content = await fs.readFile(filePath, 'utf-8');
+      return { path: filePath, content, level };
+    } catch (error) {
+      if ((error as NodeJS.ErrnoException).code !== 'ENOENT') {
+        console.error(`Failed to load memory file ${filePath}:`, error);
+      }
+      return null;
+    }
+  }
+
+  // private async loadSubdirectoryMemories(): Promise<MemoryFile[]> {
+  //   const memories: MemoryFile[] = [];
+    
+  //   if (!this.projectMemoryPath) {
+  //     return memories;
+  //   }
+
+  //   const projectRoot = path.dirname(this.projectMemoryPath);
+    
+  //   try {
+  //     const files = await glob('**/XAGENT.md', {
+  //       cwd: projectRoot,
+  //       ignore: ['node_modules/**', '.git/**', 'dist/**', 'build/**']
+  //     });
+
+  //     for (const file of files) {
+  //       if (file !== 'XAGENT.md') {
+  //         const filePath = path.join(projectRoot, file);
+  //         const memory = await this.loadMemoryFile(filePath, 'subdirectory');
+  //         if (memory) {
+  //           memories.push(memory);
+  //         }
+  //       }
+  //     }
+  //   } catch (error) {
+  //     console.error('Failed to load subdirectory memories:', error);
+  //   }
+
+  //   return memories;
+  // }
+
+  private processImports(content: string): string {
+    // Only match actual import statements: import ... from '@xxx' or import '@xxx'
+    const importRegex = /import\s+(?:(?:\{[^}]*\}|\* as \w+|\w+)\s+from\s+)?['"](@[^'"]+)['"]/g;
+    let processedContent = content;
+    const visitedFiles = new Set<string>();
+
+    const processImport = (match: string, importPath: string): string => {
+      const absolutePath = path.isAbsolute(importPath)
+        ? importPath
+        : path.resolve(process.cwd(), importPath);
+
+      if (visitedFiles.has(absolutePath)) {
+        return '';
+      }
+
+      visitedFiles.add(absolutePath);
+
+      try {
+        const importedContent = fsSync.readFileSync(absolutePath, 'utf-8');
+        return `\n\n${importedContent}\n\n`;
+      } catch (error) {
+        console.warn(`Failed to import ${importPath}:`, error);
+        return '';
+      }
+    };
+
+    let maxIterations = 10;
+    let iterations = 0;
+
+    while (importRegex.test(processedContent) && iterations < maxIterations) {
+      processedContent = processedContent.replace(importRegex, processImport);
+      iterations++;
+    }
+
+    return processedContent;
+  }
+
+  async saveMemory(content: string, scope: 'global' | 'project' = 'global'): Promise<void> {
+    const filePath = scope === 'global' ? this.globalMemoryPath : this.projectMemoryPath;
+
+    if (!filePath) {
+      throw new Error('Project memory path not set');
+    }
+
+    const dir = path.dirname(filePath);
+    await fs.mkdir(dir, { recursive: true });
+
+    await fs.writeFile(filePath, content, 'utf-8');
+
+    if (scope === 'global') {
+      const existingMemory = this.memoryFiles.find(m => m.level === 'global');
+      if (existingMemory) {
+        existingMemory.content = content;
+      } else {
+        this.memoryFiles.push({ path: filePath, content, level: 'global' });
+      }
+    } else {
+      const existingMemory = this.memoryFiles.find(m => m.level === 'project');
+      if (existingMemory) {
+        existingMemory.content = content;
+      } else {
+        this.memoryFiles.push({ path: filePath, content, level: 'project', projectRoot: this.projectRoot || undefined });
+      }
+    }
+  }
+
+  async addMemoryEntry(entry: string, scope: 'global' | 'project' = 'global'): Promise<void> {
+    const filePath = scope === 'global' ? this.globalMemoryPath : this.projectMemoryPath;
+
+    if (!filePath) {
+      throw new Error('Project memory path not set');
+    }
+
+    let existingContent = '';
+    try {
+      existingContent = await fs.readFile(filePath, 'utf-8');
+    } catch (error) {
+      if ((error as NodeJS.ErrnoException).code !== 'ENOENT') {
+        throw error;
+      }
+    }
+
+    const newContent = existingContent + `\n\n## Memory Entry (${new Date().toISOString()})\n\n${entry}\n`;
+    await this.saveMemory(newContent, scope);
+  }
+
+  getMemoryFiles(): MemoryFile[] {
+    return [...this.memoryFiles];
+  }
+
+  getMemoriesDir(): string {
+    return this.memoriesDir;
+  }
+
+  async initializeProject(projectRoot: string): Promise<void> {
+    this.projectRoot = path.resolve(projectRoot);
+    this.projectMemoryPath = this.getProjectMemoryPath(this.projectRoot);
+
+    const existingMemory = await this.loadMemoryFile(this.projectMemoryPath, 'project');
+    if (existingMemory) {
+      console.log('XAGENT.md already exists. Skipping initialization.');
+      return;
+    }
+
+    console.log('Creating XAGENT.md...');
+    await this.saveMemory('# Project Context\n\nProject-specific context will be added here.', 'project');
+    
+    console.log('Analyzing project structure...');
+    const analysis = await this.analyzeProject(projectRoot);
+    
+    console.log('Generating project-specific context...');
+    await this.saveMemory(analysis, 'project');
+    
+    console.log('✅ XAGENT.md has been successfully populated with project-specific information.');
+  }
+
+  private async analyzeProject(projectRoot: string): Promise<string> {
+    let analysis = '# Project Context\n\n';
+
+    try {
+      const packageJsonPath = path.join(projectRoot, 'package.json');
+      const packageJson = await this.loadJsonFile(packageJsonPath);
+      
+      if (packageJson) {
+        analysis += '## Project Overview\n\n';
+        analysis += `Name: ${packageJson.name || 'Unknown'}\n`;
+        analysis += `Version: ${packageJson.version || 'Unknown'}\n`;
+        analysis += `Description: ${packageJson.description || 'No description'}\n\n`;
+
+        if (packageJson.dependencies) {
+          analysis += '## Dependencies\n\n';
+          analysis += '```json\n' + JSON.stringify(packageJson.dependencies, null, 2) + '\n```\n\n';
+        }
+
+        if (packageJson.scripts) {
+          analysis += '## Scripts\n\n';
+          Object.entries(packageJson.scripts).forEach(([name, script]) => {
+            analysis += `- ${name}: ${script}\n`;
+          });
+          analysis += '\n';
+        }
+      }
+
+      const tsconfigPath = path.join(projectRoot, 'tsconfig.json');
+      const tsconfig = await this.loadJsonFile(tsconfigPath);
+      
+      if (tsconfig) {
+        analysis += '## TypeScript Configuration\n\n';
+        analysis += '```json\n' + JSON.stringify(tsconfig, null, 2) + '\n```\n\n';
+      }
+
+      analysis += '## Project Structure\n\n';
+      analysis += '```\n';
+      const structure = await this.getProjectStructure(projectRoot);
+      analysis += structure;
+      analysis += '```\n\n';
+
+      analysis += '## Development Guidelines\n\n';
+      analysis += '- Follow existing code style and conventions\n';
+      analysis += '- Write clear, descriptive commit messages\n';
+      analysis += '- Add tests for new features\n';
+      analysis += '- Update documentation as needed\n';
+    } catch (error) {
+      console.error('Failed to analyze project:', error);
+      analysis += '## Project Analysis Failed\n\n';
+      analysis += 'Unable to analyze project structure automatically.\n';
+    }
+
+    return analysis;
+  }
+
+  private async loadJsonFile(filePath: string): Promise<any> {
+    try {
+      const content = await fs.readFile(filePath, 'utf-8');
+      return JSON.parse(content);
+    } catch (error) {
+      return null;
+    }
+  }
+
+  private async getProjectStructure(rootPath: string, maxDepth: number = 3, currentDepth: number = 0): Promise<string> {
+    if (currentDepth >= maxDepth) {
+      return '';
+    }
+
+    let structure = '';
+    
+    try {
+      const entries = await fs.readdir(rootPath, { withFileTypes: true });
+      
+      for (const entry of entries) {
+        if (entry.name.startsWith('.') || entry.name === 'node_modules' || entry.name === 'dist') {
+          continue;
+        }
+
+        if (entry.isDirectory()) {
+          structure += '  '.repeat(currentDepth) + entry.name + '/\n';
+          const subPath = path.join(rootPath, entry.name);
+          structure += await this.getProjectStructure(subPath, maxDepth, currentDepth + 1);
+        } else {
+          structure += '  '.repeat(currentDepth) + entry.name + '\n';
+        }
+      }
+    } catch (error) {
+      console.error(`Failed to read directory ${rootPath}:`, error);
+    }
+
+    return structure;
+  }
+}
+
+let memoryManagerInstance: MemoryManager | null = null;
+
+export function getMemoryManager(projectRoot?: string, contextFileName?: string | string[]): MemoryManager {
+  if (!memoryManagerInstance || projectRoot) {
+    memoryManagerInstance = new MemoryManager(projectRoot, contextFileName);
+  }
+  return memoryManagerInstance;
+}