Initial commit to new repository

.decompiledproj/AxCopilot/Services/Agent/CodeSearchTool.cs

@@ -0,0 +1,127 @@
+using System.Collections.Generic;
+using System.Text;
+using System.Text.Json;
+using System.Threading;
+using System.Threading.Tasks;
+using System.Windows;
+
+namespace AxCopilot.Services.Agent;
+
+public class CodeSearchTool : IAgentTool
+{
+	private static CodeIndexService? _indexService;
+
+	private static string _lastWorkFolder = "";
+
+	public string Name => "search_codebase";
+
+	public string Description => "코드베이스를 시맨틱 검색합니다. 자연어 질문으로 관련 있는 코드를 찾습니다.\n예: '사용자 인증 로직', '데이터베이스 연결 설정', '에러 핸들링 패턴'\n인덱스는 영속 저장되어 변경된 파일만 증분 업데이트합니다.";
+
+	public ToolParameterSchema Parameters => new ToolParameterSchema
+	{
+		Properties = new Dictionary<string, ToolProperty>
+		{
+			["query"] = new ToolProperty
+			{
+				Type = "string",
+				Description = "검색할 내용 (자연어 또는 키워드)"
+			},
+			["max_results"] = new ToolProperty
+			{
+				Type = "integer",
+				Description = "최대 결과 수 (기본 5)"
+			},
+			["reindex"] = new ToolProperty
+			{
+				Type = "boolean",
+				Description = "true면 기존 인덱스를 버리고 전체 재인덱싱 (기본 false)"
+			}
+		},
+		Required = new List<string> { "query" }
+	};
+
+	public async Task<ToolResult> ExecuteAsync(JsonElement args, AgentContext context, CancellationToken ct = default(CancellationToken))
+	{
+		if (!((Application.Current as App)?.SettingsService?.Settings.Llm.Code.EnableCodeIndex ?? true))
+		{
+			return ToolResult.Ok("코드 시맨틱 검색이 비활성 상태입니다. 설정 → AX Agent → 코드에서 활성화하세요.");
+		}
+		JsonElement q;
+		string query = (args.TryGetProperty("query", out q) ? (q.GetString() ?? "") : "");
+		JsonElement m;
+		int maxResults = (args.TryGetProperty("max_results", out m) ? m.GetInt32() : 5);
+		JsonElement ri;
+		bool reindex = args.TryGetProperty("reindex", out ri) && ri.GetBoolean();
+		if (string.IsNullOrWhiteSpace(query))
+		{
+			return ToolResult.Fail("query가 필요합니다.");
+		}
+		if (string.IsNullOrEmpty(context.WorkFolder))
+		{
+			return ToolResult.Fail("작업 폴더가 설정되어 있지 않습니다.");
+		}
+		if (_indexService == null || _lastWorkFolder != context.WorkFolder || reindex)
+		{
+			_indexService?.Dispose();
+			_indexService = new CodeIndexService();
+			_lastWorkFolder = context.WorkFolder;
+			if (!reindex)
+			{
+				_indexService.TryLoadExisting(context.WorkFolder);
+			}
+		}
+		if (!_indexService.IsIndexed || reindex)
+		{
+			await _indexService.IndexAsync(context.WorkFolder, ct);
+			if (!_indexService.IsIndexed)
+			{
+				return ToolResult.Fail("코드 인덱싱에 실패했습니다.");
+			}
+		}
+		List<SearchResult> results = _indexService.Search(query, maxResults);
+		if (results.Count == 0)
+		{
+			return ToolResult.Ok("'" + query + "'에 대한 관련 코드를 찾지 못했습니다. 다른 키워드로 검색해보세요.");
+		}
+		StringBuilder sb = new StringBuilder();
+		StringBuilder stringBuilder = sb;
+		StringBuilder stringBuilder2 = stringBuilder;
+		StringBuilder.AppendInterpolatedStringHandler handler = new StringBuilder.AppendInterpolatedStringHandler(24, 3, stringBuilder);
+		handler.AppendLiteral("'");
+		handler.AppendFormatted(query);
+		handler.AppendLiteral("' 검색 결과 (");
+		handler.AppendFormatted(results.Count);
+		handler.AppendLiteral("개, 인덱스 ");
+		handler.AppendFormatted(_indexService.ChunkCount);
+		handler.AppendLiteral("개 청크):\n");
+		stringBuilder2.AppendLine(ref handler);
+		foreach (SearchResult r in results)
+		{
+			stringBuilder = sb;
+			StringBuilder stringBuilder3 = stringBuilder;
+			handler = new StringBuilder.AppendInterpolatedStringHandler(20, 4, stringBuilder);
+			handler.AppendLiteral("\ud83d\udcc1 ");
+			handler.AppendFormatted(r.FilePath);
+			handler.AppendLiteral(" (line ");
+			handler.AppendFormatted(r.StartLine);
+			handler.AppendLiteral("-");
+			handler.AppendFormatted(r.EndLine);
+			handler.AppendLiteral(", score ");
+			handler.AppendFormatted(r.Score, "F3");
+			handler.AppendLiteral(")");
+			stringBuilder3.AppendLine(ref handler);
+			sb.AppendLine("```");
+			sb.AppendLine(r.Preview);
+			sb.AppendLine("```");
+			sb.AppendLine();
+		}
+		return ToolResult.Ok(sb.ToString());
+	}
+
+	public static void InvalidateIndex()
+	{
+		_indexService?.Dispose();
+		_indexService = null;
+		_lastWorkFolder = "";
+	}
+}