Fix enum serialization (#55)

xoofx · Jun 30, 2023 · 9c71992 · 9c71992
1 parent a4faf8c
commit 9c71992
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/Tomlyn/Model/Accessors/ReflectionObjectInfo.cs b/src/Tomlyn/Model/Accessors/ReflectionObjectInfo.cs
@@ -30,7 +30,7 @@ public ReflectionObjectInfo(ReflectionObjectKind kind, Type genericArgument1, Ty
 
     public static ReflectionObjectInfo Get(Type type)
     {
-        if (type == typeof(string) || type.IsPrimitive || type == typeof(TomlDateTime) || type == typeof(DateTime) || type == typeof(DateTimeOffset)
+        if (type == typeof(string) || type.IsPrimitive || type == typeof(TomlDateTime) || type == typeof(DateTime) || type == typeof(DateTimeOffset) || type.IsEnum
 #if NET6_0_OR_GREATER
             || type == typeof(DateOnly) || type == typeof(TimeOnly)
 #endif

diff --git a/src/Tomlyn/Model/ModelToTomlTransform.cs b/src/Tomlyn/Model/ModelToTomlTransform.cs
@@ -598,6 +598,10 @@ private void WritePrimitive(object primitive, TomlPropertyDisplayKind displayKin
         {
             _writer.Write(TomlFormatHelper.ToString(dateTimeOffset, displayKind));
         }
+        else if (primitive is Enum enumValue)
+        {
+            _writer.Write(TomlFormatHelper.ToString(enumValue.ToString(), displayKind));
+        }
 #if NET6_0_OR_GREATER
         else if (primitive is DateOnly dateOnly)
         {