fix(generator): remove unused imports (#786)

widgetbook · Jun 27, 2023 · 35681ed · 35681ed
1 parent 4315e4f
commit 35681ed
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 11 deletions.
diff --git a/packages/widgetbook_generator/CHANGELOG.md b/packages/widgetbook_generator/CHANGELOG.md
@@ -1,3 +1,7 @@
+## Unreleased
+
+ - **FIX**: Remove unused imports. ([#786](https://github.com/widgetbook/widgetbook/pull/786))
+
 ## 3.0.0
 
 Check out the [migration guide](https://docs.widgetbook.io/migration/2.4.0-to-3.0.0) for more information.

diff --git a/packages/widgetbook_generator/lib/generators/app_generator.dart b/packages/widgetbook_generator/lib/generators/app_generator.dart
@@ -80,18 +80,11 @@ class AppGenerator extends GeneratorForAnnotation<App> {
   String generateImports(
     List<WidgetbookData> datas,
   ) {
-    final set = <String>{};
+    final set = <String>{
+      'package:widgetbook/widgetbook.dart',
+    };
 
-    for (final data in datas) {
-      set
-        ..add(data.importStatement)
-        ..addAll(data.dependencies);
-    }
-
-    set
-      ..add('package:flutter/material.dart')
-      ..add('package:widgetbook/widgetbook.dart')
-      ..remove('package:widgetbook_annotation/widgetbook_annotation.dart');
+    set.addAll(datas.map((data) => data.importStatement));
 
     final imports = set.map(_generateImportStatement).toList()
       ..sort((a, b) => a.compareTo(b));