fix: Enum constructor clause (#12)

Co-authored-by: Lukas Klingsbo <[email protected]>
bluefireteam · Oct 30, 2023 · 52c8cfb · 52c8cfb
1 parent 9f0cbf8
commit 52c8cfb
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 1 deletion.
diff --git a/lib/src/constructor_declaration.dart b/lib/src/constructor_declaration.dart
@@ -13,7 +13,19 @@ Map<String, dynamic>? serializeConstructorDeclaration(
  hasPrivateAnnotation(annotations)) {
  return null;
  }
- final className = (constructor.parent! as ClassDeclaration).name.lexeme;
+ final String className;
+
+ if (constructor.parent is ClassDeclaration) {
+ className = (constructor.parent! as ClassDeclaration).name.lexeme;
+ } else if (constructor.parent is EnumDeclaration) {
+ className = (constructor.parent! as EnumDeclaration).name.lexeme;
+ } else {
+ throw UnsupportedError(
+ 'Constructor parent is neither ClassDeclaration nor EnumDeclaration. '
+ 'It is ${constructor.parent.runtimeType}',
+ );
+ }
+
  final constructorName = constructor.name == null
  ? className
  : '$className.${constructor.name!.lexeme}';

diff --git a/test/enum_declaration_test.dart b/test/enum_declaration_test.dart
@@ -96,4 +96,26 @@ void main() {
  );
  });
  });
+
+ test('enum with constructor', () {
+ expect(
+ parseAsJson('''
+ enum Abc {
+ a;
+ 
+ const Abc();
+ }
+ '''),
+ {
+ 'kind': 'enum',
+ 'name': 'Abc',
+ 'values': [
+ {'name': 'a'},
+ ],
+ 'members': [
+ {'kind': 'constructor', 'name': 'Abc', 'const': true},
+ ],
+ },
+ );
+ });
 }