using System;
using System.IO;
using System.Linq;
using System.Collections.Generic;
using System.Text.RegularExpressions;
using Newtonsoft.Json.Linq;
using UnityEditor;
using UnityEngine;
using MCPForUnity.Editor.Constants;
using MCPForUnity.Editor.Helpers;
using System.Threading;
using System.Security.Cryptography;
#if USE_ROSLYN
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp;
using Microsoft.CodeAnalysis.Formatting;
#endif
#if UNITY_EDITOR
using UnityEditor.Compilation;
#endif
namespace MCPForUnity.Editor.Tools
{
/// <summary>
/// Handles CRUD operations for C# scripts within the Unity project.
/// </summary>
/// <remarks>
/// ROSLYN INSTALLATION GUIDE:
/// To enable advanced syntax validation with Roslyn compiler services:
///
/// 1. Install Microsoft.CodeAnalysis.CSharp NuGet package:
/// - Open Package Manager in Unity
/// - Follow the instruction on https://github.com/GlitchEnzo/NuGetForUnity
///
/// 2. Open NuGet Package Manager and Install Microsoft.CodeAnalysis.CSharp:
///
/// 3. Alternative: Manual DLL installation:
/// - Download Microsoft.CodeAnalysis.CSharp.dll and dependencies
/// - Place in Assets/Plugins/ folder
/// - Ensure .NET compatibility settings are correct
///
/// 4. Define USE_ROSLYN symbol:
/// - Go to Player Settings > Scripting Define Symbols
/// - Add "USE_ROSLYN" to enable Roslyn-based validation
///
/// 5. Restart Unity after installation
///
/// Note: Without Roslyn, the system falls back to basic structural validation.
/// Roslyn provides full C# compiler diagnostics with line numbers and detailed error messages.
/// </remarks>
[McpForUnityTool("manage_script", AutoRegister = false)]
public static class ManageScript
{
/// <summary>
/// Resolves a directory under Assets/, preventing traversal and escaping.
/// Returns fullPathDir on disk and canonical 'Assets/...' relative path.
/// </summary>
private static bool TryResolveUnderAssets(string relDir, out string fullPathDir, out string relPathSafe)
{
string assets = AssetPathUtility.NormalizeSeparators(Application.dataPath);
// Normalize caller path: allow both "Scripts/..." and "Assets/Scripts/..."
string rel = AssetPathUtility.NormalizeSeparators(relDir ?? "Scripts").Trim();
if (string.IsNullOrEmpty(rel)) rel = "Scripts";
// Handle both "Assets" and "Assets/" prefixes
if (rel.Equals("Assets", StringComparison.OrdinalIgnoreCase))
{
rel = string.Empty;
}
else if (rel.StartsWith("Assets/", StringComparison.OrdinalIgnoreCase))
{
rel = rel.Substring(7);
}
rel = rel.TrimStart('/');
string targetDir = AssetPathUtility.NormalizeSeparators(Path.Combine(assets, rel));
string full = AssetPathUtility.NormalizeSeparators(Path.GetFullPath(targetDir));
bool underAssets = full.StartsWith(assets + "/", StringComparison.OrdinalIgnoreCase)
|| string.Equals(full, assets, StringComparison.OrdinalIgnoreCase);
if (!underAssets)
{
fullPathDir = null;
relPathSafe = null;
return false;
}
// Best-effort symlink guard: if the directory OR ANY ANCESTOR (up to Assets/) is a reparse point/symlink, reject
try
{
var di = new DirectoryInfo(full);
while (di != null)
{
if (di.Exists && (di.Attributes & FileAttributes.ReparsePoint) != 0)
{
fullPathDir = null;
relPathSafe = null;
return false;
}
var atAssets = string.Equals(
di.FullName.Replace('\\', '/'),
assets,
StringComparison.OrdinalIgnoreCase
);
if (atAssets) break;
di = di.Parent;
}
}
catch { /* best effort; proceed */ }
fullPathDir = full;
string tail = full.Length > assets.Length ? full.Substring(assets.Length).TrimStart('/') : string.Empty;
relPathSafe = ("Assets/" + tail).TrimEnd('/');
return true;
}
/// <summary>
/// Main handler for script management actions.
/// </summary>
public static object HandleCommand(JObject @params)
{
// Handle null parameters
if (@params == null)
{
return new ErrorResponse("invalid_params", "Parameters cannot be null.");
}
var p = new ToolParams(@params);
// Extract and validate required parameters
var actionResult = p.GetRequired("action");
if (!actionResult.IsSuccess)
{
return new ErrorResponse(actionResult.ErrorMessage);
}
string action = actionResult.Value.ToLowerInvariant();
var nameResult = p.GetRequired("name");
if (!nameResult.IsSuccess)
{
return new ErrorResponse(nameResult.ErrorMessage);
}
string name = nameResult.Value;
// Optional parameters
string path = p.Get("path"); // Relative to Assets/
string contents = null;
// Check if we have base64 encoded contents
bool contentsEncoded = p.GetBool("contentsEncoded", false);
if (contentsEncoded && p.Has("encodedContents"))
{
try
{
contents = DecodeBase64(p.Get("encodedContents"));
}
catch (Exception e)
{
return new ErrorResponse($"Failed to decode script contents: {e.Message}");
}
}
else
{
contents = p.Get("contents");
}
string scriptType = p.Get("scriptType"); // For templates/validation
string namespaceName = p.Get("namespace"); // For organizing code
// Basic name validation (alphanumeric, underscores, cannot start with number)
if (!Regex.IsMatch(name, @"^[a-zA-Z_][a-zA-Z0-9_]*$", RegexOptions.CultureInvariant, TimeSpan.FromSeconds(2)))
{
return new ErrorResponse(
$"Invalid script name: '{name}'. Use only letters, numbers, underscores, and don't start with a number."
);
}
// Resolve and harden target directory under Assets/
if (!TryResolveUnderAssets(path, out string fullPathDir, out string relPathSafeDir))
{
return new ErrorResponse($"Invalid path. Target directory must be within 'Assets/'. Provided: '{(path ?? "(null)")}'");
}
// Construct file paths
string scriptFileName = $"{name}.cs";
string fullPath = Path.Combine(fullPathDir, scriptFileName);
string relativePath = AssetPathUtility.NormalizeSeparators(Path.Combine(relPathSafeDir, scriptFileName));
// Ensure the target directory exists for create/update
if (action == "create" || action == "update")
{
try
{
Directory.CreateDirectory(fullPathDir);
}
catch (Exception e)
{
return new ErrorResponse(
$"Could not create directory '{fullPathDir}': {e.Message}"
);
}
}
// Route to specific action handlers
switch (action)
{
case "create":
return CreateScript(
fullPath,
relativePath,
name,
contents,
scriptType,
namespaceName
);
case "read":
McpLog.Warn("manage_script.read is deprecated; prefer resources/read. Serving read for backward compatibility.");
return ReadScript(fullPath, relativePath);
case "update":
McpLog.Warn("manage_script.update is deprecated; prefer apply_text_edits. Serving update for backward compatibility.");
return UpdateScript(fullPath, relativePath, name, contents);
case "delete":
return DeleteScript(fullPath, relativePath);
case "apply_text_edits":
{
var textEdits = p.GetRaw("edits") as JArray;
string precondition = p.Get("precondition_sha256");
// Respect optional options (guard type before indexing)
var optionsObj = p.GetRaw("options") as JObject;
string refreshOpt = optionsObj?["refresh"]?.ToString()?.ToLowerInvariant();
string validateOpt = optionsObj?["validate"]?.ToString()?.ToLowerInvariant();
return ApplyTextEdits(fullPath, relativePath, name, textEdits, precondition, refreshOpt, validateOpt);
}
case "validate":
{
string level = p.Get("level", "standard").ToLowerInvariant();
var chosen = level switch
{
"basic" => ValidationLevel.Basic,
"standard" => ValidationLevel.Standard,
"strict" => ValidationLevel.Strict,
"comprehensive" => ValidationLevel.Comprehensive,
_ => ValidationLevel.Standard
};
string fileText;
try { fileText = File.ReadAllText(fullPath); }
catch (Exception ex) { return new ErrorResponse($"Failed to read script: {ex.Message}"); }
bool ok = ValidateScriptSyntax(fileText, chosen, out string[] diagsRaw);
var diags = (diagsRaw ?? Array.Empty<string>()).Select(s =>
{
var m = Regex.Match(
s,
@"^(ERROR|WARNING|INFO): (.*?)(?: \(Line (\d+)\))?$",
RegexOptions.CultureInvariant | RegexOptions.Multiline,
TimeSpan.FromMilliseconds(250)
);
string severity = m.Success ? m.Groups[1].Value.ToLowerInvariant() : "info";
string message = m.Success ? m.Groups[2].Value : s;
int lineNum = m.Success && int.TryParse(m.Groups[3].Value, out var l) ? l : 0;
return new { line = lineNum, col = 0, severity, message };
}).ToArray();
var result = new { diagnostics = diags };
return ok ? new SuccessResponse("Validation completed.", result)
: new ErrorResponse("Validation failed.", result);
}
case "edit":
McpLog.Warn("manage_script.edit is deprecated; prefer apply_text_edits. Serving structured edit for backward compatibility.");
var structEdits = @params["edits"] as JArray;
var options = @params["options"] as JObject;
return EditScript(fullPath, relativePath, name, structEdits, options);
case "get_sha":
{
try
{
if (!File.Exists(fullPath))
return new ErrorResponse($"Script not found at '{relativePath}'.");
string text = File.ReadAllText(fullPath);
string sha = ComputeSha256(text);
var fi = new FileInfo(fullPath);
long lengthBytes;
try { lengthBytes = new System.Text.UTF8Encoding(encoderShouldEmitUTF8Identifier: false).GetByteCount(text); }
catch { lengthBytes = fi.Exists ? fi.Length : 0; }
var data = new
{
uri = $"mcpforunity://path/{relativePath}",
path = relativePath,
sha256 = sha,
lengthBytes,
lastModifiedUtc = fi.Exists ? fi.LastWriteTimeUtc.ToString("o") : string.Empty
};
return new SuccessResponse($"SHA computed for '{relativePath}'.", data);
}
catch (Exception ex)
{
return new ErrorResponse($"Failed to compute SHA: {ex.Message}");
}
}
default:
return new ErrorResponse(
$"Unknown action: '{action}'. Valid actions are: create, delete, apply_text_edits, validate, read (deprecated), update (deprecated), edit (deprecated)."
);
}
}
/// <summary>
/// Decode base64 string to normal text
/// </summary>
private static string DecodeBase64(string encoded)
{
byte[] data = Convert.FromBase64String(encoded);
return System.Text.Encoding.UTF8.GetString(data);
}
/// <summary>
/// Encode text to base64 string
/// </summary>
private static string EncodeBase64(string text)
{
byte[] data = System.Text.Encoding.UTF8.GetBytes(text);
return Convert.ToBase64String(data);
}
private static object CreateScript(
string fullPath,
string relativePath,
string name,
string contents,
string scriptType,
string namespaceName
)
{
// Check if script already exists
if (File.Exists(fullPath))
{
return new ErrorResponse(
$"Script already exists at '{relativePath}'. Use 'update' action to modify."
);
}
// Generate default content if none provided
if (string.IsNullOrEmpty(contents))
{
contents = GenerateDefaultScriptContent(name, scriptType, namespaceName);
}
// Validate syntax with detailed error reporting using GUI setting
ValidationLevel validationLevel = GetValidationLevelFromGUI();
bool isValid = ValidateScriptSyntax(contents, validationLevel, out string[] validationErrors);
if (!isValid)
{
return new ErrorResponse("validation_failed", new { status = "validation_failed", diagnostics = validationErrors ?? Array.Empty<string>() });
}
else if (validationErrors != null && validationErrors.Length > 0)
{
// Log warnings but don't block creation
McpLog.Warn($"Script validation warnings for {name}:\n" + string.Join("\n", validationErrors));
}
try
{
// Atomic create without BOM; schedule refresh after reply
var enc = new System.Text.UTF8Encoding(encoderShouldEmitUTF8Identifier: false);
var tmp = fullPath + ".tmp";
File.WriteAllText(tmp, contents, enc);
try
{
File.Move(tmp, fullPath);
}
catch (IOException)
{
File.Copy(tmp, fullPath, overwrite: true);
try { File.Delete(tmp); } catch { }
}
var uri = $"mcpforunity://path/{relativePath}";
var ok = new SuccessResponse(
$"Script '{name}.cs' created successfully at '{relativePath}'.",
new { uri, scheduledRefresh = false }
);
ManageScriptRefreshHelpers.ImportAndRequestCompile(relativePath);
return ok;
}
catch (Exception e)
{
return new ErrorResponse($"Failed to create script '{relativePath}': {e.Message}");
}
}
private static object ReadScript(string fullPath, string relativePath)
{
if (!File.Exists(fullPath))
{
return new ErrorResponse($"Script not found at '{relativePath}'.");
}
try
{
string contents = File.ReadAllText(fullPath);
// Return both normal and encoded contents for larger files
bool isLarge = contents.Length > 10000; // If content is large, include encoded version
var uri = $"mcpforunity://path/{relativePath}";
var responseData = new
{
uri,
path = relativePath,
contents = contents,
// For large files, also include base64-encoded version
encodedContents = isLarge ? EncodeBase64(contents) : null,
contentsEncoded = isLarge,
};
return new SuccessResponse(
$"Script '{Path.GetFileName(relativePath)}' read successfully.",
responseData
);
}
catch (Exception e)
{
return new ErrorResponse($"Failed to read script '{relativePath}': {e.Message}");
}
}
private static object UpdateScript(
string fullPath,
string relativePath,
string name,
string contents
)
{
if (!File.Exists(fullPath))
{
return new ErrorResponse(
$"Script not found at '{relativePath}'. Use 'create' action to add a new script."
);
}
if (string.IsNullOrEmpty(contents))
{
return new ErrorResponse("Content is required for the 'update' action.");
}
// Validate syntax with detailed error reporting using GUI setting
ValidationLevel validationLevel = GetValidationLevelFromGUI();
bool isValid = ValidateScriptSyntax(contents, validationLevel, out string[] validationErrors);
if (!isValid)
{
return new ErrorResponse("validation_failed", new { status = "validation_failed", diagnostics = validationErrors ?? Array.Empty<string>() });
}
else if (validationErrors != null && validationErrors.Length > 0)
{
// Log warnings but don't block update
McpLog.Warn($"Script validation warnings for {name}:\n" + string.Join("\n", validationErrors));
}
try
{
// Safe write with atomic replace when available, without BOM
var encoding = new System.Text.UTF8Encoding(encoderShouldEmitUTF8Identifier: false);
string tempPath = fullPath + ".tmp";
File.WriteAllText(tempPath, contents, encoding);
string backupPath = fullPath + ".bak";
try
{
File.Replace(tempPath, fullPath, backupPath);
try { if (File.Exists(backupPath)) File.Delete(backupPath); } catch { }
}
catch (PlatformNotSupportedException)
{
File.Copy(tempPath, fullPath, true);
try { File.Delete(tempPath); } catch { }
try { if (File.Exists(backupPath)) File.Delete(backupPath); } catch { }
}
catch (IOException)
{
File.Copy(tempPath, fullPath, true);
try { File.Delete(tempPath); } catch { }
try { if (File.Exists(backupPath)) File.Delete(backupPath); } catch { }
}
// Prepare success response BEFORE any operation that can trigger a domain reload
var uri = $"mcpforunity://path/{relativePath}";
var ok = new SuccessResponse(
$"Script '{name}.cs' updated successfully at '{relativePath}'.",
new { uri, path = relativePath, scheduledRefresh = true }
);
// Schedule a debounced import/compile on next editor tick to avoid stalling the reply
ManageScriptRefreshHelpers.ScheduleScriptRefresh(relativePath);
return ok;
}
catch (Exception e)
{
return new ErrorResponse($"Failed to update script '{relativePath}': {e.Message}");
}
}
/// <summary>
/// Apply simple text edits specified by line/column ranges. Applies transactionally and validates result.
/// </summary>
private const int MaxEditPayloadBytes = 64 * 1024;
private static object ApplyTextEdits(
string fullPath,
string relativePath,
string name,
JArray edits,
string preconditionSha256,
string refreshModeFromCaller = null,
string validateMode = null)
{
if (!File.Exists(fullPath))
return new ErrorResponse($"Script not found at '{relativePath}'.");
// Refuse edits if the target or any ancestor is a symlink
try
{
var di = new DirectoryInfo(Path.GetDirectoryName(fullPath) ?? "");
while (di != null && !string.Equals(di.FullName.Replace('\\', '/'), Application.dataPath.Replace('\\', '/'), StringComparison.OrdinalIgnoreCase))
{
if (di.Exists && (di.Attributes & FileAttributes.ReparsePoint) != 0)
return new ErrorResponse("Refusing to edit a symlinked script path.");
di = di.Parent;
}
}
catch
{
// If checking attributes fails, proceed without the symlink guard
}
if (edits == null || edits.Count == 0)
return new ErrorResponse("No edits provided.");
string original;
try { original = File.ReadAllText(fullPath); }
catch (Exception ex) { return new ErrorResponse($"Failed to read script: {ex.Message}"); }
// Require precondition to avoid drift on large files
string currentSha = ComputeSha256(original);
if (string.IsNullOrEmpty(preconditionSha256))
return new ErrorResponse("precondition_required", new { status = "precondition_required", current_sha256 = currentSha });
if (!preconditionSha256.Equals(currentSha, StringComparison.OrdinalIgnoreCase))
return new ErrorResponse("stale_file", new { status = "stale_file", expected_sha256 = preconditionSha256, current_sha256 = currentSha });
// Convert edits to absolute index ranges
var spans = new List<(int start, int end, string text)>();
long totalBytes = 0;
foreach (var e in edits)
{
try
{
int sl = Math.Max(1, e.Value<int>("startLine"));
int sc = Math.Max(1, e.Value<int>("startCol"));
int el = Math.Max(1, e.Value<int>("endLine"));
int ec = Math.Max(1, e.Value<int>("endCol"));
string newText = e.Value<string>("newText") ?? string.Empty;
if (!TryIndexFromLineCol(original, sl, sc, out int sidx))
return new ErrorResponse($"apply_text_edits: start out of range (line {sl}, col {sc})");
if (!TryIndexFromLineCol(original, el, ec, out int eidx))
return new ErrorResponse($"apply_text_edits: end out of range (line {el}, col {ec})");
if (eidx < sidx) (sidx, eidx) = (eidx, sidx);
spans.Add((sidx, eidx, newText));
checked
{
totalBytes += System.Text.Encoding.UTF8.GetByteCount(newText);
}
}
catch (Exception ex)
{
return new ErrorResponse($"Invalid edit payload: {ex.Message}");
}
}
// Header guard: refuse edits that touch before the first 'using ' directive (after optional BOM) to prevent file corruption
int headerBoundary = (original.Length > 0 && original[0] == '\uFEFF') ? 1 : 0; // skip BOM once if present
// Find first top-level using (supports alias, static, and dotted namespaces)
var mUsing = System.Text.RegularExpressions.Regex.Match(
original,
@"(?m)^\s*using\s+(?:static\s+)?(?:[A-Za-z_]\w*\s*=\s*)?[A-Za-z_]\w*(?:\.[A-Za-z_]\w*)*\s*;",
System.Text.RegularExpressions.RegexOptions.CultureInvariant,
TimeSpan.FromSeconds(2)
);
if (mUsing.Success)
{
headerBoundary = Math.Min(Math.Max(headerBoundary, mUsing.Index), original.Length);
}
foreach (var sp in spans)
{
if (sp.start < headerBoundary)
{
return new ErrorResponse("using_guard", new { status = "using_guard", hint = "Refusing to edit before the first 'using'. Use anchor_insert near a method or a structured edit." });
}
}
// Attempt auto-upgrade: if a single edit targets a method header/body, re-route as structured replace_method
if (spans.Count == 1)
{
var sp = spans[0];
// Heuristic: around the start of the edit, try to match a method header in original
int searchStart = Math.Max(0, sp.start - 200);
int searchEnd = Math.Min(original.Length, sp.start + 200);
string slice = original.Substring(searchStart, searchEnd - searchStart);
var rx = new System.Text.RegularExpressions.Regex(@"(?m)^[\t ]*(?:\[[^\]]+\][\t ]*)*[\t ]*(?:public|private|protected|internal|static|virtual|override|sealed|async|extern|unsafe|new|partial)[\s\S]*?\b([A-Za-z_][A-Za-z0-9_]*)\s*\(");
var mh = rx.Match(slice);
if (mh.Success)
{
string methodName = mh.Groups[1].Value;
// Find class span containing the edit
if (TryComputeClassSpan(original, name, null, out var clsStart, out var clsLen, out _))
{
if (TryComputeMethodSpan(original, clsStart, clsLen, methodName, null, null, null, out var mStart, out var mLen, out _))
{
// If the edit overlaps the method span significantly, treat as replace_method
if (sp.start <= mStart + 2 && sp.end >= mStart + 1)
{
var structEdits = new JArray();
// Apply the edit to get a candidate string, then recompute method span on the edited text
string candidate = original.Remove(sp.start, sp.end - sp.start).Insert(sp.start, sp.text ?? string.Empty);
string replacementText;
if (TryComputeClassSpan(candidate, name, null, out var cls2Start, out var cls2Len, out _)
&& TryComputeMethodSpan(candidate, cls2Start, cls2Len, methodName, null, null, null, out var m2Start, out var m2Len, out _))
{
replacementText = candidate.Substring(m2Start, m2Len);
}
else
{
// Fallback: adjust method start by the net delta if the edit was before the method
int delta = (sp.text?.Length ?? 0) - (sp.end - sp.start);
int adjustedStart = mStart + (sp.start <= mStart ? delta : 0);
adjustedStart = Math.Max(0, Math.Min(adjustedStart, candidate.Length));
// If the edit was within the original method span, adjust the length by the delta within-method
int withinMethodDelta = 0;
if (sp.start >= mStart && sp.start <= mStart + mLen)
{
withinMethodDelta = delta;
}
int adjustedLen = mLen + withinMethodDelta;
adjustedLen = Math.Max(0, Math.Min(candidate.Length - adjustedStart, adjustedLen));
replacementText = candidate.Substring(adjustedStart, adjustedLen);
}
var op = new JObject
{
["mode"] = "replace_method",
["className"] = name,
["methodName"] = methodName,
["replacement"] = replacementText
};
structEdits.Add(op);
// Reuse structured path
return EditScript(fullPath, relativePath, name, structEdits, new JObject { ["refresh"] = "immediate", ["validate"] = "standard" });
}
}
}
}
}
if (totalBytes > MaxEditPayloadBytes)
{
return new ErrorResponse("too_large", new { status = "too_large", limitBytes = MaxEditPayloadBytes, hint = "split into smaller edits" });
}
// Ensure non-overlap and apply from back to front
spans = spans.OrderByDescending(t => t.start).ToList();
for (int i = 1; i < spans.Count; i++)
{
if (spans[i].end > spans[i - 1].start)
{
var conflict = new[] { new { startA = spans[i].start, endA = spans[i].end, startB = spans[i - 1].start, endB = spans[i - 1].end } };
return new ErrorResponse("overlap", new { status = "overlap", conflicts = conflict, hint = "Sort ranges descending by start and compute from the same snapshot." });
}
}
string working = original;
bool syntaxOnly = string.Equals(validateMode, "syntax", StringComparison.OrdinalIgnoreCase);
foreach (var sp in spans)
{
working = working.Remove(sp.start, sp.end - sp.start).Insert(sp.start, sp.text ?? string.Empty);
}
// No-op guard: if resulting text is identical, avoid writes and return explicit no-op
if (string.Equals(working, original, StringComparison.Ordinal))
{
string noChangeSha = ComputeSha256(original);
return new SuccessResponse(
$"No-op: contents unchanged for '{relativePath}'.",
new
{
uri = $"mcpforunity://path/{relativePath}",
path = relativePath,
editsApplied = 0,
no_op = true,
sha256 = noChangeSha,
evidence = new { reason = "identical_content" }
}
);
}
// Always check final structural balance
if (!CheckBalancedDelimiters(working, out int line, out char expected))
{
int startLine = Math.Max(1, line - 5);
int endLine = line + 5;
string hint = $"unbalanced_braces at line {line}. Call resources/read for lines {startLine}-{endLine} and resend a smaller apply_text_edits that restores balance.";
return new ErrorResponse(hint, new { status = "unbalanced_braces", line, expected = expected.ToString(), evidenceWindow = new { startLine, endLine } });
}
#if USE_ROSLYN
if (!syntaxOnly)
{
var tree = CSharpSyntaxTree.ParseText(working);
var diagnostics = tree.GetDiagnostics().Where(d => d.Severity == DiagnosticSeverity.Error).Take(3)
.Select(d => new {
line = d.Location.GetLineSpan().StartLinePosition.Line + 1,
col = d.Location.GetLineSpan().StartLinePosition.Character + 1,
code = d.Id,
message = d.GetMessage()
}).ToArray();
if (diagnostics.Length > 0)
{
int firstLine = diagnostics[0].line;
int startLineRos = Math.Max(1, firstLine - 5);
int endLineRos = firstLine + 5;
return new ErrorResponse("syntax_error", new { status = "syntax_error", diagnostics, evidenceWindow = new { startLine = startLineRos, endLine = endLineRos } });
}
// Optional formatting
try
{
var root = tree.GetRoot();
var workspace = new AdhocWorkspace();
root = Microsoft.CodeAnalysis.Formatting.Formatter.Format(root, workspace);
working = root.ToFullString();
}
catch { }
}
#endif
string newSha = ComputeSha256(working);
// Atomic write and schedule refresh
try
{
var enc = new System.Text.UTF8Encoding(encoderShouldEmitUTF8Identifier: false);
var tmp = fullPath + ".tmp";
File.WriteAllText(tmp, working, enc);
string backup = fullPath + ".bak";
try
{
File.Replace(tmp, fullPath, backup);
try { if (File.Exists(backup)) File.Delete(backup); } catch { /* ignore */ }
}
catch (PlatformNotSupportedException)
{
File.Copy(tmp, fullPath, true);
try { File.Delete(tmp); } catch { }
try { if (File.Exists(backup)) File.Delete(backup); } catch { }
}
catch (IOException)
{
File.Copy(tmp, fullPath, true);
try { File.Delete(tmp); } catch { }
try { if (File.Exists(backup)) File.Delete(backup); } catch { }
}
// Respect refresh mode: immediate vs debounced
bool immediate = string.Equals(refreshModeFromCaller, "immediate", StringComparison.OrdinalIgnoreCase) ||
string.Equals(refreshModeFromCaller, "sync", StringComparison.OrdinalIgnoreCase);
if (immediate)
{
McpLog.Info($"[ManageScript] ApplyTextEdits: immediate refresh for '{relativePath}'");
AssetDatabase.ImportAsset(
relativePath,
ImportAssetOptions.ForceSynchronousImport | ImportAssetOptions.ForceUpdate
);
#if UNITY_EDITOR
UnityEditor.Compilation.CompilationPipeline.RequestScriptCompilation();
#endif
}
else
{
McpLog.Info($"[ManageScript] ApplyTextEdits: debounced refresh scheduled for '{relativePath}'");
ManageScriptRefreshHelpers.ScheduleScriptRefresh(relativePath);
}
return new SuccessResponse(
$"Applied {spans.Count} text edit(s) to '{relativePath}'.",
new
{
uri = $"mcpforunity://path/{relativePath}",
path = relativePath,
editsApplied = spans.Count,
sha256 = newSha,
scheduledRefresh = !immediate
}
);
}
catch (Exception ex)
{
return new ErrorResponse($"Failed to write edits: {ex.Message}");
}
}
private static bool TryIndexFromLineCol(string text, int line1, int col1, out int index)
{
// 1-based line/col to absolute index (0-based), col positions are counted in code points
int line = 1, col = 1;
for (int i = 0; i <= text.Length; i++)
{
if (line == line1 && col == col1)
{
index = i;
return true;
}
if (i == text.Length) break;
char c = text[i];
if (c == '\r')
{
// Treat CRLF as a single newline; skip the LF if present
if (i + 1 < text.Length && text[i + 1] == '\n')
i++;
line++;
col = 1;
}
else if (c == '\n')
{
line++;
col = 1;
}
else
{
col++;
}
}
index = -1;
return false;
}
private static string ComputeSha256(string contents)
{
using (var sha = SHA256.Create())
{
var bytes = System.Text.Encoding.UTF8.GetBytes(contents);
var hash = sha.ComputeHash(bytes);
return BitConverter.ToString(hash).Replace("-", string.Empty).ToLowerInvariant();
}
}
/// <summary>
/// Lightweight lexer that tracks whether the current position is inside a
/// string literal (regular, verbatim, interpolated, raw) or a comment.
/// Callers advance character-by-character and check <see cref="InNonCode"/>.
/// </summary>
private struct CSharpLexer
{
private readonly string _text;
private int _pos;
private readonly int _end;
private int _line;
// String/comment state
private bool _inSingleComment;
private bool _inMultiComment;
public CSharpLexer(string text, int start = 0, int end = -1)
{
_text = text;
_pos = start;
_end = end < 0 ? text.Length : end;
_line = 1;
// count newlines before start
for (int i = 0; i < start && i < text.Length; i++)
if (text[i] == '\n') _line++;
_inSingleComment = false;
_inMultiComment = false;
InNonCode = false;
}
public bool InNonCode { get; private set; }
public int Position => _pos;
public int Line => _line;
/// <summary>
/// Advance to the next character, updating all state.
/// Returns false at end of range.
/// </summary>
public bool Advance(out char c)
{
if (_pos >= _end) { c = '\0'; return false; }
c = _text[_pos];
char next = _pos + 1 < _end ? _text[_pos + 1] : '\0';
if (c == '\n')
{
_line++;
if (_inSingleComment) _inSingleComment = false;
}
// Inside single-line comment
if (_inSingleComment) { InNonCode = true; _pos++; return true; }
// Inside multi-line comment
if (_inMultiComment)
{
if (c == '*' && next == '/') { _inMultiComment = false; InNonCode = true; _pos += 2; c = '/'; return true; }
InNonCode = true; _pos++; return true;
}
// Start of comment
if (c == '/' && next == '/') { _inSingleComment = true; InNonCode = true; _pos += 2; return true; }
if (c == '/' && next == '*') { _inMultiComment = true; InNonCode = true; _pos += 2; return true; }
// Interpolated raw string: $"""...""" or $$"""...""" etc. (C# 11)
// Must check BEFORE regular $" and BEFORE plain """
if (c == '$')
{
int dollarCount = 1;
while (_pos + dollarCount < _end && _text[_pos + dollarCount] == '$') dollarCount++;
int afterDollars = _pos + dollarCount;
if (afterDollars + 2 < _end && _text[afterDollars] == '"' && _text[afterDollars + 1] == '"' && _text[afterDollars + 2] == '"')
{
int q = 3;
while (afterDollars + q < _end && _text[afterDollars + q] == '"') q++;
_pos = afterDollars + q; // past all opening quotes
SkipInterpolatedRawStringBody(dollarCount, q);
InNonCode = true; return true;
}
}
// Raw string literal: """...""" (C# 11, non-interpolated)
if (c == '"' && next == '"' && _pos + 2 < _end && _text[_pos + 2] == '"')
{
int q = 3;
while (_pos + q < _end && _text[_pos + q] == '"') q++;
_pos += q; // past opening quotes
int closeCount = 0;
while (_pos < _end)
{
if (_text[_pos] == '\n') _line++;
if (_text[_pos] == '"') { closeCount++; if (closeCount >= q) { _pos++; break; } }
else closeCount = 0;
_pos++;
}
InNonCode = true; return true;
}
// Interpolated string: $"..." or $@"..." or @$"..."
if ((c == '$' && next == '"') ||
(c == '$' && next == '@' && _pos + 2 < _end && _text[_pos + 2] == '"') ||
(c == '@' && next == '$' && _pos + 2 < _end && _text[_pos + 2] == '"'))
{
bool isVerbatim = (next == '@') || (c == '@');
_pos += (c == '$' && next == '"') ? 2 : 3;
SkipInterpolatedStringBody(isVerbatim);
InNonCode = true; return true;
}
// Verbatim string: @"..."
if (c == '@' && next == '"')
{
_pos += 2;
while (_pos < _end)
{
if (_text[_pos] == '\n') _line++;
if (_text[_pos] == '"')
{
if (_pos + 1 < _end && _text[_pos + 1] == '"') { _pos += 2; continue; }
_pos++; break;
}
_pos++;
}
InNonCode = true; return true;
}
// Regular string: "..."
if (c == '"')
{
_pos++;
while (_pos < _end)
{
if (_text[_pos] == '\\') { _pos += 2; continue; }
if (_text[_pos] == '"') { _pos++; break; }
if (_text[_pos] == '\n') _line++;
_pos++;
}
InNonCode = true; return true;
}
// Char literal: '...'
if (c == '\'')
{
_pos++;
while (_pos < _end)
{
if (_text[_pos] == '\\') { _pos += 2; continue; }
if (_text[_pos] == '\'') { _pos++; break; }
_pos++;
}
InNonCode = true; return true;
}
InNonCode = false;
_pos++;
return true;
}
/// <summary>
/// Skip the body of an interpolated string, handling nested interpolation holes.
/// _pos should be right after the opening quote.
/// </summary>
private void SkipInterpolatedStringBody(bool isVerbatim)
{
int interpDepth = 0;
while (_pos < _end)
{
char ch = _text[_pos];
if (ch == '\n') _line++;
if (interpDepth > 0)
{
// Inside interpolation hole — this is code, scan for nested strings/braces
if (ch == '{') { interpDepth++; _pos++; continue; }
if (ch == '}') { interpDepth--; _pos++; continue; }
if (ch == '"')
{
// Nested string inside interpolation hole
_pos++;
while (_pos < _end)
{
if (_text[_pos] == '\\') { _pos += 2; continue; }
if (_text[_pos] == '"') { _pos++; break; }
if (_text[_pos] == '\n') _line++;
_pos++;
}
continue;
}
if (ch == '/' && _pos + 1 < _end)
{
if (_text[_pos + 1] == '/') { _pos += 2; while (_pos < _end && _text[_pos] != '\n') _pos++; continue; }
if (_text[_pos + 1] == '*') { _pos += 2; while (_pos + 1 < _end && !(_text[_pos] == '*' && _text[_pos + 1] == '/')) { if (_text[_pos] == '\n') _line++; _pos++; } if (_pos + 1 < _end) _pos += 2; continue; }
}
if (ch == '\'') { _pos++; while (_pos < _end) { if (_text[_pos] == '\\') { _pos += 2; continue; } if (_text[_pos] == '\'') { _pos++; break; } _pos++; } continue; }
_pos++;
continue;
}
// interpDepth == 0: inside string content
if (ch == '{')
{
if (_pos + 1 < _end && _text[_pos + 1] == '{') { _pos += 2; continue; } // escaped {{
interpDepth = 1; _pos++; continue;
}
if (ch == '}')
{
if (_pos + 1 < _end && _text[_pos + 1] == '}') { _pos += 2; continue; } // escaped }}
// Stray } at depth 0 — shouldn't happen in valid code, just advance
_pos++; continue;
}
if (ch == '"')
{
if (isVerbatim && _pos + 1 < _end && _text[_pos + 1] == '"') { _pos += 2; continue; } // doubled quote
_pos++; return; // closing quote
}
if (!isVerbatim && ch == '\\') { _pos += 2; continue; } // escape in regular interpolated
_pos++;
}
}
/// <summary>
/// Skip the body of an interpolated raw string ($"""...""", $$"""...""", etc.).
/// dollarCount determines how many consecutive { start an interpolation hole.
/// quoteCount is the number of " that close the string.
/// _pos should be right after the opening quotes.
/// </summary>
private void SkipInterpolatedRawStringBody(int dollarCount, int quoteCount)
{
int interpDepth = 0;
while (_pos < _end)
{
char ch = _text[_pos];
if (ch == '\n') _line++;
if (interpDepth > 0)
{
// Inside interpolation hole — code context
if (ch == '{') { interpDepth++; _pos++; continue; }
if (ch == '}') { interpDepth--; _pos++; continue; }
if (ch == '"')
{
_pos++;
while (_pos < _end)
{
if (_text[_pos] == '\\') { _pos += 2; continue; }
if (_text[_pos] == '"') { _pos++; break; }
if (_text[_pos] == '\n') _line++;
_pos++;
}
continue;
}
if (ch == '/' && _pos + 1 < _end)
{
if (_text[_pos + 1] == '/') { _pos += 2; while (_pos < _end && _text[_pos] != '\n') _pos++; continue; }
if (_text[_pos + 1] == '*') { _pos += 2; while (_pos + 1 < _end && !(_text[_pos] == '*' && _text[_pos + 1] == '/')) { if (_text[_pos] == '\n') _line++; _pos++; } if (_pos + 1 < _end) _pos += 2; continue; }
}
if (ch == '\'') { _pos++; while (_pos < _end) { if (_text[_pos] == '\\') { _pos += 2; continue; } if (_text[_pos] == '\'') { _pos++; break; } _pos++; } continue; }
_pos++;
continue;
}
// String content (interpDepth == 0)
// Check for closing quote sequence
if (ch == '"')
{
int qc = 1;
while (_pos + qc < _end && _text[_pos + qc] == '"') qc++;
if (qc >= quoteCount) { _pos += quoteCount; return; }
// Fewer quotes than needed — literal content
_pos += qc;
continue;
}
// Check for interpolation hole: dollarCount consecutive {'s
if (ch == '{')
{
int bc = 1;
while (_pos + bc < _end && _text[_pos + bc] == '{') bc++;
if (bc >= dollarCount)
{
// Exactly dollarCount opens an interpolation hole; extras are literal
_pos += dollarCount;
interpDepth = 1;
}
else
{
// Fewer than dollarCount — literal braces
_pos += bc;
}
continue;
}
// Closing braces with dollarCount threshold — literal if fewer
if (ch == '}')
{
int bc = 1;
while (_pos + bc < _end && _text[_pos + bc] == '}') bc++;
_pos += bc; // all literal at depth 0
continue;
}
_pos++;
}
}
}
private static bool CheckBalancedDelimiters(string text, out int line, out char expected)
{
var braceStack = new Stack<int>();
var parenStack = new Stack<int>();
var bracketStack = new Stack<int>();
line = 1; expected = '\0';
var lexer = new CSharpLexer(text);
while (lexer.Advance(out char c))
{
if (lexer.InNonCode) continue;
switch (c)
{
case '{': braceStack.Push(lexer.Line); break;
case '}':
if (braceStack.Count == 0) { line = lexer.Line; expected = '{'; return false; }
braceStack.Pop();
break;
case '(': parenStack.Push(lexer.Line); break;
case ')':
if (parenStack.Count == 0) { line = lexer.Line; expected = '('; return false; }
parenStack.Pop();
break;
case '[': bracketStack.Push(lexer.Line); break;
case ']':
if (bracketStack.Count == 0) { line = lexer.Line; expected = '['; return false; }
bracketStack.Pop();
break;
}
}
if (braceStack.Count > 0) { line = braceStack.Peek(); expected = '}'; return false; }
if (parenStack.Count > 0) { line = parenStack.Peek(); expected = ')'; return false; }
if (bracketStack.Count > 0) { line = bracketStack.Peek(); expected = ']'; return false; }
return true;
}
private static object DeleteScript(string fullPath, string relativePath)
{
if (!File.Exists(fullPath))
{
return new ErrorResponse($"Script not found at '{relativePath}'. Cannot delete.");
}
try
{
// Use AssetDatabase.MoveAssetToTrash for safer deletion (allows undo)
bool deleted = AssetDatabase.MoveAssetToTrash(relativePath);
if (deleted)
{
AssetDatabase.Refresh(ImportAssetOptions.ForceSynchronousImport);
return new SuccessResponse(
$"Script '{Path.GetFileName(relativePath)}' moved to trash successfully.",
new { deleted = true }
);
}
else
{
// Fallback or error if MoveAssetToTrash fails
return new ErrorResponse(
$"Failed to move script '{relativePath}' to trash. It might be locked or in use."
);
}
}
catch (Exception e)
{
return new ErrorResponse($"Error deleting script '{relativePath}': {e.Message}");
}
}
/// <summary>
/// Structured edits (AST-backed where available) on existing scripts.
/// Supports class-level replace/delete with Roslyn span computation if USE_ROSLYN is defined,
/// otherwise falls back to a conservative balanced-brace scan.
/// </summary>
private static object EditScript(
string fullPath,
string relativePath,
string name,
JArray edits,
JObject options)
{
if (!File.Exists(fullPath))
return new ErrorResponse($"Script not found at '{relativePath}'.");
// Refuse edits if the target is a symlink
try
{
var attrs = File.GetAttributes(fullPath);
if ((attrs & FileAttributes.ReparsePoint) != 0)
return new ErrorResponse("Refusing to edit a symlinked script path.");
}
catch
{
// ignore failures checking attributes and proceed
}
if (edits == null || edits.Count == 0)
return new ErrorResponse("No edits provided.");
string original;
try { original = File.ReadAllText(fullPath); }
catch (Exception ex) { return new ErrorResponse($"Failed to read script: {ex.Message}"); }
string working = original;
try
{
var replacements = new List<(int start, int length, string text)>();
int appliedCount = 0;
// Apply mode: atomic (default) computes all spans against original and applies together.
// Sequential applies each edit immediately to the current working text (useful for dependent edits).
string applyMode = options?["applyMode"]?.ToString()?.ToLowerInvariant();
bool applySequentially = applyMode == "sequential";
foreach (var e in edits)
{
var op = (JObject)e;
var mode = (op.Value<string>("mode") ?? op.Value<string>("op") ?? string.Empty).ToLowerInvariant();
switch (mode)
{
case "replace_class":
{
string className = op.Value<string>("className");
string ns = op.Value<string>("namespace");
string replacement = ExtractReplacement(op);
if (string.IsNullOrWhiteSpace(className))
return new ErrorResponse("replace_class requires 'className'.");
if (replacement == null)
return new ErrorResponse("replace_class requires 'replacement' (inline or base64).");
if (!TryComputeClassSpan(working, className, ns, out var spanStart, out var spanLength, out var why))
return new ErrorResponse($"replace_class failed: {why}");
if (!ValidateClassSnippet(replacement, className, out var vErr))
return new ErrorResponse($"Replacement snippet invalid: {vErr}");
if (applySequentially)
{
working = working.Remove(spanStart, spanLength).Insert(spanStart, NormalizeNewlines(replacement));
appliedCount++;
}
else
{
replacements.Add((spanStart, spanLength, NormalizeNewlines(replacement)));
}
break;
}
case "delete_class":
{
string className = op.Value<string>("className");
string ns = op.Value<string>("namespace");
if (string.IsNullOrWhiteSpace(className))
return new ErrorResponse("delete_class requires 'className'.");
if (!TryComputeClassSpan(working, className, ns, out var s, out var l, out var why))
return new ErrorResponse($"delete_class failed: {why}");
if (applySequentially)
{
working = working.Remove(s, l);
appliedCount++;
}
else
{
replacements.Add((s, l, string.Empty));
}
break;
}
case "replace_method":
{
string className = op.Value<string>("className");
string ns = op.Value<string>("namespace");
string methodName = op.Value<string>("methodName");
string replacement = ExtractReplacement(op);
string returnType = op.Value<string>("returnType");
string parametersSignature = op.Value<string>("parametersSignature");
string attributesContains = op.Value<string>("attributesContains");
if (string.IsNullOrWhiteSpace(className)) return new ErrorResponse("replace_method requires 'className'.");
if (string.IsNullOrWhiteSpace(methodName)) return new ErrorResponse("replace_method requires 'methodName'.");
if (replacement == null) return new ErrorResponse("replace_method requires 'replacement' (inline or base64).");
if (!TryComputeClassSpan(working, className, ns, out var clsStart, out var clsLen, out var whyClass))
return new ErrorResponse($"replace_method failed to locate class: {whyClass}");
if (!TryComputeMethodSpan(working, clsStart, clsLen, methodName, returnType, parametersSignature, attributesContains, out var mStart, out var mLen, out var whyMethod))
{
bool hasDependentInsert = edits.Any(j => j is JObject jo &&
string.Equals(jo.Value<string>("className"), className, StringComparison.Ordinal) &&
string.Equals(jo.Value<string>("methodName"), methodName, StringComparison.Ordinal) &&
((jo.Value<string>("mode") ?? jo.Value<string>("op") ?? string.Empty).ToLowerInvariant() == "insert_method"));
string hint = hasDependentInsert && !applySequentially ? " Hint: This batch inserts this method. Use options.applyMode='sequential' or split into separate calls." : string.Empty;
return new ErrorResponse($"replace_method failed: {whyMethod}.{hint}");
}
if (applySequentially)
{
working = working.Remove(mStart, mLen).Insert(mStart, NormalizeNewlines(replacement));
appliedCount++;
}
else
{
replacements.Add((mStart, mLen, NormalizeNewlines(replacement)));
}
break;
}
case "delete_method":
{
string className = op.Value<string>("className");
string ns = op.Value<string>("namespace");
string methodName = op.Value<string>("methodName");
string returnType = op.Value<string>("returnType");
string parametersSignature = op.Value<string>("parametersSignature");
string attributesContains = op.Value<string>("attributesContains");
if (string.IsNullOrWhiteSpace(className)) return new ErrorResponse("delete_method requires 'className'.");
if (string.IsNullOrWhiteSpace(methodName)) return new ErrorResponse("delete_method requires 'methodName'.");
if (!TryComputeClassSpan(working, className, ns, out var clsStart, out var clsLen, out var whyClass))
return new ErrorResponse($"delete_method failed to locate class: {whyClass}");
if (!TryComputeMethodSpan(working, clsStart, clsLen, methodName, returnType, parametersSignature, attributesContains, out var mStart, out var mLen, out var whyMethod))
{
bool hasDependentInsert = edits.Any(j => j is JObject jo &&
string.Equals(jo.Value<string>("className"), className, StringComparison.Ordinal) &&
string.Equals(jo.Value<string>("methodName"), methodName, StringComparison.Ordinal) &&
((jo.Value<string>("mode") ?? jo.Value<string>("op") ?? string.Empty).ToLowerInvariant() == "insert_method"));
string hint = hasDependentInsert && !applySequentially ? " Hint: This batch inserts this method. Use options.applyMode='sequential' or split into separate calls." : string.Empty;
return new ErrorResponse($"delete_method failed: {whyMethod}.{hint}");
}
if (applySequentially)
{
working = working.Remove(mStart, mLen);
appliedCount++;
}
else
{
replacements.Add((mStart, mLen, string.Empty));
}
break;
}
case "insert_method":
{
string className = op.Value<string>("className");
string ns = op.Value<string>("namespace");
string position = (op.Value<string>("position") ?? "end").ToLowerInvariant();
string afterMethodName = op.Value<string>("afterMethodName");
string afterReturnType = op.Value<string>("afterReturnType");
string afterParameters = op.Value<string>("afterParametersSignature");
string afterAttributesContains = op.Value<string>("afterAttributesContains");
string snippet = ExtractReplacement(op);
// Harden: refuse empty replacement for inserts
if (snippet == null || snippet.Trim().Length == 0)
return new ErrorResponse("insert_method requires a non-empty 'replacement' text.");
if (string.IsNullOrWhiteSpace(className)) return new ErrorResponse("insert_method requires 'className'.");
if (snippet == null) return new ErrorResponse("insert_method requires 'replacement' (inline or base64) containing a full method declaration.");
if (!TryComputeClassSpan(working, className, ns, out var clsStart, out var clsLen, out var whyClass))
return new ErrorResponse($"insert_method failed to locate class: {whyClass}");
if (position == "after")
{
if (string.IsNullOrEmpty(afterMethodName)) return new ErrorResponse("insert_method with position='after' requires 'afterMethodName'.");
if (!TryComputeMethodSpan(working, clsStart, clsLen, afterMethodName, afterReturnType, afterParameters, afterAttributesContains, out var aStart, out var aLen, out var whyAfter))
return new ErrorResponse($"insert_method(after) failed to locate anchor method: {whyAfter}");
int insAt = aStart + aLen;
string text = NormalizeNewlines("\n\n" + snippet.TrimEnd() + "\n");
if (applySequentially)
{
working = working.Insert(insAt, text);
appliedCount++;
}
else
{
replacements.Add((insAt, 0, text));
}
}
else if (!TryFindClassInsertionPoint(working, clsStart, clsLen, position, out var insAt, out var whyIns))
return new ErrorResponse($"insert_method failed: {whyIns}");
else
{
string text = NormalizeNewlines("\n\n" + snippet.TrimEnd() + "\n");
if (applySequentially)
{
working = working.Insert(insAt, text);
appliedCount++;
}
else
{
replacements.Add((insAt, 0, text));
}
}
break;
}
case "anchor_insert":
{
string anchor = op.Value<string>("anchor");
string position = (op.Value<string>("position") ?? "before").ToLowerInvariant();
string text = op.Value<string>("text") ?? ExtractReplacement(op);
if (string.IsNullOrWhiteSpace(anchor)) return new ErrorResponse("anchor_insert requires 'anchor' (regex).");
if (string.IsNullOrEmpty(text)) return new ErrorResponse("anchor_insert requires non-empty 'text'.");
try
{
var rx = new Regex(anchor, RegexOptions.Multiline, TimeSpan.FromSeconds(2));
var allMatches = rx.Matches(working);
if (allMatches.Count == 0) return new ErrorResponse($"anchor_insert: anchor not found: {anchor}");
var m = FindBestAnchorMatch(allMatches, working, anchor);
if (m == null) return new ErrorResponse($"anchor_insert: anchor not found (filtered): {anchor}");
int insAt = position == "after" ? m.Index + m.Length : m.Index;
string norm = NormalizeNewlines(text);
if (!norm.EndsWith("\n"))
{
norm += "\n";
}
// Duplicate guard: if identical snippet already exists within this class, skip insert
if (TryComputeClassSpan(working, name, null, out var clsStartDG, out var clsLenDG, out _))
{
string classSlice = working.Substring(clsStartDG, Math.Min(clsLenDG, working.Length - clsStartDG));
if (classSlice.IndexOf(norm, StringComparison.Ordinal) >= 0)
{
// Do not insert duplicate; treat as no-op
break;
}
}
if (applySequentially)
{
working = working.Insert(insAt, norm);
appliedCount++;
}
else
{
replacements.Add((insAt, 0, norm));
}
}
catch (Exception ex)
{
return new ErrorResponse($"anchor_insert failed: {ex.Message}");
}
break;
}
case "anchor_delete":
{
string anchor = op.Value<string>("anchor");
if (string.IsNullOrWhiteSpace(anchor)) return new ErrorResponse("anchor_delete requires 'anchor' (regex).");
try
{
var rx = new Regex(anchor, RegexOptions.Multiline, TimeSpan.FromSeconds(2));
var allDelMatches = rx.Matches(working);
if (allDelMatches.Count == 0) return new ErrorResponse($"anchor_delete: anchor not found: {anchor}");
var m = FindBestAnchorMatch(allDelMatches, working, anchor);
if (m == null) return new ErrorResponse($"anchor_delete: anchor not found (filtered): {anchor}");
int delAt = m.Index;
int delLen = m.Length;
if (applySequentially)
{
working = working.Remove(delAt, delLen);
appliedCount++;
}
else
{
replacements.Add((delAt, delLen, string.Empty));
}
}
catch (Exception ex)
{
return new ErrorResponse($"anchor_delete failed: {ex.Message}");
}
break;
}
case "anchor_replace":
{
string anchor = op.Value<string>("anchor");
string replacement = op.Value<string>("text") ?? op.Value<string>("replacement") ?? ExtractReplacement(op) ?? string.Empty;
if (string.IsNullOrWhiteSpace(anchor)) return new ErrorResponse("anchor_replace requires 'anchor' (regex).");
try
{
var rx = new Regex(anchor, RegexOptions.Multiline, TimeSpan.FromSeconds(2));
var allReplMatches = rx.Matches(working);
if (allReplMatches.Count == 0) return new ErrorResponse($"anchor_replace: anchor not found: {anchor}");
var m = FindBestAnchorMatch(allReplMatches, working, anchor);
if (m == null) return new ErrorResponse($"anchor_replace: anchor not found (filtered): {anchor}");
int at = m.Index;
int len = m.Length;
string norm = NormalizeNewlines(replacement);
if (applySequentially)
{
working = working.Remove(at, len).Insert(at, norm);
appliedCount++;
}
else
{
replacements.Add((at, len, norm));
}
}
catch (Exception ex)
{
return new ErrorResponse($"anchor_replace failed: {ex.Message}");
}
break;
}
default:
return new ErrorResponse($"Unknown edit mode: '{mode}'. Allowed: replace_class, delete_class, replace_method, delete_method, insert_method, anchor_insert, anchor_delete, anchor_replace.");
}
}
if (!applySequentially)
{
if (HasOverlaps(replacements))
{
var ordered = replacements.OrderByDescending(r => r.start).ToList();
for (int i = 1; i < ordered.Count; i++)
{
if (ordered[i].start + ordered[i].length > ordered[i - 1].start)
{
var conflict = new[] { new { startA = ordered[i].start, endA = ordered[i].start + ordered[i].length, startB = ordered[i - 1].start, endB = ordered[i - 1].start + ordered[i - 1].length } };
return new ErrorResponse("overlap", new { status = "overlap", conflicts = conflict, hint = "Sort ranges descending by start and compute from the same snapshot." });
}
}
return new ErrorResponse("overlap", new { status = "overlap" });
}
foreach (var r in replacements.OrderByDescending(r => r.start))
working = working.Remove(r.start, r.length).Insert(r.start, r.text);
appliedCount = replacements.Count;
}
// Guard against structural imbalance before validation
if (!CheckBalancedDelimiters(working, out int lineBal, out char expectedBal))
return new ErrorResponse("unbalanced_braces", new { status = "unbalanced_braces", line = lineBal, expected = expectedBal.ToString() });
// No-op guard for structured edits: if text unchanged, return explicit no-op
if (string.Equals(working, original, StringComparison.Ordinal))
{
var sameSha = ComputeSha256(original);
return new SuccessResponse(
$"No-op: contents unchanged for '{relativePath}'.",
new
{
path = relativePath,
uri = $"mcpforunity://path/{relativePath}",
editsApplied = 0,
no_op = true,
sha256 = sameSha,
evidence = new { reason = "identical_content" }
}
);
}
// Validate result using override from options if provided; otherwise GUI strictness
var level = GetValidationLevelFromGUI();
try
{
var validateOpt = options?["validate"]?.ToString()?.ToLowerInvariant();
if (!string.IsNullOrEmpty(validateOpt))
{
level = validateOpt switch
{
"basic" => ValidationLevel.Basic,
"standard" => ValidationLevel.Standard,
"comprehensive" => ValidationLevel.Comprehensive,
"strict" => ValidationLevel.Strict,
_ => level
};
}
}
catch { /* ignore option parsing issues */ }
if (!ValidateScriptSyntax(working, level, out var errors))
return new ErrorResponse("validation_failed", new { status = "validation_failed", diagnostics = errors ?? Array.Empty<string>() });
else if (errors != null && errors.Length > 0)
McpLog.Warn($"Script validation warnings for {name}:\n" + string.Join("\n", errors));
// Atomic write with backup; schedule refresh
// Decide refresh behavior
string refreshMode = options?["refresh"]?.ToString()?.ToLowerInvariant();
bool immediate = refreshMode == "immediate" || refreshMode == "sync";
// Persist changes atomically (no BOM), then compute/return new file SHA
var enc = new System.Text.UTF8Encoding(encoderShouldEmitUTF8Identifier: false);
var tmp = fullPath + ".tmp";
File.WriteAllText(tmp, working, enc);
var backup = fullPath + ".bak";
try
{
File.Replace(tmp, fullPath, backup);
try { if (File.Exists(backup)) File.Delete(backup); } catch { }
}
catch (PlatformNotSupportedException)
{
File.Copy(tmp, fullPath, true);
try { File.Delete(tmp); } catch { }
try { if (File.Exists(backup)) File.Delete(backup); } catch { }
}
catch (IOException)
{
File.Copy(tmp, fullPath, true);
try { File.Delete(tmp); } catch { }
try { if (File.Exists(backup)) File.Delete(backup); } catch { }
}
var newSha = ComputeSha256(working);
var ok = new SuccessResponse(
$"Applied {appliedCount} structured edit(s) to '{relativePath}'.",
new
{
path = relativePath,
uri = $"mcpforunity://path/{relativePath}",
editsApplied = appliedCount,
scheduledRefresh = !immediate,
sha256 = newSha
}
);
if (immediate)
{
McpLog.Info($"[ManageScript] EditScript: immediate refresh for '{relativePath}'", always: false);
ManageScriptRefreshHelpers.ImportAndRequestCompile(relativePath);
}
else
{
ManageScriptRefreshHelpers.ScheduleScriptRefresh(relativePath);
}
return ok;
}
catch (Exception ex)
{
return new ErrorResponse($"Edit failed: {ex.Message}");
}
}
private static bool HasOverlaps(IEnumerable<(int start, int length, string text)> list)
{
var arr = list.OrderBy(x => x.start).ToArray();
for (int i = 1; i < arr.Length; i++)
{
if (arr[i - 1].start + arr[i - 1].length > arr[i].start)
return true;
}
return false;
}
private static string ExtractReplacement(JObject op)
{
var inline = op.Value<string>("replacement");
if (!string.IsNullOrEmpty(inline)) return inline;
var b64 = op.Value<string>("replacementBase64");
if (!string.IsNullOrEmpty(b64))
{
try { return System.Text.Encoding.UTF8.GetString(Convert.FromBase64String(b64)); }
catch { return null; }
}
return null;
}
private static string NormalizeNewlines(string t)
{
if (string.IsNullOrEmpty(t)) return t;
return t.Replace("\r\n", "\n").Replace("\r", "\n");
}
private static bool ValidateClassSnippet(string snippet, string expectedName, out string err)
{
#if USE_ROSLYN
try
{
var tree = CSharpSyntaxTree.ParseText(snippet);
var root = tree.GetRoot();
var classes = root.DescendantNodes().OfType<Microsoft.CodeAnalysis.CSharp.Syntax.ClassDeclarationSyntax>().ToList();
if (classes.Count != 1) { err = "snippet must contain exactly one class declaration"; return false; }
// Optional: enforce expected name
// if (classes[0].Identifier.ValueText != expectedName) { err = $"snippet declares '{classes[0].Identifier.ValueText}', expected '{expectedName}'"; return false; }
err = null; return true;
}
catch (Exception ex) { err = ex.Message; return false; }
#else
if (string.IsNullOrWhiteSpace(snippet) || !snippet.Contains("class ")) { err = "no 'class' keyword found in snippet"; return false; }
err = null; return true;
#endif
}
private static bool TryComputeClassSpan(string source, string className, string ns, out int start, out int length, out string why)
{
#if USE_ROSLYN
try
{
var tree = CSharpSyntaxTree.ParseText(source);
var root = tree.GetRoot();
var classes = root.DescendantNodes()
.OfType<Microsoft.CodeAnalysis.CSharp.Syntax.ClassDeclarationSyntax>()
.Where(c => c.Identifier.ValueText == className);
if (!string.IsNullOrEmpty(ns))
{
classes = classes.Where(c =>
(c.FirstAncestorOrSelf<Microsoft.CodeAnalysis.CSharp.Syntax.NamespaceDeclarationSyntax>()?.Name?.ToString() ?? "") == ns
|| (c.FirstAncestorOrSelf<Microsoft.CodeAnalysis.CSharp.Syntax.FileScopedNamespaceDeclarationSyntax>()?.Name?.ToString() ?? "") == ns);
}
var list = classes.ToList();
if (list.Count == 0) { start = length = 0; why = $"class '{className}' not found" + (ns != null ? $" in namespace '{ns}'" : ""); return false; }
if (list.Count > 1) { start = length = 0; why = $"class '{className}' matched {list.Count} declarations (partial/nested?). Disambiguate."; return false; }
var cls = list[0];
var span = cls.FullSpan; // includes attributes & leading trivia
start = span.Start; length = span.Length; why = null; return true;
}
catch
{
// fall back below
}
#endif
return TryComputeClassSpanBalanced(source, className, ns, out start, out length, out why);
}
private static bool TryComputeClassSpanBalanced(string source, string className, string ns, out int start, out int length, out string why)
{
start = length = 0; why = null;
var idx = IndexOfClassToken(source, className);
if (idx < 0) { why = $"class '{className}' not found (balanced scan)"; return false; }
if (!string.IsNullOrEmpty(ns) && !AppearsWithinNamespaceHeader(source, idx, ns))
{ why = $"class '{className}' not under namespace '{ns}' (balanced scan)"; return false; }
// Include modifiers/attributes on the same line: back up to the start of line
int lineStart = idx;
while (lineStart > 0 && source[lineStart - 1] != '\n' && source[lineStart - 1] != '\r') lineStart--;
int i = idx;
while (i < source.Length && source[i] != '{') i++;
if (i >= source.Length) { why = "no opening brace after class header"; return false; }
int depth = 0;
int startSpan = lineStart;
var lexer = new CSharpLexer(source, i);
while (lexer.Advance(out char c))
{
if (lexer.InNonCode) continue;
if (c == '{') { depth++; }
else if (c == '}')
{
depth--;
if (depth == 0) { start = startSpan; length = (lexer.Position - 1 - startSpan) + 1; return true; }
if (depth < 0) { why = "brace underflow"; return false; }
}
}
why = "unterminated class block"; return false;
}
private static bool TryComputeMethodSpan(
string source,
int classStart,
int classLength,
string methodName,
string returnType,
string parametersSignature,
string attributesContains,
out int start,
out int length,
out string why)
{
start = length = 0; why = null;
int searchStart = classStart;
int searchEnd = Math.Min(source.Length, classStart + classLength);
// 1) Find the method header using a stricter regex (allows optional attributes above)
string rtPattern = string.IsNullOrEmpty(returnType) ? @"[^\s]+" : Regex.Escape(returnType).Replace("\\ ", "\\s+");
string namePattern = Regex.Escape(methodName);
// If a parametersSignature is provided, it may include surrounding parentheses. Strip them so
// we can safely embed the signature inside our own parenthesis group without duplicating.
string paramsPattern;
if (string.IsNullOrEmpty(parametersSignature))
{
paramsPattern = @"[\s\S]*?"; // permissive when not specified
}
else
{
string ps = parametersSignature.Trim();
if (ps.StartsWith("(") && ps.EndsWith(")") && ps.Length >= 2)
{
ps = ps.Substring(1, ps.Length - 2);
}
// Escape literal text of the signature
paramsPattern = Regex.Escape(ps);
}
string pattern =
@"(?m)^[\t ]*(?:\[[^\]]+\][\t ]*)*[\t ]*" +
@"(?:(?:public|private|protected|internal|static|virtual|override|sealed|async|extern|unsafe|new|partial|readonly|volatile|event|abstract|ref|in|out)\s+)*" +
rtPattern + @"[\t ]+" + namePattern + @"\s*(?:<[^>]+>)?\s*\(" + paramsPattern + @"\)";
string slice = source.Substring(searchStart, searchEnd - searchStart);
var headerMatch = Regex.Match(slice, pattern, RegexOptions.Multiline, TimeSpan.FromSeconds(2));
if (!headerMatch.Success)
{
why = $"method '{methodName}' header not found in class"; return false;
}
int headerIndex = searchStart + headerMatch.Index;
// Optional attributes filter: look upward from headerIndex for contiguous attribute lines
if (!string.IsNullOrEmpty(attributesContains))
{
int attrScanStart = headerIndex;
while (attrScanStart > searchStart)
{
int prevNl = source.LastIndexOf('\n', attrScanStart - 1);
if (prevNl < 0 || prevNl < searchStart) break;
string prevLine = source.Substring(prevNl + 1, attrScanStart - (prevNl + 1));
if (prevLine.TrimStart().StartsWith("[")) { attrScanStart = prevNl; continue; }
break;
}
string attrBlock = source.Substring(attrScanStart, headerIndex - attrScanStart);
if (attrBlock.IndexOf(attributesContains, StringComparison.Ordinal) < 0)
{
why = $"method '{methodName}' found but attributes filter did not match"; return false;
}
}
// backtrack to the very start of header/attributes to include in span
int lineStart = headerIndex;
while (lineStart > searchStart && source[lineStart - 1] != '\n' && source[lineStart - 1] != '\r') lineStart--;
// If previous lines are attributes, include them
int attrStart = lineStart;
int probe = lineStart - 1;
while (probe > searchStart)
{
int prevNl = source.LastIndexOf('\n', probe);
if (prevNl < 0 || prevNl < searchStart) break;
string prev = source.Substring(prevNl + 1, attrStart - (prevNl + 1));
if (prev.TrimStart().StartsWith("[")) { attrStart = prevNl + 1; probe = prevNl - 1; }
else break;
}
// 2) Walk from the end of signature to detect body style ('{' or '=> ...;') and compute end
// Find the '(' that belongs to the method signature, not attributes
int nameTokenIdx = IndexOfTokenWithin(source, methodName, headerIndex, searchEnd);
if (nameTokenIdx < 0) { why = $"method '{methodName}' token not found after header"; return false; }
int sigOpenParen = IndexOfTokenWithin(source, "(", nameTokenIdx, searchEnd);
if (sigOpenParen < 0) { why = "method parameter list '(' not found"; return false; }
int i = sigOpenParen;
int parenDepth = 0;
var parenLexer = new CSharpLexer(source, i, searchEnd);
while (parenLexer.Advance(out char pc))
{
if (parenLexer.InNonCode) continue;
if (pc == '(') parenDepth++;
if (pc == ')') { parenDepth--; if (parenDepth == 0) break; }
}
i = parenLexer.Position;
// After params: detect expression-bodied or block-bodied
// Skip whitespace/comments
for (; i < searchEnd; i++)
{
char c = source[i];
char n = i + 1 < searchEnd ? source[i + 1] : '\0';
if (char.IsWhiteSpace(c)) continue;
if (c == '/' && n == '/') { while (i < searchEnd && source[i] != '\n') i++; continue; }
if (c == '/' && n == '*') { i += 2; while (i + 1 < searchEnd && !(source[i] == '*' && source[i + 1] == '/')) i++; i++; continue; }
break;
}
// Tolerate generic constraints between params and body: multiple 'where T : ...'
for (; ; )
{
// Skip whitespace/comments before checking for 'where'
for (; i < searchEnd; i++)
{
char c = source[i];
char n = i + 1 < searchEnd ? source[i + 1] : '\0';
if (char.IsWhiteSpace(c)) continue;
if (c == '/' && n == '/') { while (i < searchEnd && source[i] != '\n') i++; continue; }
if (c == '/' && n == '*') { i += 2; while (i + 1 < searchEnd && !(source[i] == '*' && source[i + 1] == '/')) i++; i++; continue; }
break;
}
// Check word-boundary 'where'
bool hasWhere = false;
if (i + 5 <= searchEnd)
{
hasWhere = source[i] == 'w' && source[i + 1] == 'h' && source[i + 2] == 'e' && source[i + 3] == 'r' && source[i + 4] == 'e';
if (hasWhere)
{
// Left boundary
if (i - 1 >= 0)
{
char lb = source[i - 1];
if (char.IsLetterOrDigit(lb) || lb == '_') hasWhere = false;
}
// Right boundary
if (hasWhere && i + 5 < searchEnd)
{
char rb = source[i + 5];
if (char.IsLetterOrDigit(rb) || rb == '_') hasWhere = false;
}
}
}
if (!hasWhere) break;
// Advance past the entire where-constraint clause until we hit '{' or '=>' or ';'
i += 5; // past 'where'
while (i < searchEnd)
{
char c = source[i];
char n = i + 1 < searchEnd ? source[i + 1] : '\0';
if (c == '{' || c == ';' || (c == '=' && n == '>')) break;
// Skip comments inline
if (c == '/' && n == '/') { while (i < searchEnd && source[i] != '\n') i++; continue; }
if (c == '/' && n == '*') { i += 2; while (i + 1 < searchEnd && !(source[i] == '*' && source[i + 1] == '/')) i++; i++; continue; }
i++;
}
}
// Re-check for expression-bodied after constraints
if (i < searchEnd - 1 && source[i] == '=' && source[i + 1] == '>')
{
// expression-bodied method: seek to terminating semicolon
int j = i;
bool done = false;
while (j < searchEnd)
{
char c = source[j];
if (c == ';') { done = true; break; }
j++;
}
if (!done) { why = "unterminated expression-bodied method"; return false; }
start = attrStart; length = (j - attrStart) + 1; return true;
}
if (i >= searchEnd || source[i] != '{') { why = "no opening brace after method signature"; return false; }
int depth = 0;
int startSpan = attrStart;
var bodyLexer = new CSharpLexer(source, i, searchEnd);
while (bodyLexer.Advance(out char bc))
{
if (bodyLexer.InNonCode) continue;
if (bc == '{') depth++;
else if (bc == '}')
{
depth--;
if (depth == 0) { start = startSpan; length = (bodyLexer.Position - 1 - startSpan) + 1; return true; }
if (depth < 0) { why = "brace underflow in method"; return false; }
}
}
why = "unterminated method block"; return false;
}
private static int IndexOfTokenWithin(string s, string token, int start, int end)
{
int idx = s.IndexOf(token, start, StringComparison.Ordinal);
return (idx >= 0 && idx < end) ? idx : -1;
}
private static bool TryFindClassInsertionPoint(string source, int classStart, int classLength, string position, out int insertAt, out string why)
{
insertAt = 0; why = null;
int searchStart = classStart;
int searchEnd = Math.Min(source.Length, classStart + classLength);
if (position == "start")
{
// find first '{' after class header, insert just after with a newline
int i = IndexOfTokenWithin(source, "{", searchStart, searchEnd);
if (i < 0) { why = "could not find class opening brace"; return false; }
insertAt = i + 1; return true;
}
else // end
{
// walk to matching closing brace of class and insert just before it
int i = IndexOfTokenWithin(source, "{", searchStart, searchEnd);
if (i < 0) { why = "could not find class opening brace"; return false; }
int depth = 0;
var lexer = new CSharpLexer(source, i, searchEnd);
while (lexer.Advance(out char c))
{
if (lexer.InNonCode) continue;
if (c == '{') depth++;
else if (c == '}')
{
depth--;
if (depth == 0) { insertAt = lexer.Position - 1; return true; }
if (depth < 0) { why = "brace underflow while scanning class"; return false; }
}
}
why = "could not find class closing brace"; return false;
}
}
/// <summary>
/// Given multiple regex matches in C# source, pick the best one for
/// anchor-based insertions. For closing-brace patterns, uses actual
/// brace-depth analysis to prefer the outermost (class-level) brace.
/// Otherwise, returns the last match.
/// </summary>
private static Match FindBestAnchorMatch(MatchCollection matches, string text, string pattern)
{
if (matches.Count == 0) return null;
if (matches.Count == 1) return matches[0];
bool isClosingBracePattern = pattern.Contains("}") &&
(pattern.Contains("$") || pattern.EndsWith(@"\s*"));
if (isClosingBracePattern)
{
// Compute brace depth at each match's '}' position using CSharpLexer
Match best = null;
int bestDepth = int.MaxValue;
int bestPos = -1;
// Single-pass: scan text and record depth at every '}' in real code
var depthMap = new Dictionary<int, int>();
int depth = 0;
var lexer = new CSharpLexer(text);
while (lexer.Advance(out char c))
{
if (lexer.InNonCode) continue;
if (c == '{') depth++;
else if (c == '}')
{
depthMap[lexer.Position - 1] = depth;
depth = Math.Max(0, depth - 1);
}
}
foreach (Match m in matches)
{
// Find the '}' within the match span
int bracePos = -1;
for (int k = m.Index; k < m.Index + m.Length && k < text.Length; k++)
{
if (text[k] == '}') { bracePos = k; break; }
}
if (bracePos < 0) continue;
if (!depthMap.TryGetValue(bracePos, out int d)) continue; // in string/comment
// Prefer shallowest depth, then latest position
if (d < bestDepth || (d == bestDepth && bracePos > bestPos))
{
bestDepth = d;
bestPos = bracePos;
best = m;
}
}
return best ?? matches[matches.Count - 1];
}
// Default: prefer last match
return matches[matches.Count - 1];
}
private static int IndexOfClassToken(string s, string className)
{
var pattern = "class " + className;
int searchFrom = 0;
while (searchFrom < s.Length)
{
int idx = s.IndexOf(pattern, searchFrom, StringComparison.Ordinal);
if (idx < 0) return -1;
// Word boundary on left: char before "class" must not be letter/digit/_
if (idx > 0)
{
char left = s[idx - 1];
if (char.IsLetterOrDigit(left) || left == '_') { searchFrom = idx + 1; continue; }
}
// Word boundary on right: char after className must not be letter/digit/_
int afterEnd = idx + pattern.Length;
if (afterEnd < s.Length)
{
char right = s[afterEnd];
if (char.IsLetterOrDigit(right) || right == '_') { searchFrom = idx + 1; continue; }
}
// Check that this position is not inside a string or comment
var lexer = new CSharpLexer(s, 0, idx + 1);
bool inNonCode = false;
while (lexer.Advance(out _))
{
inNonCode = lexer.InNonCode;
}
// After advancing to idx+1, check if the last character processed was in non-code
if (inNonCode) { searchFrom = idx + 1; continue; }
return idx;
}
return -1;
}
private static bool AppearsWithinNamespaceHeader(string s, int pos, string ns)
{
int from = Math.Max(0, pos - 2000);
var slice = s.Substring(from, pos - from);
return slice.Contains("namespace " + ns);
}
/// <summary>
/// Generates basic C# script content based on name and type.
/// </summary>
private static string GenerateDefaultScriptContent(
string name,
string scriptType,
string namespaceName
)
{
string usingStatements = "using UnityEngine;\nusing System.Collections;\n";
string classDeclaration;
string body =
"\n // Use this for initialization\n void Start() {\n\n }\n\n // Update is called once per frame\n void Update() {\n\n }\n";
string baseClass = "";
if (!string.IsNullOrEmpty(scriptType))
{
if (scriptType.Equals("MonoBehaviour", StringComparison.OrdinalIgnoreCase))
baseClass = " : MonoBehaviour";
else if (scriptType.Equals("ScriptableObject", StringComparison.OrdinalIgnoreCase))
{
baseClass = " : ScriptableObject";
body = ""; // ScriptableObjects don't usually need Start/Update
}
else if (
scriptType.Equals("Editor", StringComparison.OrdinalIgnoreCase)
|| scriptType.Equals("EditorWindow", StringComparison.OrdinalIgnoreCase)
)
{
usingStatements += "using UnityEditor;\n";
if (scriptType.Equals("Editor", StringComparison.OrdinalIgnoreCase))
baseClass = " : Editor";
else
baseClass = " : EditorWindow";
body = ""; // Editor scripts have different structures
}
// Add more types as needed
}
classDeclaration = $"public class {name}{baseClass}";
string fullContent = $"{usingStatements}\n";
bool useNamespace = !string.IsNullOrEmpty(namespaceName);
if (useNamespace)
{
fullContent += $"namespace {namespaceName}\n{{\n";
// Indent class and body if using namespace
classDeclaration = " " + classDeclaration;
body = string.Join("\n", body.Split('\n').Select(line => " " + line));
}
fullContent += $"{classDeclaration}\n{{\n{body}\n}}";
if (useNamespace)
{
fullContent += "\n}"; // Close namespace
}
return fullContent.Trim() + "\n"; // Ensure a trailing newline
}
/// <summary>
/// Gets the validation level from the GUI settings
/// </summary>
private static ValidationLevel GetValidationLevelFromGUI()
{
int savedLevel = EditorPrefs.GetInt(EditorPrefKeys.ValidationLevel, (int)ValidationLevel.Standard);
return (ValidationLevel)Mathf.Clamp(savedLevel, 0, 3);
}
/// <summary>
/// Validates C# script syntax using multiple validation layers.
/// </summary>
private static bool ValidateScriptSyntax(string contents)
{
return ValidateScriptSyntax(contents, ValidationLevel.Standard, out _);
}
/// <summary>
/// Advanced syntax validation with detailed diagnostics and configurable strictness.
/// </summary>
private static bool ValidateScriptSyntax(string contents, ValidationLevel level, out string[] errors)
{
var errorList = new System.Collections.Generic.List<string>();
errors = null;
if (string.IsNullOrEmpty(contents))
{
return true; // Empty content is valid
}
// Basic structural validation: check balanced delimiters
if (!CheckBalancedDelimiters(contents, out int errLine, out char errExpected))
{
errorList.Add($"ERROR: Unbalanced delimiter at line {errLine} (expected '{errExpected}')");
errors = errorList.ToArray();
return false;
}
#if USE_ROSLYN
// Advanced Roslyn-based validation: only run for Standard+; fail on Roslyn errors
if (level >= ValidationLevel.Standard)
{
if (!ValidateScriptSyntaxRoslyn(contents, level, errorList))
{
errors = errorList.ToArray();
return false;
}
}
#endif
// Unity-specific validation
if (level >= ValidationLevel.Standard)
{
ValidateScriptSyntaxUnity(contents, errorList);
}
// Semantic analysis for common issues
if (level >= ValidationLevel.Comprehensive)
{
ValidateSemanticRules(contents, errorList);
}
#if USE_ROSLYN
// Full semantic compilation validation for Strict level
if (level == ValidationLevel.Strict)
{
if (!ValidateScriptSemantics(contents, errorList))
{
errors = errorList.ToArray();
return false; // Strict level fails on any semantic errors
}
}
#endif
errors = errorList.ToArray();
return errorList.Count == 0 || (level != ValidationLevel.Strict && !errorList.Any(e => e.StartsWith("ERROR:")));
}
/// <summary>
/// Validation strictness levels
/// </summary>
private enum ValidationLevel
{
Basic, // Only syntax errors
Standard, // Syntax + Unity best practices
Comprehensive, // All checks + semantic analysis
Strict // Treat all issues as errors
}
#if USE_ROSLYN
/// <summary>
/// Cached compilation references for performance
/// </summary>
private static System.Collections.Generic.List<MetadataReference> _cachedReferences = null;
private static DateTime _cacheTime = DateTime.MinValue;
private static readonly TimeSpan CacheExpiry = TimeSpan.FromMinutes(5);
/// <summary>
/// Validates syntax using Roslyn compiler services
/// </summary>
private static bool ValidateScriptSyntaxRoslyn(string contents, ValidationLevel level, System.Collections.Generic.List<string> errors)
{
try
{
var syntaxTree = CSharpSyntaxTree.ParseText(contents);
var diagnostics = syntaxTree.GetDiagnostics();
bool hasErrors = false;
foreach (var diagnostic in diagnostics)
{
string severity = diagnostic.Severity.ToString().ToUpper();
string message = $"{severity}: {diagnostic.GetMessage()}";
if (diagnostic.Severity == DiagnosticSeverity.Error)
{
hasErrors = true;
}
// Include warnings in comprehensive mode
if (level >= ValidationLevel.Standard || diagnostic.Severity == DiagnosticSeverity.Error) //Also use Standard for now
{
var location = diagnostic.Location.GetLineSpan();
if (location.IsValid)
{
message += $" (Line {location.StartLinePosition.Line + 1})";
}
errors.Add(message);
}
}
return !hasErrors;
}
catch (Exception ex)
{
errors.Add($"ERROR: Roslyn validation failed: {ex.Message}");
return false;
}
}
/// <summary>
/// Validates script semantics using full compilation context to catch namespace, type, and method resolution errors
/// </summary>
private static bool ValidateScriptSemantics(string contents, System.Collections.Generic.List<string> errors)
{
try
{
// Get compilation references with caching
var references = GetCompilationReferences();
if (references == null || references.Count == 0)
{
errors.Add("WARNING: Could not load compilation references for semantic validation");
return true; // Don't fail if we can't get references
}
// Create syntax tree
var syntaxTree = CSharpSyntaxTree.ParseText(contents);
// Create compilation with full context
var compilation = CSharpCompilation.Create(
"TempValidation",
new[] { syntaxTree },
references,
new CSharpCompilationOptions(OutputKind.DynamicallyLinkedLibrary)
);
// Get semantic diagnostics - this catches all the issues you mentioned!
var diagnostics = compilation.GetDiagnostics();
bool hasErrors = false;
foreach (var diagnostic in diagnostics)
{
if (diagnostic.Severity == DiagnosticSeverity.Error)
{
hasErrors = true;
var location = diagnostic.Location.GetLineSpan();
string locationInfo = location.IsValid ?
$" (Line {location.StartLinePosition.Line + 1}, Column {location.StartLinePosition.Character + 1})" : "";
// Include diagnostic ID for better error identification
string diagnosticId = !string.IsNullOrEmpty(diagnostic.Id) ? $" [{diagnostic.Id}]" : "";
errors.Add($"ERROR: {diagnostic.GetMessage()}{diagnosticId}{locationInfo}");
}
else if (diagnostic.Severity == DiagnosticSeverity.Warning)
{
var location = diagnostic.Location.GetLineSpan();
string locationInfo = location.IsValid ?
$" (Line {location.StartLinePosition.Line + 1}, Column {location.StartLinePosition.Character + 1})" : "";
string diagnosticId = !string.IsNullOrEmpty(diagnostic.Id) ? $" [{diagnostic.Id}]" : "";
errors.Add($"WARNING: {diagnostic.GetMessage()}{diagnosticId}{locationInfo}");
}
}
return !hasErrors;
}
catch (Exception ex)
{
errors.Add($"ERROR: Semantic validation failed: {ex.Message}");
return false;
}
}
/// <summary>
/// Gets compilation references with caching for performance
/// </summary>
private static System.Collections.Generic.List<MetadataReference> GetCompilationReferences()
{
// Check cache validity
if (_cachedReferences != null && DateTime.Now - _cacheTime < CacheExpiry)
{
return _cachedReferences;
}
try
{
var references = new System.Collections.Generic.List<MetadataReference>();
// Core .NET assemblies
references.Add(MetadataReference.CreateFromFile(typeof(object).Assembly.Location)); // mscorlib/System.Private.CoreLib
references.Add(MetadataReference.CreateFromFile(typeof(System.Linq.Enumerable).Assembly.Location)); // System.Linq
references.Add(MetadataReference.CreateFromFile(typeof(System.Collections.Generic.List<>).Assembly.Location)); // System.Collections
// Unity assemblies
try
{
references.Add(MetadataReference.CreateFromFile(typeof(UnityEngine.Debug).Assembly.Location)); // UnityEngine
}
catch (Exception ex)
{
McpLog.Warn($"Could not load UnityEngine assembly: {ex.Message}");
}
#if UNITY_EDITOR
try
{
references.Add(MetadataReference.CreateFromFile(typeof(UnityEditor.Editor).Assembly.Location)); // UnityEditor
}
catch (Exception ex)
{
McpLog.Warn($"Could not load UnityEditor assembly: {ex.Message}");
}
// Get Unity project assemblies
try
{
var assemblies = CompilationPipeline.GetAssemblies();
foreach (var assembly in assemblies)
{
if (File.Exists(assembly.outputPath))
{
references.Add(MetadataReference.CreateFromFile(assembly.outputPath));
}
}
}
catch (Exception ex)
{
McpLog.Warn($"Could not load Unity project assemblies: {ex.Message}");
}
#endif
// Cache the results
_cachedReferences = references;
_cacheTime = DateTime.Now;
return references;
}
catch (Exception ex)
{
McpLog.Error($"Failed to get compilation references: {ex.Message}");
return new System.Collections.Generic.List<MetadataReference>();
}
}
#else
private static bool ValidateScriptSyntaxRoslyn(string contents, ValidationLevel level, System.Collections.Generic.List<string> errors)
{
// Fallback when Roslyn is not available
return true;
}
#endif
/// <summary>
/// Validates Unity-specific coding rules and best practices
/// //TODO: Naive Unity Checks and not really yield any results, need to be improved
/// </summary>
private static void ValidateScriptSyntaxUnity(string contents, System.Collections.Generic.List<string> errors)
{
// Check for common Unity anti-patterns
if (contents.Contains("FindObjectOfType") && contents.Contains("Update()"))
{
errors.Add("WARNING: FindObjectOfType in Update() can cause performance issues");
}
if (contents.Contains("GameObject.Find") && contents.Contains("Update()"))
{
errors.Add("WARNING: GameObject.Find in Update() can cause performance issues");
}
// Check for proper MonoBehaviour usage
if (contents.Contains(": MonoBehaviour") && !contents.Contains("using UnityEngine"))
{
errors.Add("WARNING: MonoBehaviour requires 'using UnityEngine;'");
}
// Check for SerializeField usage
if (contents.Contains("[SerializeField]") && !contents.Contains("using UnityEngine"))
{
errors.Add("WARNING: SerializeField requires 'using UnityEngine;'");
}
// Check for proper coroutine usage
if (contents.Contains("StartCoroutine") && !contents.Contains("IEnumerator"))
{
errors.Add("WARNING: StartCoroutine typically requires IEnumerator methods");
}
// Check for Update without FixedUpdate for physics
if (contents.Contains("Rigidbody") && contents.Contains("Update()") && !contents.Contains("FixedUpdate()"))
{
errors.Add("WARNING: Consider using FixedUpdate() for Rigidbody operations");
}
// Check for missing null checks on Unity objects
if (contents.Contains("GetComponent<") && !contents.Contains("!= null"))
{
errors.Add("WARNING: Consider null checking GetComponent results");
}
// Check for proper event function signatures
if (contents.Contains("void Start(") && !contents.Contains("void Start()"))
{
errors.Add("WARNING: Start() should not have parameters");
}
if (contents.Contains("void Update(") && !contents.Contains("void Update()"))
{
errors.Add("WARNING: Update() should not have parameters");
}
// Check for inefficient string operations
if (contents.Contains("Update()") && contents.Contains("\"") && contents.Contains("+"))
{
errors.Add("WARNING: String concatenation in Update() can cause garbage collection issues");
}
}
/// <summary>
/// Validates semantic rules and common coding issues
/// </summary>
private static void ValidateSemanticRules(string contents, System.Collections.Generic.List<string> errors)
{
// Check for potential memory leaks
if (contents.Contains("new ") && contents.Contains("Update()"))
{
errors.Add("WARNING: Creating objects in Update() may cause memory issues");
}
// Check for magic numbers
var magicNumberPattern = new Regex(@"\b\d+\.?\d*f?\b(?!\s*[;})\]])", RegexOptions.CultureInvariant, TimeSpan.FromSeconds(2));
var matches = magicNumberPattern.Matches(contents);
if (matches.Count > 5)
{
errors.Add("WARNING: Consider using named constants instead of magic numbers");
}
// Check for long methods (simple line count check)
var methodPattern = new Regex(@"(public|private|protected|internal)?\s*(static)?\s*\w+\s+\w+\s*\([^)]*\)\s*{", RegexOptions.CultureInvariant, TimeSpan.FromSeconds(2));
var methodMatches = methodPattern.Matches(contents);
foreach (Match match in methodMatches)
{
int startIndex = match.Index;
int braceCount = 0;
int lineCount = 0;
bool inMethod = false;
for (int i = startIndex; i < contents.Length; i++)
{
if (contents[i] == '{')
{
braceCount++;
inMethod = true;
}
else if (contents[i] == '}')
{
braceCount--;
if (braceCount == 0 && inMethod)
break;
}
else if (contents[i] == '\n' && inMethod)
{
lineCount++;
}
}
if (lineCount > 50)
{
errors.Add("WARNING: Method is very long, consider breaking it into smaller methods");
break; // Only report once
}
}
// Check for proper exception handling
if (contents.Contains("catch") && contents.Contains("catch()"))
{
errors.Add("WARNING: Empty catch blocks should be avoided");
}
// Check for proper async/await usage
if (contents.Contains("async ") && !contents.Contains("await"))
{
errors.Add("WARNING: Async method should contain await or return Task");
}
// Check for hardcoded tags and layers
if (contents.Contains("\"Player\"") || contents.Contains("\"Enemy\""))
{
errors.Add("WARNING: Consider using constants for tags instead of hardcoded strings");
}
}
//TODO: A easier way for users to update incorrect scripts (now duplicated with the updateScript method and need to also update server side, put aside for now)
/// <summary>
/// Public method to validate script syntax with configurable validation level
/// Returns detailed validation results including errors and warnings
/// </summary>
// public static object ValidateScript(JObject @params)
// {
// string contents = @params["contents"]?.ToString();
// string validationLevel = @params["validationLevel"]?.ToString() ?? "standard";
// if (string.IsNullOrEmpty(contents))
// {
// return new ErrorResponse("Contents parameter is required for validation.");
// }
// // Parse validation level
// ValidationLevel level = ValidationLevel.Standard;
// switch (validationLevel.ToLower())
// {
// case "basic": level = ValidationLevel.Basic; break;
// case "standard": level = ValidationLevel.Standard; break;
// case "comprehensive": level = ValidationLevel.Comprehensive; break;
// case "strict": level = ValidationLevel.Strict; break;
// default:
// return new ErrorResponse($"Invalid validation level: '{validationLevel}'. Valid levels are: basic, standard, comprehensive, strict.");
// }
// // Perform validation
// bool isValid = ValidateScriptSyntax(contents, level, out string[] validationErrors);
// var errors = validationErrors?.Where(e => e.StartsWith("ERROR:")).ToArray() ?? new string[0];
// var warnings = validationErrors?.Where(e => e.StartsWith("WARNING:")).ToArray() ?? new string[0];
// var result = new
// {
// isValid = isValid,
// validationLevel = validationLevel,
// errorCount = errors.Length,
// warningCount = warnings.Length,
// errors = errors,
// warnings = warnings,
// summary = isValid
// ? (warnings.Length > 0 ? $"Validation passed with {warnings.Length} warnings" : "Validation passed with no issues")
// : $"Validation failed with {errors.Length} errors and {warnings.Length} warnings"
// };
// if (isValid)
// {
// return new SuccessResponse("Script validation completed successfully.", result);
// }
// else
// {
// return new ErrorResponse("Script validation failed.", result);
// }
// }
}
// Debounced refresh/compile scheduler to coalesce bursts of edits
static class RefreshDebounce
{
private static int _pending;
private static readonly object _lock = new object();
private static readonly HashSet<string> _paths = new HashSet<string>(StringComparer.OrdinalIgnoreCase);
// The timestamp of the most recent schedule request.
private static DateTime _lastRequest;
// Guard to ensure we only have a single ticking callback running.
private static bool _scheduled;
public static void Schedule(string relPath, TimeSpan window)
{
// Record that work is pending and track the path in a threadsafe way.
Interlocked.Exchange(ref _pending, 1);
lock (_lock)
{
_paths.Add(relPath);
_lastRequest = DateTime.UtcNow;
// If a debounce timer is already scheduled it will pick up the new request.
if (_scheduled)
return;
_scheduled = true;
}
// Kick off a ticking callback that waits until the window has elapsed
// from the last request before performing the refresh.
EditorApplication.delayCall += () => Tick(window);
// Nudge the editor loop so ticks run even if the window is unfocused
EditorApplication.QueuePlayerLoopUpdate();
}
private static void Tick(TimeSpan window)
{
bool ready;
lock (_lock)
{
// Only proceed once the debounce window has fully elapsed.
ready = (DateTime.UtcNow - _lastRequest) >= window;
if (ready)
{
_scheduled = false;
}
}
if (!ready)
{
// Window has not yet elapsed; check again on the next editor tick.
EditorApplication.delayCall += () => Tick(window);
return;
}
if (Interlocked.Exchange(ref _pending, 0) == 1)
{
string[] toImport;
lock (_lock) { toImport = _paths.ToArray(); _paths.Clear(); }
foreach (var p in toImport)
{
var sp = ManageScriptRefreshHelpers.SanitizeAssetsPath(p);
AssetDatabase.ImportAsset(sp, ImportAssetOptions.ForceUpdate | ImportAssetOptions.ForceSynchronousImport);
}
#if UNITY_EDITOR
UnityEditor.Compilation.CompilationPipeline.RequestScriptCompilation();
#endif
// Fallback if needed:
// AssetDatabase.Refresh();
}
}
}
static class ManageScriptRefreshHelpers
{
public static string SanitizeAssetsPath(string p)
{
if (string.IsNullOrEmpty(p)) return p;
p = AssetPathUtility.NormalizeSeparators(p).Trim();
if (p.StartsWith("mcpforunity://path/", StringComparison.OrdinalIgnoreCase))
p = p.Substring("mcpforunity://path/".Length);
while (p.StartsWith("Assets/Assets/", StringComparison.OrdinalIgnoreCase))
p = p.Substring("Assets/".Length);
if (!p.StartsWith("Assets/", StringComparison.OrdinalIgnoreCase))
p = "Assets/" + p.TrimStart('/');
return p;
}
public static void ScheduleScriptRefresh(string relPath)
{
var sp = SanitizeAssetsPath(relPath);
RefreshDebounce.Schedule(sp, TimeSpan.FromMilliseconds(200));
}
public static void ImportAndRequestCompile(string relPath, bool synchronous = true)
{
var sp = SanitizeAssetsPath(relPath);
var opts = ImportAssetOptions.ForceUpdate;
if (synchronous) opts |= ImportAssetOptions.ForceSynchronousImport;
AssetDatabase.ImportAsset(sp, opts);
#if UNITY_EDITOR
UnityEditor.Compilation.CompilationPipeline.RequestScriptCompilation();
#endif
}
}
}
