Skip to content

Align script management responses and validation #12

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Aug 17, 2025
Merged

Align script management responses and validation #12

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
32 changes: 23 additions & 9 deletions UnityMcpBridge/Editor/Tools/ManageScript.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -202,7 +202,7 @@ public static object HandleCommand(JObject @params)
}
case "validate":
{
string level = @params["level"]?.ToString()?.ToLowerInvariant() ?? "basic";
string level = @params["level"]?.ToString()?.ToLowerInvariant() ?? "standard";
var chosen = level switch
{
"basic" => ValidationLevel.Basic,
Expand Down Expand Up @@ -343,8 +343,10 @@ private static object ReadScript(string fullPath, string relativePath)

// Return both normal and encoded contents for larger files
bool isLarge = contents.Length > 10000; // If content is large, include encoded version
var uri = $"unity://path/{relativePath}";
var responseData = new
{
uri,
path = relativePath,
contents = contents,
// For large files, also include base64-encoded version
Expand Down Expand Up @@ -420,9 +422,10 @@ string contents
}

// Prepare success response BEFORE any operation that can trigger a domain reload
var uri = $"unity://path/{relativePath}";
var ok = Response.Success(
$"Script '{name}.cs' updated successfully at '{relativePath}'.",
new { path = relativePath, scheduledRefresh = true }
new { uri, path = relativePath, scheduledRefresh = true }
);

// Schedule a debounced import/compile on next editor tick to avoid stalling the reply
Expand Down Expand Up @@ -1523,11 +1526,14 @@ private static bool ValidateScriptSyntax(string contents, ValidationLevel level,
}

#if USE_ROSLYN
// Advanced Roslyn-based validation
if (!ValidateScriptSyntaxRoslyn(contents, level, errorList))
// Advanced Roslyn-based validation: only run for Standard+; fail on Roslyn errors
if (level >= ValidationLevel.Standard)
{
errors = errorList.ToArray();
return level != ValidationLevel.Standard; //TODO: Allow standard to run roslyn right now, might formalize it in the future
if (!ValidateScriptSyntaxRoslyn(contents, level, errorList))
{
errors = errorList.ToArray();
return false;
}
}
#endif

Expand Down Expand Up @@ -2105,20 +2111,28 @@ static class RefreshDebounce
{
private static int _pending;
private static DateTime _last;
private static readonly object _lock = new object();
private static readonly HashSet<string> _paths = new HashSet<string>(StringComparer.OrdinalIgnoreCase);
private static bool _scheduled;

public static void Schedule(string relPath, TimeSpan window)
{
Interlocked.Exchange(ref _pending, 1);
lock (_lock) { _paths.Add(relPath); }
var now = DateTime.UtcNow;
if ((now - _last) < window) return;
if (_scheduled && (now - _last) < window) return;
_last = now;
_scheduled = true;

EditorApplication.delayCall += () =>
{
_scheduled = false;
if (Interlocked.Exchange(ref _pending, 0) == 1)
{
// Prefer targeted import and script compile over full refresh
AssetDatabase.ImportAsset(relPath, ImportAssetOptions.ForceUpdate);
string[] toImport;
lock (_lock) { toImport = _paths.ToArray(); _paths.Clear(); }
foreach (var p in toImport)
AssetDatabase.ImportAsset(p, ImportAssetOptions.ForceUpdate);
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Bug: Race Condition in RefreshDebounce Method

The RefreshDebounce.Schedule method has a race condition where the _scheduled flag is accessed and modified without proper synchronization. This allows multiple threads to simultaneously pass the scheduling check, leading to redundant EditorApplication.delayCall callbacks being scheduled unnecessarily, which creates inefficiency.

Fix in Cursor Fix in Web

#if UNITY_EDITOR
UnityEditor.Compilation.CompilationPipeline.RequestScriptCompilation();
#endif
Expand Down