Implemented the casing options

Author: bratan

example/lib/localization/keys.dart

@@ -2,7 +2,7 @@ import 'package:flutter_translate_annotations/flutter_translate_annotations.dart
 
 part 'keys.g.dart';
 
-@TranslateKeysOptions(path: 'assets/i18n', caseStyle: CaseStyle.titleCase, separator: "_")
-class _$Keys
+@TranslateKeysOptions(path: 'assets/i18n', caseStyle: CaseStyle.lowerCase, separator: "_")
+class _$Keys // ignore: unused_element
 {
 }

example/lib/main.dart

@@ -1,4 +1,2 @@
-import 'package:flutter_translate_gen/flutter_translate_gen.dart';
-
 main() {
 }

flutter_translate_gen.iml

@@ -6,6 +6,7 @@
       <excludeFolder url="file://$MODULE_DIR$/.dart_tool" />
       <excludeFolder url="file://$MODULE_DIR$/.pub" />
       <excludeFolder url="file://$MODULE_DIR$/build" />
+      <excludeFolder url="file://$MODULE_DIR$/example/.dart_tool" />
       <excludeFolder url="file://$MODULE_DIR$/example/.pub" />
       <excludeFolder url="file://$MODULE_DIR$/example/build" />
     </content>

lib/flutter_translate_gen.dart

@@ -3,12 +3,13 @@ import 'dart:convert';
 import 'package:analyzer/dart/element/element.dart';
 import 'package:build/build.dart';
 import 'package:code_builder/code_builder.dart';
+import 'package:dart_casing/dart_casing.dart';
 import 'package:dart_style/dart_style.dart';
 import 'package:dart_utils/dart_utils.dart';
 import 'package:flutter_translate_annotations/flutter_translate_annotations.dart';
 import 'package:flutter_translate_gen/annotation_generator.dart';
 import 'package:flutter_translate_gen/keys_class_generator.dart';
-import 'package:flutter_translate_gen/translation.dart';
+import 'package:flutter_translate_gen/localized_item.dart';
 import 'package:glob/glob.dart';
 import 'package:source_gen/source_gen.dart';
 
@@ -27,7 +28,7 @@ class FlutterTranslateGen extends AnnotationGenerator<TranslateKeysOptions>
 
         validateClassName(className);
 
-        List<Translation> translations;
+        List<LocalizedItem> translations;
 
         try
         {
@@ -40,7 +41,7 @@ class FlutterTranslateGen extends AnnotationGenerator<TranslateKeysOptions>
 
         validateLocalizationItems(translations);
 
-        final file = Library((lb) => lb..body.addAll([generateKeysClass(translations, className)]));
+        final file = Library((lb) => lb..body.addAll([KeysClassGenerator.generateClass(options, translations, className)]));
 
         final DartEmitter emitter = DartEmitter(Allocator());
 
@@ -55,7 +56,7 @@ class FlutterTranslateGen extends AnnotationGenerator<TranslateKeysOptions>
                 separator: annotation.peek("separator")?.stringValue);
     }
 
-    Future<List<Translation>> getKeyMap(BuildStep step, TranslateKeysOptions options) async
+    Future<List<LocalizedItem>> getKeyMap(BuildStep step, TranslateKeysOptions options) async
     {
         var mapping = <String, List<String>>{};
 
@@ -70,13 +71,24 @@ class FlutterTranslateGen extends AnnotationGenerator<TranslateKeysOptions>
             translationMap.forEach((key, value) => (mapping[key] ??= <String>[]).add(value));
         }
 
-        var translations = List<Translation>();
+        var translations = List<LocalizedItem>();
 
-        mapping.forEach((id, trans) => translations.add(Translation(key: id, translations: trans, separator: options.separator),));
+        mapping.forEach((id, trans) => translations.add(LocalizedItem(id, trans, getKeyFieldName(id, options))));
 
         return translations;
     }
 
+    String getKeyFieldName(String key, TranslateKeysOptions options)
+    {
+        switch(options.caseStyle)
+        {
+            case CaseStyle.titleCase: return Casing.titleCase(key,separator: options.separator);
+            case CaseStyle.upperCase: return Casing.upperCase(key,separator: options.separator);
+            case CaseStyle.lowerCase: return Casing.lowerCase(key,separator: options.separator);
+            default: return throw InvalidGenerationSourceError("Invalid CaseStyle specified: ${options.caseStyle.toString()}");
+        }
+    }
+
     Map<String, String> getTranslationMap(Map<String, dynamic> jsonMap, {String parentKey})
     {
         final map = Map<String, String>();
@@ -101,7 +113,7 @@ class FlutterTranslateGen extends AnnotationGenerator<TranslateKeysOptions>
         return map;
     }
 
-    void validateLocalizationItems(List<Translation> translations)
+    void validateLocalizationItems(List<LocalizedItem> translations)
     {
         if (translations.isEmpty) return;
 

lib/keys_class_generator.dart

@@ -1,37 +1,31 @@
 import 'package:code_builder/code_builder.dart';
-import 'package:dart_casing/dart_casing.dart';
-import 'package:flutter_translate_gen/translation.dart';
+import 'package:flutter_translate_annotations/flutter_translate_annotations.dart';
+import 'package:flutter_translate_gen/localized_item.dart';
 
-const String strKeysClassName = "_\$Keys";
-
-Reference get stringType => TypeReference((trb) => trb..symbol = "String");
-
-Class generateKeysClass(List<Translation> translations, String className)
+class KeysClassGenerator
 {
-    return Class((x) => x
+    static Reference get stringType => TypeReference((trb) => trb..symbol = "String");
+
+    static Class generateClass(TranslateKeysOptions options, List<LocalizedItem> translations, String className)
+    {
+        return Class((x) => x
             ..docs.add("/// Contains the static localization keys")
             ..name = className.substring(2)
             ..fields.addAll(translations
-                    .map((translation) => generateField(translation))
+                    .map((translation) => generateField(translation, options))
                     .toList()),
-    );
+        );
+    }
+
+    static Field generateField(LocalizedItem item, TranslateKeysOptions options)
+    {
+        return Field((x) => x
+                ..name = item.fieldName
+                ..type = stringType
+                ..static = true
+                ..modifier = FieldModifier.constant
+                ..assignment = literalString(item.key).code,
+        );
+    }
 }
 
-Field generateField(Translation translation)
-{
-    var a = translation.keyVariable;
-
-    var key = translation.keyVariable.replaceAll("\$", "\\");
-    var name = Casing.titleCase(key, separator: translation.separator);
-
-    return Field(
-                (fb) => fb
-            ..name = name //translation.keyVariable.replaceAll("\$", '_')
-            ..type = stringType
-            ..static = true
-            ..modifier = FieldModifier.final$
-            ..assignment = literalString(translation.key).code,
-    );
-}
-
-

lib/localized_item.dart

@@ -0,0 +1,15 @@
+import 'package:dart_casing/dart_casing.dart';
+
+class LocalizedItem
+{
+    final String key;
+    final List<String> translations;
+    final String fieldName;
+
+    LocalizedItem(this.key, this.translations, this.fieldName);
+}
+
+List<String> keysOf(List<LocalizedItem> items)
+{
+    return items.map((x) => x.key).toList();
+}