Unity MCP Server

using System; using System.IO; using MCPForUnity.Editor.Constants; using MCPForUnity.Editor.Helpers; using UnityEditor; using UnityEngine; using PackageInfo = UnityEditor.PackageManager.PackageInfo; namespace MCPForUnity.Editor.Services { /// <summary> /// Handles copying a local MCPForUnity folder into the current project's package location with backup/restore support. /// </summary> public class PackageDeploymentService : IPackageDeploymentService { private const string BackupRootFolderName = "MCPForUnityDeployBackups"; public string GetStoredSourcePath() { return EditorPrefs.GetString(EditorPrefKeys.PackageDeploySourcePath, string.Empty); } public void SetStoredSourcePath(string path) { ValidateSource(path); EditorPrefs.SetString(EditorPrefKeys.PackageDeploySourcePath, Path.GetFullPath(path)); } public void ClearStoredSourcePath() { EditorPrefs.DeleteKey(EditorPrefKeys.PackageDeploySourcePath); } public string GetTargetPath() { // Prefer Package Manager resolved path for the installed package var packageInfo = PackageInfo.FindForAssembly(typeof(PackageDeploymentService).Assembly); if (packageInfo != null) { if (!string.IsNullOrEmpty(packageInfo.resolvedPath) && Directory.Exists(packageInfo.resolvedPath)) { return packageInfo.resolvedPath; } if (!string.IsNullOrEmpty(packageInfo.assetPath)) { string absoluteFromAsset = MakeAbsolute(packageInfo.assetPath); if (Directory.Exists(absoluteFromAsset)) { return absoluteFromAsset; } } } // Fallback to computed package root string packageRoot = AssetPathUtility.GetMcpPackageRootPath(); if (!string.IsNullOrEmpty(packageRoot)) { string absolutePath = MakeAbsolute(packageRoot); if (Directory.Exists(absolutePath)) { return absolutePath; } } return null; } public string GetTargetDisplayPath() { string target = GetTargetPath(); return string.IsNullOrEmpty(target) ? "Not found (check Packages/manifest.json)" : target; } public string GetLastBackupPath() { return EditorPrefs.GetString(EditorPrefKeys.PackageDeployLastBackupPath, string.Empty); } public bool HasBackup() { string path = GetLastBackupPath(); return !string.IsNullOrEmpty(path) && Directory.Exists(path); } public PackageDeploymentResult DeployFromStoredSource() { string sourcePath = GetStoredSourcePath(); if (string.IsNullOrEmpty(sourcePath)) { return Fail("Select a MCPForUnity folder first."); } string validationError = ValidateSource(sourcePath, throwOnError: false); if (!string.IsNullOrEmpty(validationError)) { return Fail(validationError); } string targetPath = GetTargetPath(); if (string.IsNullOrEmpty(targetPath)) { return Fail("Could not locate the installed MCP package. Check Packages/manifest.json."); } if (PathsEqual(sourcePath, targetPath)) { return Fail("Source and target are the same. Choose a different MCPForUnity folder."); } try { EditorUtility.DisplayProgressBar("Deploy MCP for Unity", "Creating backup...", 0.25f); string backupPath = CreateBackup(targetPath); EditorUtility.DisplayProgressBar("Deploy MCP for Unity", "Replacing package contents...", 0.7f); CopyCoreFolders(sourcePath, targetPath); EditorPrefs.SetString(EditorPrefKeys.PackageDeployLastBackupPath, backupPath); EditorPrefs.SetString(EditorPrefKeys.PackageDeployLastTargetPath, targetPath); EditorPrefs.SetString(EditorPrefKeys.PackageDeployLastSourcePath, sourcePath); AssetDatabase.Refresh(ImportAssetOptions.ForceUpdate); return Success("Deployment completed.", sourcePath, targetPath, backupPath); } catch (Exception ex) { McpLog.Error($"Deployment failed: {ex.Message}"); return Fail($"Deployment failed: {ex.Message}"); } finally { EditorUtility.ClearProgressBar(); } } public PackageDeploymentResult RestoreLastBackup() { string backupPath = GetLastBackupPath(); string targetPath = EditorPrefs.GetString(EditorPrefKeys.PackageDeployLastTargetPath, string.Empty); if (string.IsNullOrEmpty(backupPath) || !Directory.Exists(backupPath)) { return Fail("No backup available to restore."); } if (string.IsNullOrEmpty(targetPath) || !Directory.Exists(targetPath)) { targetPath = GetTargetPath(); } if (string.IsNullOrEmpty(targetPath) || !Directory.Exists(targetPath)) { return Fail("Could not locate target package path."); } try { EditorUtility.DisplayProgressBar("Restore MCP for Unity", "Restoring backup...", 0.5f); ReplaceDirectory(backupPath, targetPath); AssetDatabase.Refresh(ImportAssetOptions.ForceUpdate); return Success("Restore completed.", null, targetPath, backupPath); } catch (Exception ex) { McpLog.Error($"Restore failed: {ex.Message}"); return Fail($"Restore failed: {ex.Message}"); } finally { EditorUtility.ClearProgressBar(); } } private void CopyCoreFolders(string sourceRoot, string targetRoot) { string sourceEditor = Path.Combine(sourceRoot, "Editor"); string sourceRuntime = Path.Combine(sourceRoot, "Runtime"); ReplaceDirectory(sourceEditor, Path.Combine(targetRoot, "Editor")); ReplaceDirectory(sourceRuntime, Path.Combine(targetRoot, "Runtime")); } private static void ReplaceDirectory(string source, string destination) { if (Directory.Exists(destination)) { FileUtil.DeleteFileOrDirectory(destination); } FileUtil.CopyFileOrDirectory(source, destination); } private string CreateBackup(string targetPath) { string backupRoot = Path.Combine(GetProjectRoot(), "Library", BackupRootFolderName); Directory.CreateDirectory(backupRoot); string stamp = DateTime.Now.ToString("yyyyMMdd_HHmmss"); string backupPath = Path.Combine(backupRoot, $"backup_{stamp}"); if (Directory.Exists(backupPath)) { FileUtil.DeleteFileOrDirectory(backupPath); } FileUtil.CopyFileOrDirectory(targetPath, backupPath); return backupPath; } private static string ValidateSource(string sourcePath, bool throwOnError = true) { if (string.IsNullOrEmpty(sourcePath)) { if (throwOnError) { throw new ArgumentException("Source path cannot be empty."); } return "Source path is empty."; } if (!Directory.Exists(sourcePath)) { if (throwOnError) { throw new ArgumentException("Selected folder does not exist."); } return "Selected folder does not exist."; } bool hasEditor = Directory.Exists(Path.Combine(sourcePath, "Editor")); bool hasRuntime = Directory.Exists(Path.Combine(sourcePath, "Runtime")); if (!hasEditor || !hasRuntime) { string message = "Folder must contain Editor and Runtime subfolders."; if (throwOnError) { throw new ArgumentException(message); } return message; } return null; } private static string MakeAbsolute(string assetPath) { assetPath = assetPath.Replace('\\', '/'); if (assetPath.StartsWith("Assets/", StringComparison.OrdinalIgnoreCase)) { return Path.GetFullPath(Path.Combine(Application.dataPath, "..", assetPath)); } if (assetPath.StartsWith("Packages/", StringComparison.OrdinalIgnoreCase)) { return Path.GetFullPath(Path.Combine(Application.dataPath, "..", assetPath)); } return Path.GetFullPath(assetPath); } private static string GetProjectRoot() { return Path.GetFullPath(Path.Combine(Application.dataPath, "..")); } private static bool PathsEqual(string a, string b) { string normA = Path.GetFullPath(a).TrimEnd(Path.DirectorySeparatorChar, Path.AltDirectorySeparatorChar); string normB = Path.GetFullPath(b).TrimEnd(Path.DirectorySeparatorChar, Path.AltDirectorySeparatorChar); return string.Equals(normA, normB, StringComparison.OrdinalIgnoreCase); } private static PackageDeploymentResult Success(string message, string source, string target, string backup) { return new PackageDeploymentResult { Success = true, Message = message, SourcePath = source, TargetPath = target, BackupPath = backup }; } private static PackageDeploymentResult Fail(string message) { return new PackageDeploymentResult { Success = false, Message = message }; } } }