agent-resource-management 1.3.0

package/src/lib/validate.ts

@@ -0,0 +1,210 @@
+import { readFileSync, existsSync, statSync } from 'fs';
+import { execSync } from 'child_process';
+import { join, extname } from 'path';
+import { mkdtempSync, rmSync } from 'fs';
+
+export interface ValidationResult {
+  valid: boolean;
+  errors: string[];
+  warnings: string[];
+  metadata?: {
+    name?: string;
+    description?: string;
+    license?: string;
+    compatibility?: string;
+    allowedTools?: string[];
+    metadata?: Record<string, string>;
+  };
+}
+
+export function validateZip(filePath: string): ValidationResult {
+  const result: ValidationResult = {
+    valid: true,
+    errors: [],
+    warnings: [],
+  };
+
+  if (!existsSync(filePath)) {
+    result.valid = false;
+    result.errors.push(`文件不存在: ${filePath}`);
+    return result;
+  }
+
+  if (extname(filePath).toLowerCase() !== '.zip') {
+    result.valid = false;
+    result.errors.push('文件必须是 ZIP 格式');
+    return result;
+  }
+
+  const stats = statSync(filePath);
+  if (stats.size === 0) {
+    result.valid = false;
+    result.errors.push('ZIP 文件为空');
+    return result;
+  }
+
+  try {
+    const tempDir = mkdtempSync('/tmp/skill-validate-');
+    execSync(`unzip -o "${filePath}" -d "${tempDir}"`, { stdio: 'pipe' });
+
+    const entries = execSync(`unzip -l "${filePath}"`, { encoding: 'utf-8' });
+    const hasSkillMd = entries.includes('SKILL.md');
+
+    if (!hasSkillMd) {
+      result.valid = false;
+      result.errors.push('ZIP 包内缺少 SKILL.md 文件');
+    }
+
+    const skillMdPath = join(tempDir, 'SKILL.md');
+    if (existsSync(skillMdPath)) {
+      const skillMdContent = readFileSync(skillMdPath, 'utf-8');
+      const frontmatterMatch = skillMdContent.match(/^---\n([\s\S]*?)\n---/);
+
+      if (frontmatterMatch) {
+        const frontmatter = frontmatterMatch[1];
+        result.metadata = {};
+
+        const nameMatch = frontmatter.match(/name:\s*(.+)/);
+        const descMatch = frontmatter.match(/description:\s*(.+)/);
+        const licenseMatch = frontmatter.match(/license:\s*(.+)/);
+        const compatMatch = frontmatter.match(/compatibility:\s*(.+)/);
+        const toolsMatch = frontmatter.match(/allowed-tools:\s*(.+)/);
+        const metaMatch = frontmatter.match(/metadata:\s*([\s\S]*?)(?=\n\w|$)/);
+
+        if (nameMatch) {
+          const name = nameMatch[1].trim();
+          if (name.length < 1 || name.length > 64) {
+            result.valid = false;
+            result.errors.push('name 长度必须在 1-64 字符之间');
+          }
+          if (!/^[a-z0-9-]+$/.test(name)) {
+            result.valid = false;
+            result.errors.push('name 只能包含小写字母、数字和连字符');
+          }
+          result.metadata.name = name;
+        } else {
+          result.valid = false;
+          result.errors.push('缺少 name 字段');
+        }
+
+        if (descMatch) {
+          const desc = descMatch[1].trim();
+          if (desc.length < 1 || desc.length > 1024) {
+            result.valid = false;
+            result.errors.push('description 长度必须在 1-1024 字符之间');
+          }
+          result.metadata.description = desc;
+        } else {
+          result.valid = false;
+          result.errors.push('缺少 description 字段');
+        }
+
+        if (licenseMatch) {
+          result.metadata.license = licenseMatch[1].trim();
+        }
+
+        if (compatMatch) {
+          result.metadata.compatibility = compatMatch[1].trim();
+        }
+
+        if (toolsMatch) {
+          result.metadata.allowedTools = toolsMatch[1].split(',').map(t => t.trim());
+        }
+
+        if (metaMatch) {
+          try {
+            result.metadata.metadata = {};
+            const metaLines = metaMatch[1].split('\n').filter(l => l.trim());
+            for (const line of metaLines) {
+              const [key, ...valueParts] = line.split(':');
+              if (key && valueParts.length) {
+                result.metadata.metadata[key.trim()] = valueParts.join(':').trim();
+              }
+            }
+          } catch {
+            result.warnings.push('metadata 格式解析失败');
+          }
+        }
+      } else {
+        result.valid = false;
+        result.errors.push('SKILL.md 缺少 frontmatter (--- 包裹的 YAML)');
+      }
+    }
+
+    rmSync(tempDir, { recursive: true, force: true });
+  } catch (err) {
+    result.valid = false;
+    result.errors.push(`解压失败: ${err instanceof Error ? err.message : '未知错误'}`);
+  }
+
+  return result;
+}
+
+export function validateSkillDir(dirPath: string): ValidationResult {
+  const result: ValidationResult = {
+    valid: true,
+    errors: [],
+    warnings: [],
+  };
+
+  if (!existsSync(dirPath)) {
+    result.valid = false;
+    result.errors.push(`目录不存在: ${dirPath}`);
+    return result;
+  }
+
+  const skillMdPath = join(dirPath, 'SKILL.md');
+  if (!existsSync(skillMdPath)) {
+    result.valid = false;
+    result.errors.push('目录内缺少 SKILL.md 文件');
+    return result;
+  }
+
+  try {
+    const skillMdContent = readFileSync(skillMdPath, 'utf-8');
+    const frontmatterMatch = skillMdContent.match(/^---\n([\s\S]*?)\n---/);
+
+    if (frontmatterMatch) {
+      const frontmatter = frontmatterMatch[1];
+      result.metadata = {};
+
+      const nameMatch = frontmatter.match(/name:\s*(.+)/);
+      const descMatch = frontmatter.match(/description:\s*(.+)/);
+
+      if (nameMatch) {
+        const name = nameMatch[1].trim();
+        if (name.length < 1 || name.length > 64) {
+          result.valid = false;
+          result.errors.push('name 长度必须在 1-64 字符之间');
+        }
+        if (!/^[a-z0-9-]+$/.test(name)) {
+          result.valid = false;
+          result.errors.push('name 只能包含小写字母、数字和连字符');
+        }
+        result.metadata.name = name;
+      } else {
+        result.valid = false;
+        result.errors.push('缺少 name 字段');
+      }
+
+      if (descMatch) {
+        const desc = descMatch[1].trim();
+        if (desc.length < 1 || desc.length > 1024) {
+          result.valid = false;
+          result.errors.push('description 长度必须在 1-1024 字符之间');
+        }
+        result.metadata.description = desc;
+      } else {
+        result.valid = false;
+        result.errors.push('缺少 description 字段');
+      }
+    } else {
+      result.valid = false;
+      result.errors.push('SKILL.md 缺少 frontmatter (--- 包裹的 YAML)');
+    }
+  } catch (err) {
+    result.warnings.push(`SKILL.md 读取失败: ${err instanceof Error ? err.message : '未知错误'}`);
+  }
+
+  return result;
+}

package/src/main.ts

@@ -0,0 +1,235 @@
+import { login, logout, getCurrentUser } from './cmd/auth';
+import { listSkills, searchSkills, infoSkill, downloadSkill, uploadSkill, mySkills, deleteSkill, validateSkill } from './cmd/skill';
+import { listAgents, searchAgents, infoAgent, downloadAgent } from './cmd/agent';
+import { listKnowledge, searchKnowledge, infoKnowledge, downloadKnowledge, uploadKnowledge, myKnowledge, deleteKnowledge } from './cmd/knowledge';
+import { showServer, setServer } from './cmd/server';
+
+const args = process.argv.slice(2);
+const command = args[0];
+const subCommand = args[1];
+
+async function main() {
+  switch (command) {
+    case 'login':
+      if (!args[1] || !args[2]) {
+        console.error('用法: arm login <server-url> <api-key>');
+        process.exit(1);
+      }
+      await login(args[1], args[2]);
+      break;
+
+    case 'logout':
+      await logout();
+      break;
+
+    case 'server':
+      if (subCommand === 'set' && args[2]) {
+        setServer(args[2]);
+      } else {
+        showServer();
+      }
+      break;
+
+    case 'skill':
+      switch (subCommand) {
+        case 'ls':
+          await listSkills();
+          break;
+        case 'search':
+          if (!args[2]) {
+            console.error('用法: arm skill search <keyword>');
+            process.exit(1);
+          }
+          await searchSkills(args[2]);
+          break;
+        case 'info':
+          if (!args[2]) {
+            console.error('用法: arm skill info <name>');
+            process.exit(1);
+          }
+          await infoSkill(args[2]);
+          break;
+        case 'download':
+          if (!args[2]) {
+            console.error('用法: arm skill download <name> [output-dir]');
+            process.exit(1);
+          }
+          await downloadSkill(args[2], args[3]);
+          break;
+        case 'upload':
+          if (!args[2]) {
+            console.error('用法: arm skill upload <path>');
+            process.exit(1);
+          }
+          await uploadSkill(args[2]);
+          break;
+        case 'my':
+          await mySkills();
+          break;
+        case 'delete':
+          if (!args[2]) {
+            console.error('用法: arm skill delete <name>');
+            process.exit(1);
+          }
+          await deleteSkill(args[2]);
+          break;
+        case 'validate':
+          if (!args[2]) {
+            console.error('用法: arm skill validate <path>');
+            process.exit(1);
+          }
+          await validateSkill(args[2]);
+          break;
+        default:
+          console.log(`
+可用命令:
+  arm login <server-url> <api-key>   登录
+  arm logout                          登出
+  arm skill ls                        列出所有 Skill
+  arm skill search <keyword>          搜索 Skill
+  arm skill info <name>               查看 Skill 详情
+  arm skill download <name> [dir]     下载 Skill
+  arm skill upload <path>             上传 Skill
+  arm skill my                        我的发布
+  arm skill delete <name>             删除 Skill
+  arm skill validate <path>           验证 Skill 格式
+  arm server                          显示当前服务端
+  arm server set <url>                设置服务端
+`);
+      }
+      break;
+
+    case 'knowledge':
+      switch (subCommand) {
+        case 'ls':
+          await listKnowledge();
+          break;
+        case 'search':
+          if (!args[2]) {
+            console.error('用法: arm knowledge search <keyword>');
+            process.exit(1);
+          }
+          await searchKnowledge(args[2]);
+          break;
+        case 'info':
+          if (!args[2]) {
+            console.error('用法: arm knowledge info <name>');
+            process.exit(1);
+          }
+          await infoKnowledge(args[2]);
+          break;
+        case 'download':
+          if (!args[2]) {
+            console.error('用法: arm knowledge download <name> [output-dir]');
+            process.exit(1);
+          }
+          await downloadKnowledge(args[2], args[3]);
+          break;
+        case 'upload':
+          if (!args[2]) {
+            console.error('用法: arm knowledge upload <path>');
+            process.exit(1);
+          }
+          await uploadKnowledge(args[2]);
+          break;
+        case 'my':
+          await myKnowledge();
+          break;
+        case 'delete':
+          if (!args[2]) {
+            console.error('用法: arm knowledge delete <name>');
+            process.exit(1);
+          }
+          await deleteKnowledge(args[2]);
+          break;
+        default:
+          console.log(`
+可用命令:
+  arm knowledge ls                    列出所有 Knowledge
+  arm knowledge search <keyword>      搜索 Knowledge
+  arm knowledge info <name>            查看 Knowledge 详情
+  arm knowledge download <name> [dir]  下载 Knowledge
+  arm knowledge upload <path>          上传 Knowledge
+  arm knowledge my                     我的发布
+  arm knowledge delete <name>          删除 Knowledge
+`);
+      }
+      break;
+
+    case 'me':
+      await getCurrentUser();
+      break;
+
+    case 'agent':
+      switch (subCommand) {
+        case 'ls':
+          await listAgents();
+          break;
+        case 'search':
+          if (!args[2]) {
+            console.error('用法: arm agent search <keyword>');
+            process.exit(1);
+          }
+          await searchAgents(args[2]);
+          break;
+        case 'info':
+          if (!args[2]) {
+            console.error('用法: arm agent info <name>');
+            process.exit(1);
+          }
+          await infoAgent(args[2]);
+          break;
+        case 'download':
+          if (!args[2]) {
+            console.error('用法: arm agent download <name> [output-dir]');
+            process.exit(1);
+          }
+          await downloadAgent(args[2], args[3]);
+          break;
+        default:
+          console.log(`
+可用命令:
+  arm agent ls                        列出所有 Agent
+  arm agent search <keyword>          搜索 Agent
+  arm agent info <name>               查看 Agent 详情
+  arm agent download <name> [dir]     下载 Agent
+`);
+      }
+      break;
+
+    default:
+      console.log(`
+Agent Resource Management (arm)
+
+用法:
+  arm login <server-url> <api-key>   登录
+  arm logout                          登出
+  arm skill ls                        列出所有 Skill
+  arm skill search <keyword>          搜索 Skill
+  arm skill info <name>               查看 Skill 详情
+  arm skill download <name> [dir]    下载 Skill
+  arm skill upload <path>             上传 Skill
+  arm skill my                        我的发布
+  arm skill delete <name>             删除 Skill
+  arm skill validate <path>           验证 Skill 格式
+  arm knowledge ls                    列出所有 Knowledge
+  arm knowledge search <keyword>      搜索 Knowledge
+  arm knowledge info <name>           查看 Knowledge 详情
+  arm knowledge download <name> [dir] 下载 Knowledge
+  arm knowledge upload <path>         上传 Knowledge
+  arm knowledge my                    我的发布
+  arm knowledge delete <name>        删除 Knowledge
+  arm agent ls                        列出所有 Agent
+  arm agent search <keyword>          搜索 Agent
+  arm agent info <name>               查看 Agent 详情
+  arm agent download <name> [dir]     下载 Agent
+  arm server                          显示当前服务端
+  arm server set <url>                设置服务端
+`);
+  }
+}
+
+main().catch((err) => {
+  console.error('Error:', err);
+  process.exit(1);
+});