add code generators

workers/unity/Packages/io.improbable.gdk.debug/.codegen/Source/ComponentVisualElementJob.cs

@@ -0,0 +1,27 @@
+using System.IO;
+using System.Linq;
+using Improbable.Gdk.CodeGeneration.FileHandling;
+using Improbable.Gdk.CodeGeneration.Jobs;
+using Improbable.Gdk.CodeGeneration.Model.Details;
+
+namespace Improbable.Gdk.CodeGenerator
+{
+    public class ComponentVisualElementJob : CodegenJob
+    {
+        private const string DebugAsmdefFileName = "Improbable.Gdk.Generated.Debug.asmdef";
+        private readonly string relativeOutputPath = Path.Combine("improbable", "debugextensions");
+
+        public ComponentVisualElementJob(CodegenJobOptions options, IFileSystem fileSystem, DetailsStore detailsStore) : base(options.AsEditor(), fileSystem, detailsStore)
+        {
+            const string jobName = nameof(ComponentVisualElementJob);
+            Logger.Trace($"Initialising {jobName}.");
+
+            Logger.Trace($"Adding job target for {DebugAsmdefFileName}");
+            AddJobTarget(Path.Combine(relativeOutputPath, DebugAsmdefFileName), () => DebugAssemblyGenerator.Generate());
+
+            var componentsToGenerate = detailsStore.Components.Values.ToList();
+            AddGenerators(relativeOutputPath, componentsToGenerate, component => ($"{component.Name}Renderer.cs", ComponentVisualElementGenerator.Generate));
+            Logger.Trace($"Added job targets for {componentsToGenerate.Count} components");
+        }
+    }
+}

workers/unity/Packages/io.improbable.gdk.debug/.codegen/Source/Generators/DebugExtensions/ComponentVisualElementGenerator.cs

@@ -0,0 +1,49 @@
+using Improbable.Gdk.CodeGeneration.CodeWriter;
+using Improbable.Gdk.CodeGeneration.CodeWriter.Scopes;
+using Improbable.Gdk.CodeGeneration.Model.Details;
+
+namespace Improbable.Gdk.CodeGenerator
+{
+    public static class ComponentVisualElementGenerator
+    {
+        public static CodeWriter Generate(UnityComponentDetails details)
+        {
+            return CodeWriter.Populate(cgw =>
+            {
+                cgw.UsingDirectives(
+                    "Unity.Entities",
+                    "UnityEngine.UIElements",
+                    "Improbable.Gdk.Debug.WorkerInspector.Codegen"
+                );
+
+                cgw.Namespace(details.Namespace, ns =>
+                {
+                    ns.Type($"public class {details.Name}Renderer : ComponentVisualElement", type =>
+                    {
+                        type.Line($"public override ComponentType ComponentType {{ get; }} = ComponentType.ReadOnly<{details.Name}.Component>();");
+
+                        GenerateConstructor(type, details);
+                        GenerateUpdateMethod(type, details);
+                    });
+                });
+            });
+        }
+
+        private static void GenerateConstructor(TypeBlock typeBlock, UnityComponentDetails details)
+        {
+            typeBlock.Method($"public {details.Name}Renderer() : base()", mb =>
+            {
+                mb.Line($"ComponentFoldout.text = \"{details.Name}\";");
+                mb.Line($"AuthoritativeToggle.SetEnabled(false);");
+            });
+        }
+
+        private static void GenerateUpdateMethod(TypeBlock typeBlock, UnityComponentDetails details)
+        {
+            typeBlock.Method("public override void Update(EntityManager manager, Entity entity)", mb =>
+            {
+                mb.Line($"AuthoritativeToggle.value = manager.HasComponent<{details.Name}.HasAuthority>(entity);");
+            });
+        }
+    }
+}

workers/unity/Packages/io.improbable.gdk.debug/.codegen/Source/Generators/DebugExtensions/DebugAssemblyGenerator.cs

@@ -0,0 +1,28 @@
+namespace Improbable.Gdk.CodeGenerator
+{
+    public static class DebugAssemblyGenerator
+    {
+        public static string Generate()
+        {
+            return @"{
+    ""name"": ""Improbable.Gdk.Generated.Debug"",
+    ""references"": [
+        ""Improbable.Gdk.Generated"",
+        ""Improbable.Gdk.Debug.WorkerInspector.Codegen"",
+        ""Unity.Entities""
+    ],
+    ""includePlatforms"": [
+        ""Editor""
+    ],
+    ""excludePlatforms"": [],
+    ""allowUnsafeCode"": false,
+    ""overrideReferences"": false,
+    ""precompiledReferences"": [],
+    ""autoReferenced"": true,
+    ""defineConstraints"": [],
+    ""versionDefines"": []
+}
+";
+        }
+    }
+}

workers/unity/Packages/io.improbable.gdk.tools/.CodeGenTemplate/CodeGenerationLib/Jobs/CodegenJob.cs

@@ -93,6 +93,32 @@ protected void AddGenerators<TDetails>(IEnumerable<TDetails> details, params Gen
             }));
         }
 
+        protected void AddGenerators<TDetails>(string relativeOutputDir, IEnumerable<TDetails> details, params GeneratorSetupDelegate<TDetails, CodeWriter.CodeWriter>[] generatorSetupDelegates)
+            where TDetails : GeneratorInputDetails
+        {
+            jobTargets.AddRange(details.SelectMany(detail =>
+            {
+                return generatorSetupDelegates.Select(generatorSetup =>
+                {
+                    var (filePath, generate) = generatorSetup(detail);
+                    return new JobTarget(Path.Combine(OutputDirectory, relativeOutputDir, detail.NamespacePath, filePath), () => generate(detail));
+                });
+            }));
+        }
+
+        protected void AddGenerators<TDetails>(string relativeOutputDir, IEnumerable<TDetails> details, params GeneratorSetupDelegate<TDetails, string>[] generatorSetupDelegates)
+            where TDetails : GeneratorInputDetails
+        {
+            jobTargets.AddRange(details.SelectMany(detail =>
+            {
+                return generatorSetupDelegates.Select(generatorSetup =>
+                {
+                    var (filePath, generate) = generatorSetup(detail);
+                    return new JobTarget(Path.Combine(OutputDirectory, relativeOutputDir, detail.NamespacePath, filePath), () => generate(detail));
+                });
+            }));
+        }
+
         public void Clean()
         {
             var numRemovedDirectories = 0;