work-agent 0.1.0

package/lib/market-types.ts

@@ -0,0 +1,51 @@
+/**
+ * 技能市场类型定义
+ */
+
+export type MarketSkillStatus = 'pending' | 'approved' | 'rejected' | 'archived';
+
+export interface MarketSkill {
+  id: string;
+  name: string;
+  displayName: string;
+  description: string;
+  version: string;
+  author: string;
+  authorName?: string;
+  status: MarketSkillStatus;
+  isOfficial: boolean;
+  ratingAvg: number;
+  ratingCount: number;
+  downloadCount: number;
+  reviewedBy?: string;
+  reviewedAt?: string;
+  reviewComment?: string;
+  createdAt: string;
+  updatedAt: string;
+}
+
+export interface UploadToMarketRequest {
+  userId: string;
+  userName?: string;
+  skillName: string;  // 用户私有技能的名称
+}
+
+export interface MarketSkillDetail extends MarketSkill {
+  content: string;  // SKILL.md 内容
+  riskConfig?: any;
+}
+
+export interface SkillRating {
+  id: string;
+  skillId: string;
+  userId: string;
+  rating: number;
+  comment?: string;
+  createdAt: string;
+}
+
+export interface RateSkillRequest {
+  userId: string;
+  rating: number;  // 1-5
+  comment?: string;
+}

package/lib/memory-tool.ts

@@ -0,0 +1,211 @@
+/**
+ * 用户记忆工具
+ *
+ * 提供 save_memory 工具，让 AI 可以将用户要求记住的内容保存到记忆文件中。
+ */
+
+import { Type } from '@sinclair/typebox';
+import type { ToolDefinition, AgentToolResult, ExtensionContext } from '@mariozechner/pi-coding-agent';
+import { saveMemory, loadUserMemories } from './memory';
+
+/**
+ * 创建带用户上下文的记忆工具
+ */
+export function createMemoryTool(userId: string | undefined): ToolDefinition<typeof saveMemorySchema, SaveMemoryDetails> {
+	return {
+		name: 'save_memory',
+		label: 'Save Memory',
+		description: `将用户要求记住的内容保存到用户记忆中。
+
+使用场景：
+- 用户明确说"记下来"、"帮我记住"、"这个很重要"
+- 用户提供了重要的信息，如偏好、习惯、配置等
+- 用户要求 AI 记住某些内容以便后续使用
+
+注意：此工具用于保存用户个人的记忆，如偏好、习惯等。
+如果要保存共享知识（如解决方案、文档），请使用 save_knowledge 工具。
+
+参数说明：
+- content: 要记住的内容，应该简洁明了地描述需要记忆的信息
+- category: 可选分类，如"偏好"、"项目"、"技能"等`,
+	
+		parameters: saveMemorySchema,
+	
+		async execute(
+			toolCallId: string,
+			params: any,
+			signal: AbortSignal | undefined,
+			onUpdate: ((result: AgentToolResult<SaveMemoryDetails>) => void) | undefined,
+			ctx: ExtensionContext
+		): Promise<AgentToolResult<SaveMemoryDetails>> {
+			try {
+				const { content, category } = params;
+	
+				if (!content) {
+					throw new Error('content is required');
+				}
+	
+				// 使用工具创建时传入的 userId
+				if (!userId) {
+					throw new Error('User ID not available');
+				}
+	
+				// 保存记忆
+				saveMemory(userId, content, category);
+	
+				const result: AgentToolResult<SaveMemoryDetails> = {
+					content: [
+						{
+							type: 'text',
+							text: `✅ 已记住以下内容：
+
+${content}
+
+${category ? `\n分类: ${category}` : ''}
+
+我会在后续对话中参考这些信息。`,
+						},
+					],
+					details: {
+						success: true,
+						userId,
+						content,
+						category,
+					},
+				};
+	
+				// 通知更新
+				onUpdate?.(result);
+	
+				return result;
+			} catch (error: any) {
+				const errorResult: AgentToolResult<SaveMemoryDetails> = {
+					content: [
+						{
+							type: 'text',
+							text: `❌ 保存记忆失败: ${error.message}`,
+						},
+					],
+					details: {
+						success: false,
+						error: error.message,
+					},
+				};
+	
+				onUpdate?.(errorResult);
+	
+				return errorResult;
+			}
+		},
+	};
+}
+
+/**
+ * 获取用户记忆的工具
+ */
+export function createGetMemoriesTool(userId: string | undefined): ToolDefinition<typeof getMemoriesSchema, GetMemoriesDetails> {
+	return {
+		name: 'get_memories',
+		label: 'Get Memories',
+		description: `获取用户已保存的记忆内容。
+
+此工具用于在需要时查看用户之前保存的记忆信息。
+通常不需要主动调用，系统会在每次对话开始时自动加载用户记忆。
+
+返回格式为 Markdown 格式的用户记忆内容。`,
+	
+		parameters: getMemoriesSchema,
+	
+		async execute(
+			toolCallId: string,
+			params: any,
+			signal: AbortSignal | undefined,
+			onUpdate: ((result: AgentToolResult<GetMemoriesDetails>) => void) | undefined,
+			ctx: ExtensionContext
+		): Promise<AgentToolResult<GetMemoriesDetails>> {
+			try {
+				if (!userId) {
+					throw new Error('User ID not available');
+				}
+	
+				const memories = loadUserMemories(userId);
+	
+				const result: AgentToolResult<GetMemoriesDetails> = {
+					content: [
+						{
+							type: 'text',
+							text: memories || '暂无记忆内容',
+						},
+					],
+					details: {
+						success: true,
+						userId,
+						hasMemories: !!memories,
+						content: memories,
+					},
+				};
+	
+				onUpdate?.(result);
+	
+				return result;
+			} catch (error: any) {
+				const errorResult: AgentToolResult<GetMemoriesDetails> = {
+					content: [
+						{
+							type: 'text',
+							text: `❌ 获取记忆失败: ${error.message}`,
+						},
+					],
+					details: {
+						success: false,
+						hasMemories: false,
+						error: error.message,
+					},
+				};
+	
+				onUpdate?.(errorResult);
+	
+				return errorResult;
+			}
+		},
+	};
+}
+
+/**
+ * 记忆保存工具的参数 Schema
+ */
+const saveMemorySchema = Type.Object({
+	content: Type.String({
+		description: '要记住的内容，应该简洁明了地描述需要记忆的信息',
+	}),
+	category: Type.Optional(Type.String({
+		description: '可选分类，如"偏好"、"项目"、"技能"等',
+	})),
+});
+
+/**
+ * 记忆保存工具的详情类型
+ */
+interface SaveMemoryDetails {
+	success: boolean;
+	userId?: string;
+	content?: string;
+	category?: string;
+	error?: string;
+}
+
+/**
+ * 获取记忆工具的参数 Schema
+ */
+const getMemoriesSchema = Type.Object({});
+
+/**
+ * 获取记忆工具的详情类型
+ */
+interface GetMemoriesDetails {
+	success: boolean;
+	userId?: string;
+	hasMemories: boolean;
+	content?: string;
+	error?: string;
+}

package/lib/memory.ts

@@ -0,0 +1,197 @@
+/**
+ * 用户记忆管理模块
+ *
+ * 管理用户的长期记忆，存储在 data/users/{userId}/memory/memories.md
+ */
+
+import { mkdirSync, writeFileSync, readFileSync, existsSync } from 'fs';
+import { join } from 'path';
+
+const DATA_DIR = join(process.cwd(), 'data');
+const MEMORY_DIR_NAME = 'memory';
+const MEMORY_FILE_NAME = 'memories.md';
+
+function getUserMemoryDir(userId: string): string {
+	return join(DATA_DIR, 'users', userId, MEMORY_DIR_NAME);
+}
+
+function getUserMemoryPath(userId: string): string {
+	return join(getUserMemoryDir(userId), MEMORY_FILE_NAME);
+}
+
+function ensureMemoryDir(userId: string): string {
+	const memDir = getUserMemoryDir(userId);
+	if (!existsSync(memDir)) {
+		mkdirSync(memDir, { recursive: true });
+	}
+	return memDir;
+}
+
+/**
+ * 解析 frontmatter 和 body
+ */
+function parseFrontmatter(content: string): { fm: string | null; body: string } {
+	const match = content.match(/^---\n([\s\S]*?)\n---\n/);
+	if (match) {
+		return { fm: match[1], body: content.slice(match[0].length) };
+	}
+	return { fm: null, body: content };
+}
+
+/**
+ * 从 frontmatter 中提取 created 日期
+ */
+function extractCreated(fm: string | null): string | null {
+	if (!fm) return null;
+	const match = fm.match(/^created:\s*(.+)$/m);
+	return match ? match[1].trim() : null;
+}
+
+/**
+ * 加载用户的记忆
+ */
+export function loadUserMemories(userId: string): string {
+	if (!userId) {
+		return '';
+	}
+
+	const memPath = getUserMemoryPath(userId);
+
+	if (!existsSync(memPath)) {
+		return '';
+	}
+
+	try {
+		const content = readFileSync(memPath, 'utf-8');
+		return content;
+	} catch (error) {
+		console.error(`[memory] Failed to load user memories for ${userId}:`, error);
+		return '';
+	}
+}
+
+/**
+ * 追加新的记忆内容（AI 调用）
+ */
+export function saveMemory(userId: string, newContent: string, category?: string): void {
+	if (!userId) {
+		throw new Error('userId is required');
+	}
+
+	const memPath = getUserMemoryPath(userId);
+	ensureMemoryDir(userId);
+
+	const now = new Date().toISOString().split('T')[0];
+
+	let existing = '';
+	if (existsSync(memPath)) {
+		try {
+			existing = readFileSync(memPath, 'utf-8');
+		} catch (error) {
+			console.error(`[memory] Failed to read existing memories:`, error);
+		}
+	}
+
+	const { fm, body } = parseFrontmatter(existing);
+	const created = extractCreated(fm);
+
+	// 追加新内容到 body 末尾
+	const newSection = category
+		? `\n## ${category} - ${now}\n${newContent}\n`
+		: `\n## ${now}\n${newContent}\n`;
+
+	const newBody = body.trim() + newSection;
+
+	const newFm = `created: ${created || now}\nupdated: ${now}`;
+	const finalContent = `---
+${newFm}
+---
+
+${newBody}`;
+
+	try {
+		writeFileSync(memPath, finalContent, 'utf-8');
+		console.log(`[memory] Saved memory for user ${userId}`);
+	} catch (error) {
+		console.error(`[memory] Failed to save memory for ${userId}:`, error);
+		throw error;
+	}
+}
+
+/**
+ * 更新整个记忆文件（前端调用）
+ */
+export function updateMemories(userId: string, content: string): void {
+	if (!userId) {
+		throw new Error('userId is required');
+	}
+
+	const memPath = getUserMemoryPath(userId);
+	ensureMemoryDir(userId);
+
+	const now = new Date().toISOString().split('T')[0];
+
+	// 清理前端传入内容（移除 MDEditor 前缀）
+	let cleanContent = content.replace(/^\[Pasted[^\n]*\n*/, '').trim();
+
+	// 如果新内容带 frontmatter，提取 body
+	const { fm: newFm, body: newBody } = parseFrontmatter(cleanContent);
+	if (newFm) {
+		cleanContent = newBody;
+	}
+
+	let finalContent: string;
+
+	if (existsSync(memPath)) {
+		try {
+			const existing = readFileSync(memPath, 'utf-8');
+			const { fm: existingFm } = parseFrontmatter(existing);
+			const created = extractCreated(existingFm);
+
+			// 保留原有 created，更新 updated
+			const fmStr = `created: ${created || now}\nupdated: ${now}`;
+			finalContent = `---
+${fmStr}
+---
+
+${cleanContent}`;
+		} catch (error) {
+			console.error(`[memory] Failed to read existing memories:`, error);
+			// 读取失败，添加新 frontmatter
+			finalContent = `---
+created: ${now}
+updated: ${now}
+---
+
+${cleanContent}`;
+		}
+	} else {
+		// 文件不存在，添加 frontmatter
+		finalContent = `---
+created: ${now}
+updated: ${now}
+---
+
+${cleanContent}`;
+	}
+
+	try {
+		writeFileSync(memPath, finalContent, 'utf-8');
+		console.log(`[memory] Updated memories for user ${userId}`);
+	} catch (error) {
+		console.error(`[memory] Failed to update memories for ${userId}:`, error);
+		throw error;
+	}
+}
+
+/**
+ * 检查用户是否有记忆
+ */
+export function hasUserMemories(userId: string): boolean {
+	if (!userId) {
+		return false;
+	}
+
+	const memPath = getUserMemoryPath(userId);
+	return existsSync(memPath);
+}