skill-linker 3.0.8 → 4.0.4

package/src/commands/install.js

@@ -1,347 +1,203 @@
-const prompts = require('prompts');
-const chalk = require('chalk');
-const path = require('path');
-const { findSkills, findRepos, dirExists, ensureDir, createSymlink, listDirectories } = require('../utils/file-system');
-const { getAllAgents, detectInstalledAgents } = require('../utils/agents');
-const { DEFAULT_LIB_PATH, cloneOrUpdateRepo, pullRepo } = require('../utils/git');
+const chalk = require("chalk");
+const path = require("path");
+const {
+  dirExists,
+  ensureDir,
+  createSymlink,
+  listDirectories,
+} = require("../utils/file-system");
+const {
+  getAllAgents,
+  detectInstalledAgents,
+  findAgentIndex,
+} = require("../utils/agents");
+const { cloneOrUpdateRepo, pullRepo } = require("../utils/git");
 
 /**
- * Main install command
+ * Main install command (CLI mode only)
  * @param {Object} options - Command options
+ * @param {string} options.skill - Skill path or --from clone URL
+ * @param {string} [options.from] - GitHub URL to clone from
+ * @param {string[]} [options.agents] - Agent names to install to
+ * @param {string} [options.scope] - Scope: project, global, or both
+ * @param {boolean} [options.yes] - Auto-overwrite existing links
  */
 async function install(options) {
-    let skillPaths = [];
-
-    // Handle --from flag: Clone from GitHub
-    if (options.from) {
-        console.log(chalk.blue('[INFO]'), `Cloning from ${options.from}...`);
-
-        try {
-            const { skillPath: clonedPath, targetPath, needsUpdate, hasSubpath } = await cloneOrUpdateRepo(options.from);
-
-            if (needsUpdate) {
-                const { shouldUpdate } = await prompts({
-                    type: 'confirm',
-                    name: 'shouldUpdate',
-                    message: `Repository already exists. Update with git pull?`,
-                    initial: false
-                });
-
-                if (shouldUpdate) {
-                    await pullRepo(targetPath);
-                    console.log(chalk.green('[SUCCESS]'), 'Repository updated!');
-                }
-            }
-
-            // If no subpath, check for skills/ subdirectory
-            if (!hasSubpath && dirExists(path.join(targetPath, 'skills'))) {
-                const subSkills = listDirectories(path.join(targetPath, 'skills'));
-
-                if (subSkills.length > 0) {
-                    const { selectedSkills } = await prompts({
-                        type: 'multiselect',
-                        name: 'selectedSkills',
-                        message: 'This repo contains multiple skills. Select skills to install:',
-                        choices: [
-                            ...subSkills.map(s => ({ title: s, value: path.join(targetPath, 'skills', s) })),
-                            { title: 'Link entire repo', value: targetPath }
-                        ],
-                        hint: '- Space to select. Return to submit'
-                    });
+  let skillPaths = [];
+
+  // Handle --from flag: Clone from GitHub
+  if (options.from) {
+    console.log(chalk.blue("[INFO]"), `Cloning from ${options.from}...`);
+
+    try {
+      const {
+        skillPath: clonedPath,
+        targetPath,
+        needsUpdate,
+        hasSubpath,
+      } = await cloneOrUpdateRepo(options.from);
+
+      if (needsUpdate) {
+        if (options.yes) {
+          await pullRepo(targetPath);
+          console.log(chalk.green("[SUCCESS]"), "Repository updated!");
+        } else {
+          console.log(
+            chalk.yellow("[WARNING]"),
+            "Repository already exists. Use --yes to update.",
+          );
+        }
+      }
 
-                    if (selectedSkills && selectedSkills.length > 0) {
-                        skillPaths = selectedSkills;
-                    } else {
-                        // If nothing selected, maybe they just hit enter without selection? Default to repo path?
-                        // Or better, error out if multiselect returns empty.
-                        // Let's assume empty selection means exit or user made mistake.
-                        // But to be safe, if they chose "Link entire repo" in multiselect (which is weird), handle it.
-                        // Actually multiselect is better for picking subsets.
-                        // If empty, let's fall back to entire repo (or maybe error).
-                        // Let's error to be consistent with agent selection.
-                    }
-                } else {
-                    skillPaths = [targetPath];
-                }
-            } else {
-                skillPaths = [clonedPath];
-            }
+      // If no subpath, check for skills/ subdirectory
+      if (!hasSubpath && dirExists(path.join(targetPath, "skills"))) {
+        const subSkills = listDirectories(path.join(targetPath, "skills"));
 
-            console.log(chalk.green('[SUCCESS]'), 'Clone completed!');
-        } catch (error) {
-            console.error(chalk.red('[ERROR]'), error.message);
-            process.exit(1);
+        if (subSkills.length > 0) {
+          // Install all skills from skills/ directory
+          skillPaths = subSkills.map((s) => path.join(targetPath, "skills", s));
+        } else {
+          skillPaths = [targetPath];
         }
+      } else {
+        skillPaths = [clonedPath];
+      }
+
+      console.log(chalk.green("[SUCCESS]"), "Clone completed!");
+    } catch (error) {
+      console.error(chalk.red("[ERROR]"), error.message);
+      process.exit(1);
     }
-
-    // If no skill path provided, show library selection
-    if (skillPaths.length === 0 && options.skill) {
-        skillPaths = [options.skill];
+  }
+
+  // If no skill path provided via --from, use --skill
+  if (skillPaths.length === 0 && options.skill) {
+    skillPaths = [options.skill];
+  }
+
+  // Validate skill paths
+  for (const p of skillPaths) {
+    if (!dirExists(p)) {
+      console.error(chalk.red("[ERROR]"), `Skill directory not found: ${p}`);
+      process.exit(1);
     }
-
-    if (skillPaths.length === 0) {
-        // First, ask user to choose source: local library or GitHub
-        const hasLocalLibrary = dirExists(DEFAULT_LIB_PATH) && findRepos(DEFAULT_LIB_PATH).length > 0;
-        
-        const sourceChoices = [
-            { title: 'Clone from GitHub', value: 'github' }
-        ];
-        
-        if (hasLocalLibrary) {
-            sourceChoices.unshift({ title: 'Select from local library', value: 'local' });
-        }
-        
-        const { source } = await prompts({
-            type: 'select',
-            name: 'source',
-            message: 'Where do you want to get skills from?',
-            choices: sourceChoices
-        });
-        
-        if (!source) {
-            console.log(chalk.yellow('[WARNING]'), 'No source selected. Exiting.');
-            process.exit(0);
+  }
+
+  if (skillPaths.length > 1) {
+    console.log(chalk.blue("[INFO]"), `Selected ${skillPaths.length} skills`);
+  } else {
+    const skillName = path.basename(skillPaths[0]);
+    console.log(
+      chalk.blue("[INFO]"),
+      `Selected Skill: ${chalk.cyan(skillName)} (${skillPaths[0]})`,
+    );
+  }
+
+  // Agent selection
+  const agents = getAllAgents();
+  const installedIndices = detectInstalledAgents();
+
+  let selectedAgents = [];
+
+  // Use provided agents list
+  if (options.agents && options.agents.length > 0) {
+    selectedAgents = options.agents
+      .map((agentName) => {
+        const idx = findAgentIndex(agentName);
+        if (idx === null) {
+          console.log(
+            chalk.yellow("[WARNING]"),
+            `Unknown agent: ${agentName}, skipping...`,
+          );
+          return null;
         }
-        
-        // Handle GitHub source
-        if (source === 'github') {
-            const { githubUrl } = await prompts({
-                type: 'text',
-                name: 'githubUrl',
-                message: 'Enter GitHub URL:',
-                validate: value => value.trim() !== '' || 'Please enter a valid GitHub URL'
-            });
-            
-            if (!githubUrl) {
-                console.log(chalk.yellow('[WARNING]'), 'No URL provided. Exiting.');
-                process.exit(0);
-            }
-            
-            // Use the same logic as --from flag
-            console.log(chalk.blue('[INFO]'), `Cloning from ${githubUrl}...`);
-            
-            try {
-                const { skillPath: clonedPath, targetPath, needsUpdate, hasSubpath } = await cloneOrUpdateRepo(githubUrl);
-                
-                if (needsUpdate) {
-                    const { shouldUpdate } = await prompts({
-                        type: 'confirm',
-                        name: 'shouldUpdate',
-                        message: `Repository already exists. Update with git pull?`,
-                        initial: false
-                    });
-                    
-                    if (shouldUpdate) {
-                        await pullRepo(targetPath);
-                        console.log(chalk.green('[SUCCESS]'), 'Repository updated!');
-                    }
-                }
-                
-                // If no subpath, check for skills/ subdirectory
-                if (!hasSubpath && dirExists(path.join(targetPath, 'skills'))) {
-                    const subSkills = listDirectories(path.join(targetPath, 'skills'));
-                    
-                    if (subSkills.length > 0) {
-                        const { selectedSkills } = await prompts({
-                            type: 'multiselect',
-                            name: 'selectedSkills',
-                            message: 'This repo contains multiple skills. Select skills to install:',
-                            choices: [
-                                ...subSkills.map(s => ({ title: s, value: path.join(targetPath, 'skills', s) })),
-                                { title: 'Link entire repo', value: targetPath }
-                            ],
-                            hint: '- Space to select. Return to submit'
-                        });
-                        
-                        if (selectedSkills && selectedSkills.length > 0) {
-                            skillPaths = selectedSkills;
-                        }
-                    } else {
-                        skillPaths = [targetPath];
-                    }
-                } else {
-                    skillPaths = [clonedPath];
-                }
-                
-                console.log(chalk.green('[SUCCESS]'), 'Clone completed!');
-            } catch (error) {
-                console.error(chalk.red('[ERROR]'), error.message);
-                process.exit(1);
-            }
-        }
-        
-        // Handle local library source
-        if (source === 'local') {
-            const repos = findRepos(DEFAULT_LIB_PATH);
-            
-            if (repos.length === 0) {
-                console.error(chalk.red('[ERROR]'), `No repos found in ${DEFAULT_LIB_PATH}`);
-                console.log(chalk.blue('[INFO]'), 'Use --from <github_url> to clone skills first.');
-                process.exit(1);
-            }
-            
-            console.log('');
-            
-            // 1. Select Repository
-            const { selectedRepo } = await prompts({
-                type: 'autocomplete',
-                name: 'selectedRepo',
-                message: 'Select a repository:',
-                choices: repos.map(repo => ({
-                    title: `${repo.name}${repo.hasSkillsDir ? chalk.dim(' (has skills/)') : ''}`,
-                    value: repo
-                })),
-                suggest: (input, choices) => {
-                    const inputLower = input.toLowerCase();
-                    return Promise.resolve(
-                        choices.filter(choice => choice.title.toLowerCase().includes(inputLower))
-                    );
-                }
-            });
-
-            if (!selectedRepo) {
-                console.log(chalk.yellow('[WARNING]'), 'No repository selected. Exiting.');
-                process.exit(0);
-            }
+        return idx;
+      })
+      .filter((idx) => idx !== null);
 
-            // 2. Select Sub-skills (if applicable)
-            if (selectedRepo.hasSkillsDir) {
-                const skillsDir = path.join(selectedRepo.path, 'skills');
-                const subSkills = listDirectories(skillsDir);
-
-                if (subSkills.length > 0) {
-                    const { selectedSubSkills } = await prompts({
-                        type: 'multiselect',
-                        name: 'selectedSubSkills',
-                        message: `Select skills from ${chalk.cyan(selectedRepo.name)} (Space to select):`,
-                        choices: [
-                            ...subSkills.map(s => ({ title: s, value: path.join(skillsDir, s) })),
-                            { title: 'Link entire repo', value: selectedRepo.path }
-                        ],
-                        hint: '- Space to select. Return to submit'
-                    });
-
-                    if (!selectedSubSkills || selectedSubSkills.length === 0) {
-                        console.log(chalk.yellow('[WARNING]'), 'No skills selected. Exiting.');
-                        process.exit(0);
-                    }
-                    skillPaths = selectedSubSkills;
-                } else {
-                    skillPaths = [selectedRepo.path];
-                }
-            } else {
-                skillPaths = [selectedRepo.path];
-            }
-        }
+    if (selectedAgents.length === 0) {
+      console.error(chalk.red("[ERROR]"), "No valid agents specified");
+      process.exit(1);
     }
-
-    // Validate skill paths
-    for (const p of skillPaths) {
-        if (!dirExists(p)) {
-            console.error(chalk.red('[ERROR]'), `Skill directory not found: ${p}`);
-            process.exit(1);
-        }
+  } else {
+    // Use all installed agents
+    selectedAgents = installedIndices;
+    if (selectedAgents.length === 0) {
+      console.error(
+        chalk.red("[ERROR]"),
+        "No installed agents detected. Please specify --agent.",
+      );
+      process.exit(1);
     }
-
-    if (skillPaths.length > 1) {
-        console.log(chalk.blue('[INFO]'), `Selected ${skillPaths.length} skills`);
-    } else {
-        const skillName = path.basename(skillPaths[0]);
-        console.log(chalk.blue('[INFO]'), `Selected Skill: ${chalk.cyan(skillName)} (${skillPaths[0]})`);
+  }
+
+  console.log(
+    chalk.blue("[INFO]"),
+    `Installing to ${selectedAgents.length} agent(s): ${selectedAgents.map((i) => agents[i].name).join(", ")}`,
+  );
+
+  // Determine scope
+  let scope = options.scope ? options.scope.toLowerCase() : "both";
+  if (!["project", "global", "both"].includes(scope)) {
+    console.error(
+      chalk.red("[ERROR]"),
+      `Invalid scope: ${scope}. Use: project, global, or both`,
+    );
+    process.exit(1);
+  }
+  console.log(chalk.blue("[INFO]"), `Scope: ${scope}`);
+
+  // Process each selected agent
+  for (const agentIndex of selectedAgents) {
+    const agent = agents[agentIndex];
+
+    console.log("");
+    console.log(
+      chalk.blue("[INFO]"),
+      `Configuring for ${chalk.cyan(agent.name)}...`,
+    );
+
+    const targets = [];
+    if (scope === "project" || scope === "both") {
+      targets.push(path.join(process.cwd(), agent.projectDir));
     }
-
-    // Agent selection
-    const agents = getAllAgents();
-    const installedIndices = detectInstalledAgents();
-
-    const { selectedAgents } = await prompts({
-        type: 'multiselect',
-        name: 'selectedAgents',
-        message: 'Select agents to install to (Space to select, Enter to confirm):',
-        choices: agents.map((agent, index) => ({
-            title: agent.name + (installedIndices.includes(index) ? chalk.green(' (Installed)') : ''),
-            value: index,
-            selected: installedIndices.includes(index)
-        })),
-        hint: '- Space to select. Return to submit'
-    });
-
-    if (!selectedAgents || selectedAgents.length === 0) {
-        console.log(chalk.yellow('[WARNING]'), 'No agents selected. Exiting.');
-        process.exit(0);
+    if (scope === "global" || scope === "both") {
+      targets.push(agent.globalDir);
     }
 
-    // Process each selected agent
-    for (const agentIndex of selectedAgents) {
-        const agent = agents[agentIndex];
-
-        console.log('');
-        console.log(chalk.blue('[INFO]'), `Configuring for ${chalk.cyan(agent.name)}...`);
-
-        const { scope } = await prompts({
-            type: 'select',
-            name: 'scope',
-            message: 'Select scope:',
-            choices: [
-                { title: `Project (${agent.projectDir})`, value: 'project' },
-                { title: `Global (${agent.globalDir})`, value: 'global' },
-                { title: 'Both', value: 'both' },
-                { title: 'Skip', value: 'skip' }
-            ]
-        });
-
-        if (scope === 'skip') continue;
-
-        const targets = [];
-        if (scope === 'project' || scope === 'both') {
-            targets.push(path.join(process.cwd(), agent.projectDir));
-        }
-        if (scope === 'global' || scope === 'both') {
-            targets.push(agent.globalDir);
+    for (const targetBase of targets) {
+      ensureDir(targetBase);
+
+      // Loop through all selected skills and link them
+      for (const sPath of skillPaths) {
+        const sName = path.basename(sPath);
+        const targetLink = path.join(targetBase, sName);
+
+        if (dirExists(targetLink)) {
+          if (!options.yes) {
+            console.log(
+              chalk.yellow("[WARNING]"),
+              `Already exists: ${targetLink}. Use --yes to overwrite.`,
+            );
+            continue;
+          }
+          console.log(
+            chalk.blue("[INFO]"),
+            `Overwriting existing: ${targetLink}`,
+          );
         }
 
-        for (const targetBase of targets) {
-            ensureDir(targetBase);
-
-            // Loop through all selected skills and link them
-            for (const sPath of skillPaths) {
-                const sName = path.basename(sPath);
-                const targetLink = path.join(targetBase, sName);
-
-                if (dirExists(targetLink)) {
-                    // Check if already correct link to avoid prompt
-                    // But for simplicity, we prompt or skip. 
-                    // To avoid spamming prompts for multiple skills, maybe auto-overwrite or ask once?
-                    // Let's ask individually for safety for now, or maybe just log and skip if overwrite not confirmed.
-                    // Actually, prompting for every file in a loop is annoying. 
-                    // Let's check overlap first? Or just try createSymlink which handles unlink.
-
-                    // Let's prompt once per agent/target if any conflicts? Too complex.
-                    // Simple approach: Prompt for each conflict.
-                    const { overwrite } = await prompts({
-                        type: 'confirm',
-                        name: 'overwrite',
-                        message: `${targetLink} already exists. Overwrite?`,
-                        initial: false
-                    });
-
-                    if (!overwrite) {
-                        console.log(chalk.blue('[INFO]'), `Skipping ${sName}...`);
-                        continue;
-                    }
-                }
-
-                if (createSymlink(sPath, targetLink)) {
-                    console.log(chalk.green('[SUCCESS]'), `Linked ${sName}`);
-                } else {
-                    console.error(chalk.red('[ERROR]'), `Failed to link ${sName}`);
-                }
-            }
+        if (createSymlink(sPath, targetLink)) {
+          console.log(chalk.green("[SUCCESS]"), `Linked ${sName}`);
+        } else {
+          console.error(chalk.red("[ERROR]"), `Failed to link ${sName}`);
         }
+      }
     }
+  }
 
-    console.log('');
-    console.log(chalk.green('[SUCCESS]'), 'All operations completed.');
+  console.log("");
+  console.log(chalk.green("[SUCCESS]"), "All operations completed.");
 }
 
 module.exports = install;