From b2cd1d0fcbb4a0f6686afb04850e1687475b1fb1 Mon Sep 17 00:00:00 2001
From: "Andrew J. Mauer" <ajmauer@gmail.com>
Date: Fri, 16 Feb 2024 20:33:27 +0000
Subject: [PATCH] eliminate the usage of jdt internal API to get list of
 imports in a java file

---
 .../jdt/internal/DiagnosticUtils.java         | 66 +++++--------------
 1 file changed, 16 insertions(+), 50 deletions(-)

diff --git a/jakarta.jdt/org.eclipse.lsp4jakarta.jdt.core/src/main/java/org/eclipse/lsp4jakarta/jdt/internal/DiagnosticUtils.java b/jakarta.jdt/org.eclipse.lsp4jakarta.jdt.core/src/main/java/org/eclipse/lsp4jakarta/jdt/internal/DiagnosticUtils.java
index a3a10d08..4af128a6 100644
--- a/jakarta.jdt/org.eclipse.lsp4jakarta.jdt.core/src/main/java/org/eclipse/lsp4jakarta/jdt/internal/DiagnosticUtils.java
+++ b/jakarta.jdt/org.eclipse.lsp4jakarta.jdt.core/src/main/java/org/eclipse/lsp4jakarta/jdt/internal/DiagnosticUtils.java
@@ -29,8 +29,6 @@
 import org.eclipse.jdt.core.IType;
 import org.eclipse.jdt.core.ITypeHierarchy;
 import org.eclipse.jdt.core.JavaModelException;
-import org.eclipse.jdt.internal.core.ImportContainerInfo;
-import org.eclipse.jdt.internal.core.JavaModelManager;
 import org.eclipse.jdt.internal.corext.util.JavaModelUtil;
 import org.eclipse.lsp4jakarta.jdt.core.JakartaCorePlugin;
 
@@ -113,35 +111,19 @@ public static boolean isMatchedJavaElement(IType type, String javaElementName, S
      *         otherwise.
      */
     public static boolean isImportedJavaElement(ICompilationUnit unit, String javaElementFQName) throws JavaModelException {
+    	
+    	if (!unit.isOpen()) {
+    		unit.open(null);
+    	}
+    	
         IImportContainer container = unit.getImportContainer();
         if (container == null) {
             return false;
         }
 
-        // The following code uses JDT internal class and looks like
-        // ICompilationUnit#getImports()
-        // To avoid creating an array of IImportDeclaration, we do the following code:
-        JavaModelManager manager = JavaModelManager.getJavaModelManager();
-        Object info = manager.getInfo(container);
-        if (info == null) {
-            if (manager.getInfo(unit) != null) {
-                // CU was opened, but no import container, then no imports
-                return false;
-            } else {
-                try {
-                    unit.open(null);
-                } catch (JavaModelException e) {
-                    e.printStackTrace();
-                } // force opening of CU
-                info = manager.getInfo(container);
-                if (info == null)
-                    // after opening, if no import container, then no imports
-                    return false;
-            }
-        }
-        IJavaElement[] elements = ((ImportContainerInfo) info).getChildren();
-        for (IJavaElement child : elements) {
-            IImportDeclaration importDeclaration = (IImportDeclaration) child;
+        IImportDeclaration[] importDeclArray = unit.getImports();
+
+        for (IImportDeclaration importDeclaration : importDeclArray) {
             if (importDeclaration.isOnDemand()) {
                 String fqn = importDeclaration.getElementName();
                 String qualifier = fqn.substring(0, fqn.lastIndexOf('.'));
@@ -165,35 +147,18 @@ public static boolean isImportedJavaElement(ICompilationUnit unit, String javaEl
      *         false otherwise.
      */
     protected static boolean isImportedJavaElement(ICompilationUnit unit, String[] javaElementFQNames) throws JavaModelException {
+    	if (!unit.isOpen()) {
+    		unit.open(null);
+    	}
+    	
         IImportContainer container = unit.getImportContainer();
         if (container == null) {
             return false;
         }
 
-        // The following code uses JDT internal class and looks like
-        // ICompilationUnit#getImports()
-        // To avoid creating an array of IImportDeclaration, we do the following code:
-        JavaModelManager manager = JavaModelManager.getJavaModelManager();
-        Object info = manager.getInfo(container);
-        if (info == null) {
-            if (manager.getInfo(unit) != null) {
-                // CU was opened, but no import container, then no imports
-                return false;
-            } else {
-                try {
-                    unit.open(null);
-                } catch (JavaModelException e) {
-                    e.printStackTrace();
-                } // force opening of CU
-                info = manager.getInfo(container);
-                if (info == null)
-                    // after opening, if no import container, then no imports
-                    return false;
-            }
-        }
-        IJavaElement[] elements = ((ImportContainerInfo) info).getChildren();
-        for (IJavaElement child : elements) {
-            IImportDeclaration importDeclaration = (IImportDeclaration) child;
+        IImportDeclaration[] importDeclArray = unit.getImports();
+
+        for (IImportDeclaration importDeclaration : importDeclArray) {
             if (importDeclaration.isOnDemand()) {
                 String fqn = importDeclaration.getElementName();
                 String qualifier = fqn.substring(0, fqn.lastIndexOf('.'));
@@ -210,6 +175,7 @@ protected static boolean isImportedJavaElement(ICompilationUnit unit, String[] j
             }
         }
         return false;
+
     }
 
     /**