100 lines
4.2 KiB
C#
100 lines
4.2 KiB
C#
using System.IO;
|
|
using System.Text;
|
|
using System.Text.Json;
|
|
|
|
namespace AxCopilot.Services.Agent;
|
|
|
|
/// <summary>
|
|
/// 배치파일(.bat) / PowerShell 스크립트(.ps1)를 생성하는 내장 스킬.
|
|
/// 파일 생성만 수행하며 자동 실행하지 않습니다.
|
|
/// 시스템 수준 명령(레지스트리, 서비스, 드라이버 등)은 차단합니다.
|
|
/// </summary>
|
|
public class BatchSkill : IAgentTool
|
|
{
|
|
public string Name => "script_create";
|
|
public string Description => "Create a batch (.bat) or PowerShell (.ps1) script file. The script is ONLY created, NOT executed. System-level commands are blocked.";
|
|
|
|
public ToolParameterSchema Parameters => new()
|
|
{
|
|
Properties = new()
|
|
{
|
|
["path"] = new() { Type = "string", Description = "Output file path (.bat or .ps1). Relative to work folder." },
|
|
["content"] = new() { Type = "string", Description = "Script content. Each line should have Korean comments explaining the command." },
|
|
["description"] = new() { Type = "string", Description = "Brief description of what this script does." },
|
|
},
|
|
Required = ["path", "content"]
|
|
};
|
|
|
|
// 시스템 수준 명령 차단 목록
|
|
private static readonly string[] BlockedCommands =
|
|
[
|
|
"reg ", "reg.exe", "regedit",
|
|
"sc ", "sc.exe",
|
|
"net stop", "net start", "net user",
|
|
"bcdedit", "diskpart", "format ",
|
|
"shutdown", "schtasks",
|
|
"wmic", "powercfg",
|
|
"Set-Service", "Stop-Service", "Start-Service",
|
|
"New-Service", "Remove-Service",
|
|
"Set-ItemProperty.*HKLM", "Set-ItemProperty.*HKCU",
|
|
"Remove-Item.*-Recurse.*-Force",
|
|
];
|
|
|
|
public async Task<ToolResult> ExecuteAsync(JsonElement args, AgentContext context, CancellationToken ct)
|
|
{
|
|
var path = args.GetProperty("path").GetString() ?? "";
|
|
var content = args.GetProperty("content").GetString() ?? "";
|
|
var desc = args.TryGetProperty("description", out var d) ? d.GetString() ?? "" : "";
|
|
|
|
var fullPath = FileReadTool.ResolvePath(path, context.WorkFolder);
|
|
if (context.ActiveTab == "Cowork") fullPath = AgentContext.EnsureTimestampedPath(fullPath);
|
|
|
|
// 확장자 검증
|
|
var ext = Path.GetExtension(fullPath).ToLowerInvariant();
|
|
if (ext != ".bat" && ext != ".ps1" && ext != ".cmd")
|
|
return ToolResult.Fail("지원하는 스크립트 형식: .bat, .cmd, .ps1");
|
|
|
|
if (!context.IsPathAllowed(fullPath))
|
|
return ToolResult.Fail($"경로 접근 차단: {fullPath}");
|
|
|
|
// 시스템 명령 차단 검사
|
|
var contentLower = content.ToLowerInvariant();
|
|
foreach (var blocked in BlockedCommands)
|
|
{
|
|
if (contentLower.Contains(blocked.ToLowerInvariant()))
|
|
return ToolResult.Fail($"시스템 수준 명령이 포함되어 차단됨: {blocked.Trim()}");
|
|
}
|
|
|
|
if (!await context.CheckWritePermissionAsync(Name, fullPath))
|
|
return ToolResult.Fail($"쓰기 권한 거부: {fullPath}");
|
|
|
|
try
|
|
{
|
|
var dir = Path.GetDirectoryName(fullPath);
|
|
if (!string.IsNullOrEmpty(dir)) Directory.CreateDirectory(dir);
|
|
|
|
// 파일 상단에 설명 주석 추가
|
|
var sb = new StringBuilder();
|
|
if (!string.IsNullOrEmpty(desc))
|
|
{
|
|
var commentPrefix = ext == ".ps1" ? "#" : "REM";
|
|
sb.AppendLine($"{commentPrefix} === {desc} ===");
|
|
sb.AppendLine($"{commentPrefix} 이 스크립트는 AX Copilot에 의해 생성되었습니다.");
|
|
sb.AppendLine($"{commentPrefix} 실행 전 내용을 반드시 확인하세요.");
|
|
sb.AppendLine();
|
|
}
|
|
sb.Append(content);
|
|
|
|
await File.WriteAllTextAsync(fullPath, sb.ToString(), new UTF8Encoding(false), ct);
|
|
|
|
return ToolResult.Ok(
|
|
$"스크립트 파일 생성 완료: {fullPath}\n형식: {ext}, 설명: {(string.IsNullOrEmpty(desc) ? "(없음)" : desc)}\n⚠ 자동 실행되지 않습니다. 내용을 확인한 후 직접 실행하세요.",
|
|
fullPath);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
return ToolResult.Fail($"스크립트 생성 실패: {ex.Message}");
|
|
}
|
|
}
|
|
}
|