feat: Phase1 项目骨架 - Maven项目结构、核心AgentLoop、6个工具、命令系统、控制台渲染、REPL会话
实现内容: - pom.xml: JDK25 + Spring AI 2.0.0-M4 + JLine3 + Picocli - core/AgentLoop: 基于ChatModel的显式工具循环（非ChatClient） - tool/: Tool接口 + ToolRegistry + ToolCallbackAdapter（适配Spring AI） - tool/impl/: BashTool, FileReadTool, FileWriteTool, FileEditTool, GlobTool, GrepTool - command/: SlashCommand接口 + CommandRegistry + /help, /clear, /exit - console/: AnsiStyle, BannerPrinter, ToolStatusRenderer, ThinkingRenderer, SpinnerAnimation, MarkdownRenderer - context/: SystemPromptBuilder, ClaudeMdLoader（多级CLAUDE.md加载） - repl/ReplSession: REPL主循环（Scanner降级方案） - config/AppConfig: Spring Bean装配 - application.yml: Anthropic/OpenAI双模型配置 Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
This commit is contained in:
liuzh
@@ -0,0 +1,169 @@
|
||||
package com.claudecode.tool.impl;
|
||||
|
||||
import com.claudecode.tool.Tool;
|
||||
import com.claudecode.tool.ToolContext;
|
||||
|
||||
import java.io.BufferedReader;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStreamReader;
|
||||
import java.nio.file.Path;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
/**
|
||||
* Grep 搜索工具 —— 对应 claude-code/src/tools/grep/GrepTool.ts。
|
||||
* <p>
|
||||
* 在文件中搜索文本模式（正则），优先使用 ripgrep（rg），降级为系统 grep。
|
||||
*/
|
||||
public class GrepTool implements Tool {
|
||||
|
||||
private static final int MAX_RESULTS = 100;
|
||||
|
||||
@Override
|
||||
public String name() {
|
||||
return "Grep";
|
||||
}
|
||||
|
||||
@Override
|
||||
public String description() {
|
||||
return """
|
||||
Search for a pattern in file contents using regex. Returns matching lines with \
|
||||
file paths and line numbers. Uses ripgrep (rg) if available, falls back to grep.""";
|
||||
}
|
||||
|
||||
@Override
|
||||
public String inputSchema() {
|
||||
return """
|
||||
{
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"pattern": {
|
||||
"type": "string",
|
||||
"description": "Regular expression pattern to search for"
|
||||
},
|
||||
"path": {
|
||||
"type": "string",
|
||||
"description": "Directory or file to search in (default: working directory)"
|
||||
},
|
||||
"include": {
|
||||
"type": "string",
|
||||
"description": "File glob pattern to include (e.g., '*.java')"
|
||||
}
|
||||
},
|
||||
"required": ["pattern"]
|
||||
}""";
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean isReadOnly() {
|
||||
return true;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String execute(Map<String, Object> input, ToolContext context) {
|
||||
String pattern = (String) input.get("pattern");
|
||||
String searchPath = (String) input.getOrDefault("path", ".");
|
||||
String include = (String) input.getOrDefault("include", null);
|
||||
Path baseDir = context.getWorkDir().resolve(searchPath).normalize();
|
||||
|
||||
try {
|
||||
List<String> cmd = buildCommand(pattern, baseDir.toString(), include);
|
||||
ProcessBuilder pb = new ProcessBuilder(cmd);
|
||||
pb.directory(context.getWorkDir().toFile());
|
||||
pb.redirectErrorStream(true);
|
||||
|
||||
Process process = pb.start();
|
||||
List<String> lines = new ArrayList<>();
|
||||
|
||||
try (BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()))) {
|
||||
String line;
|
||||
while ((line = reader.readLine()) != null && lines.size() < MAX_RESULTS) {
|
||||
lines.add(line);
|
||||
}
|
||||
}
|
||||
|
||||
process.waitFor(30, TimeUnit.SECONDS);
|
||||
|
||||
if (lines.isEmpty()) {
|
||||
return "No matches found for pattern: " + pattern;
|
||||
}
|
||||
|
||||
StringBuilder sb = new StringBuilder();
|
||||
for (String line : lines) {
|
||||
sb.append(line).append("\n");
|
||||
}
|
||||
if (lines.size() >= MAX_RESULTS) {
|
||||
sb.append("... (results truncated at ").append(MAX_RESULTS).append(")\n");
|
||||
}
|
||||
return sb.toString().stripTrailing();
|
||||
|
||||
} catch (Exception e) {
|
||||
return "Error searching: " + e.getMessage();
|
||||
}
|
||||
}
|
||||
|
||||
/** 构建搜索命令（优先 rg，降级 grep/findstr） */
|
||||
private List<String> buildCommand(String pattern, String path, String include) {
|
||||
boolean isWindows = System.getProperty("os.name").toLowerCase().contains("win");
|
||||
List<String> cmd = new ArrayList<>();
|
||||
|
||||
// 尝试 ripgrep
|
||||
if (isCommandAvailable("rg")) {
|
||||
cmd.add("rg");
|
||||
cmd.add("--no-heading");
|
||||
cmd.add("--line-number");
|
||||
cmd.add("--color=never");
|
||||
cmd.add("--max-count=100");
|
||||
if (include != null) {
|
||||
cmd.add("--glob=" + include);
|
||||
}
|
||||
cmd.add(pattern);
|
||||
cmd.add(path);
|
||||
} else if (isWindows) {
|
||||
// Windows 降级到 findstr（功能有限）
|
||||
cmd.add("findstr");
|
||||
cmd.add("/s");
|
||||
cmd.add("/n");
|
||||
cmd.add("/r");
|
||||
cmd.add(pattern);
|
||||
if (include != null) {
|
||||
cmd.add(path + "\\" + include);
|
||||
} else {
|
||||
cmd.add(path + "\\*");
|
||||
}
|
||||
} else {
|
||||
cmd.add("grep");
|
||||
cmd.add("-rn");
|
||||
cmd.add("--color=never");
|
||||
if (include != null) {
|
||||
cmd.add("--include=" + include);
|
||||
}
|
||||
cmd.add(pattern);
|
||||
cmd.add(path);
|
||||
}
|
||||
|
||||
return cmd;
|
||||
}
|
||||
|
||||
private boolean isCommandAvailable(String command) {
|
||||
try {
|
||||
boolean isWindows = System.getProperty("os.name").toLowerCase().contains("win");
|
||||
Process p;
|
||||
if (isWindows) {
|
||||
p = new ProcessBuilder("where", command).start();
|
||||
} else {
|
||||
p = new ProcessBuilder("which", command).start();
|
||||
}
|
||||
return p.waitFor(5, TimeUnit.SECONDS) && p.exitValue() == 0;
|
||||
} catch (Exception e) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public String activityDescription(Map<String, Object> input) {
|
||||
return "🔍 Searching for '" + input.getOrDefault("pattern", "...") + "'";
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user