Disable aot query codelenses for Gradle project in VSCode

Signed-off-by: BoykoAlex <alex.boyko@broadcom.com>

Author: BoykoAlex

headless-services/spring-boot-language-server/src/main/java/org/springframework/ide/vscode/boot/java/data/DataRepositoryAotMetadataCodeLensProvider.java

@@ -130,10 +130,6 @@ protected void provideCodeLens(CancelChecker cancelToken, MethodDeclaration node
 		if (project == null || !QueryMethodCodeActionProvider.isValidProject(project)) {
 			return;
 		}
-		if (!repositoryMetadataService.isAotGenerationAvailable(project)) {
-			// The build cannot generate the AOT metadata, e.g. the Gradle project has no `processAot` task
-			return;
-		}
 
 		IMethodBinding methodBinding = node.resolveBinding();
 

headless-services/spring-boot-language-server/src/main/java/org/springframework/ide/vscode/boot/java/data/DataRepositoryAotMetadataService.java

@@ -234,7 +234,7 @@ private static String buildType(IJavaProject project) {
 	 * GraalVM native build tools plugin is applied. The IDE reports the tasks of the build without running Gradle.
 	 * If it does not know them (yet) the project is given the benefit of the doubt.
 	 */
-	public boolean isAotGenerationAvailable(IJavaProject project) {
+	public static boolean isAotGenerationAvailable(IJavaProject project) {
 		IProjectBuild build = project.getProjectBuild();
 		if (build != null && ProjectBuild.GRADLE_PROJECT_TYPE.equals(build.getType())) {
 			Set<String> tasks = build.getTasks();

headless-services/spring-boot-language-server/src/main/java/org/springframework/ide/vscode/boot/java/data/QueryMethodCodeActionProvider.java

@@ -27,8 +27,12 @@
 import org.springframework.ide.vscode.boot.java.jdt.refactoring.JdtRefactorings;
 import org.springframework.ide.vscode.commons.Version;
 import org.springframework.ide.vscode.commons.java.IJavaProject;
+import org.springframework.ide.vscode.commons.java.IProjectBuild;
 import org.springframework.ide.vscode.commons.java.SpringProjectUtil;
 import org.springframework.ide.vscode.commons.languageserver.reconcile.ICollector;
+import org.springframework.ide.vscode.commons.languageserver.util.LspClient;
+import org.springframework.ide.vscode.commons.languageserver.util.LspClient.Client;
+import org.springframework.ide.vscode.commons.protocol.java.ProjectBuild;
 import org.springframework.ide.vscode.commons.util.text.IRegion;
 import org.springframework.ide.vscode.commons.util.text.TextDocument;
 
@@ -52,12 +56,19 @@ public boolean isApplicable(IJavaProject project) {
 	}
 
 	static boolean isValidProject(IJavaProject project) {
+		IProjectBuild build = project.getProjectBuild();
+		if (LspClient.currentClient() == Client.VSCODE && build != null && ProjectBuild.GRADLE_PROJECT_TYPE.equals(build.getType())) {
+			// TODO: remove once `gradle.runBuild` of vscode-gradle takes the arguments of the AOT refresh command (GH-1722)
+			return false;
+		}
 		Version springDataJpaVersion = SpringProjectUtil.getDependencyVersionByPrefix(project, "spring-data-jpa");
 		Version springDataMongoDbVersion = SpringProjectUtil.getDependencyVersionByPrefix(project, "spring-data-mongodb");
 		Version springDataJddbcVersion = SpringProjectUtil.getDependencyVersionByPrefix(project, "spring-data-jdbc");
-		return (springDataJpaVersion != null && springDataJpaVersion.getMajor() >= 4)
+		boolean supportedSpringData = (springDataJpaVersion != null && springDataJpaVersion.getMajor() >= 4)
 				|| (springDataMongoDbVersion != null && springDataMongoDbVersion.getMajor() >= 5)
 				|| (springDataJddbcVersion != null && springDataJddbcVersion.getMajor() >= 4);
+		// The AOT metadata must be possible to generate, e.g. a Gradle build without the `processAot` task cannot
+		return supportedSpringData && DataRepositoryAotMetadataService.isAotGenerationAvailable(project);
 	}
 
 	@Override

headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/java/data/DataRepositoryAotMetadataServiceRefreshCommandTest.java

@@ -160,4 +160,19 @@ void clientWithoutSupportGetsNoCommand() throws Exception {
 		assertFalse(service.regenerateMetadataCommand(project).isPresent());
 	}
 
+	@Test
+	void gradleProjectsAreNotValidInVSCodeUntilVSCodeGradleSupportsTheCommand() throws Exception {
+		String client = System.getProperty("sts.lsp.client");
+		System.setProperty("sts.lsp.client", "vscode");
+		try {
+			assertFalse(QueryMethodCodeActionProvider.isValidProject(gradleProject("spring-boot-4.0.0.jar", projectDir.toFile())));
+		} finally {
+			if (client == null) {
+				System.clearProperty("sts.lsp.client");
+			} else {
+				System.setProperty("sts.lsp.client", client);
+			}
+		}
+	}
+
 }

headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/java/data/test/DataRepositoryAotMetadataServiceTest.java

@@ -219,11 +219,11 @@ private static IJavaProject gradleProject(Set<String> tasks) {
 
 		@Test
 		void processAotAvailability() {
-			assertTrue(service.isAotGenerationAvailable(gradleProject(Set.of("build", "processAot", "aotClasses"))));
-			assertFalse(service.isAotGenerationAvailable(gradleProject(Set.of("build", "classes", "bootJar"))));
-			assertFalse(service.isAotGenerationAvailable(gradleProject(Set.of())));
+			assertTrue(DataRepositoryAotMetadataService.isAotGenerationAvailable(gradleProject(Set.of("build", "processAot", "aotClasses"))));
+			assertFalse(DataRepositoryAotMetadataService.isAotGenerationAvailable(gradleProject(Set.of("build", "classes", "bootJar"))));
+			assertFalse(DataRepositoryAotMetadataService.isAotGenerationAvailable(gradleProject(Set.of())));
 			// unknown tasks get the benefit of the doubt
-			assertTrue(service.isAotGenerationAvailable(gradleProject(null)));
+			assertTrue(DataRepositoryAotMetadataService.isAotGenerationAvailable(gradleProject(null)));
 		}
 
 		@Test
@@ -233,9 +233,9 @@ void otherProjectsAreNotAffected() {
 			IJavaProject untyped = mock(IJavaProject.class);
 			when(untyped.getProjectBuild()).thenReturn(IProjectBuild.create(null, null));
 
-			assertTrue(service.isAotGenerationAvailable(maven));
-			assertTrue(service.isAotGenerationAvailable(untyped));
-			assertTrue(service.isAotGenerationAvailable(mock(IJavaProject.class)));
+			assertTrue(DataRepositoryAotMetadataService.isAotGenerationAvailable(maven));
+			assertTrue(DataRepositoryAotMetadataService.isAotGenerationAvailable(untyped));
+			assertTrue(DataRepositoryAotMetadataService.isAotGenerationAvailable(mock(IJavaProject.class)));
 		}
 
 		@Test