Files
AX-Copilot-Codex/.decompiledproj/AxCopilot/Services/Agent/FileInfoTool.cs

179 lines
7.4 KiB
C#

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Text.Json;
using System.Threading;
using System.Threading.Tasks;
namespace AxCopilot.Services.Agent;
public class FileInfoTool : IAgentTool
{
public string Name => "file_info";
public string Description => "Get file or directory metadata without reading contents. Returns: size, created/modified dates, line count (files), item count (directories), encoding hint.";
public ToolParameterSchema Parameters
{
get
{
ToolParameterSchema obj = new ToolParameterSchema
{
Properties = new Dictionary<string, ToolProperty> { ["path"] = new ToolProperty
{
Type = "string",
Description = "File or directory path"
} }
};
int num = 1;
List<string> list = new List<string>(num);
CollectionsMarshal.SetCount(list, num);
CollectionsMarshal.AsSpan(list)[0] = "path";
obj.Required = list;
return obj;
}
}
public Task<ToolResult> ExecuteAsync(JsonElement args, AgentContext context, CancellationToken ct = default(CancellationToken))
{
string text = args.GetProperty("path").GetString() ?? "";
string text2 = (Path.IsPathRooted(text) ? text : Path.Combine(context.WorkFolder, text));
if (!context.IsPathAllowed(text2))
{
return Task.FromResult(ToolResult.Fail("경로 접근 차단: " + text2));
}
try
{
if (File.Exists(text2))
{
FileInfo fileInfo = new FileInfo(text2);
StringBuilder stringBuilder = new StringBuilder();
stringBuilder.AppendLine("Type: File");
StringBuilder stringBuilder2 = stringBuilder;
StringBuilder stringBuilder3 = stringBuilder2;
StringBuilder.AppendInterpolatedStringHandler handler = new StringBuilder.AppendInterpolatedStringHandler(6, 1, stringBuilder2);
handler.AppendLiteral("Path: ");
handler.AppendFormatted(fileInfo.FullName);
stringBuilder3.AppendLine(ref handler);
stringBuilder2 = stringBuilder;
StringBuilder stringBuilder4 = stringBuilder2;
handler = new StringBuilder.AppendInterpolatedStringHandler(15, 2, stringBuilder2);
handler.AppendLiteral("Size: ");
handler.AppendFormatted(FormatSize(fileInfo.Length));
handler.AppendLiteral(" (");
handler.AppendFormatted(fileInfo.Length, "N0");
handler.AppendLiteral(" bytes)");
stringBuilder4.AppendLine(ref handler);
stringBuilder2 = stringBuilder;
StringBuilder stringBuilder5 = stringBuilder2;
handler = new StringBuilder.AppendInterpolatedStringHandler(11, 1, stringBuilder2);
handler.AppendLiteral("Extension: ");
handler.AppendFormatted(fileInfo.Extension);
stringBuilder5.AppendLine(ref handler);
stringBuilder2 = stringBuilder;
StringBuilder stringBuilder6 = stringBuilder2;
handler = new StringBuilder.AppendInterpolatedStringHandler(9, 1, stringBuilder2);
handler.AppendLiteral("Created: ");
handler.AppendFormatted(fileInfo.CreationTime, "yyyy-MM-dd HH:mm:ss");
stringBuilder6.AppendLine(ref handler);
stringBuilder2 = stringBuilder;
StringBuilder stringBuilder7 = stringBuilder2;
handler = new StringBuilder.AppendInterpolatedStringHandler(10, 1, stringBuilder2);
handler.AppendLiteral("Modified: ");
handler.AppendFormatted(fileInfo.LastWriteTime, "yyyy-MM-dd HH:mm:ss");
stringBuilder7.AppendLine(ref handler);
stringBuilder2 = stringBuilder;
StringBuilder stringBuilder8 = stringBuilder2;
handler = new StringBuilder.AppendInterpolatedStringHandler(10, 1, stringBuilder2);
handler.AppendLiteral("ReadOnly: ");
handler.AppendFormatted(fileInfo.IsReadOnly);
stringBuilder8.AppendLine(ref handler);
HashSet<string> hashSet = new HashSet<string>(StringComparer.OrdinalIgnoreCase)
{
".cs", ".py", ".js", ".ts", ".java", ".cpp", ".c", ".h", ".xml", ".json",
".yaml", ".yml", ".md", ".txt", ".csv", ".html", ".htm", ".css", ".sql", ".sh",
".bat", ".ps1", ".config", ".ini", ".log", ".xaml"
};
if (hashSet.Contains(fileInfo.Extension) && fileInfo.Length < 52428800)
{
int num = File.ReadLines(text2).Take(1000000).Count();
stringBuilder2 = stringBuilder;
StringBuilder stringBuilder9 = stringBuilder2;
handler = new StringBuilder.AppendInterpolatedStringHandler(7, 2, stringBuilder2);
handler.AppendLiteral("Lines: ");
handler.AppendFormatted(num, "N0");
handler.AppendFormatted((num >= 1000000) ? "+" : "");
stringBuilder9.AppendLine(ref handler);
}
return Task.FromResult(ToolResult.Ok(stringBuilder.ToString()));
}
if (Directory.Exists(text2))
{
DirectoryInfo directoryInfo = new DirectoryInfo(text2);
FileInfo[] files = directoryInfo.GetFiles("*", SearchOption.TopDirectoryOnly);
DirectoryInfo[] directories = directoryInfo.GetDirectories();
long bytes = files.Sum((FileInfo f) => f.Length);
StringBuilder stringBuilder10 = new StringBuilder();
stringBuilder10.AppendLine("Type: Directory");
StringBuilder stringBuilder2 = stringBuilder10;
StringBuilder stringBuilder11 = stringBuilder2;
StringBuilder.AppendInterpolatedStringHandler handler = new StringBuilder.AppendInterpolatedStringHandler(6, 1, stringBuilder2);
handler.AppendLiteral("Path: ");
handler.AppendFormatted(directoryInfo.FullName);
stringBuilder11.AppendLine(ref handler);
stringBuilder2 = stringBuilder10;
StringBuilder stringBuilder12 = stringBuilder2;
handler = new StringBuilder.AppendInterpolatedStringHandler(7, 1, stringBuilder2);
handler.AppendLiteral("Files: ");
handler.AppendFormatted(files.Length);
stringBuilder12.AppendLine(ref handler);
stringBuilder2 = stringBuilder10;
StringBuilder stringBuilder13 = stringBuilder2;
handler = new StringBuilder.AppendInterpolatedStringHandler(16, 1, stringBuilder2);
handler.AppendLiteral("Subdirectories: ");
handler.AppendFormatted(directories.Length);
stringBuilder13.AppendLine(ref handler);
stringBuilder2 = stringBuilder10;
StringBuilder stringBuilder14 = stringBuilder2;
handler = new StringBuilder.AppendInterpolatedStringHandler(30, 1, stringBuilder2);
handler.AppendLiteral("Total Size (top-level files): ");
handler.AppendFormatted(FormatSize(bytes));
stringBuilder14.AppendLine(ref handler);
stringBuilder2 = stringBuilder10;
StringBuilder stringBuilder15 = stringBuilder2;
handler = new StringBuilder.AppendInterpolatedStringHandler(9, 1, stringBuilder2);
handler.AppendLiteral("Created: ");
handler.AppendFormatted(directoryInfo.CreationTime, "yyyy-MM-dd HH:mm:ss");
stringBuilder15.AppendLine(ref handler);
stringBuilder2 = stringBuilder10;
StringBuilder stringBuilder16 = stringBuilder2;
handler = new StringBuilder.AppendInterpolatedStringHandler(10, 1, stringBuilder2);
handler.AppendLiteral("Modified: ");
handler.AppendFormatted(directoryInfo.LastWriteTime, "yyyy-MM-dd HH:mm:ss");
stringBuilder16.AppendLine(ref handler);
return Task.FromResult(ToolResult.Ok(stringBuilder10.ToString()));
}
return Task.FromResult(ToolResult.Fail("경로를 찾을 수 없습니다: " + text2));
}
catch (Exception ex)
{
return Task.FromResult(ToolResult.Fail("정보 조회 오류: " + ex.Message));
}
}
private static string FormatSize(long bytes)
{
if (1 == 0)
{
}
string result = ((bytes < 1048576) ? ((bytes >= 1024) ? $"{(double)bytes / 1024.0:F1}KB" : $"{bytes}B") : ((bytes >= 1073741824) ? $"{(double)bytes / 1073741824.0:F2}GB" : $"{(double)bytes / 1048576.0:F1}MB"));
if (1 == 0)
{
}
return result;
}
}