work-agent 0.1.0

package/lib/file-tool.ts

@@ -0,0 +1,583 @@
+/**
+ * 文件管理工具
+ *
+ * 提供文件上传、下载、列表、删除等操作。
+ */
+
+import { Type } from '@sinclair/typebox';
+import type { ToolDefinition, AgentToolResult, ExtensionContext } from '@mariozechner/pi-coding-agent';
+import { getDocuments, createDocument, deleteDocument, updateDocument, getDocument } from './db';
+import { getStorageProvider, MAX_FILE_SIZE, isAllowedMimeType } from './file-storage';
+import type { FileMetadata } from './db';
+
+function formatFileSizeLocal(bytes: number): string {
+	if (bytes === 0) return '0 B';
+	const k = 1024;
+	const sizes = ['B', 'KB', 'MB', 'GB'];
+	const i = Math.floor(Math.log(bytes) / Math.log(k));
+	return `${parseFloat((bytes / Math.pow(k, i)).toFixed(2))} ${sizes[i]}`;
+}
+
+const listFilesSchema = Type.Object({
+	search: Type.Optional(Type.String({ description: '搜索关键词（可选）' })),
+	limit: Type.Optional(Type.Number({ description: '返回数量限制，默认20' })),
+	userId: Type.String({ description: '用户ID' }),
+});
+
+interface ListFilesDetails {
+	success: boolean;
+	count?: number;
+	files?: Array<{
+		id: string;
+		title: string;
+		size: string;
+		mimeType: string;
+		isPublic: boolean;
+		createdAt: string;
+	}>;
+	error?: string;
+}
+
+export const listFilesTool: ToolDefinition<typeof listFilesSchema, ListFilesDetails> = {
+	name: 'list_user_files',
+	label: 'List User Files',
+	description: '列出用户上传的所有文件。可以搜索特定文件名。',
+	parameters: listFilesSchema,
+
+	async execute(
+		toolCallId: string,
+		params: any,
+		signal: AbortSignal | undefined,
+		onUpdate: ((result: AgentToolResult<ListFilesDetails>) => void) | undefined,
+		ctx: ExtensionContext
+	): Promise<AgentToolResult<ListFilesDetails>> {
+		try {
+			const { search, limit = 20, userId } = params;
+
+			const documents = getDocuments({
+				author: userId,
+				type: 'file',
+				limit,
+				search,
+			}) as any[];
+
+			const files = documents.map(doc => ({
+				id: doc.id,
+				title: doc.title,
+				size: formatFileSizeLocal(doc.fileMetadata?.size || 0),
+				mimeType: doc.fileMetadata?.mimeType || 'unknown',
+				isPublic: doc.isPublic,
+				createdAt: doc.createdAt.toISOString(),
+			}));
+
+			const result: AgentToolResult<ListFilesDetails> = {
+				content: [{ type: 'text', text: `找到 ${files.length} 个文件:\n${files.map(f => `- ${f.title} (${f.size})`).join('\n')}` }],
+				details: { success: true, count: files.length, files },
+			};
+
+			onUpdate?.(result);
+			return result;
+		} catch (error: any) {
+			const errorResult: AgentToolResult<ListFilesDetails> = {
+				content: [{ type: 'text', text: `❌ 列出文件失败: ${error.message}` }],
+				details: { success: false, error: error.message },
+			};
+			onUpdate?.(errorResult);
+			return errorResult;
+		}
+	},
+};
+
+const getFileInfoSchema = Type.Object({
+	fileId: Type.String({ description: '文件ID' }),
+	userId: Type.String({ description: '用户ID' }),
+});
+
+interface GetFileInfoDetails {
+	success: boolean;
+	file?: {
+		id: string;
+		title: string;
+		description?: string;
+		size: string;
+		sizeBytes: number;
+		mimeType?: string;
+		md5?: string;
+		isPublic?: boolean;
+		createdAt: string;
+		updatedAt: string;
+	};
+	error?: string;
+}
+
+export const getFileInfoTool: ToolDefinition<typeof getFileInfoSchema, GetFileInfoDetails> = {
+	name: 'get_file_info',
+	label: 'Get File Info',
+	description: '获取文件的详细信息，包括元数据。',
+	parameters: getFileInfoSchema,
+
+	async execute(
+		toolCallId: string,
+		params: any,
+		signal: AbortSignal | undefined,
+		onUpdate: ((result: AgentToolResult<GetFileInfoDetails>) => void) | undefined,
+		ctx: ExtensionContext
+	): Promise<AgentToolResult<GetFileInfoDetails>> {
+		try {
+			const { fileId, userId } = params;
+			const doc = getDocument(fileId);
+
+			if (!doc) {
+				const errorResult: AgentToolResult<GetFileInfoDetails> = {
+					content: [{ type: 'text', text: '❌ 文件未找到' }],
+					details: { success: false, error: 'File not found' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			if (doc.author !== userId) {
+				const errorResult: AgentToolResult<GetFileInfoDetails> = {
+					content: [{ type: 'text', text: '❌ 无权访问此文件' }],
+					details: { success: false, error: 'Permission denied' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			const result: AgentToolResult<GetFileInfoDetails> = {
+				content: [{ type: 'text', text: `文件信息:\n- 名称: ${doc.title}\n- 大小: ${formatFileSizeLocal(doc.fileMetadata?.size || 0)}\n- 类型: ${doc.fileMetadata?.mimeType || 'unknown'}` }],
+				details: {
+					success: true,
+					file: {
+						id: doc.id,
+						title: doc.title,
+						description: doc.content,
+						size: formatFileSizeLocal(doc.fileMetadata?.size || 0),
+						sizeBytes: doc.fileMetadata?.size || 0,
+						mimeType: doc.fileMetadata?.mimeType,
+						md5: doc.fileMetadata?.md5,
+						isPublic: doc.isPublic,
+						createdAt: doc.createdAt.toISOString(),
+						updatedAt: doc.updatedAt.toISOString(),
+					},
+				},
+			};
+
+			onUpdate?.(result);
+			return result;
+		} catch (error: any) {
+			const errorResult: AgentToolResult<GetFileInfoDetails> = {
+				content: [{ type: 'text', text: `❌ 获取文件信息失败: ${error.message}` }],
+				details: { success: false, error: error.message },
+			};
+			onUpdate?.(errorResult);
+			return errorResult;
+		}
+	},
+};
+
+const downloadFileSchema = Type.Object({
+	fileId: Type.String({ description: '文件ID' }),
+	userId: Type.String({ description: '用户ID' }),
+	maxSize: Type.Optional(Type.Number({ description: '最大允许下载的文件大小（字节），默认1MB' })),
+});
+
+interface DownloadFileDetails {
+	success: boolean;
+	file?: {
+		id: string;
+		title: string;
+		mimeType: string;
+		size: number;
+		base64Content: string;
+	};
+	error?: string;
+}
+
+export const downloadFileTool: ToolDefinition<typeof downloadFileSchema, DownloadFileDetails> = {
+	name: 'download_file',
+	label: 'Download File',
+	description: '下载文件内容。返回文件的Base64编码内容和元数据。适用于小文件（<1MB）。',
+	parameters: downloadFileSchema,
+
+	async execute(
+		toolCallId: string,
+		params: any,
+		signal: AbortSignal | undefined,
+		onUpdate: ((result: AgentToolResult<DownloadFileDetails>) => void) | undefined,
+		ctx: ExtensionContext
+	): Promise<AgentToolResult<DownloadFileDetails>> {
+		try {
+			const { fileId, userId, maxSize = 1024 * 1024 } = params;
+			const doc = getDocument(fileId);
+
+			if (!doc) {
+				const errorResult: AgentToolResult<DownloadFileDetails> = {
+					content: [{ type: 'text', text: '❌ 文件未找到' }],
+					details: { success: false, error: 'File not found' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			if (doc.author !== userId && !doc.isPublic) {
+				const errorResult: AgentToolResult<DownloadFileDetails> = {
+					content: [{ type: 'text', text: '❌ 无权访问此文件' }],
+					details: { success: false, error: 'Permission denied' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			if (!doc.fileMetadata?.storageKey) {
+				const errorResult: AgentToolResult<DownloadFileDetails> = {
+					content: [{ type: 'text', text: '❌ 文件存储信息缺失' }],
+					details: { success: false, error: 'File storage key not found' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			if (doc.fileMetadata.size > maxSize) {
+				const errorResult: AgentToolResult<DownloadFileDetails> = {
+					content: [{ type: 'text', text: `❌ 文件太大 (${formatFileSizeLocal(doc.fileMetadata.size)})，最大允许 ${formatFileSizeLocal(maxSize)}` }],
+					details: { success: false, error: 'File too large' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			const storage = getStorageProvider();
+			const buffer = await storage.download(doc.fileMetadata.storageKey);
+			const base64 = buffer.toString('base64');
+
+			const result: AgentToolResult<DownloadFileDetails> = {
+				content: [{ type: 'text', text: `✅ 已下载文件 "${doc.title}" (${formatFileSizeLocal(buffer.length)})` }],
+				details: {
+					success: true,
+					file: {
+						id: doc.id,
+						title: doc.title,
+						mimeType: doc.fileMetadata.mimeType,
+						size: buffer.length,
+						base64Content: base64,
+					},
+				},
+			};
+
+			onUpdate?.(result);
+			return result;
+		} catch (error: any) {
+			const errorResult: AgentToolResult<DownloadFileDetails> = {
+				content: [{ type: 'text', text: `❌ 下载文件失败: ${error.message}` }],
+				details: { success: false, error: error.message },
+			};
+			onUpdate?.(errorResult);
+			return errorResult;
+		}
+	},
+};
+
+const uploadFileSchema = Type.Object({
+	filename: Type.String({ description: '文件名' }),
+	userId: Type.String({ description: '用户ID' }),
+	content: Type.Optional(Type.String({ description: '文件内容（Base64编码）' })),
+	url: Type.Optional(Type.String({ description: '文件URL（可选，优先使用）' })),
+	mimeType: Type.Optional(Type.String({ description: '文件MIME类型（可选）' })),
+	description: Type.Optional(Type.String({ description: '文件描述（可选）' })),
+	isPublic: Type.Optional(Type.Boolean({ description: '是否公开分享（默认false）' })),
+});
+
+interface UploadFileDetails {
+	success: boolean;
+	file?: {
+		id: string;
+		title: string;
+		size: string;
+		mimeType: string;
+		isPublic?: boolean;
+	};
+	error?: string;
+}
+
+export const uploadFileTool: ToolDefinition<typeof uploadFileSchema, UploadFileDetails> = {
+	name: 'upload_file',
+	label: 'Upload File',
+	description: '上传文件。可以从URL下载文件，或从Base64内容创建文件。',
+	parameters: uploadFileSchema,
+
+	async execute(
+		toolCallId: string,
+		params: any,
+		signal: AbortSignal | undefined,
+		onUpdate: ((result: AgentToolResult<UploadFileDetails>) => void) | undefined,
+		ctx: ExtensionContext
+	): Promise<AgentToolResult<UploadFileDetails>> {
+		try {
+			const { filename, userId, content, url, mimeType, description, isPublic } = params;
+			let buffer: Buffer;
+			let detectedMimeType = mimeType || 'application/octet-stream';
+
+			if (url) {
+				const response = await fetch(url);
+				if (!response.ok) {
+					const errorResult: AgentToolResult<UploadFileDetails> = {
+						content: [{ type: 'text', text: `❌ 下载URL失败: ${response.status}` }],
+						details: { success: false, error: `Failed to fetch URL: ${response.status}` },
+					};
+					onUpdate?.(errorResult);
+					return errorResult;
+				}
+				const arrayBuffer = await response.arrayBuffer();
+				buffer = Buffer.from(arrayBuffer);
+				detectedMimeType = response.headers.get('content-type') || detectedMimeType;
+			} else if (content) {
+				buffer = Buffer.from(content, 'base64');
+			} else {
+				const errorResult: AgentToolResult<UploadFileDetails> = {
+					content: [{ type: 'text', text: '❌ 需要提供 url 或 content 参数' }],
+					details: { success: false, error: 'Either url or content is required' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			if (buffer.length > MAX_FILE_SIZE) {
+				const errorResult: AgentToolResult<UploadFileDetails> = {
+					content: [{ type: 'text', text: `❌ 文件太大 (${formatFileSizeLocal(buffer.length)})，最大允许 ${formatFileSizeLocal(MAX_FILE_SIZE)}` }],
+					details: { success: false, error: 'File too large' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			if (!isAllowedMimeType(detectedMimeType)) {
+				const errorResult: AgentToolResult<UploadFileDetails> = {
+					content: [{ type: 'text', text: `❌ 不支持的文件类型: ${detectedMimeType}` }],
+					details: { success: false, error: `File type "${detectedMimeType}" is not allowed` },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			const storage = getStorageProvider();
+			const { storageKey, md5 } = await storage.upload({
+				userId,
+				filename,
+				mimeType: detectedMimeType,
+				buffer,
+			});
+
+			const fileMetadata: FileMetadata = {
+				size: buffer.length,
+				mimeType: detectedMimeType,
+				md5,
+				storageKey,
+				storageType: 'local',
+				originalName: filename,
+			};
+
+			const document = createDocument({
+				title: filename,
+				content: description || '',
+				author: userId,
+				type: 'file',
+				fileMetadata,
+				isPublic: isPublic || false,
+				isActive: true,
+			});
+
+			const result: AgentToolResult<UploadFileDetails> = {
+				content: [{ type: 'text', text: `✅ 文件上传成功: ${filename} (${formatFileSizeLocal(buffer.length)})` }],
+				details: {
+					success: true,
+					file: {
+						id: document.id,
+						title: document.title,
+						size: formatFileSizeLocal(buffer.length),
+						mimeType: detectedMimeType,
+						isPublic: document.isPublic,
+					},
+				},
+			};
+
+			onUpdate?.(result);
+			return result;
+		} catch (error: any) {
+			const errorResult: AgentToolResult<UploadFileDetails> = {
+				content: [{ type: 'text', text: `❌ 上传文件失败: ${error.message}` }],
+				details: { success: false, error: error.message },
+			};
+			onUpdate?.(errorResult);
+			return errorResult;
+		}
+	},
+};
+
+const deleteFileSchema = Type.Object({
+	fileId: Type.String({ description: '文件ID' }),
+	userId: Type.String({ description: '用户ID' }),
+});
+
+interface DeleteFileDetails {
+	success: boolean;
+	message?: string;
+	error?: string;
+}
+
+export const deleteFileTool: ToolDefinition<typeof deleteFileSchema, DeleteFileDetails> = {
+	name: 'delete_file',
+	label: 'Delete File',
+	description: '删除文件。',
+	parameters: deleteFileSchema,
+
+	async execute(
+		toolCallId: string,
+		params: any,
+		signal: AbortSignal | undefined,
+		onUpdate: ((result: AgentToolResult<DeleteFileDetails>) => void) | undefined,
+		ctx: ExtensionContext
+	): Promise<AgentToolResult<DeleteFileDetails>> {
+		try {
+			const { fileId, userId } = params;
+			const doc = getDocument(fileId);
+
+			if (!doc) {
+				const errorResult: AgentToolResult<DeleteFileDetails> = {
+					content: [{ type: 'text', text: '❌ 文件未找到' }],
+					details: { success: false, error: 'File not found' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			if (doc.author !== userId) {
+				const errorResult: AgentToolResult<DeleteFileDetails> = {
+					content: [{ type: 'text', text: '❌ 无权删除此文件' }],
+					details: { success: false, error: 'Permission denied' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			if (doc.fileMetadata?.storageKey) {
+				const storage = getStorageProvider();
+				await storage.delete(doc.fileMetadata.storageKey);
+			}
+
+			deleteDocument(fileId);
+
+			const result: AgentToolResult<DeleteFileDetails> = {
+				content: [{ type: 'text', text: `✅ 文件 "${doc.title}" 已删除` }],
+				details: { success: true, message: `File "${doc.title}" deleted` },
+			};
+
+			onUpdate?.(result);
+			return result;
+		} catch (error: any) {
+			const errorResult: AgentToolResult<DeleteFileDetails> = {
+				content: [{ type: 'text', text: `❌ 删除文件失败: ${error.message}` }],
+				details: { success: false, error: error.message },
+			};
+			onUpdate?.(errorResult);
+			return errorResult;
+		}
+	},
+};
+
+const updateFileInfoSchema = Type.Object({
+	fileId: Type.String({ description: '文件ID' }),
+	userId: Type.String({ description: '用户ID' }),
+	title: Type.Optional(Type.String({ description: '新标题（可选）' })),
+	description: Type.Optional(Type.String({ description: '新描述（可选）' })),
+	isPublic: Type.Optional(Type.Boolean({ description: '是否公开分享（可选）' })),
+});
+
+interface UpdateFileInfoDetails {
+	success: boolean;
+	file?: {
+		id?: string;
+		title?: string;
+		description?: string;
+		isPublic?: boolean;
+	};
+	error?: string;
+}
+
+export const updateFileInfoTool: ToolDefinition<typeof updateFileInfoSchema, UpdateFileInfoDetails> = {
+	name: 'update_file_info',
+	label: 'Update File Info',
+	description: '更新文件信息（标题、描述、公开状态）。',
+	parameters: updateFileInfoSchema,
+
+	async execute(
+		toolCallId: string,
+		params: any,
+		signal: AbortSignal | undefined,
+		onUpdate: ((result: AgentToolResult<UpdateFileInfoDetails>) => void) | undefined,
+		ctx: ExtensionContext
+	): Promise<AgentToolResult<UpdateFileInfoDetails>> {
+		try {
+			const { fileId, userId, title, description, isPublic } = params;
+			const doc = getDocument(fileId);
+
+			if (!doc) {
+				const errorResult: AgentToolResult<UpdateFileInfoDetails> = {
+					content: [{ type: 'text', text: '❌ 文件未找到' }],
+					details: { success: false, error: 'File not found' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			if (doc.author !== userId) {
+				const errorResult: AgentToolResult<UpdateFileInfoDetails> = {
+					content: [{ type: 'text', text: '❌ 无权修改此文件' }],
+					details: { success: false, error: 'Permission denied' },
+				};
+				onUpdate?.(errorResult);
+				return errorResult;
+			}
+
+			const updates: any = {};
+			if (title !== undefined) updates.title = title;
+			if (description !== undefined) updates.content = description;
+			if (isPublic !== undefined) updates.isPublic = isPublic;
+
+			const updated = updateDocument(fileId, updates);
+
+			const result: AgentToolResult<UpdateFileInfoDetails> = {
+				content: [{ type: 'text', text: `✅ 文件信息已更新` }],
+				details: {
+					success: true,
+					file: {
+						id: updated?.id,
+						title: updated?.title,
+						description: updated?.content,
+						isPublic: updated?.isPublic,
+					},
+				},
+			};
+
+			onUpdate?.(result);
+			return result;
+		} catch (error: any) {
+			const errorResult: AgentToolResult<UpdateFileInfoDetails> = {
+				content: [{ type: 'text', text: `❌ 更新文件信息失败: ${error.message}` }],
+				details: { success: false, error: error.message },
+			};
+			onUpdate?.(errorResult);
+			return errorResult;
+		}
+	},
+};
+
+export const fileTools = [
+	listFilesTool,
+	getFileInfoTool,
+	downloadFileTool,
+	uploadFileTool,
+	deleteFileTool,
+	updateFileInfoTool,
+];