ArabDesert/Packages/com.vrchat.core.vpm-resolver/Editor/Resolver/ResolverWindow.cs

292 lines
10 KiB
C#

using System.Collections.Generic;
using System.Text;
using System.Threading.Tasks;
using UnityEditor;
using UnityEditor.UIElements;
using UnityEngine;
using UnityEngine.UIElements;
using VRC.PackageManagement.Core;
using VRC.PackageManagement.Core.Types;
using VRC.PackageManagement.Core.Types.Packages;
using Version = VRC.PackageManagement.Core.Types.VPMVersion.Version;
namespace VRC.PackageManagement.Resolver
{
public class ResolverWindow : EditorWindow
{
// VisualElements
private static VisualElement _rootView;
private static Button _refreshButton;
private static Button _createButton;
private static Button _resolveButton;
private static Box _manifestInfo;
private static Label _manifestLabel;
private static bool _isUpdating;
private static Color _colorPositive = Color.green;
private static Color _colorNegative = new Color(1, 0.3f, 0.3f);
[MenuItem("VRChat SDK/Utilities/Package Resolver")]
public static void ShowWindow()
{
ResolverWindow wnd = GetWindow<ResolverWindow>();
wnd.titleContent = new GUIContent("Package Resolver");
}
public static void Refresh()
{
if (_rootView == null || string.IsNullOrWhiteSpace(Resolver.ProjectDir)) return;
_manifestInfo.SetEnabled(!_isUpdating);
_refreshButton.SetEnabled(!_isUpdating);
_manifestLabel.text = (_isUpdating ? "Working ..." : "Required Packages");
_manifestInfo.Clear();
_manifestInfo.Add(_manifestLabel);
bool needsResolve = VPMProjectManifest.ResolveIsNeeded(Resolver.ProjectDir);
string resolveStatus = needsResolve ? "Please press \"Resolve\" to Download them." : "All of them are in the project.";
// check for vpm dependencies
if (!Resolver.VPMManifestExists())
{
TextElement noManifestText = new TextElement();
noManifestText.text = "No VPM Manifest";
noManifestText.style.color = _colorNegative;
_manifestInfo.Add(noManifestText);
}
else
{
var manifest = VPMProjectManifest.Load(Resolver.ProjectDir);
var project = new UnityProject(Resolver.ProjectDir);
// Here is where we detect if all dependencies are installed
var allDependencies = (manifest.locked != null && manifest.locked.Count > 0)
? manifest.locked
: manifest.dependencies;
foreach (var pair in allDependencies)
{
var id = pair.Key;
var version = pair.Value.version;
IVRCPackage package = project.VPMProvider.GetPackage(id, version);
_manifestInfo.Add(CreateDependencyRow(id, version, project, (package != null)));
}
}
_resolveButton.SetEnabled(needsResolve);
Resolver.ForceRefresh();
}
/// <summary>
/// Unity calls the CreateGUI method automatically when the window needs to display
/// </summary>
private void CreateGUI()
{
_rootView = rootVisualElement;
_rootView.name = "root-view";
_rootView.styleSheets.Add((StyleSheet)Resources.Load("ResolverWindowStyle"));
// Main Container
var container = new Box()
{
name = "buttons"
};
_rootView.Add(container);
// Create Button
if (!Resolver.VPMManifestExists())
{
_createButton = new Button(Resolver.CreateManifest)
{
text = "Create",
name = "create-button-base"
};
container.Add(_createButton);
}
else
{
_resolveButton = new Button(Resolver.ResolveManifest)
{
text = "Resolve All",
name = "resolve-button-base"
};
container.Add(_resolveButton);
}
// Manifest Info
_manifestInfo = new Box()
{
name = "manifest-info",
};
_manifestLabel = (new Label("Required Packages") { name = "manifest-header" });
_rootView.Add(_manifestInfo);
// Refresh Button
var refreshBox = new Box();
_refreshButton = new Button(Refresh)
{
text = "Refresh",
name = "refresh-button-base"
};
refreshBox.Add(_refreshButton);
_rootView.Add(refreshBox);
Refresh();
}
private static VisualElement CreateDependencyRow(string id, string version, UnityProject project, bool havePackage)
{
// Table
VisualElement row = new Box() { name = "package-box" };
VisualElement column1 = new Box() { name = "package-box" };
VisualElement column2 = new Box() { name = "package-box" };
VisualElement column3 = new Box() { name = "package-box" };
VisualElement column4 = new Box() { name = "package-box" };
column1.style.minWidth = 200;
column2.style.minWidth = 100;
column3.style.minWidth = 100;
column4.style.minWidth = 100;
row.Add(column1);
row.Add(column2);
row.Add(column3);
row.Add(column4);
// Package Name + Status
TextElement text = new TextElement { text = $"{id} {version} " };
column1.Add(text);
if (!havePackage)
{
TextElement missingText = new TextElement { text = "MISSING" };
missingText.style.color = _colorNegative;
missingText.style.display = (_isUpdating ? DisplayStyle.None : DisplayStyle.Flex);
column2.Add(missingText);
}
// Version Popup
var choices = new List<string>();
foreach (string n in Resolver.GetAllVersionsOf(id))
{
choices.Add(n);
}
var popupField = new PopupField<string>(choices, 0);
popupField.value = choices[0];
popupField.style.display = (_isUpdating ? DisplayStyle.None : DisplayStyle.Flex);
column3.Add(popupField);
// Button
Button updateButton = new Button() { text = "Update" };
if (havePackage)
RefreshUpdateButton(updateButton, version, choices[0]);
else
RefreshMissingButton(updateButton);
updateButton.clicked += (() =>
{
IVRCPackage package = Repos.GetPackageWithVersionMatch(id, popupField.value);
// Check and warn on Dependencies if Updating or Downgrading
if (Version.TryParse(version, out var currentVersion) &&
Version.TryParse(popupField.value, out var newVersion))
{
Dictionary<string, string> dependencies = new Dictionary<string, string>();
StringBuilder dialogMsg = new StringBuilder();
List<string> affectedPackages = Resolver.GetAffectedPackageList(package);
for (int v = 0; v < affectedPackages.Count; v++)
{
dialogMsg.Append(affectedPackages[v]);
}
if (affectedPackages.Count > 1)
{
dialogMsg.Insert(0, "This will update multiple packages:\n\n");
dialogMsg.AppendLine("\nAre you sure?");
if (EditorUtility.DisplayDialog("Package Has Dependencies", dialogMsg.ToString(), "OK", "Cancel"))
OnUpdatePackageClicked(project, package);
}
else
{
OnUpdatePackageClicked(project, package);
}
}
});
column4.Add(updateButton);
popupField.RegisterCallback<ChangeEvent<string>>((evt) =>
{
if (havePackage)
RefreshUpdateButton(updateButton, version, evt.newValue);
else
RefreshMissingButton(updateButton);
});
return row;
}
private static void RefreshUpdateButton(Button button, string currentVersion, string highestAvailableVersion)
{
if (currentVersion == highestAvailableVersion)
{
button.style.display = DisplayStyle.None;
}
else
{
button.style.display = (_isUpdating ? DisplayStyle.None : DisplayStyle.Flex);
if (Version.TryParse(currentVersion, out var currentVersionObject) &&
Version.TryParse(highestAvailableVersion, out var highestAvailableVersionObject))
{
if (currentVersionObject < highestAvailableVersionObject)
{
SetButtonColor(button, _colorPositive);
button.text = "Update";
}
else
{
SetButtonColor(button, _colorNegative);
button.text = "Downgrade";
}
}
}
}
private static void RefreshMissingButton(Button button)
{
button.text = "Resolve";
SetButtonColor(button, Color.white);
button.style.display = (_isUpdating ? DisplayStyle.None : DisplayStyle.Flex);
}
private static void SetButtonColor(Button button, Color color)
{
button.style.color = color;
color.a = 0.25f;
button.style.borderRightColor =
button.style.borderLeftColor =
button.style.borderTopColor =
button.style.borderBottomColor =
color;
}
private static async void OnUpdatePackageClicked(UnityProject project, IVRCPackage package)
{
_isUpdating = true;
Refresh();
await Task.Delay(500);
await Task.Run(() => project.UpdateVPMPackage(package));
_isUpdating = false;
Refresh();
}
}
}