byteplan-cli 1.0.2 → 1.2.0

package/src/commands/skills.js

@@ -0,0 +1,279 @@
+#!/usr/bin/env node
+
+/**
+ * Skills management commands for BytePlan CLI
+ */
+
+import { Command } from 'commander';
+import { homedir } from 'os';
+import path from 'path';
+import { existsSync, mkdirSync, readdirSync, cpSync, readFileSync } from 'fs';
+import { createRequire } from 'module';
+
+const require = createRequire(import.meta.url);
+
+// Skills 源目录（在 npm 包中）
+const getSkillsSourceDir = () => {
+  // 获取当前模块所在目录，然后向上找到项目根目录
+  const currentDir = path.dirname(new URL(import.meta.url).pathname);
+  // 从 src/commands/skills.js -> src -> 项目根目录 -> skills
+  return path.join(currentDir, '..', '..', 'skills');
+};
+
+// Skills 默认目标目录（仅用于提示）
+const getDefaultSkillsTargetDir = () => {
+  return path.join(homedir(), '.claude', 'skills');
+};
+
+function printJSON(data) {
+  console.log(JSON.stringify(data, null, 2));
+}
+
+function printError(error) {
+  printJSON({ error: true, message: error.message });
+  process.exit(1);
+}
+
+// 获取 skill 信息
+function getSkillInfo(skillPath) {
+  const skillMdPath = path.join(skillPath, 'SKILL.md');
+  if (!existsSync(skillMdPath)) {
+    return null;
+  }
+
+  const content = readFileSync(skillMdPath, 'utf-8');
+
+  // 解析 frontmatter
+  const frontmatterMatch = content.match(/^---\n([\s\S]*?)\n---/);
+  if (!frontmatterMatch) {
+    return null;
+  }
+
+  const frontmatter = frontmatterMatch[1];
+  const nameMatch = frontmatter.match(/name:\s*(.+)/);
+  const descMatch = frontmatter.match(/description:\s*(.+)/);
+
+  return {
+    name: nameMatch ? nameMatch[1].trim() : path.basename(skillPath),
+    description: descMatch ? descMatch[1].trim() : '',
+    path: skillPath,
+  };
+}
+
+// 获取所有可用 skills
+function getAvailableSkills() {
+  const sourceDir = getSkillsSourceDir();
+  if (!existsSync(sourceDir)) {
+    return [];
+  }
+
+  const skills = [];
+  const dirs = readdirSync(sourceDir, { withFileTypes: true });
+
+  for (const dir of dirs) {
+    if (dir.isDirectory()) {
+      const skillPath = path.join(sourceDir, dir.name);
+      const info = getSkillInfo(skillPath);
+      if (info) {
+        skills.push(info);
+      }
+    }
+  }
+
+  return skills;
+}
+
+// 获取已安装的 skills（支持自定义目录）
+function getInstalledSkills(targetDir) {
+  if (!existsSync(targetDir)) {
+    return [];
+  }
+
+  const skills = [];
+  const dirs = readdirSync(targetDir, { withFileTypes: true });
+
+  for (const dir of dirs) {
+    if (dir.isDirectory() && dir.name.startsWith('byteplan-')) {
+      const skillPath = path.join(targetDir, dir.name);
+      const info = getSkillInfo(skillPath);
+      if (info) {
+        skills.push(info);
+      }
+    }
+  }
+
+  return skills;
+}
+
+// Skills 命令
+const skillsCmd = new Command('skills')
+  .description('Manage BytePlan Claude Code skills')
+  .requiredOption('-d, --dir <directory>', 'Target installation directory (required)')
+  .addHelpText('after', `
+Commands:
+  skills list              List all available skills
+  skills installed         List installed skills (requires -d)
+  skills install [names]   Install skills to target directory (requires -d)
+  skills install-all       Install all BytePlan skills (requires -d)
+
+Examples:
+  $ byteplan skills list
+  $ byteplan skills installed -d ~/.claude/skills
+  $ byteplan skills install byteplan-api byteplan-analysis -d ~/.claude/skills
+  $ byteplan skills install-all -d ~/.claude/skills
+  $ byteplan skills install-all -d /custom/path/skills
+`);
+
+// skills list 子命令
+skillsCmd
+  .command('list')
+  .description('List all available BytePlan skills')
+  .action(() => {
+    try {
+      const skills = getAvailableSkills();
+
+      printJSON({
+        skills: skills.map(s => ({
+          name: s.name,
+          description: s.description,
+        })),
+        total: skills.length,
+        sourceDir: getSkillsSourceDir(),
+      });
+    } catch (error) {
+      printError(error);
+    }
+  });
+
+// skills installed 子命令
+skillsCmd
+  .command('installed')
+  .description('List installed BytePlan skills in target directory')
+  .action(() => {
+    try {
+      // 获取父命令的 -d 选项（requiredOption 已确保必传）
+      const parentOptions = skillsCmd.opts();
+      const targetDir = parentOptions.dir;
+      const skills = getInstalledSkills(targetDir);
+
+      printJSON({
+        skills: skills.map(s => ({
+          name: s.name,
+          description: s.description,
+        })),
+        total: skills.length,
+        targetDir: targetDir,
+      });
+    } catch (error) {
+      printError(error);
+    }
+  });
+
+// skills install 子命令
+skillsCmd
+  .command('install [names...]')
+  .description('Install specified skills to target directory')
+  .option('-a, --all', 'Install all BytePlan skills', false)
+  .addHelpText('after', `
+Examples:
+  $ byteplan skills install byteplan-api byteplan-analysis -d ~/.claude/skills
+  $ byteplan skills install --all -d ~/.claude/skills
+  $ byteplan skills install byteplan-api -d /custom/path/skills
+`)
+  .action(async (names, options) => {
+    try {
+      // 获取父命令的 -d 选项（requiredOption 已确保必传）
+      const parentOptions = skillsCmd.opts();
+      const targetDir = parentOptions.dir;
+      const sourceDir = getSkillsSourceDir();
+
+      // 确保目标目录存在
+      if (!existsSync(targetDir)) {
+        mkdirSync(targetDir, { recursive: true });
+      }
+
+      // 获取要安装的 skills
+      let toInstall = names || [];
+      if (options.all || toInstall.length === 0) {
+        const available = getAvailableSkills();
+        toInstall = available.map(s => s.name);
+      }
+
+      // 验证 skills 存在
+      const availableSkills = getAvailableSkills();
+      const availableNames = availableSkills.map(s => s.name);
+
+      const invalid = toInstall.filter(n => !availableNames.includes(n));
+      if (invalid.length > 0) {
+        printJSON({
+          error: true,
+          message: `Unknown skills: ${invalid.join(', ')}`,
+          available: availableNames,
+        });
+        process.exit(1);
+      }
+
+      // 安装 skills
+      const installed = [];
+      for (const skillName of toInstall) {
+        const skillSourcePath = path.join(sourceDir, skillName);
+        const skillTargetPath = path.join(targetDir, skillName);
+
+        // 复制 skill 目录
+        cpSync(skillSourcePath, skillTargetPath, { recursive: true });
+        installed.push(skillName);
+      }
+
+      printJSON({
+        success: true,
+        installed: installed,
+        total: installed.length,
+        targetDir: targetDir,
+      });
+    } catch (error) {
+      printError(error);
+    }
+  });
+
+// skills install-all 子命令（快捷方式）
+skillsCmd
+  .command('install-all')
+  .description('Install all BytePlan skills to target directory')
+  .action(async () => {
+    try {
+      // 获取父命令的 -d 选项（requiredOption 已确保必传）
+      const parentOptions = skillsCmd.opts();
+      const targetDir = parentOptions.dir;
+      const sourceDir = getSkillsSourceDir();
+
+      // 确保目标目录存在
+      if (!existsSync(targetDir)) {
+        mkdirSync(targetDir, { recursive: true });
+      }
+
+      // 获取所有 skills
+      const availableSkills = getAvailableSkills();
+
+      // 安装所有 skills
+      const installed = [];
+      for (const skill of availableSkills) {
+        const skillSourcePath = path.join(sourceDir, skill.name);
+        const skillTargetPath = path.join(targetDir, skill.name);
+
+        // 复制 skill 目录
+        cpSync(skillSourcePath, skillTargetPath, { recursive: true });
+        installed.push(skill.name);
+      }
+
+      printJSON({
+        success: true,
+        installed: installed,
+        total: installed.length,
+        targetDir: targetDir,
+      });
+    } catch (error) {
+      printError(error);
+    }
+  });
+
+export { skillsCmd };