refactor

Author: danylo-safonov-solid

tools/add_imports/bin/add_imports.dart

@@ -25,20 +25,10 @@ void main(List<String> arguments) {
 
     if (classes.isEmpty) return;
 
-    final imports = classes.map(
-      (classname) {
-        final filename = config.filenameForClass(classname);
-        final url = '${config.fileUrlPrefix}$filename';
-
-        return 'import { $classname } from "$url";';
-      },
-    );
-
-    final inits = classes.map((e) => 'self.$e = $e');
-
     final newSource = [
-      ...imports,
-      ...inits,
+      ...[config.importStringForClass, (e) => 'self.$e = $e']
+          .map(classes.map)
+          .flattened,
       sourceString,
     ].join('\n');
 

tools/add_imports/lib/src/config.dart

@@ -25,6 +25,13 @@ class Config {
     );
   }
 
-  String filenameForClass(String classname) =>
+  String _filenameForClass(String classname) =>
       classesMap.entries.firstWhere((e) => e.value.contains(classname)).key;
+
+  String importStringForClass(String classname) {
+    final filename = _filenameForClass(classname);
+    final url = '$fileUrlPrefix$filename';
+
+    return 'import { $classname } from "$url";';
+  }
 }