Generate const values if possible

SteveDunn · May 31, 2024 · 64f0830 · 64f0830
1 parent c32c024
commit 64f0830
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/src/Intellenum/Generators/ClassGenerator.cs b/src/Intellenum/Generators/ClassGenerator.cs
@@ -26,6 +26,8 @@ public string BuildClass(VoWorkItem item, TypeDeclarationSyntax tds, bool isNetF
         global::System.IComparable, 
         global::System.IComparable<{className}> 
     {{
+        {MemberGeneration.GenerateConstValuesIfPossible(item)}
+
         {Util.GenerateLazyLookupsIfNeeded(item)}
 
 #if DEBUG    
@@ -230,4 +232,4 @@ private static string GenerateNullCheckIfNeeded(VoWorkItem voWorkItem) =>
                 throw new {nameof(IntellenumCreationFailedException)}(""Cannot create an Intellenum member with a null."");
             }}
 ";
-}
+}
diff --git a/src/Intellenum/MemberGeneration.cs b/src/Intellenum/MemberGeneration.cs
@@ -199,6 +199,23 @@ public static BuildResult TryBuildMemberValueAsText(string propertyName, object
                 $"Member '{propertyName}' has a value type '{propertyValue.GetType()}' of '{propertyValue}' which cannot be converted to the underlying type of '{underlyingType}' - {e.Message}");
         }
     }
+
+    public static string GenerateConstValuesIfPossible(VoWorkItem item)
+    {
+        if (!item.IsConstant || item.MemberProperties.Count == 0)
+        {
+            return string.Empty;
+        }
+
+        StringBuilder sb = new StringBuilder("// const fields...");
+        sb.AppendLine();
+        foreach (var memberProperties in item.MemberProperties)
+        {
+            sb.AppendLine($"public const {item.UnderlyingTypeFullName} {memberProperties.FieldName}Const = {memberProperties.ValueAsText};");
+        }
+
+        return sb.ToString();
+    }
 
     public static string GeneratePrivateConstructionInitialisationIfNeeded(VoWorkItem item)
     {
@@ -226,4 +243,4 @@ public static string GeneratePrivateConstructionInitialisationIfNeeded(VoWorkIte
 """);
         return sb.ToString();
     }
-}
+}