work-agent 0.1.0

package/lib/skill-loader.ts

@@ -0,0 +1,234 @@
+/**
+ * 统一技能加载器
+ *
+ * 支持从多个来源加载技能：
+ * 1. 系统内置技能 (.pi/skills/)
+ * 2. 用户私有技能 (data/users/{userId}/skills/)
+ * 3. 市场已安装技能 (data/users/{userId}/skills/)
+ */
+
+import { existsSync, readFileSync, readdirSync } from 'fs';
+import { join } from 'path';
+
+/**
+ * 技能来源枚举
+ */
+export enum SkillSource {
+  System = 'system',      // 系统内置
+  User = 'user',          // 用户私有
+  Market = 'market'       // 市场已安装
+}
+
+/**
+ * 技能位置信息
+ */
+export interface SkillLocation {
+  source: SkillSource;
+  path: string;           // 实际文件系统路径
+  userId?: string;        // 对于用户技能
+  marketId?: string;      // 对于市场技能
+}
+
+/**
+ * 获取系统内置技能列表
+ */
+function loadSystemSkills(skills: Map<string, SkillLocation>): void {
+  const systemSkillsDir = join(process.cwd(), '.pi', 'skills');
+
+  if (!existsSync(systemSkillsDir)) {
+    return;
+  }
+
+  const entries = readdirSync(systemSkillsDir, { withFileTypes: true });
+
+  for (const entry of entries) {
+    if (!entry.isDirectory()) continue;
+
+    const skillName = entry.name;
+    const skillPath = join(systemSkillsDir, skillName);
+
+    // 验证技能文件存在
+    if (existsSync(join(skillPath, 'SKILL.md'))) {
+      skills.set(skillName, {
+        source: SkillSource.System,
+        path: skillPath,
+      });
+    }
+  }
+}
+
+/**
+ * 获取用户私有技能列表
+ */
+function loadUserSkills(userId: string | undefined, skills: Map<string, SkillLocation>): void {
+  // 如果没有userId，跳过用户技能加载
+  if (!userId) {
+    return;
+  }
+
+  const userSkillsDir = join(process.cwd(), 'data', 'skills', 'users', userId);
+
+  if (!existsSync(userSkillsDir)) {
+    return;
+  }
+
+  const entries = readdirSync(userSkillsDir, { withFileTypes: true });
+
+  for (const entry of entries) {
+    if (!entry.isDirectory()) continue;
+
+    const skillDirName = entry.name;
+    const skillPath = join(userSkillsDir, skillDirName);
+
+    // 验证技能文件存在
+    if (existsSync(join(skillPath, 'SKILL.md'))) {
+      // 读取 SKILL.json 获取显示名称
+      const skillJsonPath = join(skillPath, 'SKILL.json');
+      let displayName = skillDirName;
+      try {
+        if (existsSync(skillJsonPath)) {
+          const skillData = JSON.parse(readFileSync(skillJsonPath, 'utf-8'));
+          displayName = skillData.skillName || skillData.displayName || skillDirName;
+        }
+      } catch {}
+
+      // 同时用目录名和显示名称作为 key
+      skills.set(skillDirName, {
+        source: SkillSource.User,
+        path: skillPath,
+        userId,
+      });
+      if (displayName !== skillDirName) {
+        skills.set(displayName, {
+          source: SkillSource.User,
+          path: skillPath,
+          userId,
+        });
+      }
+    }
+  }
+}
+
+/**
+ * 获取市场已安装技能列表
+ */
+function loadMarketInstalledSkills(userId: string | undefined, skills: Map<string, SkillLocation>): void {
+  // 如果没有userId，跳过市场技能加载
+  if (!userId) {
+    return;
+  }
+
+  try {
+    // 从数据库查询用户已安装的技能
+    const { getUserInstalledSkills, getMarketSkillById } = require('./market-db');
+    const installedSkillIds = getUserInstalledSkills(userId);
+
+    for (const skillId of installedSkillIds) {
+      const skill = getMarketSkillById(skillId);
+      if (!skill) {
+        console.warn(`[skill-loader] Market skill ${skillId} not found in database`);
+        continue;
+      }
+
+      // 市场技能解压到用户目录下，使用技能名称
+      const userSkillsDir = join(process.cwd(), 'data', 'users', userId, 'skills');
+      const skillPath = join(userSkillsDir, skill.name);
+
+      // 验证技能文件存在
+      if (existsSync(join(skillPath, 'SKILL.md'))) {
+        skills.set(skill.name, {
+          source: SkillSource.Market,
+          path: skillPath,
+          userId,
+          marketId: skillId,
+        });
+      } else {
+        console.warn(`[skill-loader] Market skill files not found: ${skillPath}`);
+      }
+    }
+  } catch (error) {
+    console.error('[skill-loader] Failed to load market installed skills:', error);
+  }
+}
+
+/**
+ * 获取用户可用的所有技能位置
+ *
+ * @param userId 用户ID（可选，如果不提供则只加载系统技能）
+ * @returns Map<技能名称, 技能位置>
+ */
+export function getUserAvailableSkills(userId?: string): Map<string, SkillLocation> {
+  const skills = new Map<string, SkillLocation>();
+
+  // 1. 加载系统内置技能（优先级最低，放在前面被覆盖）
+  loadSystemSkills(skills);
+
+  // 2. 加载用户私有技能（会覆盖同名系统技能）
+  loadUserSkills(userId, skills);
+
+  // 3. 加载市场已安装技能（优先级最高）
+  loadMarketInstalledSkills(userId, skills);
+
+  return skills;
+}
+
+/**
+ * 从路径加载技能内容
+ *
+ * @param skillPath 技能目录路径
+ * @returns SKILL.md 内容，如果不存在则返回 null
+ */
+export function loadSkillFromPath(skillPath: string): string | null {
+  const skillMdPath = join(skillPath, 'SKILL.md');
+
+  if (!existsSync(skillMdPath)) {
+    return null;
+  }
+
+  try {
+    return readFileSync(skillMdPath, 'utf-8');
+  } catch (error) {
+    console.error(`[skill-loader] Failed to load skill from ${skillPath}:`, error);
+    return null;
+  }
+}
+
+/**
+ * 根据技能名称和用户ID加载技能内容
+ *
+ * @param skillName 技能名称
+ * @param userId 用户ID（可选）
+ * @returns SKILL.md 内容，如果不存在则返回 null
+ */
+export function loadSkillContentByName(skillName: string, userId?: string): string | null {
+  const locations = getUserAvailableSkills(userId);
+  const location = locations.get(skillName);
+
+  if (!location) {
+    return null;
+  }
+
+  return loadSkillFromPath(location.path);
+}
+
+/**
+ * 获取技能元数据 (SKILL.json)
+ *
+ * @param skillPath 技能目录路径
+ * @returns SKILL.json 内容，如果不存在则返回 null
+ */
+export function loadSkillMetaFromPath(skillPath: string): any | null {
+  const skillJsonPath = join(skillPath, 'SKILL.json');
+
+  if (!existsSync(skillJsonPath)) {
+    return null;
+  }
+
+  try {
+    const content = readFileSync(skillJsonPath, 'utf-8');
+    return JSON.parse(content);
+  } catch (error) {
+    console.error(`[skill-loader] Failed to load skill meta from ${skillPath}:`, error);
+    return null;
+  }
+}

package/lib/skill-tool.ts

@@ -0,0 +1,188 @@
+/**
+ * Skill创建工具
+ *
+ * 提供 create_skill 工具，让 AI 可以通过对话创建通用Skill。
+ */
+
+import { Type } from '@sinclair/typebox';
+import type { ToolDefinition, AgentToolResult, ExtensionContext } from '@mariozechner/pi-coding-agent';
+import { createSkill } from './skill-generator';
+import type { CreateSkillRequest, SkillType } from './skill-types';
+
+/**
+ * Skill创建工具的参数 Schema
+ */
+const createSkillSchema = Type.Object({
+  name: Type.String({
+    description: 'Skill名称（小写字母、数字和短横线，如 "docker-ops"）',
+  }),
+  displayName: Type.Optional(Type.String({
+    description: '显示名称（如 "Docker容器管理"）',
+  })),
+  description: Type.String({
+    description: '自然语言描述skill的功能和用途',
+  }),
+  type: Type.Optional(Type.Union([
+    Type.Literal('documentation'),
+    Type.Literal('command'),
+    Type.Literal('workflow'),
+    Type.Literal('hybrid'),
+  ], {
+    description: 'Skill类型（可选，会根据description自动检测）：documentation=知识型, command=命令型, workflow=工作流型, hybrid=混合型',
+  })),
+  knowledge: Type.Optional(Type.String({
+    description: '额外的知识内容或AI指令（用于documentation类型）',
+  })),
+  commandPatterns: Type.Optional(Type.Array(Type.String({
+    description: '命令正则模式（如 "^kubectl\\\\s+"）',
+  }))),
+  riskConfig: Type.Optional(Type.Object({
+    default: Type.Union([
+      Type.Literal('low'),
+      Type.Literal('medium'),
+      Type.Literal('high'),
+      Type.Literal('critical'),
+    ], {
+      description: '默认风险级别',
+    }),
+    overrides: Type.Optional(Type.Record(Type.String(), Type.Union([
+      Type.Literal('low'),
+      Type.Literal('medium'),
+      Type.Literal('high'),
+      Type.Literal('critical'),
+    ], {
+      description: '命令对应的风险级别覆盖',
+    }))),
+  })),
+  examples: Type.Optional(Type.Array(Type.Object({
+    command: Type.String({
+      description: '示例命令',
+    }),
+    description: Type.String({
+      description: '示例描述',
+    }),
+    riskLevel: Type.Optional(Type.Union([
+      Type.Literal('low'),
+      Type.Literal('medium'),
+      Type.Literal('high'),
+      Type.Literal('critical'),
+    ], {
+      description: '风险级别',
+    })),
+  }))),
+  author: Type.Optional(Type.String({
+    description: '作者名称',
+  })),
+});
+
+/**
+ * Skill创建工具的详情类型
+ */
+interface CreateSkillDetails {
+  success: boolean;
+  skill?: {
+    name: string;
+    displayName: string;
+    type: SkillType;
+    path: string;
+    files: string[];
+  };
+  error?: string;
+}
+
+/**
+ * 创建带用户上下文的 Skill 工具
+ */
+export function createSkillTool(userId: string | undefined): ToolDefinition<typeof createSkillSchema, CreateSkillDetails> {
+  return {
+    name: 'create_skill',
+    label: 'Create Skill',
+    description: `通过自然语言创建新的通用Skill。
+
+Skill类型说明：
+- documentation: 知识型skill，为AI提供领域知识和指令
+- command: 命令型skill，定义命令模式和风险配置
+- workflow: 工作流型skill，多步骤自动化操作
+- hybrid: 混合型skill
+
+创建的Skill会保存到用户私有目录。
+
+示例用法：
+- 创建Docker管理skill: name="docker-ops", description="管理Docker容器的skill"
+- 创建K8s运维skill: name="k8s-ops", description="Kubernetes运维skill", type="command"
+
+注意：此工具用于创建通用skill。如需从Swagger/OpenAPI生成API skill，请使用generate-api-system功能。`,
+
+    parameters: createSkillSchema,
+
+    async execute(
+      toolCallId: string,
+      params: any,
+      signal: AbortSignal | undefined,
+      onUpdate: ((result: AgentToolResult<CreateSkillDetails>) => void) | undefined,
+      ctx: ExtensionContext
+    ): Promise<AgentToolResult<CreateSkillDetails>> {
+      try {
+        const request: CreateSkillRequest = {
+          name: params.name,
+          displayName: params.displayName || params.name,
+          description: params.description,
+          userId: userId || undefined,
+          type: params.type,
+          knowledge: params.knowledge,
+          commandPatterns: params.commandPatterns,
+          riskConfig: params.riskConfig,
+          examples: params.examples,
+          author: params.author,
+        };
+
+        const result = await createSkill(request);
+
+        if (!result.success) {
+          const errorResult: AgentToolResult<CreateSkillDetails> = {
+            content: [{ type: 'text', text: `❌ 创建Skill失败: ${result.error}` }],
+            details: { success: false, error: result.error },
+          };
+          onUpdate?.(errorResult);
+          return errorResult;
+        }
+
+        const successResult: AgentToolResult<CreateSkillDetails> = {
+          content: [{
+            type: 'text',
+            text: `✅ Skill已创建成功！
+
+**名称:** ${result.skillName}
+**类型:** ${result.type}
+**路径:** ${result.skillPath}
+**文件:** ${result.files.join(', ')}
+
+Skill已保存到您的私有目录，可以在会话中通过 @${result.skillName} 激活使用。`,
+          }],
+          details: {
+            success: true,
+            skill: {
+              name: result.skillName,
+              displayName: params.displayName || params.name,
+              type: result.type,
+              path: result.skillPath,
+              files: result.files,
+            },
+          },
+        };
+
+        onUpdate?.(successResult);
+        return successResult;
+      } catch (error: any) {
+        const errorResult: AgentToolResult<CreateSkillDetails> = {
+          content: [{ type: 'text', text: `❌ 创建Skill失败: ${error.message}` }],
+          details: { success: false, error: error.message },
+        };
+        onUpdate?.(errorResult);
+        return errorResult;
+      }
+    },
+  };
+}
+
+export default createSkillTool;

package/lib/skill-types.ts

@@ -0,0 +1,82 @@
+/**
+ * 通用Skill类型定义
+ *
+ * 定义了skill创建、管理所需的类型接口
+ */
+
+/**
+ * Skill类型枚举
+ */
+export type SkillType =
+  | 'documentation'  // 知识型：提供AI指令和知识
+  | 'command'        // 命令型：命令模式+风险配置
+  | 'workflow'       // 工作流型：多步骤操作
+  | 'hybrid';        // 混合型
+
+/**
+ * Skill风险配置
+ */
+export interface SkillRiskConfig {
+  default: 'low' | 'medium' | 'high' | 'critical';
+  overrides?: Record<string, 'low' | 'medium' | 'high' | 'critical'>;  // 命令 -> 风险级别
+}
+
+/**
+ * 示例命令定义
+ */
+export interface SkillExample {
+  command: string;
+  description: string;
+  riskLevel?: 'low' | 'medium' | 'high' | 'critical';
+}
+
+/**
+ * Skill创建请求
+ */
+export interface CreateSkillRequest {
+  name: string;              // skill名称 (小写, 短横线)
+  displayName: string;       // 显示名称
+  description: string;       // 自然语言描述skill功能
+  userId?: string;           // 用户ID（可选，不提供则保存到系统目录）
+  type?: SkillType;          // 类型（可选，自动检测）
+  author?: string;
+
+  // 知识内容（用于documentation类型）
+  knowledge?: string;        // 额外的知识/指令
+
+  // 命令模式（用于command类型）
+  commandPatterns?: string[];  // 如 ["^kubectl\\s+", "^docker\\s+"]
+
+  // 风险配置
+  riskConfig?: SkillRiskConfig;
+
+  // 示例命令
+  examples?: SkillExample[];
+}
+
+/**
+ * Skill创建结果
+ */
+export interface CreateSkillResult {
+  success: boolean;
+  skillName: string;
+  skillPath: string;
+  type: SkillType;
+  files: string[];
+  error?: string;
+}
+
+/**
+ * Skill元数据（SKILL.json的结构）
+ */
+export interface SkillMetadata {
+  name: string;
+  displayName: string;
+  description: string;
+  version: string;
+  author: string;
+  type: SkillType;
+  createdAt: string;
+  commandPatterns?: string[];
+  riskConfig?: SkillRiskConfig;
+}

package/lib/skills-init.ts

@@ -0,0 +1,58 @@
+/**
+ * Skills Directory Initialization
+ *
+ * Ensures that all required skills directories exist at runtime.
+ * This module is imported early to create directories before they're needed.
+ */
+
+import { mkdirSync, existsSync } from 'fs';
+import { join } from 'path';
+
+/**
+ * Base directory for skills data
+ */
+export const SKILLS_BASE_DIR = join(process.cwd(), 'data', 'skills');
+
+/**
+ * Subdirectories for different skill types
+ */
+export const SKILLS_DIRS = {
+	users: join(SKILLS_BASE_DIR, 'users'),      // User private skills
+	market: join(SKILLS_BASE_DIR, 'market'),    // Skills from marketplace
+	tmp: join(SKILLS_BASE_DIR, 'tmp'),          // Temporary files for uploads/downloads
+} as const;
+
+/**
+ * Initialize skills directories
+ * Creates all required directories if they don't exist
+ *
+ * This function is safe to call multiple times - it will not fail
+ * if directories already exist.
+ */
+export function initializeSkillsDirs(): void {
+	for (const [name, dir] of Object.entries(SKILLS_DIRS)) {
+		if (!existsSync(dir)) {
+			try {
+				mkdirSync(dir, { recursive: true });
+				console.log(`[skills-init] Created directory: ${name} -> ${dir}`);
+			} catch (error) {
+				console.error(`[skills-init] Failed to create directory ${name}:`, error);
+				// Re-throw critical errors so the app can fail fast
+				throw error;
+			}
+		}
+	}
+
+	// Also ensure base directory exists
+	if (!existsSync(SKILLS_BASE_DIR)) {
+		try {
+			mkdirSync(SKILLS_BASE_DIR, { recursive: true });
+		} catch (error) {
+			console.error('[skills-init] Failed to create base directory:', error);
+			throw error;
+		}
+	}
+}
+
+// Auto-initialize on module import
+initializeSkillsDirs();