Improve ResolveMainClassHandler.resolveMainClassUnderPaths (#596)

snjeza · wenytang-ms · web-flow · commit a209e39b9e07 · 2025-08-13T15:18:31.000+08:00
Co-authored-by: wenyt &lt;75360946+wenytang-ms@users.noreply.github.com&gt;
diff --git a/com.microsoft.java.debug.plugin/src/main/java/com/microsoft/java/debug/plugin/internal/ResolveMainClassHandler.java b/com.microsoft.java.debug.plugin/src/main/java/com/microsoft/java/debug/plugin/internal/ResolveMainClassHandler.java
@@ -35,6 +35,7 @@
 import org.eclipse.core.runtime.IPath;
 import org.eclipse.jdt.core.IJavaProject;
 import org.eclipse.jdt.core.IMethod;
+import org.eclipse.jdt.core.JavaCore;
 import org.eclipse.jdt.core.JavaModelException;
 import org.eclipse.jdt.core.search.IJavaSearchConstants;
 import org.eclipse.jdt.core.search.IJavaSearchScope;
@@ -100,8 +101,18 @@ private List<ResolutionItem> resolveMainClassCore(List<Object> arguments) {
 
     private List<ResolutionItem> resolveMainClassUnderPaths(List<IPath> parentPaths) {
         // Limit to search main method from source code only.
-        IJavaSearchScope searchScope = SearchEngine.createJavaSearchScope(ProjectUtils.getJavaProjects(),
-            IJavaSearchScope.REFERENCED_PROJECTS | IJavaSearchScope.SOURCES);
+        IJavaProject[] projects;
+        if (parentPaths == null || parentPaths.isEmpty()) {
+            projects = ProjectUtils.getJavaProjects();
+        } else {
+            projects = Stream.of(ProjectUtils.getAllProjects())
+                .filter(p -> ProjectUtils.isJavaProject(p)  && p.getLocation() != null && ResourceUtils.isContainedIn(p.getLocation(), parentPaths))
+                .map(p -> JavaCore.create(p))
+                .filter(p -> p.exists())
+                .toArray(IJavaProject[]::new);
+        }
+        IJavaSearchScope searchScope = SearchEngine.createJavaSearchScope(projects,
+            IJavaSearchScope.SOURCES);
         SearchPattern pattern = createMainMethodSearchPattern();
         final List<ResolutionItem> res = new ArrayList<>();
         SearchRequestor requestor = new SearchRequestor() {
