tools-cc 1.0.3 → 1.0.5

package/src/core/project.ts

@@ -1,87 +1,161 @@
 import fs from 'fs-extra';
 import path from 'path';
-import { ProjectConfig } from '../types';
+import { ProjectConfig, LegacyProjectConfig, normalizeProjectConfig, SourceSelection, ExportConfig } from '../types';
 import { loadManifest } from './manifest';
 import { getToolsccDir, getProjectConfigPath } from '../utils/path';
 
+/**
+ * 默认选择配置 - 导入所有内容
+ */
+const DEFAULT_SELECTION: SourceSelection = {
+  skills: ['*'],
+  commands: ['*'],
+  agents: ['*']
+};
+
 export async function initProject(projectDir: string): Promise<void> {
   const toolsccDir = getToolsccDir(projectDir);
   const configFile = getProjectConfigPath(projectDir);
-  
+
   // Create .toolscc directory structure
   await fs.ensureDir(path.join(toolsccDir, 'skills'));
   await fs.ensureDir(path.join(toolsccDir, 'commands'));
   await fs.ensureDir(path.join(toolsccDir, 'agents'));
-  
+
   // Create project config if not exists
   if (!(await fs.pathExists(configFile))) {
     const config: ProjectConfig = {
-      sources: [],
+      sources: {},
       links: []
     };
     await fs.writeJson(configFile, config, { spaces: 2 });
   }
 }
 
+/**
+ * 读取项目配置，自动处理新旧格式
+ */
+async function readProjectConfig(configFile: string): Promise<ProjectConfig> {
+  const rawConfig = await fs.readJson(configFile);
+  return normalizeProjectConfig(rawConfig);
+}
+
+/**
+ * 获取源名称列表（兼容新旧格式）
+ */
+function getSourceNames(config: ProjectConfig): string[] {
+  return Object.keys(config.sources);
+}
+
+/**
+ * 检查是否应该复制某项（根据选择配置）
+ */
+function shouldInclude(itemName: string, selection: string[]): boolean {
+  // 如果选择包含通配符，包含所有项
+  if (selection.includes('*')) {
+    return true;
+  }
+  // 否则检查是否在选择列表中
+  return selection.includes(itemName);
+}
+
 export async function useSource(
   sourceName: string,
   sourceDir: string,
-  projectDir: string
+  projectDir: string,
+  selection?: SourceSelection
 ): Promise<void> {
   // Input validation
   if (!sourceName || !sourceName.trim()) {
     throw new Error('Source name is required');
   }
-  
+
   // Check source directory existence
   if (!(await fs.pathExists(sourceDir))) {
     throw new Error(`Source directory does not exist: ${sourceDir}`);
   }
-  
+
   const toolsccDir = getToolsccDir(projectDir);
   const manifest = await loadManifest(sourceDir);
-  
+
   // Ensure project is initialized
   await initProject(projectDir);
-  
+
+  // 使用传入的选择配置或默认配置
+  const effectiveSelection: SourceSelection = selection ?? DEFAULT_SELECTION;
+
   // Copy/link skills (flattened with prefix)
   const sourceSkillsDir = path.join(sourceDir, 'skills');
   if (await fs.pathExists(sourceSkillsDir)) {
     const skills = await fs.readdir(sourceSkillsDir);
     for (const skill of skills) {
+      // 检查是否应该包含此 skill
+      if (!shouldInclude(skill, effectiveSelection.skills)) {
+        continue;
+      }
+
       const srcPath = path.join(sourceSkillsDir, skill);
-      const destPath = path.join(toolsccDir, 'skills', `${sourceName}-${skill}`);
-      
+      const name = `${sourceName}` == `${skill}` ? skill : `${sourceName}-${skill}`;
+      const destPath = path.join(toolsccDir, 'skills', name);
+
       // Remove existing if exists
       await fs.remove(destPath);
-      
+
       // Copy directory
       await fs.copy(srcPath, destPath);
     }
   }
-  
+
   // Copy commands (in subdirectory by source name)
   const sourceCommandsDir = path.join(sourceDir, 'commands');
   if (await fs.pathExists(sourceCommandsDir)) {
-    const destDir = path.join(toolsccDir, 'commands', sourceName);
-    await fs.remove(destDir);
-    await fs.copy(sourceCommandsDir, destDir);
+    // 检查是否有选择 commands
+    if (effectiveSelection.commands.includes('*')) {
+      // 复制所有 commands
+      const destDir = path.join(toolsccDir, 'commands', sourceName);
+      await fs.remove(destDir);
+      await fs.copy(sourceCommandsDir, destDir);
+    } else if (effectiveSelection.commands.length > 0) {
+      // 只复制选中的 commands
+      const destDir = path.join(toolsccDir, 'commands', sourceName);
+      await fs.ensureDir(destDir);
+      
+      for (const cmdName of effectiveSelection.commands) {
+        const srcFile = path.join(sourceCommandsDir, `${cmdName}.md`);
+        if (await fs.pathExists(srcFile)) {
+          await fs.copy(srcFile, path.join(destDir, `${cmdName}.md`));
+        }
+      }
+    }
   }
-  
+
   // Copy agents (in subdirectory by source name)
   const sourceAgentsDir = path.join(sourceDir, 'agents');
   if (await fs.pathExists(sourceAgentsDir)) {
-    const destDir = path.join(toolsccDir, 'agents', sourceName);
-    await fs.remove(destDir);
-    await fs.copy(sourceAgentsDir, destDir);
+    // 检查是否有选择 agents
+    if (effectiveSelection.agents.includes('*')) {
+      // 复制所有 agents
+      const destDir = path.join(toolsccDir, 'agents', sourceName);
+      await fs.remove(destDir);
+      await fs.copy(sourceAgentsDir, destDir);
+    } else if (effectiveSelection.agents.length > 0) {
+      // 只复制选中的 agents
+      const destDir = path.join(toolsccDir, 'agents', sourceName);
+      await fs.ensureDir(destDir);
+      
+      for (const agentName of effectiveSelection.agents) {
+        const srcFile = path.join(sourceAgentsDir, `${agentName}.md`);
+        if (await fs.pathExists(srcFile)) {
+          await fs.copy(srcFile, path.join(destDir, `${agentName}.md`));
+        }
+      }
+    }
   }
-  
-  // Update project config
+
+  // Update project config - 保存实际使用的选择配置
   const configFile = getProjectConfigPath(projectDir);
-  const config: ProjectConfig = await fs.readJson(configFile);
-  if (!config.sources.includes(sourceName)) {
-    config.sources.push(sourceName);
-  }
+  const config = await readProjectConfig(configFile);
+  config.sources[sourceName] = effectiveSelection;
   await fs.writeJson(configFile, config, { spaces: 2 });
 }
 
@@ -90,10 +164,10 @@ export async function unuseSource(sourceName: string, projectDir: string): Promi
   if (!sourceName || !sourceName.trim()) {
     throw new Error('Source name is required');
   }
-  
+
   const toolsccDir = getToolsccDir(projectDir);
   const configFile = getProjectConfigPath(projectDir);
-  
+
   // Remove skills with prefix
   const skillsDir = path.join(toolsccDir, 'skills');
   if (await fs.pathExists(skillsDir)) {
@@ -104,33 +178,100 @@ export async function unuseSource(sourceName: string, projectDir: string): Promi
       }
     }
   }
-  
+
   // Remove commands subdirectory
   await fs.remove(path.join(toolsccDir, 'commands', sourceName));
-  
+
   // Remove agents subdirectory
   await fs.remove(path.join(toolsccDir, 'agents', sourceName));
-  
+
   // Update project config with error handling
   let config: ProjectConfig;
   try {
-    config = await fs.readJson(configFile);
+    config = await readProjectConfig(configFile);
   } catch (error) {
     // If config file doesn't exist or is invalid, nothing to update
     return;
   }
-  
-  config.sources = config.sources.filter(s => s !== sourceName);
+
+  delete config.sources[sourceName];
   await fs.writeJson(configFile, config, { spaces: 2 });
 }
 
 export async function listUsedSources(projectDir: string): Promise<string[]> {
   const configFile = getProjectConfigPath(projectDir);
-  
+
   if (!(await fs.pathExists(configFile))) {
     return [];
   }
-  
-  const config: ProjectConfig = await fs.readJson(configFile);
-  return config.sources;
+
+  const config = await readProjectConfig(configFile);
+  return getSourceNames(config);
+}
+
+/**
+ * 导出项目配置到 JSON 文件
+ */
+export async function exportProjectConfig(
+  projectDir: string,
+  outputPath: string
+): Promise<void> {
+  const configFile = getProjectConfigPath(projectDir);
+
+  if (!(await fs.pathExists(configFile))) {
+    throw new Error('Project not initialized. Use `tools-cc use <source>` to get started.');
+  }
+
+  const config = await readProjectConfig(configFile);
+
+  const exportConfig: ExportConfig = {
+    version: '1.0',
+    type: 'project',
+    config,
+    exportedAt: new Date().toISOString()
+  };
+
+  await fs.writeJson(outputPath, exportConfig, { spaces: 2 });
 }
+
+/**
+ * 从 JSON 文件导入项目配置
+ */
+export async function importProjectConfig(
+  configPath: string,
+  projectDir: string,
+  resolveSourcePath: (sourceName: string) => Promise<string>
+): Promise<void> {
+  if (!(await fs.pathExists(configPath))) {
+    throw new Error(`Config file not found: ${configPath}`);
+  }
+
+  const exportConfig: ExportConfig = await fs.readJson(configPath);
+
+  // Validate version
+  if (exportConfig.version !== '1.0') {
+    throw new Error(`Unsupported config version: ${exportConfig.version}`);
+  }
+
+  // Validate type
+  if (exportConfig.type !== 'project') {
+    throw new Error(`Invalid config type: ${exportConfig.type}. Expected 'project'.`);
+  }
+
+  // Initialize project
+  await initProject(projectDir);
+
+  // Apply each source
+  for (const [sourceName, selection] of Object.entries(exportConfig.config.sources)) {
+    const sourceDir = await resolveSourcePath(sourceName);
+    await useSource(sourceName, sourceDir, projectDir, selection);
+  }
+
+  // Update links if present
+  if (exportConfig.config.links && exportConfig.config.links.length > 0) {
+    const configFile = getProjectConfigPath(projectDir);
+    const config = await readProjectConfig(configFile);
+    config.links = exportConfig.config.links;
+    await fs.writeJson(configFile, config, { spaces: 2 });
+  }
+}