fix: fix enum generation #663

trevorwang · Jul 18, 2024 · 30f37b4 · 30f37b4
1 parent b24d210
commit 30f37b4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/generator/lib/src/generator.dart b/generator/lib/src/generator.dart
@@ -1999,7 +1999,7 @@ ${bodyName.displayName} == null
           } else {
             throw Exception('Unknown error!');
           }
-        } else if (_isBasicType(p.type)) {
+        } else if (_isBasicType(p.type) || _isEnum(p.type)) {
           if (p.type.nullabilitySuffix == NullabilitySuffix.question) {
             blocks.add(Code('if (${p.displayName} != null) {'));
           }
@@ -2009,6 +2009,8 @@ ${bodyName.displayName} == null
                 literal(fieldName),
                 if (_typeChecker(String).isExactlyType(p.type))
                   refer(p.displayName)
+                else if (_isEnum(p.type))
+                  refer(p.displayName).property('name')
                 else
                   refer(p.displayName).property('toString').call([])
               ])