升级XR插件版本
This commit is contained in:
Sora丶kong
304
Packages/MCPForUnity/Editor/Services/PackageDeploymentService.cs
Normal file
304
Packages/MCPForUnity/Editor/Services/PackageDeploymentService.cs
Normal file
@@ -0,0 +1,304 @@
|
||||
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();
|
||||
if (string.IsNullOrEmpty(target))
|
||||
return "Not found (check Packages/manifest.json)";
|
||||
// Use forward slashes to avoid backslash escape sequence issues in UI text
|
||||
return target.Replace('\\', '/');
|
||||
}
|
||||
|
||||
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
|
||||
};
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user