@aicgen/aicgen 1.0.0-beta.1 → 1.0.0-beta.2

package/src/services/assistant-file-writer.ts

@@ -1,612 +0,0 @@
-import { writeFile, mkdir } from 'fs/promises';
-import { join, dirname } from 'path';
-import { AIAssistant } from '../models/project.js';
-import { ProfileSelection } from '../models/profile.js';
-import { GuidelineLoader } from './guideline-loader.js';
-import { HookGenerator } from './hook-generator.js';
-import { SubAgentGenerator } from './subagent-generator.js';
-
-export interface GeneratedFile {
-  path: string;
-  content: string;
-  type: 'main' | 'guideline' | 'config' | 'agent' | 'universal';
-}
-
-export class AssistantFileWriter {
-  private guidelineLoader: GuidelineLoader;
-  private hookGenerator: HookGenerator;
-  private subAgentGenerator: SubAgentGenerator;
-
-  static async create(): Promise<AssistantFileWriter> {
-    const guidelineLoader = await GuidelineLoader.create();
-    return new AssistantFileWriter(guidelineLoader);
-  }
-
-  private constructor(guidelineLoader: GuidelineLoader) {
-    this.guidelineLoader = guidelineLoader;
-    this.hookGenerator = new HookGenerator();
-    this.subAgentGenerator = new SubAgentGenerator();
-  }
-
-  async generateFiles(
-    assistant: AIAssistant,
-    guidelineIds: string[],
-    selection: ProfileSelection,
-    projectPath: string
-  ): Promise<GeneratedFile[]> {
-    const files: GeneratedFile[] = [];
-
-    const categoryTree = this.organizeByCategory(guidelineIds);
-
-    switch (assistant) {
-      case 'claude-code':
-        files.push(...await this.generateClaudeCodeFiles(categoryTree, guidelineIds, selection, projectPath));
-        break;
-      case 'copilot':
-        files.push(...await this.generateCopilotFiles(categoryTree, selection));
-        break;
-      case 'gemini':
-        files.push(...await this.generateGeminiFiles(categoryTree, selection));
-        break;
-      case 'antigravity':
-        files.push(...await this.generateAntigravityFiles(categoryTree, selection));
-        break;
-      case 'codex':
-        files.push(...await this.generateCodexFiles(categoryTree, selection));
-        break;
-    }
-
-    files.push(this.generateUniversalAgentsFile(categoryTree, selection));
-
-    return files.map(file => ({
-      ...file,
-      path: join(projectPath, file.path)
-    }));
-  }
-
-  async writeFiles(files: GeneratedFile[]): Promise<void> {
-    for (const file of files) {
-      await mkdir(dirname(file.path), { recursive: true });
-      await writeFile(file.path, file.content, 'utf-8');
-    }
-  }
-
-  private organizeByCategory(guidelineIds: string[]): Map<string, string[]> {
-    const byCategory = new Map<string, string[]>();
-
-    for (const id of guidelineIds) {
-      const mapping = this.guidelineLoader.getMapping(id);
-      if (mapping) {
-        const category = mapping.category || 'General';
-        if (!byCategory.has(category)) {
-          byCategory.set(category, []);
-        }
-        byCategory.get(category)!.push(id);
-      }
-    }
-
-    return byCategory;
-  }
-
-  private async generateClaudeCodeFiles(
-    categoryTree: Map<string, string[]>,
-    guidelineIds: string[],
-    selection: ProfileSelection,
-    projectPath: string
-  ): Promise<GeneratedFile[]> {
-    const files: GeneratedFile[] = [];
-
-    const mainReferences: string[] = [];
-    const guidelines: string[] = [];
-
-    for (const [category, ids] of categoryTree) {
-      const categoryFile = category.toLowerCase().replace(/\s+/g, '-');
-      const categoryContent = ids.map(id => {
-        const content = this.guidelineLoader.loadGuideline(id);
-        guidelines.push(content);
-        return content;
-      }).join('\n\n---\n\n');
-
-      files.push({
-        path: `.claude/guidelines/${categoryFile}.md`,
-        content: `# ${category}\n\n${categoryContent}`,
-        type: 'guideline'
-      });
-
-      mainReferences.push(`- **${category}**: @.claude/guidelines/${categoryFile}.md`);
-    }
-
-    const archDescription = selection.architecture === 'other'
-      ? ''
-      : ` and ${selection.architecture} architecture`;
-
-    const mainContent = `# ${selection.projectName} - Development Guidelines
-
-**Role:** You are an expert software engineer specialized in ${selection.language}${archDescription}.
-**User's Goal:** Build high-quality, maintainable software following strict project guidelines.
-
-## Guidelines
-
-This project follows structured coding guidelines organized by category:
-
-${mainReferences.join('\n')}
-
-## Quick Reference
-
-- Run tests: Check package.json scripts
-- Build: Check package.json scripts
-- Code style: See Code Style guidelines above
-- Architecture: See Architecture guidelines above
-
-## Important Notes
-
-- **Follow the guidelines** referenced above.
-- **Verification**: Use sub-agents in \`.claude/agents/\` to verify compliance.
-- **Constraints**: Hooks in \`.claude/settings.json\` enforce critical rules.
-
----
-*Generated by aicgen*
-`;
-
-    files.push({
-      path: 'CLAUDE.md',
-      content: mainContent,
-      type: 'main'
-    });
-
-    const hooks = await this.hookGenerator.generateHooks(guidelineIds);
-    const settingsContent = this.hookGenerator.generateClaudeCodeSettings(hooks, projectPath, selection.level);
-
-    files.push({
-      path: '.claude/settings.json',
-      content: settingsContent,
-      type: 'config'
-    });
-
-    const subAgents = await this.subAgentGenerator.generateSubAgents(guidelineIds);
-    for (const agent of subAgents) {
-      files.push({
-        path: `.claude/agents/${agent.name}.md`,
-        content: agent.content,
-        type: 'agent'
-      });
-    }
-
-    return files;
-  }
-
-  private async generateCopilotFiles(
-    categoryTree: Map<string, string[]>,
-    selection: ProfileSelection
-  ): Promise<GeneratedFile[]> {
-    const files: GeneratedFile[] = [];
-
-    const mainReferences: string[] = [];
-
-    for (const [category, ids] of categoryTree) {
-      const categoryFile = category.toLowerCase().replace(/\s+/g, '-');
-      const categoryContent = ids.map(id =>
-        this.guidelineLoader.loadGuideline(id)
-      ).join('\n\n---\n\n');
-
-      let applyTo = "**/*";
-
-      // Smart globs for specific categories
-      if (category.toLowerCase().includes('testing')) {
-        applyTo = "**/*.test.ts, **/*.spec.ts, **/tests/**";
-      } else if (category.toLowerCase().includes('language') || category.toLowerCase().includes('style')) {
-        applyTo = "**/*.ts, **/*.js, **/*.tsx, **/*.jsx"; // Adjust based on language in future
-      }
-
-      const instructionsContent = `---
-applyTo: "${applyTo}"
-description: "${category} guidelines"
----
-
-# ${category}
-
-${categoryContent}
-`;
-
-      files.push({
-        path: `.github/instructions/${categoryFile}.instructions.md`,
-        content: instructionsContent,
-        type: 'guideline'
-      });
-
-      mainReferences.push(`- ${category}: @.github/instructions/${categoryFile}.instructions.md`);
-    }
-
-    const archDescription = selection.architecture === 'other'
-      ? ''
-      : ` and ${selection.architecture}`;
-
-    const mainContent = `# GitHub Copilot Instructions
-
-**Role:** You are an expert AI pair programmer specialized in ${selection.language}${archDescription}.
-
-## Guidelines
-
-${mainReferences.join('\n')}
-
-## Development
-
-See the instruction files above for detailed guidelines on:
-- Code style and naming conventions
-- Architecture patterns and best practices
-- Testing requirements
-- Security considerations
-
----
-*Generated by aicgen*
-`;
-
-    files.push({
-      path: '.github/copilot-instructions.md',
-      content: mainContent,
-      type: 'main'
-    });
-
-    return files;
-  }
-
-  private async generateGeminiFiles(
-    categoryTree: Map<string, string[]>,
-    selection: ProfileSelection
-  ): Promise<GeneratedFile[]> {
-    const files: GeneratedFile[] = [];
-
-    let allGuidelines = '';
-    for (const [category, ids] of categoryTree) {
-      const categoryContent = ids.map(id =>
-        this.guidelineLoader.loadGuideline(id)
-      ).join('\n\n');
-
-      allGuidelines += `\n\n## ${category}\n\n${categoryContent}\n\n---\n`;
-    }
-
-    const archDescription = selection.architecture === 'other'
-      ? ''
-      : ` and ${selection.architecture} architecture`;
-
-    const content = `# Gemini Development Guide
-
-**Role:** You are an expert software engineer specialized in ${selection.language}${archDescription}.
-**Objective:** Write clean, efficient, and maintainable code following the guidelines below.
-
-## Guiding Principles
-1. **Quality First**: Prioritize code quality and maintainability over speed.
-2. **Step-by-Step**: Think through problems systematically.
-3. **Verify**: Double-check your code against the guidelines.
-
-${allGuidelines}
-
----
-*Generated by aicgen*
-`;
-
-    files.push({
-      path: '.gemini/instructions.md',
-      content,
-      type: 'main'
-    });
-
-    return files;
-  }
-
-  private async generateAntigravityFiles(
-    categoryTree: Map<string, string[]>,
-    selection: ProfileSelection
-  ): Promise<GeneratedFile[]> {
-    const files: GeneratedFile[] = [];
-    const ruleFiles: string[] = [];
-
-    // Generate separate rule files for each category
-    for (const [category, ids] of categoryTree) {
-      const guidelines = ids.map(id => {
-        const guideline = this.guidelineLoader.loadGuideline(id);
-        return guideline;
-      }).filter(Boolean).join('\n\n---\n\n');
-
-      if (guidelines) {
-        const fileName = category.toLowerCase().replace(/\s+/g, '-');
-
-        const content = `# ${category} Rules
-
-${guidelines}
-
----
-*Generated by aicgen*
-`;
-
-        files.push({
-          path: `.agent/rules/${fileName}.md`,
-          content,
-          type: 'guideline'
-        });
-
-        ruleFiles.push(`- **${category}**: @.agent/rules/${fileName}.md`);
-      }
-    }
-
-    // Generate main instruction file (Index/System Prompt)
-    const archDescription = selection.architecture === 'other'
-      ? ''
-      : ` and ${selection.architecture} architecture`;
-
-    const mainContent = `# Google Antigravity Instructions
-
-**Role:** You are an expert software engineer specialized in ${selection.language}${archDescription}.
-**Objective:** Assist the user in building high-quality, maintainable software following strict project guidelines.
-
-## Rule Index
-
-This project is governed by the following rule sets:
-
-${ruleFiles.join('\n')}
-
-## Core Principles
-1. **Act as an Agent**: Don't just answer; perform actions, verify results, and course-correct.
-2. **Follow Guidelines**: Strictly adhere to the rules linked above.
-3. **Think Step-by-Step**: Break down complex tasks into smaller, verifiable steps.
-
----
-*Generated by aicgen*
-`;
-
-    files.push({
-      path: '.agent/rules/instructions.md',
-      content: mainContent,
-      type: 'main'
-    });
-
-    // Generate workflows based on instruction level
-    const workflows = this.getWorkflowsForLevel(selection.level);
-    for (const workflow of workflows) {
-      files.push({
-        path: `.agent/workflows/${workflow.name}.md`,
-        content: workflow.content,
-        type: 'config'
-      });
-    }
-
-    return files;
-  }
-
-
-  private getWorkflowsForLevel(level: string): Array<{ name: string; content: string }> {
-    const workflows: Array<{ name: string; content: string }> = [];
-
-    // Basic workflows for all levels
-    workflows.push({
-      name: 'generate-unit-tests',
-      content: `---
-description: Generate comprehensive unit tests for all functions and methods
----
-
-* Analyze the selected file or directory for testable code
-* Generate test files with appropriate naming conventions
-* Create test cases covering happy path, edge cases, and error handling
-* Mock external dependencies appropriately
-* Follow language-specific testing best practices
-* Aim for >80% code coverage
-* Include setup and teardown methods where needed`
-    });
-
-    workflows.push({
-      name: 'add-documentation',
-      content: `---
-description: Add or update comprehensive documentation for code
----
-
-* Analyze the selected code for documentation needs
-* Add inline documentation with clear descriptions
-* Include parameter types and return value documentation
-* Add usage examples for complex functions
-* Follow language-specific documentation standards
-* Update README.md if adding new features`
-    });
-
-    // Standard level and above
-    if (level === 'standard' || level === 'expert' || level === 'full') {
-      workflows.push({
-        name: 'refactor-extract-module',
-        content: `---
-description: Extract code into a separate, reusable module
----
-
-* Identify the code section to extract
-* Analyze dependencies and determine module interface
-* Create a new module file following project conventions
-* Maintain original functionality and behavior
-* Update imports and exports appropriately
-* Ensure no circular dependencies
-* Add documentation to the new module
-* Verify all tests still pass`
-      });
-
-      workflows.push({
-        name: 'generate-integration-tests',
-        content: `---
-description: Generate integration tests for API endpoints and system components
----
-
-* Identify integration points (APIs, databases, services)
-* Create integration test files
-* Test end-to-end workflows
-* Use realistic test data and fixtures
-* Include proper setup and teardown
-* Ensure tests are idempotent`
-      });
-    }
-
-    // Expert and full levels
-    if (level === 'expert' || level === 'full') {
-      workflows.push({
-        name: 'security-audit',
-        content: `---
-description: Perform comprehensive security audit of the codebase
----
-
-* Scan for common vulnerabilities (SQL injection, XSS, etc.)
-* Check authentication and authorization logic
-* Review input validation and sanitization
-* Examine error handling for information leakage
-* Verify secure handling of sensitive data
-* Check third-party dependencies
-* Suggest remediation steps
-* Prioritize findings by severity`
-      });
-
-      workflows.push({
-        name: 'performance-audit',
-        content: `---
-description: Analyze code for performance bottlenecks and optimization opportunities
----
-
-* Profile code to identify hotspots
-* Check for N+1 queries and inefficient algorithms
-* Analyze caching opportunities
-* Review resource management
-* Suggest specific optimizations
-* Estimate performance impact
-* Prioritize recommendations`
-      });
-    }
-
-    return workflows;
-  }
-
-  private async generateCodexFiles(
-    categoryTree: Map<string, string[]>,
-    selection: ProfileSelection
-  ): Promise<GeneratedFile[]> {
-    const files: GeneratedFile[] = [];
-
-    let allGuidelines = '';
-    for (const [category, ids] of categoryTree) {
-      const categoryContent = ids.map(id =>
-        this.guidelineLoader.loadGuideline(id)
-      ).join('\n\n');
-
-      allGuidelines += `\n\n## ${category}\n\n${categoryContent}\n\n---\n`;
-    }
-
-    const content = `# Development Guide
-
-**Role:** You are an expert software engineer specialized in ${selection.language}.
-**Objective:** Assist the user in writing high-quality code.
-
-**Language:** ${selection.language}
-**Type:** ${selection.projectType}
-
-${allGuidelines}
-
----
-*Generated by aicgen*
-`;
-
-    files.push({
-      path: '.codex/instructions.md',
-      content,
-      type: 'main'
-    });
-
-    return files;
-  }
-
-  private generateUniversalAgentsFile(
-    categoryTree: Map<string, string[]>,
-    selection: ProfileSelection
-  ): GeneratedFile {
-    const categories: string[] = [];
-
-    for (const [category, ids] of categoryTree) {
-      const examples = ids.slice(0, 3).map(id => {
-        const mapping = this.guidelineLoader.getMapping(id);
-        const fileName = mapping?.path.split('/').pop()?.replace('.md', '') || id;
-        return `- ${fileName}`;
-      }).join('\n  ');
-
-      categories.push(`### ${category}\n\n  ${examples}\n`);
-    }
-
-    const archLine = selection.architecture === 'other'
-      ? ''
-      : `**Architecture:** ${selection.architecture}\n`;
-
-    const archSection = selection.architecture === 'other'
-      ? ''
-      : `## Architecture
-
-This project follows **${selection.architecture}** architecture. See architecture guidelines in tool-specific files.
-
-`;
-
-    const content = `# AGENTS.md
-
-## Project Overview
-
-**Language:** ${selection.language}
-**Type:** ${selection.projectType}
-${archLine}
-## Development Guidelines
-
-This project follows structured coding guidelines across multiple categories:
-
-${categories.join('\n')}
-
-## Commands
-
-**Install dependencies:**
-\`\`\`bash
-# Check package.json for package manager
-npm install
-# or yarn install
-# or pnpm install
-# or bun install
-\`\`\`
-
-**Run tests:**
-\`\`\`bash
-# Check package.json scripts section
-npm test
-\`\`\`
-
-**Build:**
-\`\`\`bash
-# Check package.json scripts section
-npm run build
-\`\`\`
-
-## Code Style
-
-See tool-specific instruction files for detailed code style guidelines:
-- Claude Code: \`CLAUDE.md\`
-- GitHub Copilot: \`.github/copilot-instructions.md\`
-- Gemini: \`.gemini/instructions.md\`
-- Antigravity: \`.agent/rules/instructions.md\`
-
-${archSection}## Testing
-
-Follow testing guidelines in tool-specific instruction files.
-
-## Git Workflow
-
-- Write clear commit messages
-- Follow conventional commits if configured
-- Run tests before pushing
-- Keep PRs focused and reviewable
-
----
-
-*Generated by aicgen - Universal AI agent instructions following the AGENTS.md standard*
-`;
-
-    return {
-      path: 'AGENTS.md',
-      content,
-      type: 'universal'
-    };
-  }
-}