Add Configuration for Kotlin's All-Open Plugin for JPA Entities

initializr-generator-spring/src/main/java/io/spring/initializr/generator/spring/build/BuildMetadataResolver.java

@@ -29,6 +29,7 @@
  *
  * @author Stephane Nicoll
  * @author Moritz Halbritter
+ * @author Sijun Yang
  */
 public class BuildMetadataResolver {
 
@@ -65,4 +66,16 @@ public boolean hasFacet(Build build, String facet) {
  return dependencies(build).anyMatch((dependency) -> dependency.getFacets().contains(facet));
  }
 
+ /**
+ * Checks if the given {@link Build} contains dependencies with the given
+ * {@code groupId}.
+ * @param build the build to query
+ * @param groupId the groupId to query
+ * @return {@code true} if this build defines at least a dependency with that groupId,
+ * {@code false} otherwise
+ */
+ public boolean hasGroupId(Build build, String groupId) {
+ return dependencies(build).anyMatch((dependency) -> dependency.getGroupId().equals(groupId));
+ }
+
 }

initializr-generator-spring/src/main/java/io/spring/initializr/generator/spring/code/kotlin/KotlinJpaGradleBuildCustomizer.java

@@ -28,25 +28,38 @@
  * related dependency is present.
  *
  * @author Madhura Bhave
+ * @author Sijun Yang
  */
 public class KotlinJpaGradleBuildCustomizer implements BuildCustomizer<GradleBuild> {
 
  private final BuildMetadataResolver buildMetadataResolver;
 
  private final KotlinProjectSettings settings;
 
+ private final char quote;
+
  public KotlinJpaGradleBuildCustomizer(InitializrMetadata metadata, KotlinProjectSettings settings,
- ProjectDescription projectDescription) {
+ ProjectDescription projectDescription, char quote) {
  this.buildMetadataResolver = new BuildMetadataResolver(metadata, projectDescription.getPlatformVersion());
  this.settings = settings;
+ this.quote = quote;
  }
 
  @Override
  public void customize(GradleBuild build) {
  if (this.buildMetadataResolver.hasFacet(build, "jpa")) {
  build.plugins()
  .add("org.jetbrains.kotlin.plugin.jpa", (plugin) -> plugin.setVersion(this.settings.getVersion()));
+ build.extensions().customize("allOpen", (allOpen) -> {
+ allOpen.invoke("annotation", quote("jakarta.persistence.Entity"));
+ allOpen.invoke("annotation", quote("jakarta.persistence.MappedSuperclass"));
+ allOpen.invoke("annotation", quote("jakarta.persistence.Embeddable"));
+ });
  }
  }
 
+ private String quote(String element) {
+ return this.quote + element + this.quote;
+ }
+
 }

initializr-generator-spring/src/main/java/io/spring/initializr/generator/spring/code/kotlin/KotlinJpaMavenBuildCustomizer.java

@@ -28,6 +28,7 @@
  *
  * @author Madhura Bhave
  * @author Sebastien Deleuze
+ * @author Sijun Yang
  */
 public class KotlinJpaMavenBuildCustomizer implements BuildCustomizer<MavenBuild> {
 
@@ -41,8 +42,15 @@ public KotlinJpaMavenBuildCustomizer(InitializrMetadata metadata, ProjectDescrip
  public void customize(MavenBuild build) {
  if (this.buildMetadataResolver.hasFacet(build, "jpa")) {
  build.plugins().add("org.jetbrains.kotlin", "kotlin-maven-plugin", (kotlinPlugin) -> {
- kotlinPlugin.configuration((configuration) -> configuration.configure("compilerPlugins",
- (compilerPlugins) -> compilerPlugins.add("plugin", "jpa")));
+ kotlinPlugin.configuration((configuration) -> {
+ configuration.configure("compilerPlugins",
+ (compilerPlugins) -> compilerPlugins.add("plugin", "jpa"));
+ configuration.configure("pluginOptions", (option) -> {
+ option.add("option", "all-open:annotation=jakarta.persistence.Entity");
+ option.add("option", "all-open:annotation=jakarta.persistence.MappedSuperclass");
+ option.add("option", "all-open:annotation=jakarta.persistence.Embeddable");
+ });
+ });
  kotlinPlugin.dependency("org.jetbrains.kotlin", "kotlin-maven-noarg", "${kotlin.version}");
  });
  }

initializr-generator-spring/src/main/java/io/spring/initializr/generator/spring/code/kotlin/KotlinProjectGenerationDefaultContributorsConfiguration.java

@@ -47,6 +47,7 @@
  * @author Stephane Nicoll
  * @author Jean-Baptiste Nizet
  * @author Moritz Halbritter
+ * @author Sijun Yang
  */
 @Configuration
 class KotlinProjectGenerationDefaultContributorsConfiguration {
@@ -67,10 +68,17 @@ BuildCustomizer<Build> kotlinDependenciesConfigurer() {
  }
 
  @Bean
- @ConditionalOnBuildSystem(GradleBuildSystem.ID)
- KotlinJpaGradleBuildCustomizer kotlinJpaGradleBuildCustomizer(InitializrMetadata metadata,
+ @ConditionalOnBuildSystem(value = GradleBuildSystem.ID, dialect = GradleBuildSystem.DIALECT_KOTLIN)
+ KotlinJpaGradleBuildCustomizer kotlinJpaGradleBuildCustomizerKotlinDsl(InitializrMetadata metadata,
+ KotlinProjectSettings settings, ProjectDescription projectDescription) {
+ return new KotlinJpaGradleBuildCustomizer(metadata, settings, projectDescription, '\"');
+ }
+
+ @Bean
+ @ConditionalOnBuildSystem(value = GradleBuildSystem.ID, dialect = GradleBuildSystem.DIALECT_GROOVY)
+ KotlinJpaGradleBuildCustomizer kotlinJpaGradleBuildCustomizerGroovyDsl(InitializrMetadata metadata,
  KotlinProjectSettings settings, ProjectDescription projectDescription) {
- return new KotlinJpaGradleBuildCustomizer(metadata, settings, projectDescription);
+ return new KotlinJpaGradleBuildCustomizer(metadata, settings, projectDescription, '\'');
  }
 
  @Bean

initializr-generator-spring/src/test/java/io/spring/initializr/generator/spring/code/kotlin/KotlinJpaGradleBuildCustomizerTests.java

@@ -17,6 +17,8 @@
 package io.spring.initializr.generator.spring.code.kotlin;
 
 import java.util.Collections;
+import java.util.List;
+import java.util.Objects;
 
 import io.spring.initializr.generator.buildsystem.gradle.GradleBuild;
 import io.spring.initializr.generator.buildsystem.gradle.GradlePlugin;
@@ -35,6 +37,7 @@
  * Tests for {@link KotlinJpaGradleBuildCustomizer}.
  *
  * @author Madhura Bhave
+ * @author Sijun Yang
  */
 class KotlinJpaGradleBuildCustomizerTests {
 
@@ -58,6 +61,32 @@ void customizeWhenJpaFacetAbsentShouldNotAddKotlinJpaPlugin() {
  assertThat(build.plugins().values()).isEmpty();
  }
 
+ @Test
+ void customizeWhenJpaFacetPresentShouldCustomizeAllOpen() {
+ Dependency dependency = Dependency.withId("foo");
+ dependency.setFacets(Collections.singletonList("jpa"));
+ GradleBuild build = getCustomizedBuild(dependency);
+ assertThat(build.extensions().values()).singleElement().satisfies((extension) -> {
+ assertThat(extension.getName()).isEqualTo("allOpen");
+ assertThat(extension.getInvocations())
+ .filteredOn((invocation) -> Objects.equals(invocation.getTarget(), "annotation"))
+ .extracting("arguments")
+ .containsExactlyInAnyOrder(List.of("\"jakarta.persistence.Entity\""),
+ List.of("\"jakarta.persistence.MappedSuperclass\""),
+ List.of("\"jakarta.persistence.Embeddable\""));
+ });
+ }
+
+ @Test
+ void customizeWhenJpaFacetAbsentShouldNotCustomizeAllOpen() {
+ Dependency dependency = Dependency.withId("foo");
+ GradleBuild build = getCustomizedBuild(dependency);
+ assertThat(build.extensions().values())
+ .filteredOn((extension) -> Objects.equals(extension.getName(), "allOpen"))
+ .isEmpty();
+ assertThat(build.extensions().values()).isEmpty();
+ }
+
  private GradleBuild getCustomizedBuild(Dependency dependency) {
  InitializrMetadata metadata = InitializrMetadataTestBuilder.withDefaults()
  .addDependencyGroup("test", dependency)
@@ -66,7 +95,7 @@ private GradleBuild getCustomizedBuild(Dependency dependency) {
  MutableProjectDescription projectDescription = new MutableProjectDescription();
  projectDescription.setPlatformVersion(Version.parse("1.0.0"));
  KotlinJpaGradleBuildCustomizer customizer = new KotlinJpaGradleBuildCustomizer(metadata, settings,
- projectDescription);
+ projectDescription, '\"');
  GradleBuild build = new GradleBuild(new MetadataBuildItemResolver(metadata, Version.parse("2.0.0.RELEASE")));
  build.dependencies().add("foo");
  customizer.customize(build);

initializr-generator-spring/src/test/java/io/spring/initializr/generator/spring/code/kotlin/KotlinJpaMavenBuildCustomizerTests.java

@@ -17,6 +17,8 @@
 package io.spring.initializr.generator.spring.code.kotlin;
 
 import java.util.Collections;
+import java.util.List;
+import java.util.Objects;
 
 import io.spring.initializr.generator.buildsystem.maven.MavenBuild;
 import io.spring.initializr.generator.buildsystem.maven.MavenPlugin;
@@ -36,6 +38,7 @@
  *
  * @author Madhura Bhave
  * @author Sebastien Deleuze
+ * @author Sijun Yang
  */
 class KotlinJpaMavenBuildCustomizerTests {
 
@@ -67,6 +70,34 @@ void customizeWhenJpaFacetAbsentShouldNotAddKotlinJpaPlugin() {
  assertThat(build.plugins().isEmpty()).isTrue();
  }
 
+ @Test
+ void customizeWhenJpaFacetPresentShouldCustomizeAllOpen() {
+ Dependency dependency = Dependency.withId("foo");
+ dependency.setFacets(Collections.singletonList("jpa"));
+ MavenBuild build = getCustomizedBuild(dependency);
+
+ assertThat(build.plugins().values()).singleElement().satisfies((plugin) -> {
+ MavenPlugin.Configuration configuration = plugin.getConfiguration();
+
+ assertThat(configuration.getSettings()).filteredOn((setting) -> setting.getName().equals("pluginOptions"))
+ .isNotEmpty()
+ .first()
+ .satisfies((pluginOptions) -> assertThat(((List<MavenPlugin.Setting>) pluginOptions.getValue()))
+ .filteredOn((option) -> Objects.equals(option.getName(), "option"))
+ .map(MavenPlugin.Setting::getValue)
+ .containsExactlyInAnyOrder("all-open:annotation=jakarta.persistence.Entity",
+ "all-open:annotation=jakarta.persistence.MappedSuperclass",
+ "all-open:annotation=jakarta.persistence.Embeddable"));
+ });
+ }
+
+ @Test
+ void customizeWhenJpaFacetAbsentShouldNotCustomizeAllOpen() {
+ Dependency dependency = Dependency.withId("foo");
+ MavenBuild build = getCustomizedBuild(dependency);
+ assertThat(build.plugins().values()).isEmpty();
+ }
+
  private MavenBuild getCustomizedBuild(Dependency dependency) {
  InitializrMetadata metadata = InitializrMetadataTestBuilder.withDefaults()
  .addDependencyGroup("test", dependency)