#135: adde simple enum assignment

Author: jorre127

lib/config/yml_generator_config.dart

@@ -163,24 +163,47 @@ class YmlGeneratorConfig {
           ));
         });
 
-        final values = value['values'] as YamlMap?;
+        final values = value['values'];
         if (values == null) {
           throw Exception('Values can not be null. model: $key');
         }
 
+        ItemType? simpleType;
+
         values.forEach((key, value) {
-          final properties = value?['properties'] as YamlMap?;
-          final description = value?['description'];
           final enumValues = <EnumValue>[];
+          String? description;
+          if (value is YamlMap) {
+            final properties = value['properties'] as YamlMap?;
+            description = value['description'];
+
+            properties?.forEach((key, value) {
+              enumValues.add(
+                EnumValue(
+                  value: value.toString(),
+                  propertyName: key,
+                ),
+              );
+            });
+          } else if (value != null) {
+            final valueType = switch (value.runtimeType) {
+              String => StringType(),
+              int => IntegerType(),
+              double => DoubleType(),
+              _ => StringType(),
+            };
+            if (simpleType == null) {
+              simpleType = valueType;
+            } else if (simpleType?.name != valueType.name) {
+              throw Exception(
+                  'All values in a simple enum declaration should have the same value type, value ${simpleType?.name} is not ${valueType.name}. enum value: $key');
+            }
+            enumValues.add(EnumValue(
+              value: value.toString(),
+              propertyName: 'jsonValue',
+            ));
+          }
 
-          properties?.forEach((key, value) {
-            enumValues.add(
-              EnumValue(
-                value: value.toString(),
-                propertyName: key,
-              ),
-            );
-          });
           fields.add(EnumField(
             name: uppercaseEnums ? key.toUpperCase() : CaseUtil(key).camelCase,
             rawName: key,
@@ -189,6 +212,21 @@ class YmlGeneratorConfig {
           ));
         });
 
+        if (simpleType != null) {
+          if (enumProperties.isNotEmpty) {
+            throw Exception(
+                'Simple enum declaration only works if no properties are defined, model: $key');
+          }
+          enumProperties.add(
+            EnumProperty(
+              name: 'jsonValue',
+              type: simpleType!,
+              isOptional: false,
+              isJsonvalue: true,
+            ),
+          );
+        }
+
         final enumModel = EnumModel(
           addJsonValueToProperties: value['use_default_json_value'] ?? true,
           generateExtension: value['generate_extension'] == true,

test/writer/enum_model_reader_test.dart

@@ -356,6 +356,38 @@ Gender:
     FEMALE:
       properties:
         isMale: hello
+""",
+            ));
+    test(
+        'Simple enum declaration only works with empty properties',
+        () => testEnumError(
+              expectedError:
+                  'Exception: Simple enum declaration only works if no properties are defined, model: Gender',
+              enumYml: """
+Gender:
+  path: user/person/
+  type: enum
+  properties:
+    isMale:
+      is_json_key: true
+      type: String
+  values:
+    MALE: hello
+    FEMALE: hello
+""",
+            ));
+    test(
+        'Simple enum declaration only works with String, int and double',
+        () => testEnumError(
+              expectedError:
+                  'Exception: All values in a simple enum declaration should have the same value type, value int is not String. enum value: FEMALE',
+              enumYml: """
+Gender:
+  path: user/person/
+  type: enum
+  values:
+    MALE: 1
+    FEMALE: female
 """,
             ));
   });

test/writer/enum_model_writer/simple_enum_double/config.txt

@@ -0,0 +1,6 @@
+MyEnumModel:
+  path: test/enum/
+  type: enum
+  values:
+    MY_VALUE_1: 1.2
+    MY_VALUE_2: 2.2

test/writer/enum_model_writer/simple_enum_double/output.txt

@@ -0,0 +1,20 @@
+// GENERATED CODE - DO NOT MODIFY BY HAND
+
+import 'package:json_annotation/json_annotation.dart';
+
+enum MyEnumModel {
+  @JsonValue(1.2)
+  myValue1(
+    jsonValue: 1.2,
+  ),
+  @JsonValue(2.2)
+  myValue2(
+    jsonValue: 2.2,
+  );
+
+  final double jsonValue;
+
+  const MyEnumModel({
+    required this.jsonValue,
+  });
+}

test/writer/enum_model_writer/simple_enum_int/config.txt

@@ -0,0 +1,6 @@
+MyEnumModel:
+  path: test/enum/
+  type: enum
+  values:
+    MY_VALUE_1: 1
+    MY_VALUE_2: 2

test/writer/enum_model_writer/simple_enum_int/output.txt

@@ -0,0 +1,20 @@
+// GENERATED CODE - DO NOT MODIFY BY HAND
+
+import 'package:json_annotation/json_annotation.dart';
+
+enum MyEnumModel {
+  @JsonValue(1)
+  myValue1(
+    jsonValue: 1,
+  ),
+  @JsonValue(2)
+  myValue2(
+    jsonValue: 2,
+  );
+
+  final int jsonValue;
+
+  const MyEnumModel({
+    required this.jsonValue,
+  });
+}

test/writer/enum_model_writer/simple_enum_string/config.txt

@@ -0,0 +1,6 @@
+MyEnumModel:
+  path: test/enum/
+  type: enum
+  values:
+    MY_VALUE_1: my_value1
+    MY_VALUE_2: my_value2

test/writer/enum_model_writer/simple_enum_string/output.txt

@@ -0,0 +1,20 @@
+// GENERATED CODE - DO NOT MODIFY BY HAND
+
+import 'package:json_annotation/json_annotation.dart';
+
+enum MyEnumModel {
+  @JsonValue('my_value1')
+  myValue1(
+    jsonValue: 'my_value1',
+  ),
+  @JsonValue('my_value2')
+  myValue2(
+    jsonValue: 'my_value2',
+  );
+
+  final String jsonValue;
+
+  const MyEnumModel({
+    required this.jsonValue,
+  });
+}