addons: implement automatic class exporter

2 files changed, 101 insertions, 0 deletions

diff --git a/addons/ClassExporter/Plugin.cs b/addons/ClassExporter/Plugin.cs
new file mode 100644
index 0000000..72df4bf
--- /dev/null
+++ b/addons/ClassExporter/Plugin.cs
@@ -0,0 +1,94 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Reflection;
+using Godot;
+
+#if TOOLS
+
+namespace Texty.addons.ClassExporter
+{
+    [Tool]
+    public class Plugin : EditorPlugin
+    {
+        private const string LoadTypesMenuItem = "Load types";
+        private const string ReloadTypesMenuItem = "Reload types";
+        private readonly List<Type> _types = new List<Type>();
+
+        public override void _EnterTree()
+        {
+            Connect("resource_saved", this, nameof(OnResourceSaved));
+            AddToolMenuItem(LoadTypesMenuItem, this, nameof(OnReloadTypes), false);
+            AddToolMenuItem(ReloadTypesMenuItem, this, nameof(OnReloadTypes), true);
+            RegisterTypes();
+        }
+
+        public override void _ExitTree()
+        {
+            DeRegisterTypes();
+            RemoveToolMenuItem(LoadTypesMenuItem);
+            RemoveToolMenuItem(ReloadTypesMenuItem);
+            Disconnect("resource_saved", this, nameof(OnResourceSaved));
+        }
+
+        // ReSharper disable once UnusedParameter.Local
+        private void OnResourceSaved(Resource resource)
+        {
+            RegisterTypes();
+        }
+
+        private void OnReloadTypes(bool reload)
+        {
+            if (reload) DeRegisterTypes();
+            RegisterTypes();
+        }
+
+        private void RegisterTypes()
+        {
+            var assembly = Assembly.GetExecutingAssembly();
+            var newTypes = from type in assembly.GetTypes()
+                where type.IsSubclassOf(typeof(Resource)) || type.IsSubclassOf(typeof(Node))
+                where !type.IsSubclassOf(typeof(EditorPlugin))
+                where !_types.Contains(type)
+                select new RegistrableType(type);
+
+            foreach (var type in newTypes)
+            {
+                GD.Print($"Exporting class {type.Name}: {type.Script.ResourcePath}");
+                AddCustomType(type.Name, type.Base, type.Script, null);
+                _types.Add(type.Type);
+            }
+        }
+
+        private void DeRegisterTypes()
+        {
+            _types.ForEach(type => RemoveCustomType(type.Name));
+            _types.Clear();
+        }
+
+        private readonly struct RegistrableType
+        {
+            public RegistrableType(Type type)
+            {
+                Type = type;
+            }
+
+            public readonly Type Type;
+
+            public CSharpScript Script
+            {
+                get
+                {
+                    var components = (Type.FullName ?? "").Split('.');
+                    var path = string.Join("/", components.Skip(1)) + ".cs";
+                    return ResourceLoader.Load<CSharpScript>(path);
+                }
+            }
+
+            public string Name => Type.Name;
+            public string Base => Type.BaseType?.Name ?? "";
+        }
+    }
+}
+
+#endif
\ No newline at end of file
diff --git a/addons/ClassExporter/plugin.cfg b/addons/ClassExporter/plugin.cfg
new file mode 100644
index 0000000..bc7c3b7
--- /dev/null
+++ b/addons/ClassExporter/plugin.cfg
@@ -0,0 +1,7 @@
+[plugin]
+
+name = "C# Class Exporter"
+description = "Add C# Node and Resource classes to Godot"
+author = "Sophia Pearson"
+version = "1.0.0"
+script = "Plugin.cs"
\ No newline at end of file