refactor(core): 移除不再需要的文件系统工具
删除以下工具及相关文件: - copy_file: 复制文件 - create_directory: 创建目录 - delete_file: 删除文件 - move_file: 移动文件 - search_files: 搜索文件 清理范围: - 工具实现文件 (5个) - 工具描述文件 (5个) - 单元测试文件 (6个) - Agent presets 中的引用 - Checkpoint 系统中的触发类型 - Hook 系统中的相关处理
This commit is contained in:
@@ -319,11 +319,10 @@ export class AgentExecutor {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 文件写入权限检查
|
// 文件写入权限检查
|
||||||
if (['write_file', 'edit_file', 'delete_file'].includes(toolName)) {
|
if (['write_file', 'edit_file'].includes(toolName)) {
|
||||||
const filePermission = permission.file;
|
const filePermission = permission.file;
|
||||||
if (filePermission) {
|
if (filePermission) {
|
||||||
const operation = toolName === 'write_file' ? 'write' :
|
const operation = toolName === 'write_file' ? 'write' : 'edit';
|
||||||
toolName === 'edit_file' ? 'edit' : 'delete';
|
|
||||||
const action = filePermission[operation];
|
const action = filePermission[operation];
|
||||||
if (action === 'deny') {
|
if (action === 'deny') {
|
||||||
return { allowed: false, reason: `${operation} 操作被禁止` };
|
return { allowed: false, reason: `${operation} 操作被禁止` };
|
||||||
|
|||||||
@@ -159,14 +159,9 @@ export const buildAgent: Omit<AgentInfo, 'name'> = {
|
|||||||
'edit_file',
|
'edit_file',
|
||||||
'multi_edit',
|
'multi_edit',
|
||||||
'list_directory',
|
'list_directory',
|
||||||
'create_directory',
|
|
||||||
'search_files',
|
|
||||||
'glob',
|
'glob',
|
||||||
'grep',
|
'grep',
|
||||||
'get_file_info',
|
'get_file_info',
|
||||||
'move_file',
|
|
||||||
'copy_file',
|
|
||||||
'delete_file',
|
|
||||||
|
|
||||||
// ============ Shell ============
|
// ============ Shell ============
|
||||||
'bash',
|
'bash',
|
||||||
|
|||||||
@@ -48,7 +48,7 @@ export const codeReviewerAgent: Omit<AgentInfo, 'name'> = {
|
|||||||
enabled: [
|
enabled: [
|
||||||
'read_file',
|
'read_file',
|
||||||
'list_directory',
|
'list_directory',
|
||||||
'search_files',
|
'glob',
|
||||||
'grep',
|
'grep',
|
||||||
'git_status',
|
'git_status',
|
||||||
'git_diff',
|
'git_diff',
|
||||||
|
|||||||
@@ -48,7 +48,6 @@ export const exploreAgent: Omit<AgentInfo, 'name'> = {
|
|||||||
- **glob**: 文件模式匹配 (*.ts, src/**/*.tsx)
|
- **glob**: 文件模式匹配 (*.ts, src/**/*.tsx)
|
||||||
- **grep**: 代码内容搜索
|
- **grep**: 代码内容搜索
|
||||||
- **read_file**: 读取文件内容
|
- **read_file**: 读取文件内容
|
||||||
- **search_files**: 文件名搜索
|
|
||||||
- **list_directory**: 目录列表
|
- **list_directory**: 目录列表
|
||||||
- **bash**: 只读命令 (ls, tree, find, git log/status/diff)
|
- **bash**: 只读命令 (ls, tree, find, git log/status/diff)
|
||||||
|
|
||||||
@@ -81,10 +80,6 @@ export const exploreAgent: Omit<AgentInfo, 'name'> = {
|
|||||||
// 禁用所有写入操作
|
// 禁用所有写入操作
|
||||||
'write_file',
|
'write_file',
|
||||||
'edit_file',
|
'edit_file',
|
||||||
'delete_file',
|
|
||||||
'move_file',
|
|
||||||
'copy_file',
|
|
||||||
'create_directory',
|
|
||||||
'multi_edit',
|
'multi_edit',
|
||||||
// 禁用 Todo(由父 Agent 管理)
|
// 禁用 Todo(由父 Agent 管理)
|
||||||
'todo_read',
|
'todo_read',
|
||||||
|
|||||||
@@ -78,10 +78,6 @@ export const guideAgent: Omit<AgentInfo, 'name'> = {
|
|||||||
// 禁用所有写入操作
|
// 禁用所有写入操作
|
||||||
'write_file',
|
'write_file',
|
||||||
'edit_file',
|
'edit_file',
|
||||||
'delete_file',
|
|
||||||
'move_file',
|
|
||||||
'copy_file',
|
|
||||||
'create_directory',
|
|
||||||
'multi_edit',
|
'multi_edit',
|
||||||
// 禁用 Todo(由父 Agent 管理)
|
// 禁用 Todo(由父 Agent 管理)
|
||||||
'todo_read',
|
'todo_read',
|
||||||
|
|||||||
@@ -90,7 +90,6 @@ export const planAgent: Omit<AgentInfo, 'name'> = {
|
|||||||
'read_file',
|
'read_file',
|
||||||
'write_file',
|
'write_file',
|
||||||
'list_directory',
|
'list_directory',
|
||||||
'search_files',
|
|
||||||
'glob',
|
'glob',
|
||||||
'grep',
|
'grep',
|
||||||
'get_file_info',
|
'get_file_info',
|
||||||
|
|||||||
@@ -213,11 +213,6 @@ export class CheckpointStore {
|
|||||||
return autoCheckpoint.beforeWrite;
|
return autoCheckpoint.beforeWrite;
|
||||||
case 'edit_file':
|
case 'edit_file':
|
||||||
return autoCheckpoint.beforeEdit;
|
return autoCheckpoint.beforeEdit;
|
||||||
case 'delete_file':
|
|
||||||
return autoCheckpoint.beforeDelete;
|
|
||||||
case 'move_file':
|
|
||||||
case 'copy_file':
|
|
||||||
return autoCheckpoint.beforeMove;
|
|
||||||
case 'bash':
|
case 'bash':
|
||||||
return autoCheckpoint.beforeBash;
|
return autoCheckpoint.beforeBash;
|
||||||
default:
|
default:
|
||||||
@@ -234,12 +229,6 @@ export class CheckpointStore {
|
|||||||
return `Write file: ${params.file_path || params.path}`;
|
return `Write file: ${params.file_path || params.path}`;
|
||||||
case 'edit_file':
|
case 'edit_file':
|
||||||
return `Edit file: ${params.file_path || params.path}`;
|
return `Edit file: ${params.file_path || params.path}`;
|
||||||
case 'delete_file':
|
|
||||||
return `Delete file: ${params.file_path || params.path}`;
|
|
||||||
case 'move_file':
|
|
||||||
return `Move: ${params.source} -> ${params.destination}`;
|
|
||||||
case 'copy_file':
|
|
||||||
return `Copy: ${params.source} -> ${params.destination}`;
|
|
||||||
case 'bash':
|
case 'bash':
|
||||||
return `Bash: ${String(params.command).slice(0, 50)}`;
|
return `Bash: ${String(params.command).slice(0, 50)}`;
|
||||||
default:
|
default:
|
||||||
|
|||||||
@@ -47,9 +47,6 @@ export class CommitMessageGenerator {
|
|||||||
manual: 'checkpoint',
|
manual: 'checkpoint',
|
||||||
'tool:write_file': 'write',
|
'tool:write_file': 'write',
|
||||||
'tool:edit_file': 'edit',
|
'tool:edit_file': 'edit',
|
||||||
'tool:delete_file': 'delete',
|
|
||||||
'tool:move_file': 'move',
|
|
||||||
'tool:copy_file': 'copy',
|
|
||||||
'tool:bash': 'bash',
|
'tool:bash': 'bash',
|
||||||
task_start: 'session-start',
|
task_start: 'session-start',
|
||||||
task_complete: 'session-end',
|
task_complete: 'session-end',
|
||||||
|
|||||||
@@ -63,8 +63,6 @@ export class CheckpointManager {
|
|||||||
autoCheckpoint: {
|
autoCheckpoint: {
|
||||||
beforeWrite: true,
|
beforeWrite: true,
|
||||||
beforeEdit: true,
|
beforeEdit: true,
|
||||||
beforeDelete: true,
|
|
||||||
beforeMove: true,
|
|
||||||
beforeBash: false,
|
beforeBash: false,
|
||||||
},
|
},
|
||||||
maxCheckpoints: 100,
|
maxCheckpoints: 100,
|
||||||
|
|||||||
@@ -175,9 +175,6 @@ export class CheckpointSafetyChecker {
|
|||||||
const aiTriggers = [
|
const aiTriggers = [
|
||||||
'tool:write_file',
|
'tool:write_file',
|
||||||
'tool:edit_file',
|
'tool:edit_file',
|
||||||
'tool:delete_file',
|
|
||||||
'tool:move_file',
|
|
||||||
'tool:copy_file',
|
|
||||||
'tool:bash',
|
'tool:bash',
|
||||||
'task_start',
|
'task_start',
|
||||||
'task_complete',
|
'task_complete',
|
||||||
|
|||||||
@@ -11,9 +11,6 @@ export type CheckpointTrigger =
|
|||||||
| 'manual' // 用户手动
|
| 'manual' // 用户手动
|
||||||
| 'tool:write_file' // 写文件前
|
| 'tool:write_file' // 写文件前
|
||||||
| 'tool:edit_file' // 编辑文件前
|
| 'tool:edit_file' // 编辑文件前
|
||||||
| 'tool:delete_file' // 删除文件前
|
|
||||||
| 'tool:move_file' // 移动文件前
|
|
||||||
| 'tool:copy_file' // 复制文件前
|
|
||||||
| 'tool:bash' // bash 命令前
|
| 'tool:bash' // bash 命令前
|
||||||
| 'task_start' // 任务开始
|
| 'task_start' // 任务开始
|
||||||
| 'task_complete' // 任务完成
|
| 'task_complete' // 任务完成
|
||||||
@@ -64,10 +61,6 @@ export interface CheckpointConfig {
|
|||||||
beforeWrite: boolean;
|
beforeWrite: boolean;
|
||||||
/** 编辑文件前创建检查点 */
|
/** 编辑文件前创建检查点 */
|
||||||
beforeEdit: boolean;
|
beforeEdit: boolean;
|
||||||
/** 删除文件前创建检查点 */
|
|
||||||
beforeDelete: boolean;
|
|
||||||
/** 移动/复制文件前创建检查点 */
|
|
||||||
beforeMove: boolean;
|
|
||||||
/** bash 命令前创建检查点 */
|
/** bash 命令前创建检查点 */
|
||||||
beforeBash: boolean;
|
beforeBash: boolean;
|
||||||
};
|
};
|
||||||
@@ -87,8 +80,6 @@ export const DEFAULT_CHECKPOINT_CONFIG: CheckpointConfig = {
|
|||||||
autoCheckpoint: {
|
autoCheckpoint: {
|
||||||
beforeWrite: true,
|
beforeWrite: true,
|
||||||
beforeEdit: true,
|
beforeEdit: true,
|
||||||
beforeDelete: true,
|
|
||||||
beforeMove: true,
|
|
||||||
beforeBash: false,
|
beforeBash: false,
|
||||||
},
|
},
|
||||||
maxCheckpoints: 100,
|
maxCheckpoints: 100,
|
||||||
|
|||||||
@@ -346,15 +346,6 @@ export class AgentToolExecutor {
|
|||||||
if (gitManager) {
|
if (gitManager) {
|
||||||
await gitManager.onFileChanged(filePath, 'modify');
|
await gitManager.onFileChanged(filePath, 'modify');
|
||||||
}
|
}
|
||||||
} else if (toolName === 'delete_file') {
|
|
||||||
await hookManager.triggerFileDeleted({
|
|
||||||
path: filePath,
|
|
||||||
tool: toolName,
|
|
||||||
sessionId,
|
|
||||||
});
|
|
||||||
if (gitManager) {
|
|
||||||
await gitManager.onFileChanged(filePath, 'delete');
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1 +0,0 @@
|
|||||||
复制文件或目录。支持递归复制整个目录结构。
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
创建新目录。支持递归创建父目录。如果目录已存在则不会报错。
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
删除文件或目录。删除目录时可以选择是否递归删除。需要谨慎使用。
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
移动或重命名文件/目录。可以将文件移动到新位置或更改文件名。
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
在目录中搜索匹配模式的文件
|
|
||||||
@@ -1,140 +0,0 @@
|
|||||||
import * as fs from 'fs/promises';
|
|
||||||
import * as path from 'path';
|
|
||||||
import type { ToolResult } from '../../types/index.js';
|
|
||||||
import type { ToolWithMetadata } from '../types.js';
|
|
||||||
import { loadDescription } from '../load_description.js';
|
|
||||||
import { getPermissionManager } from '../../permission/index.js';
|
|
||||||
|
|
||||||
async function copyRecursive(source: string, dest: string): Promise<void> {
|
|
||||||
const stats = await fs.stat(source);
|
|
||||||
|
|
||||||
if (stats.isDirectory()) {
|
|
||||||
await fs.mkdir(dest, { recursive: true });
|
|
||||||
const entries = await fs.readdir(source);
|
|
||||||
for (const entry of entries) {
|
|
||||||
await copyRecursive(path.join(source, entry), path.join(dest, entry));
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
await fs.copyFile(source, dest);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
export const copyFileTool: ToolWithMetadata = {
|
|
||||||
name: 'copy_file',
|
|
||||||
description: loadDescription('copy_file'),
|
|
||||||
metadata: {
|
|
||||||
name: 'copy_file',
|
|
||||||
category: 'filesystem',
|
|
||||||
description: '复制文件或目录',
|
|
||||||
keywords: ['copy', 'file', 'cp', 'duplicate', '复制', '文件', '拷贝'],
|
|
||||||
deferLoading: true,
|
|
||||||
},
|
|
||||||
parameters: {
|
|
||||||
source: {
|
|
||||||
type: 'string',
|
|
||||||
description: '源文件或目录的路径',
|
|
||||||
required: true,
|
|
||||||
},
|
|
||||||
destination: {
|
|
||||||
type: 'string',
|
|
||||||
description: '目标路径',
|
|
||||||
required: true,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
execute: async (params: Record<string, unknown>): Promise<ToolResult> => {
|
|
||||||
const source = params.source as string;
|
|
||||||
const destination = params.destination as string;
|
|
||||||
const cwd = process.cwd();
|
|
||||||
|
|
||||||
const absoluteSource = path.isAbsolute(source)
|
|
||||||
? source
|
|
||||||
: path.join(cwd, source);
|
|
||||||
|
|
||||||
const absoluteDest = path.isAbsolute(destination)
|
|
||||||
? destination
|
|
||||||
: path.join(cwd, destination);
|
|
||||||
|
|
||||||
// 权限检查 - 源文件需要 read 权限
|
|
||||||
const permissionManager = getPermissionManager();
|
|
||||||
const sourcePermResult = await permissionManager.checkFilePermission({
|
|
||||||
operation: 'read',
|
|
||||||
path: absoluteSource,
|
|
||||||
workdir: cwd,
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!sourcePermResult.allowed) {
|
|
||||||
if (sourcePermResult.needsConfirmation) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `需要用户确认: 读取 ${absoluteSource}\n原因: ${sourcePermResult.reason || '需要权限确认'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `权限被拒绝: ${sourcePermResult.reason || '不允许读取此文件'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
// 权限检查 - 目标位置需要 copy 权限
|
|
||||||
const destPermResult = await permissionManager.checkFilePermission({
|
|
||||||
operation: 'copy',
|
|
||||||
path: absoluteDest,
|
|
||||||
workdir: cwd,
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!destPermResult.allowed) {
|
|
||||||
if (destPermResult.needsConfirmation) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `需要用户确认: 复制到 ${absoluteDest}\n原因: ${destPermResult.reason || '需要权限确认'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `权限被拒绝: ${destPermResult.reason || '不允许复制到此位置'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
// 检查源文件是否存在
|
|
||||||
const sourceStats = await fs.stat(absoluteSource);
|
|
||||||
|
|
||||||
// 检查目标是否是目录
|
|
||||||
let finalDest = absoluteDest;
|
|
||||||
try {
|
|
||||||
const destStats = await fs.stat(absoluteDest);
|
|
||||||
if (destStats.isDirectory()) {
|
|
||||||
// 如果目标是目录,将源文件复制到该目录下
|
|
||||||
finalDest = path.join(absoluteDest, path.basename(absoluteSource));
|
|
||||||
}
|
|
||||||
} catch {
|
|
||||||
// 目标不存在,直接使用目标路径
|
|
||||||
}
|
|
||||||
|
|
||||||
// 确保目标目录存在
|
|
||||||
await fs.mkdir(path.dirname(finalDest), { recursive: true });
|
|
||||||
|
|
||||||
// 执行复制
|
|
||||||
if (sourceStats.isDirectory()) {
|
|
||||||
await copyRecursive(absoluteSource, finalDest);
|
|
||||||
} else {
|
|
||||||
await fs.copyFile(absoluteSource, finalDest);
|
|
||||||
}
|
|
||||||
|
|
||||||
return {
|
|
||||||
success: true,
|
|
||||||
output: `已复制: ${absoluteSource} -> ${finalDest}`,
|
|
||||||
};
|
|
||||||
} catch (error) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: error instanceof Error ? error.message : String(error),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
},
|
|
||||||
};
|
|
||||||
@@ -1,91 +0,0 @@
|
|||||||
import * as fs from 'fs/promises';
|
|
||||||
import * as path from 'path';
|
|
||||||
import type { ToolResult } from '../../types/index.js';
|
|
||||||
import type { ToolWithMetadata } from '../types.js';
|
|
||||||
import { loadDescription } from '../load_description.js';
|
|
||||||
import { getPermissionManager } from '../../permission/index.js';
|
|
||||||
|
|
||||||
export const createDirectoryTool: ToolWithMetadata = {
|
|
||||||
name: 'create_directory',
|
|
||||||
description: loadDescription('create_directory'),
|
|
||||||
metadata: {
|
|
||||||
name: 'create_directory',
|
|
||||||
category: 'filesystem',
|
|
||||||
description: '创建目录',
|
|
||||||
keywords: ['create', 'directory', 'mkdir', 'folder', 'new', '创建', '目录', '文件夹', '新建'],
|
|
||||||
deferLoading: true,
|
|
||||||
},
|
|
||||||
parameters: {
|
|
||||||
path: {
|
|
||||||
type: 'string',
|
|
||||||
description: '要创建的目录路径',
|
|
||||||
required: true,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
execute: async (params: Record<string, unknown>): Promise<ToolResult> => {
|
|
||||||
const dirPath = params.path as string;
|
|
||||||
const cwd = process.cwd();
|
|
||||||
|
|
||||||
const absolutePath = path.isAbsolute(dirPath)
|
|
||||||
? dirPath
|
|
||||||
: path.join(cwd, dirPath);
|
|
||||||
|
|
||||||
// 权限检查
|
|
||||||
const permissionManager = getPermissionManager();
|
|
||||||
const permResult = await permissionManager.checkFilePermission({
|
|
||||||
operation: 'mkdir',
|
|
||||||
path: absolutePath,
|
|
||||||
workdir: cwd,
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!permResult.allowed) {
|
|
||||||
if (permResult.needsConfirmation) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `需要用户确认: 创建目录 ${absolutePath}\n原因: ${permResult.reason || '需要权限确认'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `权限被拒绝: ${permResult.reason || '不允许创建此目录'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
// 检查目录是否已存在
|
|
||||||
try {
|
|
||||||
const stats = await fs.stat(absolutePath);
|
|
||||||
if (stats.isDirectory()) {
|
|
||||||
return {
|
|
||||||
success: true,
|
|
||||||
output: `目录已存在: ${absolutePath}`,
|
|
||||||
};
|
|
||||||
} else {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `路径已存在且不是目录: ${absolutePath}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
} catch {
|
|
||||||
// 目录不存在,继续创建
|
|
||||||
}
|
|
||||||
|
|
||||||
// 创建目录(递归创建父目录)
|
|
||||||
await fs.mkdir(absolutePath, { recursive: true });
|
|
||||||
|
|
||||||
return {
|
|
||||||
success: true,
|
|
||||||
output: `已创建目录: ${absolutePath}`,
|
|
||||||
};
|
|
||||||
} catch (error) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: error instanceof Error ? error.message : String(error),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
},
|
|
||||||
};
|
|
||||||
@@ -1,99 +0,0 @@
|
|||||||
import * as fs from 'fs/promises';
|
|
||||||
import * as path from 'path';
|
|
||||||
import type { ToolResult } from '../../types/index.js';
|
|
||||||
import type { ToolWithMetadata } from '../types.js';
|
|
||||||
import { loadDescription } from '../load_description.js';
|
|
||||||
import { getPermissionManager } from '../../permission/index.js';
|
|
||||||
|
|
||||||
export const deleteFileTool: ToolWithMetadata = {
|
|
||||||
name: 'delete_file',
|
|
||||||
description: loadDescription('delete_file'),
|
|
||||||
metadata: {
|
|
||||||
name: 'delete_file',
|
|
||||||
category: 'filesystem',
|
|
||||||
description: '删除文件或目录',
|
|
||||||
keywords: ['delete', 'remove', 'file', 'rm', '删除', '移除', '文件'],
|
|
||||||
deferLoading: true,
|
|
||||||
},
|
|
||||||
parameters: {
|
|
||||||
path: {
|
|
||||||
type: 'string',
|
|
||||||
description: '要删除的文件或目录的路径',
|
|
||||||
required: true,
|
|
||||||
},
|
|
||||||
recursive: {
|
|
||||||
type: 'boolean',
|
|
||||||
description: '是否递归删除目录(默认 false)',
|
|
||||||
required: false,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
execute: async (params: Record<string, unknown>): Promise<ToolResult> => {
|
|
||||||
const filePath = params.path as string;
|
|
||||||
const recursive = (params.recursive as boolean) || false;
|
|
||||||
const cwd = process.cwd();
|
|
||||||
|
|
||||||
const absolutePath = path.isAbsolute(filePath)
|
|
||||||
? filePath
|
|
||||||
: path.join(cwd, filePath);
|
|
||||||
|
|
||||||
// 权限检查
|
|
||||||
const permissionManager = getPermissionManager();
|
|
||||||
const permResult = await permissionManager.checkFilePermission({
|
|
||||||
operation: 'delete',
|
|
||||||
path: absolutePath,
|
|
||||||
workdir: cwd,
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!permResult.allowed) {
|
|
||||||
if (permResult.needsConfirmation) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `需要用户确认: 删除 ${absolutePath}\n原因: ${permResult.reason || '需要权限确认'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `权限被拒绝: ${permResult.reason || '不允许删除此文件'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
const stats = await fs.stat(absolutePath);
|
|
||||||
|
|
||||||
if (stats.isDirectory()) {
|
|
||||||
if (!recursive) {
|
|
||||||
// 检查目录是否为空
|
|
||||||
const entries = await fs.readdir(absolutePath);
|
|
||||||
if (entries.length > 0) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `目录不为空。如需删除非空目录,请设置 recursive: true`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
await fs.rmdir(absolutePath);
|
|
||||||
} else {
|
|
||||||
await fs.rm(absolutePath, { recursive: true });
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
success: true,
|
|
||||||
output: `已删除目录: ${absolutePath}`,
|
|
||||||
};
|
|
||||||
} else {
|
|
||||||
await fs.unlink(absolutePath);
|
|
||||||
return {
|
|
||||||
success: true,
|
|
||||||
output: `已删除文件: ${absolutePath}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
} catch (error) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: error instanceof Error ? error.message : String(error),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
},
|
|
||||||
};
|
|
||||||
@@ -6,17 +6,10 @@ export { multiEditTool } from './multi_edit.js';
|
|||||||
|
|
||||||
// 目录操作
|
// 目录操作
|
||||||
export { listDirTool } from './list_directory.js';
|
export { listDirTool } from './list_directory.js';
|
||||||
export { createDirectoryTool } from './create_directory.js';
|
|
||||||
|
|
||||||
// 搜索
|
// 搜索
|
||||||
export { searchFilesTool } from './search_files.js';
|
|
||||||
export { globTool } from './glob.js';
|
export { globTool } from './glob.js';
|
||||||
export { grepTool } from './grep.js';
|
export { grepTool } from './grep.js';
|
||||||
|
|
||||||
// 文件信息
|
// 文件信息
|
||||||
export { getFileInfoTool } from './get_file_info.js';
|
export { getFileInfoTool } from './get_file_info.js';
|
||||||
|
|
||||||
// 文件管理
|
|
||||||
export { moveFileTool } from './move_file.js';
|
|
||||||
export { copyFileTool } from './copy_file.js';
|
|
||||||
export { deleteFileTool } from './delete_file.js';
|
|
||||||
|
|||||||
@@ -1,122 +0,0 @@
|
|||||||
import * as fs from 'fs/promises';
|
|
||||||
import * as path from 'path';
|
|
||||||
import type { ToolResult } from '../../types/index.js';
|
|
||||||
import type { ToolWithMetadata } from '../types.js';
|
|
||||||
import { loadDescription } from '../load_description.js';
|
|
||||||
import { getPermissionManager } from '../../permission/index.js';
|
|
||||||
|
|
||||||
export const moveFileTool: ToolWithMetadata = {
|
|
||||||
name: 'move_file',
|
|
||||||
description: loadDescription('move_file'),
|
|
||||||
metadata: {
|
|
||||||
name: 'move_file',
|
|
||||||
category: 'filesystem',
|
|
||||||
description: '移动或重命名文件/目录',
|
|
||||||
keywords: ['move', 'rename', 'file', 'mv', '移动', '重命名', '文件'],
|
|
||||||
deferLoading: true,
|
|
||||||
},
|
|
||||||
parameters: {
|
|
||||||
source: {
|
|
||||||
type: 'string',
|
|
||||||
description: '源文件或目录的路径',
|
|
||||||
required: true,
|
|
||||||
},
|
|
||||||
destination: {
|
|
||||||
type: 'string',
|
|
||||||
description: '目标路径',
|
|
||||||
required: true,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
execute: async (params: Record<string, unknown>): Promise<ToolResult> => {
|
|
||||||
const source = params.source as string;
|
|
||||||
const destination = params.destination as string;
|
|
||||||
const cwd = process.cwd();
|
|
||||||
|
|
||||||
const absoluteSource = path.isAbsolute(source)
|
|
||||||
? source
|
|
||||||
: path.join(cwd, source);
|
|
||||||
|
|
||||||
const absoluteDest = path.isAbsolute(destination)
|
|
||||||
? destination
|
|
||||||
: path.join(cwd, destination);
|
|
||||||
|
|
||||||
// 权限检查 - 源文件需要 move 权限
|
|
||||||
const permissionManager = getPermissionManager();
|
|
||||||
const sourcePermResult = await permissionManager.checkFilePermission({
|
|
||||||
operation: 'move',
|
|
||||||
path: absoluteSource,
|
|
||||||
workdir: cwd,
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!sourcePermResult.allowed) {
|
|
||||||
if (sourcePermResult.needsConfirmation) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `需要用户确认: 移动 ${absoluteSource}\n原因: ${sourcePermResult.reason || '需要权限确认'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `权限被拒绝: ${sourcePermResult.reason || '不允许移动此文件'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
// 权限检查 - 目标位置需要 write 权限
|
|
||||||
const destPermResult = await permissionManager.checkFilePermission({
|
|
||||||
operation: 'write',
|
|
||||||
path: absoluteDest,
|
|
||||||
workdir: cwd,
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!destPermResult.allowed) {
|
|
||||||
if (destPermResult.needsConfirmation) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `需要用户确认: 写入到 ${absoluteDest}\n原因: ${destPermResult.reason || '需要权限确认'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `权限被拒绝: ${destPermResult.reason || '不允许写入到此位置'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
// 检查源文件是否存在
|
|
||||||
await fs.access(absoluteSource);
|
|
||||||
|
|
||||||
// 检查目标是否是目录
|
|
||||||
let finalDest = absoluteDest;
|
|
||||||
try {
|
|
||||||
const destStats = await fs.stat(absoluteDest);
|
|
||||||
if (destStats.isDirectory()) {
|
|
||||||
// 如果目标是目录,将源文件移动到该目录下
|
|
||||||
finalDest = path.join(absoluteDest, path.basename(absoluteSource));
|
|
||||||
}
|
|
||||||
} catch {
|
|
||||||
// 目标不存在,直接使用目标路径
|
|
||||||
}
|
|
||||||
|
|
||||||
// 确保目标目录存在
|
|
||||||
await fs.mkdir(path.dirname(finalDest), { recursive: true });
|
|
||||||
|
|
||||||
// 执行移动
|
|
||||||
await fs.rename(absoluteSource, finalDest);
|
|
||||||
|
|
||||||
return {
|
|
||||||
success: true,
|
|
||||||
output: `已移动: ${absoluteSource} -> ${finalDest}`,
|
|
||||||
};
|
|
||||||
} catch (error) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: error instanceof Error ? error.message : String(error),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
},
|
|
||||||
};
|
|
||||||
@@ -1,107 +0,0 @@
|
|||||||
import * as fs from 'fs/promises';
|
|
||||||
import * as path from 'path';
|
|
||||||
import type { ToolResult } from '../../types/index.js';
|
|
||||||
import type { ToolWithMetadata } from '../types.js';
|
|
||||||
import { loadDescription } from '../load_description.js';
|
|
||||||
import { getPermissionManager } from '../../permission/index.js';
|
|
||||||
|
|
||||||
export const searchFilesTool: ToolWithMetadata = {
|
|
||||||
name: 'search_files',
|
|
||||||
description: loadDescription('search_files'),
|
|
||||||
metadata: {
|
|
||||||
name: 'search_files',
|
|
||||||
category: 'filesystem',
|
|
||||||
description: '按文件名搜索文件',
|
|
||||||
keywords: ['search', 'file', 'find', 'glob', 'pattern', '搜索', '文件', '查找', '匹配'],
|
|
||||||
deferLoading: true,
|
|
||||||
},
|
|
||||||
parameters: {
|
|
||||||
directory: {
|
|
||||||
type: 'string',
|
|
||||||
description: '搜索的起始目录',
|
|
||||||
required: true,
|
|
||||||
},
|
|
||||||
pattern: {
|
|
||||||
type: 'string',
|
|
||||||
description: '文件名匹配模式(支持 glob 模式,如 *.ts)',
|
|
||||||
required: true,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
execute: async (params: Record<string, unknown>): Promise<ToolResult> => {
|
|
||||||
const directory = params.directory as string;
|
|
||||||
const pattern = params.pattern as string;
|
|
||||||
const cwd = process.cwd();
|
|
||||||
const absolutePath = path.isAbsolute(directory)
|
|
||||||
? directory
|
|
||||||
: path.join(cwd, directory);
|
|
||||||
|
|
||||||
// 权限检查
|
|
||||||
const permissionManager = getPermissionManager();
|
|
||||||
const permResult = await permissionManager.checkFilePermission({
|
|
||||||
operation: 'search',
|
|
||||||
path: absolutePath,
|
|
||||||
workdir: cwd,
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!permResult.allowed) {
|
|
||||||
if (permResult.needsConfirmation) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `需要用户确认: 搜索目录 ${absolutePath}\n原因: ${permResult.reason || '需要权限确认'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: `权限被拒绝: ${permResult.reason || '不允许搜索此目录'}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
const matches: string[] = [];
|
|
||||||
const regex = new RegExp(
|
|
||||||
pattern.replace(/\*/g, '.*').replace(/\?/g, '.'),
|
|
||||||
'i'
|
|
||||||
);
|
|
||||||
|
|
||||||
async function searchRecursive(dir: string, depth = 0): Promise<void> {
|
|
||||||
if (depth > 10) return;
|
|
||||||
|
|
||||||
try {
|
|
||||||
const entries = await fs.readdir(dir, { withFileTypes: true });
|
|
||||||
for (const entry of entries) {
|
|
||||||
const fullPath = path.join(dir, entry.name);
|
|
||||||
|
|
||||||
if (entry.name.startsWith('.') || entry.name === 'node_modules') {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (entry.isDirectory()) {
|
|
||||||
await searchRecursive(fullPath, depth + 1);
|
|
||||||
} else if (regex.test(entry.name)) {
|
|
||||||
matches.push(fullPath);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} catch {
|
|
||||||
// 忽略权限错误
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
await searchRecursive(absolutePath);
|
|
||||||
return {
|
|
||||||
success: true,
|
|
||||||
output:
|
|
||||||
matches.length > 0
|
|
||||||
? matches.join('\n')
|
|
||||||
: '没有找到匹配的文件',
|
|
||||||
};
|
|
||||||
} catch (error) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
output: '',
|
|
||||||
error: error instanceof Error ? error.message : String(error),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
},
|
|
||||||
};
|
|
||||||
@@ -21,14 +21,9 @@ import {
|
|||||||
editFileTool,
|
editFileTool,
|
||||||
multiEditTool,
|
multiEditTool,
|
||||||
listDirTool,
|
listDirTool,
|
||||||
createDirectoryTool,
|
|
||||||
searchFilesTool,
|
|
||||||
globTool,
|
globTool,
|
||||||
grepTool,
|
grepTool,
|
||||||
getFileInfoTool,
|
getFileInfoTool,
|
||||||
moveFileTool,
|
|
||||||
copyFileTool,
|
|
||||||
deleteFileTool,
|
|
||||||
} from './filesystem/index.js';
|
} from './filesystem/index.js';
|
||||||
|
|
||||||
// Web 工具
|
// Web 工具
|
||||||
@@ -87,14 +82,9 @@ const allToolsWithMetadata: ToolWithMetadata[] = [
|
|||||||
editFileTool,
|
editFileTool,
|
||||||
multiEditTool,
|
multiEditTool,
|
||||||
listDirTool,
|
listDirTool,
|
||||||
createDirectoryTool,
|
|
||||||
searchFilesTool,
|
|
||||||
globTool,
|
globTool,
|
||||||
grepTool,
|
grepTool,
|
||||||
getFileInfoTool,
|
getFileInfoTool,
|
||||||
moveFileTool,
|
|
||||||
copyFileTool,
|
|
||||||
deleteFileTool,
|
|
||||||
|
|
||||||
// Web 工具 (deferLoading: false)
|
// Web 工具 (deferLoading: false)
|
||||||
webSearchTool,
|
webSearchTool,
|
||||||
|
|||||||
@@ -15,14 +15,9 @@ const TOOL_CATEGORY_MAP: Record<string, string> = {
|
|||||||
edit_file: 'filesystem',
|
edit_file: 'filesystem',
|
||||||
multi_edit: 'filesystem',
|
multi_edit: 'filesystem',
|
||||||
list_directory: 'filesystem',
|
list_directory: 'filesystem',
|
||||||
create_directory: 'filesystem',
|
|
||||||
search_files: 'filesystem',
|
|
||||||
glob: 'filesystem',
|
glob: 'filesystem',
|
||||||
grep: 'filesystem',
|
grep: 'filesystem',
|
||||||
get_file_info: 'filesystem',
|
get_file_info: 'filesystem',
|
||||||
move_file: 'filesystem',
|
|
||||||
copy_file: 'filesystem',
|
|
||||||
delete_file: 'filesystem',
|
|
||||||
// web
|
// web
|
||||||
web_search: 'web',
|
web_search: 'web',
|
||||||
web_extract: 'web',
|
web_extract: 'web',
|
||||||
|
|||||||
@@ -218,7 +218,6 @@ describe('CheckpointManager', () => {
|
|||||||
it('should determine if checkpoint should be created for tool', () => {
|
it('should determine if checkpoint should be created for tool', () => {
|
||||||
expect(manager.shouldCreateCheckpoint('write_file')).toBe(true);
|
expect(manager.shouldCreateCheckpoint('write_file')).toBe(true);
|
||||||
expect(manager.shouldCreateCheckpoint('edit_file')).toBe(true);
|
expect(manager.shouldCreateCheckpoint('edit_file')).toBe(true);
|
||||||
expect(manager.shouldCreateCheckpoint('delete_file')).toBe(true);
|
|
||||||
expect(manager.shouldCreateCheckpoint('bash')).toBe(false); // 默认禁用
|
expect(manager.shouldCreateCheckpoint('bash')).toBe(false); // 默认禁用
|
||||||
expect(manager.shouldCreateCheckpoint('read_file')).toBe(false);
|
expect(manager.shouldCreateCheckpoint('read_file')).toBe(false);
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -328,25 +328,6 @@ describe('HookManager', () => {
|
|||||||
expect(triggered).toBe(true);
|
expect(triggered).toBe(true);
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should trigger file.deleted hook', async () => {
|
|
||||||
let triggered = false;
|
|
||||||
const hooks: Hooks = {
|
|
||||||
'file.deleted': async (input, output) => {
|
|
||||||
triggered = true;
|
|
||||||
expect(input.path).toBe('/test/old-file.ts');
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
manager.registerHooks(hooks);
|
|
||||||
|
|
||||||
await manager.triggerFileDeleted({
|
|
||||||
path: '/test/old-file.ts',
|
|
||||||
tool: 'delete_file',
|
|
||||||
sessionId: 'test-session',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(triggered).toBe(true);
|
|
||||||
});
|
|
||||||
});
|
});
|
||||||
|
|
||||||
describe('Config Hooks', () => {
|
describe('Config Hooks', () => {
|
||||||
|
|||||||
@@ -1,192 +0,0 @@
|
|||||||
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
|
||||||
|
|
||||||
// Mock fs/promises
|
|
||||||
vi.mock('fs/promises', () => ({
|
|
||||||
stat: vi.fn(),
|
|
||||||
copyFile: vi.fn().mockResolvedValue(undefined),
|
|
||||||
mkdir: vi.fn().mockResolvedValue(undefined),
|
|
||||||
readdir: vi.fn().mockResolvedValue([]),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// 可变状态
|
|
||||||
let mockCheckResults: Array<{
|
|
||||||
allowed: boolean;
|
|
||||||
action?: string;
|
|
||||||
reason?: string;
|
|
||||||
needsConfirmation?: boolean;
|
|
||||||
}> = [];
|
|
||||||
let checkCallIndex = 0;
|
|
||||||
|
|
||||||
// Mock permission manager
|
|
||||||
vi.mock('../../../../src/permission/index.js', () => ({
|
|
||||||
getPermissionManager: () => ({
|
|
||||||
checkFilePermission: vi.fn(async () => {
|
|
||||||
const result = mockCheckResults[checkCallIndex] || { allowed: true };
|
|
||||||
checkCallIndex++;
|
|
||||||
return result;
|
|
||||||
}),
|
|
||||||
}),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Mock loadDescription
|
|
||||||
vi.mock('../../../../src/tools/load_description.js', () => ({
|
|
||||||
loadDescription: vi.fn(() => '复制文件'),
|
|
||||||
}));
|
|
||||||
|
|
||||||
import { copyFileTool } from '../../../../src/tools/filesystem/copy_file.js';
|
|
||||||
import * as fs from 'fs/promises';
|
|
||||||
|
|
||||||
describe('copyFileTool - 扩展测试', () => {
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
mockCheckResults = [{ allowed: true }, { allowed: true }];
|
|
||||||
checkCallIndex = 0;
|
|
||||||
// 重置 mock 默认值
|
|
||||||
vi.mocked(fs.stat).mockReset();
|
|
||||||
vi.mocked(fs.copyFile).mockReset().mockResolvedValue(undefined);
|
|
||||||
vi.mocked(fs.mkdir).mockReset().mockResolvedValue(undefined);
|
|
||||||
vi.mocked(fs.readdir).mockReset().mockResolvedValue([]);
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('递归复制目录', () => {
|
|
||||||
it('递归复制包含文件的目录', async () => {
|
|
||||||
vi.mocked(fs.stat)
|
|
||||||
.mockResolvedValueOnce({ isDirectory: () => true } as any)
|
|
||||||
.mockRejectedValueOnce(new Error('ENOENT'))
|
|
||||||
.mockResolvedValueOnce({ isDirectory: () => false } as any);
|
|
||||||
|
|
||||||
vi.mocked(fs.readdir).mockResolvedValueOnce(['file1.txt'] as any);
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'src_dir',
|
|
||||||
destination: 'dest_dir',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(fs.mkdir).toHaveBeenCalled();
|
|
||||||
expect(fs.copyFile).toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('递归复制包含子目录的目录', async () => {
|
|
||||||
// 调用顺序:
|
|
||||||
// 1. execute: stat(source) - 检查源是否存在
|
|
||||||
// 2. execute: stat(dest) - 检查目标(ENOENT)
|
|
||||||
// 3. copyRecursive: stat(source) - 判断是否是目录
|
|
||||||
// 4. copyRecursive: stat(source/subdir) - 递归判断子目录
|
|
||||||
vi.mocked(fs.stat)
|
|
||||||
.mockResolvedValueOnce({ isDirectory: () => true } as any) // source 存在且是目录
|
|
||||||
.mockRejectedValueOnce(new Error('ENOENT')) // dest 不存在
|
|
||||||
.mockResolvedValueOnce({ isDirectory: () => true } as any) // copyRecursive(source)
|
|
||||||
.mockResolvedValueOnce({ isDirectory: () => true } as any); // copyRecursive(source/subdir)
|
|
||||||
|
|
||||||
vi.mocked(fs.readdir)
|
|
||||||
.mockResolvedValueOnce(['subdir'] as any) // 第一层目录
|
|
||||||
.mockResolvedValueOnce([] as any); // 子目录为空
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'src_dir',
|
|
||||||
destination: 'dest_dir',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(fs.mkdir).toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('目标位置权限', () => {
|
|
||||||
it('目标位置需要确认时返回错误', async () => {
|
|
||||||
mockCheckResults = [
|
|
||||||
{ allowed: true },
|
|
||||||
{ allowed: false, action: 'ask', needsConfirmation: true, reason: '首次复制到此位置' },
|
|
||||||
];
|
|
||||||
|
|
||||||
vi.mocked(fs.stat).mockResolvedValueOnce({ isDirectory: () => false } as any);
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'src.txt',
|
|
||||||
destination: '/new/location/dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('需要用户确认');
|
|
||||||
expect(result.error).toContain('首次复制到此位置');
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('绝对路径处理', () => {
|
|
||||||
it('源和目标都是绝对路径', async () => {
|
|
||||||
vi.mocked(fs.stat)
|
|
||||||
.mockResolvedValueOnce({ isDirectory: () => false } as any)
|
|
||||||
.mockRejectedValueOnce(new Error('ENOENT'));
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: '/absolute/source/file.txt',
|
|
||||||
destination: '/absolute/destination/file.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('/absolute/source/file.txt');
|
|
||||||
expect(result.output).toContain('/absolute/destination/file.txt');
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('权限检查细节', () => {
|
|
||||||
it('源文件权限被拒绝(无原因)', async () => {
|
|
||||||
mockCheckResults = [
|
|
||||||
{ allowed: false, action: 'deny' },
|
|
||||||
];
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'src.txt',
|
|
||||||
destination: 'dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('不允许读取此文件');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('源文件需要确认(无原因)', async () => {
|
|
||||||
mockCheckResults = [
|
|
||||||
{ allowed: false, action: 'ask', needsConfirmation: true },
|
|
||||||
];
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'src.txt',
|
|
||||||
destination: 'dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('需要权限确认');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('目标权限被拒绝(无原因)', async () => {
|
|
||||||
mockCheckResults = [
|
|
||||||
{ allowed: true },
|
|
||||||
{ allowed: false, action: 'deny' },
|
|
||||||
];
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'src.txt',
|
|
||||||
destination: 'dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('不允许复制到此位置');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('目标需要确认(无原因)', async () => {
|
|
||||||
mockCheckResults = [
|
|
||||||
{ allowed: true },
|
|
||||||
{ allowed: false, action: 'ask', needsConfirmation: true },
|
|
||||||
];
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'src.txt',
|
|
||||||
destination: 'dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('需要权限确认');
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,173 +0,0 @@
|
|||||||
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
|
||||||
|
|
||||||
// Mock fs/promises
|
|
||||||
vi.mock('fs/promises', () => ({
|
|
||||||
stat: vi.fn(),
|
|
||||||
copyFile: vi.fn().mockResolvedValue(undefined),
|
|
||||||
mkdir: vi.fn().mockResolvedValue(undefined),
|
|
||||||
readdir: vi.fn().mockResolvedValue([]),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Mock permission manager
|
|
||||||
vi.mock('../../../../src/permission/index.js', () => ({
|
|
||||||
getPermissionManager: vi.fn(() => ({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: true,
|
|
||||||
action: 'allow',
|
|
||||||
}),
|
|
||||||
})),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Mock loadDescription
|
|
||||||
vi.mock('../../../../src/tools/load_description.js', () => ({
|
|
||||||
loadDescription: vi.fn(() => '复制文件'),
|
|
||||||
}));
|
|
||||||
|
|
||||||
import { copyFileTool } from '../../../../src/tools/filesystem/copy_file.js';
|
|
||||||
import * as fs from 'fs/promises';
|
|
||||||
import { getPermissionManager } from '../../../../src/permission/index.js';
|
|
||||||
|
|
||||||
describe('copyFileTool - 文件复制工具', () => {
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
vi.mocked(fs.stat).mockResolvedValue({
|
|
||||||
isDirectory: () => false,
|
|
||||||
isFile: () => true,
|
|
||||||
} as any);
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('工具定义', () => {
|
|
||||||
it('有正确的名称', () => {
|
|
||||||
expect(copyFileTool.name).toBe('copy_file');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('有正确的元数据', () => {
|
|
||||||
expect(copyFileTool.metadata.category).toBe('filesystem');
|
|
||||||
expect(copyFileTool.metadata.keywords).toContain('copy');
|
|
||||||
expect(copyFileTool.metadata.keywords).toContain('cp');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('定义了必需参数', () => {
|
|
||||||
expect(copyFileTool.parameters.source.required).toBe(true);
|
|
||||||
expect(copyFileTool.parameters.destination.required).toBe(true);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('execute - 执行', () => {
|
|
||||||
it('成功复制文件', async () => {
|
|
||||||
// 第一次调用检查源文件,第二次调用检查目标是否是目录
|
|
||||||
vi.mocked(fs.stat)
|
|
||||||
.mockResolvedValueOnce({ isDirectory: () => false } as any)
|
|
||||||
.mockRejectedValueOnce(new Error('ENOENT')); // 目标不存在
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'src.txt',
|
|
||||||
destination: 'dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('已复制');
|
|
||||||
expect(fs.copyFile).toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('复制到已存在的目录', async () => {
|
|
||||||
vi.mocked(fs.stat)
|
|
||||||
.mockResolvedValueOnce({ isDirectory: () => false } as any) // 源文件
|
|
||||||
.mockResolvedValueOnce({ isDirectory: () => true } as any); // 目标是目录
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'file.txt',
|
|
||||||
destination: '/target/dir',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('file.txt');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('递归复制目录', async () => {
|
|
||||||
vi.mocked(fs.stat)
|
|
||||||
.mockResolvedValueOnce({ isDirectory: () => true } as any) // 源是目录
|
|
||||||
.mockRejectedValueOnce(new Error('ENOENT')); // 目标不存在
|
|
||||||
|
|
||||||
vi.mocked(fs.readdir).mockResolvedValueOnce([]);
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'src_dir',
|
|
||||||
destination: 'dest_dir',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(fs.mkdir).toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('源文件读取权限被拒绝', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: false,
|
|
||||||
action: 'deny',
|
|
||||||
reason: '不允许读取',
|
|
||||||
}),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: '/etc/passwd',
|
|
||||||
destination: 'copy.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('权限被拒绝');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('目标位置写入权限被拒绝', async () => {
|
|
||||||
const mockCheck = vi.fn()
|
|
||||||
.mockResolvedValueOnce({ allowed: true }) // 读取权限
|
|
||||||
.mockResolvedValueOnce({ allowed: false, action: 'deny', reason: '不允许写入' }); // 复制权限
|
|
||||||
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: mockCheck,
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'src.txt',
|
|
||||||
destination: '/protected/dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('权限被拒绝');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('源文件需要确认', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: false,
|
|
||||||
action: 'ask',
|
|
||||||
needsConfirmation: true,
|
|
||||||
}),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: '/sensitive/file',
|
|
||||||
destination: 'dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('需要用户确认');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('源文件不存在返回错误', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({ allowed: true }),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
vi.mocked(fs.stat).mockRejectedValue(new Error('ENOENT: no such file'));
|
|
||||||
|
|
||||||
const result = await copyFileTool.execute({
|
|
||||||
source: 'nonexistent.txt',
|
|
||||||
destination: 'dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('ENOENT');
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,156 +0,0 @@
|
|||||||
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
|
||||||
|
|
||||||
// Mock fs/promises
|
|
||||||
vi.mock('fs/promises', () => ({
|
|
||||||
stat: vi.fn(),
|
|
||||||
mkdir: vi.fn().mockResolvedValue(undefined),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Mock permission manager
|
|
||||||
vi.mock('../../../../src/permission/index.js', () => ({
|
|
||||||
getPermissionManager: vi.fn(() => ({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: true,
|
|
||||||
action: 'allow',
|
|
||||||
}),
|
|
||||||
})),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Mock loadDescription
|
|
||||||
vi.mock('../../../../src/tools/load_description.js', () => ({
|
|
||||||
loadDescription: vi.fn(() => '创建目录'),
|
|
||||||
}));
|
|
||||||
|
|
||||||
import { createDirectoryTool } from '../../../../src/tools/filesystem/create_directory.js';
|
|
||||||
import * as fs from 'fs/promises';
|
|
||||||
import { getPermissionManager } from '../../../../src/permission/index.js';
|
|
||||||
|
|
||||||
describe('createDirectoryTool - 创建目录工具', () => {
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
// 默认目录不存在
|
|
||||||
vi.mocked(fs.stat).mockRejectedValue(new Error('ENOENT'));
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('工具定义', () => {
|
|
||||||
it('有正确的名称', () => {
|
|
||||||
expect(createDirectoryTool.name).toBe('create_directory');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('有正确的元数据', () => {
|
|
||||||
expect(createDirectoryTool.metadata.category).toBe('filesystem');
|
|
||||||
expect(createDirectoryTool.metadata.keywords).toContain('create');
|
|
||||||
expect(createDirectoryTool.metadata.keywords).toContain('directory');
|
|
||||||
expect(createDirectoryTool.metadata.keywords).toContain('mkdir');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('定义了必需的 path 参数', () => {
|
|
||||||
expect(createDirectoryTool.parameters.path.required).toBe(true);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('execute - 执行', () => {
|
|
||||||
it('成功创建目录', async () => {
|
|
||||||
const result = await createDirectoryTool.execute({ path: 'new_dir' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('已创建目录');
|
|
||||||
expect(fs.mkdir).toHaveBeenCalledWith(
|
|
||||||
expect.any(String),
|
|
||||||
{ recursive: true }
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('目录已存在返回成功', async () => {
|
|
||||||
vi.mocked(fs.stat).mockResolvedValue({
|
|
||||||
isDirectory: () => true,
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await createDirectoryTool.execute({ path: 'existing_dir' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('目录已存在');
|
|
||||||
expect(fs.mkdir).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('路径是文件返回错误', async () => {
|
|
||||||
vi.mocked(fs.stat).mockResolvedValue({
|
|
||||||
isDirectory: () => false,
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await createDirectoryTool.execute({ path: 'file.txt' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('路径已存在且不是目录');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('权限被拒绝时返回错误', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: false,
|
|
||||||
action: 'deny',
|
|
||||||
reason: '不允许创建目录',
|
|
||||||
}),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await createDirectoryTool.execute({ path: '/protected/dir' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('权限被拒绝');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('需要确认时返回提示', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: false,
|
|
||||||
action: 'ask',
|
|
||||||
needsConfirmation: true,
|
|
||||||
}),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await createDirectoryTool.execute({ path: 'new_dir' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('需要用户确认');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('创建嵌套目录', async () => {
|
|
||||||
// 确保权限检查通过
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({ allowed: true }),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await createDirectoryTool.execute({ path: 'a/b/c/d' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(fs.mkdir).toHaveBeenCalledWith(
|
|
||||||
expect.any(String),
|
|
||||||
{ recursive: true }
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('传递正确参数给权限检查', async () => {
|
|
||||||
const mockCheck = vi.fn().mockResolvedValue({ allowed: true });
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: mockCheck,
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
await createDirectoryTool.execute({ path: 'test_dir' });
|
|
||||||
|
|
||||||
expect(mockCheck).toHaveBeenCalledWith(
|
|
||||||
expect.objectContaining({
|
|
||||||
operation: 'mkdir',
|
|
||||||
})
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('处理创建错误', async () => {
|
|
||||||
vi.mocked(fs.mkdir).mockRejectedValue(new Error('Permission denied'));
|
|
||||||
|
|
||||||
const result = await createDirectoryTool.execute({ path: 'new_dir' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('Permission denied');
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,173 +0,0 @@
|
|||||||
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
|
||||||
|
|
||||||
// Mock fs/promises
|
|
||||||
vi.mock('fs/promises', () => ({
|
|
||||||
stat: vi.fn(),
|
|
||||||
unlink: vi.fn().mockResolvedValue(undefined),
|
|
||||||
rmdir: vi.fn().mockResolvedValue(undefined),
|
|
||||||
rm: vi.fn().mockResolvedValue(undefined),
|
|
||||||
readdir: vi.fn().mockResolvedValue([]),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Mock permission manager
|
|
||||||
vi.mock('../../../../src/permission/index.js', () => ({
|
|
||||||
getPermissionManager: vi.fn(() => ({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: true,
|
|
||||||
action: 'allow',
|
|
||||||
}),
|
|
||||||
})),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Mock loadDescription
|
|
||||||
vi.mock('../../../../src/tools/load_description.js', () => ({
|
|
||||||
loadDescription: vi.fn(() => '删除文件'),
|
|
||||||
}));
|
|
||||||
|
|
||||||
import { deleteFileTool } from '../../../../src/tools/filesystem/delete_file.js';
|
|
||||||
import * as fs from 'fs/promises';
|
|
||||||
import { getPermissionManager } from '../../../../src/permission/index.js';
|
|
||||||
|
|
||||||
describe('deleteFileTool - 文件删除工具', () => {
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('工具定义', () => {
|
|
||||||
it('有正确的名称', () => {
|
|
||||||
expect(deleteFileTool.name).toBe('delete_file');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('有正确的元数据', () => {
|
|
||||||
expect(deleteFileTool.metadata.category).toBe('filesystem');
|
|
||||||
expect(deleteFileTool.metadata.keywords).toContain('delete');
|
|
||||||
expect(deleteFileTool.metadata.keywords).toContain('remove');
|
|
||||||
expect(deleteFileTool.metadata.keywords).toContain('rm');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('定义了必需的 path 参数', () => {
|
|
||||||
expect(deleteFileTool.parameters.path.required).toBe(true);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('定义了可选的 recursive 参数', () => {
|
|
||||||
expect(deleteFileTool.parameters.recursive.required).toBe(false);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('execute - 执行', () => {
|
|
||||||
it('成功删除文件', async () => {
|
|
||||||
vi.mocked(fs.stat).mockResolvedValue({
|
|
||||||
isDirectory: () => false,
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await deleteFileTool.execute({ path: 'file.txt' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('已删除文件');
|
|
||||||
expect(fs.unlink).toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('删除空目录', async () => {
|
|
||||||
vi.mocked(fs.stat).mockResolvedValue({
|
|
||||||
isDirectory: () => true,
|
|
||||||
} as any);
|
|
||||||
vi.mocked(fs.readdir).mockResolvedValue([]);
|
|
||||||
|
|
||||||
const result = await deleteFileTool.execute({ path: 'empty_dir' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('已删除目录');
|
|
||||||
expect(fs.rmdir).toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('非空目录无 recursive 返回错误', async () => {
|
|
||||||
vi.mocked(fs.stat).mockResolvedValue({
|
|
||||||
isDirectory: () => true,
|
|
||||||
} as any);
|
|
||||||
vi.mocked(fs.readdir).mockResolvedValue(['file1.txt', 'file2.txt'] as any);
|
|
||||||
|
|
||||||
const result = await deleteFileTool.execute({ path: 'nonempty_dir' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('目录不为空');
|
|
||||||
expect(result.error).toContain('recursive: true');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('递归删除非空目录', async () => {
|
|
||||||
vi.mocked(fs.stat).mockResolvedValue({
|
|
||||||
isDirectory: () => true,
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await deleteFileTool.execute({
|
|
||||||
path: 'nonempty_dir',
|
|
||||||
recursive: true,
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('已删除目录');
|
|
||||||
expect(fs.rm).toHaveBeenCalledWith(
|
|
||||||
expect.any(String),
|
|
||||||
{ recursive: true }
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it('权限被拒绝时返回错误', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: false,
|
|
||||||
action: 'deny',
|
|
||||||
reason: '不允许删除',
|
|
||||||
}),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await deleteFileTool.execute({ path: '/protected/file' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('权限被拒绝');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('需要确认时返回提示', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: false,
|
|
||||||
action: 'ask',
|
|
||||||
needsConfirmation: true,
|
|
||||||
}),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await deleteFileTool.execute({ path: 'important.txt' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('需要用户确认');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('文件不存在返回错误', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({ allowed: true }),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
vi.mocked(fs.stat).mockRejectedValue(new Error('ENOENT'));
|
|
||||||
|
|
||||||
const result = await deleteFileTool.execute({ path: 'nonexistent.txt' });
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('ENOENT');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('传递正确参数给权限检查', async () => {
|
|
||||||
const mockCheck = vi.fn().mockResolvedValue({ allowed: true });
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: mockCheck,
|
|
||||||
} as any);
|
|
||||||
vi.mocked(fs.stat).mockResolvedValue({ isDirectory: () => false } as any);
|
|
||||||
|
|
||||||
await deleteFileTool.execute({ path: 'test.txt' });
|
|
||||||
|
|
||||||
expect(mockCheck).toHaveBeenCalledWith(
|
|
||||||
expect.objectContaining({
|
|
||||||
operation: 'delete',
|
|
||||||
})
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,171 +0,0 @@
|
|||||||
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
|
||||||
|
|
||||||
// Mock fs/promises
|
|
||||||
vi.mock('fs/promises', () => ({
|
|
||||||
access: vi.fn().mockResolvedValue(undefined),
|
|
||||||
stat: vi.fn(),
|
|
||||||
rename: vi.fn().mockResolvedValue(undefined),
|
|
||||||
mkdir: vi.fn().mockResolvedValue(undefined),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Mock permission manager
|
|
||||||
vi.mock('../../../../src/permission/index.js', () => ({
|
|
||||||
getPermissionManager: vi.fn(() => ({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: true,
|
|
||||||
action: 'allow',
|
|
||||||
}),
|
|
||||||
})),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Mock loadDescription
|
|
||||||
vi.mock('../../../../src/tools/load_description.js', () => ({
|
|
||||||
loadDescription: vi.fn(() => '移动文件'),
|
|
||||||
}));
|
|
||||||
|
|
||||||
import { moveFileTool } from '../../../../src/tools/filesystem/move_file.js';
|
|
||||||
import * as fs from 'fs/promises';
|
|
||||||
import { getPermissionManager } from '../../../../src/permission/index.js';
|
|
||||||
|
|
||||||
describe('moveFileTool - 文件移动工具', () => {
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('工具定义', () => {
|
|
||||||
it('有正确的名称', () => {
|
|
||||||
expect(moveFileTool.name).toBe('move_file');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('有正确的元数据', () => {
|
|
||||||
expect(moveFileTool.metadata.category).toBe('filesystem');
|
|
||||||
expect(moveFileTool.metadata.keywords).toContain('move');
|
|
||||||
expect(moveFileTool.metadata.keywords).toContain('rename');
|
|
||||||
expect(moveFileTool.metadata.keywords).toContain('mv');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('定义了必需参数', () => {
|
|
||||||
expect(moveFileTool.parameters.source.required).toBe(true);
|
|
||||||
expect(moveFileTool.parameters.destination.required).toBe(true);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('execute - 执行', () => {
|
|
||||||
it('成功移动文件', async () => {
|
|
||||||
vi.mocked(fs.stat).mockRejectedValue(new Error('ENOENT')); // 目标不存在
|
|
||||||
|
|
||||||
const result = await moveFileTool.execute({
|
|
||||||
source: 'old.txt',
|
|
||||||
destination: 'new.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('已移动');
|
|
||||||
expect(fs.rename).toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('移动到已存在的目录', async () => {
|
|
||||||
vi.mocked(fs.stat).mockResolvedValue({ isDirectory: () => true } as any);
|
|
||||||
|
|
||||||
const result = await moveFileTool.execute({
|
|
||||||
source: 'file.txt',
|
|
||||||
destination: '/target/dir',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('file.txt');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('源文件移动权限被拒绝', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: false,
|
|
||||||
action: 'deny',
|
|
||||||
reason: '不允许移动',
|
|
||||||
}),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await moveFileTool.execute({
|
|
||||||
source: '/protected/file',
|
|
||||||
destination: 'dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('权限被拒绝');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('目标位置写入权限被拒绝', async () => {
|
|
||||||
const mockCheck = vi.fn()
|
|
||||||
.mockResolvedValueOnce({ allowed: true }) // 移动权限
|
|
||||||
.mockResolvedValueOnce({ allowed: false, action: 'deny', reason: '不允许写入' }); // 写入权限
|
|
||||||
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: mockCheck,
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await moveFileTool.execute({
|
|
||||||
source: 'src.txt',
|
|
||||||
destination: '/protected/dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('权限被拒绝');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('需要确认时返回提示', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: false,
|
|
||||||
action: 'ask',
|
|
||||||
needsConfirmation: true,
|
|
||||||
}),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await moveFileTool.execute({
|
|
||||||
source: 'file.txt',
|
|
||||||
destination: 'new.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('需要用户确认');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('源文件不存在返回错误', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({ allowed: true }),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
vi.mocked(fs.access).mockRejectedValue(new Error('ENOENT'));
|
|
||||||
|
|
||||||
const result = await moveFileTool.execute({
|
|
||||||
source: 'nonexistent.txt',
|
|
||||||
destination: 'dest.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('ENOENT');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('创建目标目录', async () => {
|
|
||||||
// 确保权限检查通过
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({ allowed: true }),
|
|
||||||
} as any);
|
|
||||||
// 源文件存在
|
|
||||||
vi.mocked(fs.access).mockResolvedValue(undefined);
|
|
||||||
// 目标不存在
|
|
||||||
vi.mocked(fs.stat).mockRejectedValue(new Error('ENOENT'));
|
|
||||||
|
|
||||||
const result = await moveFileTool.execute({
|
|
||||||
source: 'file.txt',
|
|
||||||
destination: '/new/path/file.txt',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(fs.mkdir).toHaveBeenCalledWith(
|
|
||||||
expect.any(String),
|
|
||||||
{ recursive: true }
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,199 +0,0 @@
|
|||||||
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
|
||||||
|
|
||||||
// Mock fs/promises
|
|
||||||
vi.mock('fs/promises', () => ({
|
|
||||||
readdir: vi.fn(),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Mock permission manager
|
|
||||||
vi.mock('../../../../src/permission/index.js', () => ({
|
|
||||||
getPermissionManager: vi.fn(() => ({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: true,
|
|
||||||
action: 'allow',
|
|
||||||
}),
|
|
||||||
})),
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Mock loadDescription
|
|
||||||
vi.mock('../../../../src/tools/load_description.js', () => ({
|
|
||||||
loadDescription: vi.fn(() => '按文件名搜索文件'),
|
|
||||||
}));
|
|
||||||
|
|
||||||
import { searchFilesTool } from '../../../../src/tools/filesystem/search_files.js';
|
|
||||||
import * as fs from 'fs/promises';
|
|
||||||
import { getPermissionManager } from '../../../../src/permission/index.js';
|
|
||||||
|
|
||||||
describe('searchFilesTool - 文件搜索工具', () => {
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('工具定义', () => {
|
|
||||||
it('有正确的名称', () => {
|
|
||||||
expect(searchFilesTool.name).toBe('search_files');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('有正确的元数据', () => {
|
|
||||||
expect(searchFilesTool.metadata.category).toBe('filesystem');
|
|
||||||
expect(searchFilesTool.metadata.keywords).toContain('search');
|
|
||||||
expect(searchFilesTool.metadata.keywords).toContain('find');
|
|
||||||
expect(searchFilesTool.metadata.keywords).toContain('glob');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('定义了必需参数', () => {
|
|
||||||
expect(searchFilesTool.parameters.directory.required).toBe(true);
|
|
||||||
expect(searchFilesTool.parameters.pattern.required).toBe(true);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('execute - 执行', () => {
|
|
||||||
it('成功搜索并返回匹配文件', async () => {
|
|
||||||
vi.mocked(fs.readdir).mockResolvedValue([
|
|
||||||
{ name: 'test.ts', isDirectory: () => false, isFile: () => true },
|
|
||||||
{ name: 'test.js', isDirectory: () => false, isFile: () => true },
|
|
||||||
] as any);
|
|
||||||
|
|
||||||
const result = await searchFilesTool.execute({
|
|
||||||
directory: '.',
|
|
||||||
pattern: '*.ts',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('test.ts');
|
|
||||||
expect(result.output).not.toContain('test.js');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('没有匹配时返回提示', async () => {
|
|
||||||
vi.mocked(fs.readdir).mockResolvedValue([
|
|
||||||
{ name: 'test.js', isDirectory: () => false, isFile: () => true },
|
|
||||||
] as any);
|
|
||||||
|
|
||||||
const result = await searchFilesTool.execute({
|
|
||||||
directory: '.',
|
|
||||||
pattern: '*.tsx',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('没有找到匹配的文件');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('递归搜索子目录', async () => {
|
|
||||||
vi.mocked(fs.readdir)
|
|
||||||
.mockResolvedValueOnce([
|
|
||||||
{ name: 'src', isDirectory: () => true, isFile: () => false },
|
|
||||||
{ name: 'index.ts', isDirectory: () => false, isFile: () => true },
|
|
||||||
] as any)
|
|
||||||
.mockResolvedValueOnce([
|
|
||||||
{ name: 'app.ts', isDirectory: () => false, isFile: () => true },
|
|
||||||
] as any);
|
|
||||||
|
|
||||||
const result = await searchFilesTool.execute({
|
|
||||||
directory: '.',
|
|
||||||
pattern: '*.ts',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('index.ts');
|
|
||||||
expect(result.output).toContain('app.ts');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('跳过隐藏文件和 node_modules', async () => {
|
|
||||||
// 第一次调用返回根目录内容,第二次调用返回 src 目录内容
|
|
||||||
vi.mocked(fs.readdir)
|
|
||||||
.mockResolvedValueOnce([
|
|
||||||
{ name: '.git', isDirectory: () => true, isFile: () => false },
|
|
||||||
{ name: 'node_modules', isDirectory: () => true, isFile: () => false },
|
|
||||||
{ name: 'src', isDirectory: () => true, isFile: () => false },
|
|
||||||
] as any)
|
|
||||||
.mockResolvedValueOnce([
|
|
||||||
{ name: 'index.ts', isDirectory: () => false, isFile: () => true },
|
|
||||||
] as any);
|
|
||||||
|
|
||||||
await searchFilesTool.execute({
|
|
||||||
directory: '.',
|
|
||||||
pattern: '*',
|
|
||||||
});
|
|
||||||
|
|
||||||
// 不应该进入隐藏目录或 node_modules,只进入 src
|
|
||||||
expect(fs.readdir).toHaveBeenCalledTimes(2); // 根目录 + src
|
|
||||||
});
|
|
||||||
|
|
||||||
it('权限被拒绝时返回错误', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: false,
|
|
||||||
action: 'deny',
|
|
||||||
reason: '不允许搜索',
|
|
||||||
}),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await searchFilesTool.execute({
|
|
||||||
directory: '/protected',
|
|
||||||
pattern: '*',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('权限被拒绝');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('需要确认时返回提示', async () => {
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({
|
|
||||||
allowed: false,
|
|
||||||
action: 'ask',
|
|
||||||
needsConfirmation: true,
|
|
||||||
}),
|
|
||||||
} as any);
|
|
||||||
|
|
||||||
const result = await searchFilesTool.execute({
|
|
||||||
directory: '.',
|
|
||||||
pattern: '*',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(false);
|
|
||||||
expect(result.error).toContain('需要用户确认');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('支持 glob 模式匹配', async () => {
|
|
||||||
// 恢复权限检查
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: vi.fn().mockResolvedValue({ allowed: true }),
|
|
||||||
} as any);
|
|
||||||
vi.mocked(fs.readdir).mockResolvedValue([
|
|
||||||
{ name: 'component.tsx', isDirectory: () => false, isFile: () => true },
|
|
||||||
{ name: 'helper.ts', isDirectory: () => false, isFile: () => true },
|
|
||||||
{ name: 'style.css', isDirectory: () => false, isFile: () => true },
|
|
||||||
] as any);
|
|
||||||
|
|
||||||
// *.tsx 模式会匹配 component.tsx
|
|
||||||
const result = await searchFilesTool.execute({
|
|
||||||
directory: '.',
|
|
||||||
pattern: '*.tsx',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.success).toBe(true);
|
|
||||||
expect(result.output).toContain('component.tsx');
|
|
||||||
expect(result.output).not.toContain('style.css');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('传递正确参数给权限检查', async () => {
|
|
||||||
const mockCheck = vi.fn().mockResolvedValue({ allowed: true });
|
|
||||||
vi.mocked(getPermissionManager).mockReturnValue({
|
|
||||||
checkFilePermission: mockCheck,
|
|
||||||
} as any);
|
|
||||||
vi.mocked(fs.readdir).mockResolvedValue([]);
|
|
||||||
|
|
||||||
await searchFilesTool.execute({
|
|
||||||
directory: 'src',
|
|
||||||
pattern: '*.ts',
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(mockCheck).toHaveBeenCalledWith(
|
|
||||||
expect.objectContaining({
|
|
||||||
operation: 'search',
|
|
||||||
})
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -165,13 +165,9 @@ describe('loadDescription', () => {
|
|||||||
{ tool: 'write_file', category: 'filesystem' },
|
{ tool: 'write_file', category: 'filesystem' },
|
||||||
{ tool: 'edit_file', category: 'filesystem' },
|
{ tool: 'edit_file', category: 'filesystem' },
|
||||||
{ tool: 'list_directory', category: 'filesystem' },
|
{ tool: 'list_directory', category: 'filesystem' },
|
||||||
{ tool: 'create_directory', category: 'filesystem' },
|
{ tool: 'glob', category: 'filesystem' },
|
||||||
{ tool: 'search_files', category: 'filesystem' },
|
|
||||||
{ tool: 'grep', category: 'filesystem' },
|
{ tool: 'grep', category: 'filesystem' },
|
||||||
{ tool: 'get_file_info', category: 'filesystem' },
|
{ tool: 'get_file_info', category: 'filesystem' },
|
||||||
{ tool: 'move_file', category: 'filesystem' },
|
|
||||||
{ tool: 'copy_file', category: 'filesystem' },
|
|
||||||
{ tool: 'delete_file', category: 'filesystem' },
|
|
||||||
// web
|
// web
|
||||||
{ tool: 'web_search', category: 'web' },
|
{ tool: 'web_search', category: 'web' },
|
||||||
{ tool: 'web_extract', category: 'web' },
|
{ tool: 'web_extract', category: 'web' },
|
||||||
|
|||||||
Reference in New Issue
Block a user