feat: Rebuild MCP server installation with embedded source

Refactored the server repair logic to implement a full rebuild of the MCP server installation using the embedded source. The new RebuildMcpServer method now:
- Uses embedded server source instead of attempting repair of existing installation
- Deletes the entire existing server directory before re-copying
- Handles UV process cleanup for the target path
- Simplifies the installation flow by removing the complex Python environment repair logic
- Maintains the same installation behavior but with a cleaner, more reliable rebuild approach

This change improves reliability of server installations by ensuring a clean slate rebuild rather than attempting to repair potentially corrupted environments.

Author: msanatan

UnityMcpBridge/Editor/Helpers/ServerInstaller.cs

@@ -423,105 +423,61 @@ private static void CopyDirectoryRecursive(string sourceDir, string destinationD
             }
         }
 
-        public static bool RepairPythonEnvironment()
+        public static bool RebuildMcpServer()
         {
             try
             {
-                string serverSrc = GetServerPath();
-                bool hasServer = File.Exists(Path.Combine(serverSrc, "server.py"));
-                if (!hasServer)
-                {
-                    // In dev mode or if not installed yet, try the embedded/dev source
-                    if (TryGetEmbeddedServerSource(out string embeddedSrc) && File.Exists(Path.Combine(embeddedSrc, "server.py")))
-                    {
-                        serverSrc = embeddedSrc;
-                        hasServer = true;
-                    }
-                    else
-                    {
-                        // Attempt to install then retry
-                        EnsureServerInstalled();
-                        serverSrc = GetServerPath();
-                        hasServer = File.Exists(Path.Combine(serverSrc, "server.py"));
-                    }
-                }
-
-                if (!hasServer)
-                {
-                    Debug.LogWarning("RepairPythonEnvironment: server.py not found; ensure server is installed first.");
-                    return false;
-                }
-
-                // Remove stale venv and pinned version file if present
-                string venvPath = Path.Combine(serverSrc, ".venv");
-                if (Directory.Exists(venvPath))
-                {
-                    try { Directory.Delete(venvPath, recursive: true); } catch (Exception ex) { Debug.LogWarning($"Failed to delete .venv: {ex.Message}"); }
-                }
-                string pyPin = Path.Combine(serverSrc, ".python-version");
-                if (File.Exists(pyPin))
-                {
-                    try { File.Delete(pyPin); } catch (Exception ex) { Debug.LogWarning($"Failed to delete .python-version: {ex.Message}"); }
-                }
-
-                string uvPath = FindUvPath();
-                if (uvPath == null)
+                // Find embedded source
+                if (!TryGetEmbeddedServerSource(out string embeddedSrc))
                 {
-                    Debug.LogError("UV not found. Please install uv (https://docs.astral.sh/uv/).");
+                    Debug.LogError("RebuildMcpServer: Could not find embedded server source.");
                     return false;
                 }
 
-                var psi = new System.Diagnostics.ProcessStartInfo
-                {
-                    FileName = uvPath,
-                    Arguments = "sync",
-                    WorkingDirectory = serverSrc,
-                    UseShellExecute = false,
-                    RedirectStandardOutput = true,
-                    RedirectStandardError = true,
-                    CreateNoWindow = true
-                };
+                string saveLocation = GetSaveLocation();
+                string destRoot = Path.Combine(saveLocation, ServerFolder);
+                string destSrc = Path.Combine(destRoot, "src");
 
-                using var proc = new System.Diagnostics.Process { StartInfo = psi };
-                var sbOut = new StringBuilder();
-                var sbErr = new StringBuilder();
-                proc.OutputDataReceived += (_, e) => { if (e.Data != null) sbOut.AppendLine(e.Data); };
-                proc.ErrorDataReceived += (_, e) => { if (e.Data != null) sbErr.AppendLine(e.Data); };
+                // Kill any running uv processes for this server
+                TryKillUvForPath(destSrc);
 
-                if (!proc.Start())
+                // Delete the entire installed server directory
+                if (Directory.Exists(destRoot))
                 {
-                    Debug.LogError("Failed to start uv process.");
-                    return false;
+                    try
+                    {
+                        Directory.Delete(destRoot, recursive: true);
+                        Debug.Log($"<b><color=#2EA3FF>MCP-FOR-UNITY</color></b>: Deleted existing server at {destRoot}");
+                    }
+                    catch (Exception ex)
+                    {
+                        Debug.LogError($"Failed to delete existing server: {ex.Message}");
+                        return false;
+                    }
                 }
 
-                proc.BeginOutputReadLine();
-                proc.BeginErrorReadLine();
+                // Re-copy from embedded source
+                string embeddedRoot = Path.GetDirectoryName(embeddedSrc) ?? embeddedSrc;
+                Directory.CreateDirectory(destRoot);
+                CopyDirectoryRecursive(embeddedRoot, destRoot);
 
-                if (!proc.WaitForExit(60000))
+                // Write version file
+                string embeddedVer = ReadVersionFile(Path.Combine(embeddedSrc, VersionFileName)) ?? "unknown";
+                try
                 {
-                    try { proc.Kill(); } catch { }
-                    Debug.LogError("uv sync timed out.");
-                    return false;
+                    File.WriteAllText(Path.Combine(destSrc, VersionFileName), embeddedVer);
                 }
-
-                // Ensure async buffers flushed
-                proc.WaitForExit();
-
-                string stdout = sbOut.ToString();
-                string stderr = sbErr.ToString();
-
-                if (proc.ExitCode != 0)
+                catch (Exception ex)
                 {
-                    Debug.LogError($"uv sync failed: {stderr}\n{stdout}");
-                    return false;
+                    Debug.LogWarning($"Failed to write version file: {ex.Message}");
                 }
 
-                Debug.Log("<b><color=#2EA3FF>MCP-FOR-UNITY</color></b>: Python environment repaired successfully.");
+                Debug.Log($"<b><color=#2EA3FF>MCP-FOR-UNITY</color></b>: Server rebuilt successfully at {destRoot} (version {embeddedVer})");
                 return true;
             }
             catch (Exception ex)
             {
-                Debug.LogError($"RepairPythonEnvironment failed: {ex.Message}");
+                Debug.LogError($"RebuildMcpServer failed: {ex.Message}");
                 return false;
             }
         }

UnityMcpBridge/Editor/Windows/MCPForUnityEditorWindow.cs

@@ -368,25 +368,25 @@ private void DrawServerStatusSection()
             }
             EditorGUILayout.Space(4);
 
-            // Repair Python Env button with tooltip tag
+            // Rebuild MCP Server button with tooltip tag
             using (new EditorGUILayout.HorizontalScope())
             {
                 GUILayout.FlexibleSpace();
                 GUIContent repairLabel = new GUIContent(
-                    "Repair Python Env",
-                    "Deletes the server's .venv and runs 'uv sync' to rebuild a clean environment. Use this if modules are missing or Python upgraded."
+                    "Rebuild MCP Server",
+                    "Deletes the installed server and re-copies it from the package. Use this to update the server after making source code changes or if the installation is corrupted."
                 );
                 if (GUILayout.Button(repairLabel, GUILayout.Width(160), GUILayout.Height(22)))
                 {
-                    bool ok = global::MCPForUnity.Editor.Helpers.ServerInstaller.RepairPythonEnvironment();
+                    bool ok = global::MCPForUnity.Editor.Helpers.ServerInstaller.RebuildMcpServer();
                     if (ok)
                     {
-                        EditorUtility.DisplayDialog("MCP For Unity", "Python environment repaired.", "OK");
+                        EditorUtility.DisplayDialog("MCP For Unity", "Server rebuilt successfully.", "OK");
                         UpdatePythonServerInstallationStatus();
                     }
                     else
                     {
-                        EditorUtility.DisplayDialog("MCP For Unity", "Repair failed. Please check Console for details.", "OK");
+                        EditorUtility.DisplayDialog("MCP For Unity", "Rebuild failed. Please check Console for details.", "OK");
                     }
                 }
             }

UnityMcpBridge/README.md

@@ -29,7 +29,7 @@ The window has four areas: Server Status, Unity Bridge, MCP Client Configuration
   - Ports: Unity (varies; shown in UI), MCP 6500.
 - Actions:
   - Auto-Setup: Registers/updates your selected MCP client(s), ensures bridge connectivity. Shows “Connected ✓” after success.
-  - Repair Python Env: Rebuilds a clean Python environment (deletes `.venv`, runs `uv sync`).
+  - Rebuild MCP Server: Rebuilds the Python based MCP server
   - Select server folder…: Choose the folder containing `server.py`.
   - Verify again: Re-checks server presence.
   - If Python isn’t detected, use “Open Install Instructions”.