refactor

solid-software · Nov 16, 2023 · 9e2678b · 9e2678b
1 parent ffe7f59
commit 9e2678b
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/lib/src/add_imports/config.dart b/lib/src/add_imports/config.dart
@@ -20,12 +20,9 @@ class Config {
         'file_url_prefix': String fileUrlPrefix,
       } = loadYaml(yamlString) as YamlMap;
 
-      final classesMap = classesYamlMap.map(
-        (key, value) => MapEntry(
-          key as String,
-          ClassInteropData.fromYamlList(value as YamlList),
-        ),
-      );
+      final classesMap = classesYamlMap
+          .cast<String, YamlList>()
+          .mapValues(ClassInteropData.fromYamlList);
 
       return Config(
         classesMap: classesMap,
@@ -49,3 +46,8 @@ class Config {
     return 'import { $classname } from "$url";';
   }
 }
+
+extension MapMapValues<K, V> on Map<K, V> {
+  Map<K, V1> mapValues<V1>(V1 Function(V) f) =>
+      map((k, v) => MapEntry(k, f(v)));
+}