Ensure inheritance attribute is generated

bonsai-rx · Jan 16, 2024 · 4bfc0a5 · 4bfc0a5
1 parent 486d0d1
commit 4bfc0a5
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/Bonsai.Sgen/CSharpCodeDomGenerator.cs b/Bonsai.Sgen/CSharpCodeDomGenerator.cs
@@ -82,6 +82,24 @@ public override IEnumerable<CodeArtifact> GenerateTypes()
         {
             var types = base.GenerateTypes();
             var extraTypes = new List<CodeArtifact>();
+
+            if (!Settings.SerializerLibraries.HasFlag(SerializerLibraries.NewtonsoftJson) &&
+                types.Any(r => r.Code.Contains(nameof(JsonInheritanceAttribute))))
+            {
+                if (Settings.ExcludedTypeNames?.Contains(nameof(JsonInheritanceAttribute)) != true)
+                {
+                    var model = new JsonInheritanceConverterTemplateModel(Settings);
+                    var template = Settings.TemplateFactory.CreateTemplate("CSharp", nameof(JsonInheritanceAttribute), model);
+                    var artifact = new CodeArtifact(
+                        nameof(JsonInheritanceAttribute),
+                        CodeArtifactType.Class,
+                        CodeArtifactLanguage.CSharp,
+                        CodeArtifactCategory.Utility,
+                        template);
+                    extraTypes.Add(ReplaceInitOnlyProperties(artifact));
+                }
+            }
+
             var schema = (JsonSchema)RootObject;
             var classTypes = (from type in types
                               let classType = type as CSharpClassCodeArtifact