@saber2pr/ai-agent 0.0.43 → 0.0.44

package/lib/tools/filesystem/index.js

@@ -4,6 +4,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getFilesystemTools = void 0;
+/* eslint-disable no-continue */
 const promises_1 = __importDefault(require("fs/promises"));
 const minimatch_1 = require("minimatch");
 const path_1 = __importDefault(require("path"));
@@ -14,13 +15,13 @@ const lib_1 = require("./lib");
 const ReadTextFileArgsSchema = zod_1.z.object({
     path: zod_1.z.string(),
     tail: zod_1.z.number().optional().describe('If provided, returns only the last N lines of the file'),
-    head: zod_1.z.number().optional().describe('If provided, returns only the first N lines of the file')
+    head: zod_1.z.number().optional().describe('If provided, returns only the first N lines of the file'),
 });
 const ReadMultipleFilesArgsSchema = zod_1.z.object({
     paths: zod_1.z
         .array(zod_1.z.string())
-        .min(1, "At least one file path must be provided")
-        .describe("Array of file paths to read. Each path must be a string pointing to a valid file within allowed directories."),
+        .min(1, 'At least one file path must be provided')
+        .describe('Array of file paths to read. Each path must be a string pointing to a valid file within allowed directories.'),
 });
 const WriteFileArgsSchema = zod_1.z.object({
     path: zod_1.z.string(),
@@ -28,12 +29,16 @@ const WriteFileArgsSchema = zod_1.z.object({
 });
 const EditOperation = zod_1.z.object({
     oldText: zod_1.z.string().describe('Text to search for - must match exactly'),
-    newText: zod_1.z.string().describe('Text to replace with')
+    newText: zod_1.z.string().describe('Text to replace with'),
 });
 const EditFileArgsSchema = zod_1.z.object({
     path: zod_1.z.string(),
     edits: zod_1.z.array(EditOperation),
-    dryRun: zod_1.z.boolean().optional().default(false).describe('Preview changes using git-style diff format')
+    dryRun: zod_1.z
+        .boolean()
+        .optional()
+        .default(false)
+        .describe('Preview changes using git-style diff format'),
 });
 const CreateDirectoryArgsSchema = zod_1.z.object({
     path: zod_1.z.string(),
@@ -43,11 +48,15 @@ const ListDirectoryArgsSchema = zod_1.z.object({
 });
 const ListDirectoryWithSizesArgsSchema = zod_1.z.object({
     path: zod_1.z.string(),
-    sortBy: zod_1.z.enum(['name', 'size']).optional().default('name').describe('Sort entries by name or size'),
+    sortBy: zod_1.z
+        .enum(['name', 'size'])
+        .optional()
+        .default('name')
+        .describe('Sort entries by name or size'),
 });
 const DirectoryTreeArgsSchema = zod_1.z.object({
     path: zod_1.z.string(),
-    excludePatterns: zod_1.z.array(zod_1.z.string()).optional().default([])
+    excludePatterns: zod_1.z.array(zod_1.z.string()).optional().default([]),
 });
 const MoveFileArgsSchema = zod_1.z.object({
     source: zod_1.z.string(),
@@ -56,18 +65,23 @@ const MoveFileArgsSchema = zod_1.z.object({
 const SearchFilesArgsSchema = zod_1.z.object({
     path: zod_1.z.string(),
     pattern: zod_1.z.string(),
-    excludePatterns: zod_1.z.array(zod_1.z.string()).optional().default([])
+    excludePatterns: zod_1.z.array(zod_1.z.string()).optional().default([]),
 });
 const GetFileInfoArgsSchema = zod_1.z.object({
     path: zod_1.z.string(),
 });
+const GrepSearchArgsSchema = zod_1.z.object({
+    path: zod_1.z.string().describe('The starting directory path for searching content'),
+    query: zod_1.z.string().describe('The text keyword to search for'),
+    includePattern: zod_1.z.string().optional().default('**/*').describe('Only search in files matching this pattern, for example "**/*.ts"'),
+});
 const getFilesystemTools = (targetDir) => {
     (0, lib_1.setAllowedDirectories)([targetDir]);
     // read_file (deprecated) and read_text_file
     const readTextFileHandler = async (args) => {
         const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
         if (args.head && args.tail) {
-            throw new Error("Cannot specify both head and tail parameters simultaneously");
+            throw new Error('Cannot specify both head and tail parameters simultaneously');
         }
         let content;
         if (args.tail) {
@@ -82,23 +96,23 @@ const getFilesystemTools = (targetDir) => {
         return content;
     };
     const readTextFileTool = (0, createTool_1.createTool)({
-        name: "read_text_file",
-        description: "Read the complete contents of a file from the file system as text. " +
-            "Handles various text encodings and provides detailed error messages " +
-            "if the file cannot be read. Use this tool when you need to examine " +
+        name: 'read_text_file',
+        description: 'Read the complete contents of a file from the file system as text. ' +
+            'Handles various text encodings and provides detailed error messages ' +
+            'if the file cannot be read. Use this tool when you need to examine ' +
             "the contents of a single file. Use the 'head' parameter to read only " +
             "the first N lines of a file, or the 'tail' parameter to read only " +
-            "the last N lines of a file. Operates on the file as text regardless of extension.",
+            'the last N lines of a file. Operates on the file as text regardless of extension.',
         parameters: ReadTextFileArgsSchema,
-        handler: readTextFileHandler
+        handler: readTextFileHandler,
     });
     const readMultipleFilesTool = (0, createTool_1.createTool)({
-        name: "read_multiple_files",
-        description: "Read the contents of multiple files simultaneously. This is more " +
-            "efficient than reading files one by one when you need to analyze " +
+        name: 'read_multiple_files',
+        description: 'Read the contents of multiple files simultaneously. This is more ' +
+            'efficient than reading files one by one when you need to analyze ' +
             "or compare multiple files. Each file's content is returned with its " +
             "path as a reference. Failed reads for individual files won't stop " +
-            "the entire operation. Only works within allowed directories.",
+            'the entire operation. Only works within allowed directories.',
         parameters: ReadMultipleFilesArgsSchema,
         handler: async (args) => {
             const results = await Promise.all(args.paths.map(async (filePath) => {
@@ -112,71 +126,71 @@ const getFilesystemTools = (targetDir) => {
                     return `${filePath}: Error - ${errorMessage}`;
                 }
             }));
-            const text = results.join("\n---\n");
+            const text = results.join('\n---\n');
             return text;
-        }
+        },
     });
     const writeFileTool = (0, createTool_1.createTool)({
-        name: "write_file",
-        description: "Create a new file or completely overwrite an existing file with new content. " +
-            "Use with caution as it will overwrite existing files without warning. " +
-            "Handles text content with proper encoding. Only works within allowed directories.",
+        name: 'write_file',
+        description: 'Create a new file or completely overwrite an existing file with new content. ' +
+            'Use with caution as it will overwrite existing files without warning. ' +
+            'Handles text content with proper encoding. Only works within allowed directories.',
         parameters: WriteFileArgsSchema,
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
             await (0, lib_1.writeFileContent)(validPath, args.content);
             const text = `Successfully wrote to ${args.path}`;
             return text;
-        }
+        },
     });
     const editFileTool = (0, createTool_1.createTool)({
-        name: "edit_file",
-        description: "Make line-based edits to a text file. Each edit replaces exact line sequences " +
-            "with new content. Returns a git-style diff showing the changes made. " +
-            "Only works within allowed directories.",
+        name: 'edit_file',
+        description: 'Make line-based edits to a text file. Each edit replaces exact line sequences ' +
+            'with new content. Returns a git-style diff showing the changes made. ' +
+            'Only works within allowed directories.',
         parameters: EditFileArgsSchema,
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
             const result = await (0, lib_1.applyFileEdits)(validPath, args.edits, args.dryRun);
             return result;
-        }
+        },
     });
     const createDirectoryTool = (0, createTool_1.createTool)({
-        name: "create_directory",
-        description: "Create a new directory or ensure a directory exists. Can create multiple " +
-            "nested directories in one operation. If the directory already exists, " +
-            "this operation will succeed silently. Perfect for setting up directory " +
-            "structures for projects or ensuring required paths exist. Only works within allowed directories.",
+        name: 'create_directory',
+        description: 'Create a new directory or ensure a directory exists. Can create multiple ' +
+            'nested directories in one operation. If the directory already exists, ' +
+            'this operation will succeed silently. Perfect for setting up directory ' +
+            'structures for projects or ensuring required paths exist. Only works within allowed directories.',
         parameters: CreateDirectoryArgsSchema,
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
             await promises_1.default.mkdir(validPath, { recursive: true });
             const text = `Successfully created directory ${args.path}`;
             return text;
-        }
+        },
     });
     const listDirectoryTool = (0, createTool_1.createTool)({
-        name: "list_directory",
-        description: "Get a detailed listing of all files and directories in a specified path. " +
-            "Results clearly distinguish between files and directories with [FILE] and [DIR] " +
-            "prefixes. This tool is essential for understanding directory structure and " +
-            "finding specific files within a directory. Only works within allowed directories.",
+        name: 'list_directory',
+        description: 'Get a detailed listing of all files and directories in a specified path. ' +
+            'Results clearly distinguish between files and directories with [FILE] and [DIR] ' +
+            'prefixes. This tool is essential for understanding directory structure and ' +
+            'finding specific files within a directory. Only works within allowed directories.',
         parameters: ListDirectoryArgsSchema,
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
             const entries = await promises_1.default.readdir(validPath, { withFileTypes: true });
             const formatted = entries
-                .map((entry) => `${entry.isDirectory() ? "[DIR]" : "[FILE]"} ${entry.name}`)
-                .join("\n");
+                .map(entry => `${entry.isDirectory() ? '[DIR]' : '[FILE]'} ${entry.name}`)
+                .join('\n');
             return formatted;
-        }
+        },
     });
     const listDirectoryWithSizesTool = (0, createTool_1.createTool)({
-        name: "list_directory_with_sizes",
-        description: "Get a detailed listing of all files and directories in a specified path, including sizes. " +
-            "Results clearly distinguish between files and directories with [FILE] and [DIR] " +
-            "prefixes. This tool is useful for understanding directory structure and " +
-            "finding specific files within a directory. Only works within allowed directories.",
+        name: 'list_directory_with_sizes',
+        description: 'Get a detailed listing of all files and directories in a specified path, including sizes. ' +
+            'Results clearly distinguish between files and directories with [FILE] and [DIR] ' +
+            'prefixes. This tool is useful for understanding directory structure and ' +
+            'finding specific files within a directory. Only works within allowed directories.',
         parameters: ListDirectoryWithSizesArgsSchema,
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
@@ -190,7 +204,7 @@ const getFilesystemTools = (targetDir) => {
                         name: entry.name,
                         isDirectory: entry.isDirectory(),
                         size: stats.size,
-                        mtime: stats.mtime
+                        mtime: stats.mtime,
                     };
                 }
                 catch (error) {
@@ -198,7 +212,7 @@ const getFilesystemTools = (targetDir) => {
                         name: entry.name,
                         isDirectory: entry.isDirectory(),
                         size: 0,
-                        mtime: new Date(0)
+                        mtime: new Date(0),
                     };
                 }
             }));
@@ -211,26 +225,26 @@ const getFilesystemTools = (targetDir) => {
                 return a.name.localeCompare(b.name);
             });
             // Format the output
-            const formattedEntries = sortedEntries.map(entry => `${entry.isDirectory ? "[DIR]" : "[FILE]"} ${entry.name.padEnd(30)} ${entry.isDirectory ? "" : (0, lib_1.formatSize)(entry.size).padStart(10)}`);
+            const formattedEntries = sortedEntries.map(entry => `${entry.isDirectory ? '[DIR]' : '[FILE]'} ${entry.name.padEnd(30)} ${entry.isDirectory ? '' : (0, lib_1.formatSize)(entry.size).padStart(10)}`);
             // Add summary
             const totalFiles = detailedEntries.filter(e => !e.isDirectory).length;
             const totalDirs = detailedEntries.filter(e => e.isDirectory).length;
             const totalSize = detailedEntries.reduce((sum, entry) => sum + (entry.isDirectory ? 0 : entry.size), 0);
             const summary = [
-                "",
+                '',
                 `Total: ${totalFiles} files, ${totalDirs} directories`,
-                `Combined size: ${(0, lib_1.formatSize)(totalSize)}`
+                `Combined size: ${(0, lib_1.formatSize)(totalSize)}`,
             ];
-            const text = [...formattedEntries, ...summary].join("\n");
+            const text = [...formattedEntries, ...summary].join('\n');
             return text;
-        }
+        },
     });
     const directoryTreeTool = (0, createTool_1.createTool)({
-        name: "directory_tree",
-        description: "Get a recursive tree view of files and directories as a JSON structure. " +
+        name: 'directory_tree',
+        description: 'Get a recursive tree view of files and directories as a JSON structure. ' +
             "Each entry includes 'name', 'type' (file/directory), and 'children' for directories. " +
-            "Files have no children array, while directories always have a children array (which may be empty). " +
-            "The output is formatted with 2-space indentation for readability. Only works within allowed directories.",
+            'Files have no children array, while directories always have a children array (which may be empty). ' +
+            'The output is formatted with 2-space indentation for readability. Only works within allowed directories.',
         parameters: DirectoryTreeArgsSchema,
         handler: async (args) => {
             const rootPath = args.path;
@@ -246,15 +260,15 @@ const getFilesystemTools = (targetDir) => {
                         }
                         // For files: match exact name or as part of path
                         // For directories: match as directory path
-                        return (0, minimatch_1.minimatch)(relativePath, pattern, { dot: true }) ||
+                        return ((0, minimatch_1.minimatch)(relativePath, pattern, { dot: true }) ||
                             (0, minimatch_1.minimatch)(relativePath, `**/${pattern}`, { dot: true }) ||
-                            (0, minimatch_1.minimatch)(relativePath, `**/${pattern}/**`, { dot: true });
+                            (0, minimatch_1.minimatch)(relativePath, `**/${pattern}/**`, { dot: true }));
                     });
                     if (shouldExclude)
                         continue;
                     const entryData = {
                         name: entry.name,
-                        type: entry.isDirectory() ? 'directory' : 'file'
+                        type: entry.isDirectory() ? 'directory' : 'file',
                     };
                     if (entry.isDirectory()) {
                         const subPath = path_1.default.join(currentPath, entry.name);
@@ -267,14 +281,14 @@ const getFilesystemTools = (targetDir) => {
             const treeData = await buildTree(rootPath, args.excludePatterns);
             const text = JSON.stringify(treeData, null, 2);
             return text;
-        }
+        },
     });
     const moveFileTool = (0, createTool_1.createTool)({
-        name: "move_file",
-        description: "Move or rename files and directories. Can move files between directories " +
-            "and rename them in a single operation. If the destination exists, the " +
-            "operation will fail. Works across different directories and can be used " +
-            "for simple renaming within the same directory. Both source and destination must be within allowed directories.",
+        name: 'move_file',
+        description: 'Move or rename files and directories. Can move files between directories ' +
+            'and rename them in a single operation. If the destination exists, the ' +
+            'operation will fail. Works across different directories and can be used ' +
+            'for simple renaming within the same directory. Both source and destination must be within allowed directories.',
         parameters: MoveFileArgsSchema,
         handler: async (args) => {
             const validSourcePath = await (0, lib_1.validatePath)(targetDir, args.source);
@@ -282,33 +296,72 @@ const getFilesystemTools = (targetDir) => {
             await promises_1.default.rename(validSourcePath, validDestPath);
             const text = `Successfully moved ${args.source} to ${args.destination}`;
             return text;
-        }
+        },
     });
     const searchFilesTool = (0, createTool_1.createTool)({
-        name: "search_files",
-        description: "Search for files matching a specific pattern in a specified path. " +
-            "Returns a list of files that match the pattern. Only works within allowed directories.",
+        name: 'search_files',
+        description: 'Search for files matching a specific pattern in a specified path. ' +
+            'Returns a list of files that match the pattern. Only works within allowed directories.',
         parameters: SearchFilesArgsSchema,
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
-            const results = await (0, lib_1.searchFilesWithValidation)(targetDir, validPath, args.pattern, [], { excludePatterns: args.excludePatterns });
-            const text = results.length > 0 ? results.join("\n") : "No matches found";
+            const results = await (0, lib_1.searchFilesWithValidation)(targetDir, validPath, args.pattern, [targetDir], {
+                excludePatterns: ['node_modules', ...((args === null || args === void 0 ? void 0 : args.excludePatterns) || [])],
+            });
+            const text = results.length > 0 ? results.join('\n') : 'No matches found';
             return text;
-        }
+        },
     });
     const getFileInfoTool = (0, createTool_1.createTool)({
-        name: "get_file_info",
-        description: "Get detailed information about a file, including its size, last modified time, and type. " +
-            "Only works within allowed directories.",
+        name: 'get_file_info',
+        description: 'Get detailed information about a file, including its size, last modified time, and type. ' +
+            'Only works within allowed directories.',
         parameters: GetFileInfoArgsSchema,
         handler: async (args) => {
             const validPath = await (0, lib_1.validatePath)(targetDir, args.path);
             const info = await (0, lib_1.getFileStats)(validPath);
             const text = Object.entries(info)
                 .map(([key, value]) => `${key}: ${value}`)
-                .join("\n");
+                .join('\n');
             return text;
-        }
+        },
+    });
+    const grepSearchTool = (0, createTool_1.createTool)({
+        name: 'grep_search',
+        description: 'Search for a specific keyword in the content of files in a specified directory. ' +
+            'This tool will recursively read files and return the paths of all files containing the keyword. ' +
+            'Useful for finding variable definitions, function calls, or specific text.',
+        parameters: GrepSearchArgsSchema,
+        handler: async (args) => {
+            const startPath = await (0, lib_1.validatePath)(targetDir, args.path);
+            // 1. 利用你现有的工具先筛选出文件列表
+            const allFiles = await (0, lib_1.searchFilesWithValidation)(targetDir, startPath, args.includePattern, [targetDir], { excludePatterns: ['node_modules', 'dist', '.git'] });
+            const matches = [];
+            // 2. 并发读取文件内容进行关键词匹配 (限制并发数为 10，防止内存或句柄爆炸)
+            const concurrencyLimit = 10;
+            for (let i = 0; i < allFiles.length; i += concurrencyLimit) {
+                const chunk = allFiles.slice(i, i + concurrencyLimit);
+                await Promise.all(chunk.map(async (filePath) => {
+                    try {
+                        // 注意：这里确保只搜索文件，不搜索目录
+                        const stats = await promises_1.default.stat(filePath);
+                        if (!stats.isFile())
+                            return;
+                        const content = await (0, lib_1.readFileContent)(filePath);
+                        if (content.includes(args.query)) {
+                            // 返回相对路径，方便查看
+                            matches.push(path_1.default.relative(targetDir, filePath));
+                        }
+                    }
+                    catch {
+                        // 忽略读取失败的文件（如二进制文件或无权限文件）
+                    }
+                }));
+            }
+            return matches.length > 0
+                ? `Found keyword "${args.query}" in the following files:\n${matches.join('\n')}`
+                : `No content containing "${args.query}" found in the specified range.`;
+        },
     });
     return [
         readTextFileTool,
@@ -322,6 +375,7 @@ const getFilesystemTools = (targetDir) => {
         moveFileTool,
         searchFilesTool,
         getFileInfoTool,
+        grepSearchTool,
     ];
 };
 exports.getFilesystemTools = getFilesystemTools;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saber2pr/ai-agent",
-  "version": "0.0.43",
+  "version": "0.0.44",
   "description": "AI Assistant CLI.",
   "author": "saber2pr",
   "license": "ISC",