refactor: enhance import handling in analyzing_builder.dart

- Update import retrieval to use the library's compilation unit for better accuracy - Replace direct access to libraryImports and libraryExports with compilationUnit properties - Improve compatibility with the analyzer package by ensuring correct import processing - Remove lint warnings by properly handling nullable types and type casting
schultek · Jan 3, 2025 · fb53533 · fb53533
1 parent c759ba4
commit fb53533
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/jaspr_builder/lib/src/imports/analyzing_builder.dart b/packages/jaspr_builder/lib/src/imports/analyzing_builder.dart
@@ -29,9 +29,11 @@ class ImportsModuleBuilder implements Builder {
       var outputId = buildStep.inputId.changeExtension('.imports.json');
       var partId = buildStep.inputId.changeExtension('.imports.dart');
 
-      var import = lib.libraryImports
+      // Get the library's compilation unit and use its imports instead
+      var compilationUnit = lib.definingCompilationUnit;
+      var import = compilationUnit.libraryImports
           .cast<Element>()
-          .followedBy(lib.libraryExports)
+          .followedBy(compilationUnit.libraryExports)
           .where((Element e) => importChecker.hasAnnotationOf(e))
           .where((Element e) {
         var uri = switch (e) {