diff --git a/tools/cli/installers/lib/custom/handler.js b/tools/cli/installers/lib/custom/handler.js index 1982d3b3a..fbd6c728f 100644 --- a/tools/cli/installers/lib/custom/handler.js +++ b/tools/cli/installers/lib/custom/handler.js @@ -2,17 +2,11 @@ const path = require('node:path'); const fs = require('fs-extra'); const yaml = require('yaml'); const prompts = require('../../../lib/prompts'); -const { FileOps } = require('../../../lib/file-ops'); - /** * Handler for custom content (custom.yaml) - * Installs custom agents and workflows without requiring a full module structure + * Discovers custom agents and workflows in the project */ class CustomHandler { - constructor() { - this.fileOps = new FileOps(); - } - /** * Find all custom.yaml files in the project * @param {string} projectRoot - Project root directory @@ -113,172 +107,6 @@ class CustomHandler { return null; } } - - /** - * Install custom content - * @param {string} customPath - Path to custom content directory - * @param {string} bmadDir - Target bmad directory - * @param {Object} config - Configuration from custom.yaml - * @param {Function} fileTrackingCallback - Optional callback to track installed files - * @returns {Object} Installation result - */ - async install(customPath, bmadDir, config, fileTrackingCallback = null) { - const results = { - agentsInstalled: 0, - workflowsInstalled: 0, - filesCopied: 0, - preserved: 0, - errors: [], - }; - - try { - // Create custom directories in bmad - const bmadCustomDir = path.join(bmadDir, 'custom'); - const bmadAgentsDir = path.join(bmadCustomDir, 'agents'); - const bmadWorkflowsDir = path.join(bmadCustomDir, 'workflows'); - - await fs.ensureDir(bmadCustomDir); - await fs.ensureDir(bmadAgentsDir); - await fs.ensureDir(bmadWorkflowsDir); - - // Copy agents directory - const agentsDir = path.join(customPath, 'agents'); - if (await fs.pathExists(agentsDir)) { - await this.copyDirectory(agentsDir, bmadAgentsDir, results, fileTrackingCallback, config); - - // Count agent files - const agentFiles = await this.findFilesRecursively(agentsDir, ['.md']); - results.agentsInstalled = agentFiles.length; - } - - // Process workflows - copy entire workflows directory structure - const workflowsDir = path.join(customPath, 'workflows'); - if (await fs.pathExists(workflowsDir)) { - await this.copyDirectory(workflowsDir, bmadWorkflowsDir, results, fileTrackingCallback, config); - - // Count workflow files - const workflowFiles = await this.findFilesRecursively(workflowsDir, ['.md']); - results.workflowsInstalled = workflowFiles.length; - } - - // Process any additional files at root - const entries = await fs.readdir(customPath, { withFileTypes: true }); - for (const entry of entries) { - if (entry.isFile() && entry.name !== 'custom.yaml' && !entry.name.startsWith('.') && !entry.name.endsWith('.md')) { - // Skip .md files at root as they're likely docs - const sourcePath = path.join(customPath, entry.name); - const targetPath = path.join(bmadCustomDir, entry.name); - - try { - // Check if file already exists - if (await fs.pathExists(targetPath)) { - // File already exists, preserve it - results.preserved = (results.preserved || 0) + 1; - } else { - await fs.copy(sourcePath, targetPath); - results.filesCopied++; - - if (fileTrackingCallback) { - fileTrackingCallback(targetPath); - } - } - } catch (error) { - results.errors.push(`Failed to copy file ${entry.name}: ${error.message}`); - } - } - } - } catch (error) { - results.errors.push(`Installation failed: ${error.message}`); - } - - return results; - } - - /** - * Find all files with specific extensions recursively - * @param {string} dir - Directory to search - * @param {Array} extensions - File extensions to match - * @returns {Array} List of matching files - */ - async findFilesRecursively(dir, extensions) { - const files = []; - - async function search(currentDir) { - const entries = await fs.readdir(currentDir, { withFileTypes: true }); - - for (const entry of entries) { - const fullPath = path.join(currentDir, entry.name); - - if (entry.isDirectory()) { - await search(fullPath); - } else if (extensions.some((ext) => entry.name.endsWith(ext))) { - files.push(fullPath); - } - } - } - - await search(dir); - return files; - } - - /** - * Recursively copy a directory - * @param {string} sourceDir - Source directory - * @param {string} targetDir - Target directory - * @param {Object} results - Results object to update - * @param {Function} fileTrackingCallback - Optional callback - * @param {Object} config - Configuration for placeholder replacement - */ - async copyDirectory(sourceDir, targetDir, results, fileTrackingCallback, config) { - await fs.ensureDir(targetDir); - const entries = await fs.readdir(sourceDir, { withFileTypes: true }); - - for (const entry of entries) { - const sourcePath = path.join(sourceDir, entry.name); - const targetPath = path.join(targetDir, entry.name); - - if (entry.isDirectory()) { - await this.copyDirectory(sourcePath, targetPath, results, fileTrackingCallback, config); - } else { - try { - // Check if file already exists - if (await fs.pathExists(targetPath)) { - // File already exists, preserve it - results.preserved = (results.preserved || 0) + 1; - } else { - // Copy with placeholder replacement for text files - const textExtensions = ['.md', '.yaml', '.yml', '.txt', '.json']; - if (textExtensions.some((ext) => entry.name.endsWith(ext))) { - // Read source content - let content = await fs.readFile(sourcePath, 'utf8'); - - // Replace placeholders - content = content.replaceAll('{user_name}', config.user_name || 'User'); - content = content.replaceAll('{communication_language}', config.communication_language || 'English'); - content = content.replaceAll('{output_folder}', config.output_folder || 'docs'); - - // Write to target - await fs.ensureDir(path.dirname(targetPath)); - await fs.writeFile(targetPath, content, 'utf8'); - } else { - // Copy binary files as-is - await fs.copy(sourcePath, targetPath); - } - - results.filesCopied++; - if (entry.name.endsWith('.md')) { - results.workflowsInstalled++; - } - if (fileTrackingCallback) { - fileTrackingCallback(targetPath); - } - } - } catch (error) { - results.errors.push(`Failed to copy ${entry.name}: ${error.message}`); - } - } - } - } } module.exports = { CustomHandler };