feat: add Godot C# compatibility and shared codegen weaving

2026-09-05 02:37:43 +08:00
parent 60cf457230
commit 88ba3cdc48
95 changed files with 2712 additions and 19 deletions
@@ -0,0 +1,133 @@
+#nullable enable
+
+using System;
+using System.IO;
+using System.Linq;
+using System.Text.RegularExpressions;
+using Godot;
+using ShrinkSDK.Installer;
+
+namespace ShrinkSDK.Godot.Editor;
+
+[Tool]
+public partial class ShrinkGodotInstallerPlugin : EditorPlugin
+{
+    private EditorDock? _panel;
+    private Label? _status;
+    private RichTextLabel? _output;
+    private readonly System.Collections.Generic.Dictionary<string, Button> _installActions =
+        new(StringComparer.OrdinalIgnoreCase);
+    private readonly System.Collections.Generic.Dictionary<string, Button> _removeActions =
+        new(StringComparer.OrdinalIgnoreCase);
+    private string ProjectDirectory => ProjectSettings.GlobalizePath("res://");
+    private string ProjectPath => Directory.GetFiles(ProjectDirectory, "*.csproj", SearchOption.TopDirectoryOnly).Single();
+
+    public override void _EnterTree()
+    {
+        _panel = new EditorDock { Name = "ShrinkSDK", Title = "ShrinkSDK", DefaultSlot = EditorDock.DockSlot.LeftBr };
+        var content = new VBoxContainer();
+        _panel.AddChild(content);
+        content.AddChild(new Label { Text = "ShrinkSDK Packages" });
+        _installActions.Clear();
+        _removeActions.Clear();
+        foreach (var package in ShrinkProjectPackageEditor.RecommendedPackages)
+        {
+            var row = new HBoxContainer();
+            row.AddChild(new Label { Text = $"{package.Key}  {package.Value}", SizeFlagsHorizontal = Control.SizeFlags.ExpandFill });
+            var install = new Button { Text = "Install / Update" };
+            install.Pressed += () => ChangePackage(package.Key, package.Value);
+            row.AddChild(install);
+            _installActions[package.Key] = install;
+            var remove = new Button { Text = "Remove" };
+            remove.Pressed += () => ChangePackage(package.Key, null);
+            row.AddChild(remove);
+            _removeActions[package.Key] = remove;
+            content.AddChild(row);
+        }
+        var restore = new Button { Text = "Restore and Build" };
+        restore.Pressed += BuildProject;
+        content.AddChild(restore);
+        _status = new Label { Text = "CodeGen: not built" };
+        content.AddChild(_status);
+        _output = new RichTextLabel { FitContent = true, CustomMinimumSize = new Vector2(360, 180) };
+        content.AddChild(_output);
+        AddDock(_panel);
+        RefreshInstalledState();
+    }
+
+    public override void _ExitTree()
+    {
+        if (_panel == null) return;
+        RemoveDock(_panel);
+        _panel.QueueFree();
+        _panel = null;
+    }
+
+    private async void ChangePackage(string packageId, string? version)
+    {
+        try
+        {
+            ShrinkProjectPackageEditor.EnsureShrinkFeed(Path.Combine(ProjectDirectory, "NuGet.Config"));
+            ShrinkProjectPackageEditor.SetPackageReference(ProjectPath, packageId, version);
+            await RunBuild("restore");
+        }
+        catch (Exception exception) { ShowFailure(exception); }
+    }
+
+    private async void BuildProject()
+    {
+        try { await RunBuild("build"); }
+        catch (Exception exception) { ShowFailure(exception); }
+    }
+
+    private async System.Threading.Tasks.Task RunBuild(string command)
+    {
+        _status!.Text = $"CodeGen: running dotnet {command}";
+        var result = await ShrinkProjectPackageEditor.RunDotnetAsync(ProjectDirectory, $"{command} \"{ProjectPath}\" /nr:false");
+        _output!.Text = result.Output;
+        var match = Regex.Matches(result.Output, @"\[ShrinkSDK\.CodeGen\].*woven: instance=(\d+), static=(\d+), registry=(\d+)").Cast<Match>().LastOrDefault();
+        _status.Text = result.ExitCode == 0
+            ? match == null ? "CodeGen: build succeeded (no attributed registrations)" : $"CodeGen: woven; instance={match.Groups[1].Value}, static={match.Groups[2].Value}, registry={match.Groups[3].Value}"
+            : "CodeGen: build failed";
+        RefreshInstalledState();
+    }
+
+    private void RefreshInstalledState()
+    {
+        if (_output == null) return;
+        var installed = File.Exists(ProjectPath)
+            ? ShrinkProjectPackageEditor.ReadPackageReferences(ProjectPath)
+            : new System.Collections.Generic.Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
+        foreach (var package in ShrinkProjectPackageEditor.RecommendedPackages)
+        {
+            var hasInstalled = installed.TryGetValue(package.Key, out var installedVersion);
+            var relation = hasInstalled
+                ? ShrinkPackageVersion.Classify(installedVersion, package.Value)
+                : ShrinkPackageVersionRelation.Unknown;
+            if (_installActions.TryGetValue(package.Key, out var install))
+            {
+                install.Text = !hasInstalled
+                    ? $"Install {package.Value}"
+                    : relation switch
+                    {
+                        ShrinkPackageVersionRelation.Equal => $"Installed {installedVersion}",
+                        ShrinkPackageVersionRelation.Newer => $"Newer {installedVersion}",
+                        ShrinkPackageVersionRelation.Older => $"Upgrade {installedVersion} → {package.Value}",
+                        _ => $"Update {installedVersion} → {package.Value}"
+                    };
+                install.Disabled = hasInstalled && relation is
+                    ShrinkPackageVersionRelation.Equal or ShrinkPackageVersionRelation.Newer;
+            }
+            if (_removeActions.TryGetValue(package.Key, out var remove))
+                remove.Disabled = !hasInstalled;
+        }
+        _output.Text = "Direct packages: " + string.Join(", ", installed.Select(item => $"{item.Key}@{item.Value}"));
+    }
+
+    private void ShowFailure(Exception exception)
+    {
+        _status!.Text = "CodeGen: operation failed";
+        _output!.Text = exception.ToString();
+        GD.PushError(exception.ToString());
+    }
+}