revert: parse import

heowc · Nov 17, 2024 · d30dd04 · d30dd04
1 parent 2af34d0
commit d30dd04
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/heo-core/src/main/java/dev/heowc/heo/core/analysis/domain/DependencyMapper.java b/heo-core/src/main/java/dev/heowc/heo/core/analysis/domain/DependencyMapper.java
@@ -20,6 +20,7 @@
 import com.github.javaparser.ParseResult;
 import com.github.javaparser.ParseStart;
 import com.github.javaparser.Providers;
+import com.github.javaparser.ast.CompilationUnit;
 import com.github.javaparser.ast.ImportDeclaration;
 import com.github.javaparser.ast.nodeTypes.NodeWithName;
 
@@ -90,15 +91,17 @@ private DependentModule createDependentModule(Module module, Map<String, Module>
 
     private Stream<ImportDeclaration> parseImports(Module module) {
         try {
-            final ParseResult<ImportDeclaration> parsed =
-                    javaParser.parse(ParseStart.IMPORT_DECLARATION,
+            final ParseResult<CompilationUnit> parsed =
+                    javaParser.parse(ParseStart.COMPILATION_UNIT,
                                      Providers.provider(module.getPath(),
                                                         javaParser.getParserConfiguration().getCharacterEncoding()));
 
             if (!parsed.isSuccessful()) {
                 return Stream.empty();
             }
-            return parsed.getResult().stream();
+            return parsed.getResult().orElseThrow()
+                         .getImports()
+                         .stream();
         } catch (IOException e) {
             return Stream.empty();
         }