From 459d899ed98835f3ad89d600a281bd16236986d4 Mon Sep 17 00:00:00 2001
From: Scott Frederick <scott.frederick@broadcom.com>
Date: Wed, 21 Aug 2024 18:03:57 -0500
Subject: [PATCH] Add release type attribute for Antora documentation
 generation

Fixes gh-41993
---
 .../antora/AntoraAsciidocAttributes.java      |  5 +++--
 .../antora/AntoraAsciidocAttributesTests.java | 21 +++++++++++++++++++
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/buildSrc/src/main/java/org/springframework/boot/build/antora/AntoraAsciidocAttributes.java b/buildSrc/src/main/java/org/springframework/boot/build/antora/AntoraAsciidocAttributes.java
index 7526f89a3969..6745386e1da5 100644
--- a/buildSrc/src/main/java/org/springframework/boot/build/antora/AntoraAsciidocAttributes.java
+++ b/buildSrc/src/main/java/org/springframework/boot/build/antora/AntoraAsciidocAttributes.java
@@ -77,7 +77,7 @@ public Map<String, String> get() {
 		Map<String, String> attributes = new LinkedHashMap<>();
 		addGitHubAttributes(attributes);
 		addVersionAttributes(attributes);
-		addUrlArtifactRepository(attributes);
+		addArtifactAttributes(attributes);
 		addUrlJava(attributes);
 		addUrlLibraryLinkAttributes(attributes);
 		addPropertyAttributes(attributes);
@@ -139,8 +139,9 @@ private void addDependencyVersion(Map<String, String> attributes, String name, S
 		attributes.put("version-" + name, version);
 	}
 
-	private void addUrlArtifactRepository(Map<String, String> attributes) {
+	private void addArtifactAttributes(Map<String, String> attributes) {
 		attributes.put("url-artifact-repository", this.artifactRelease.getDownloadRepo());
+		attributes.put("artifact-release-type", this.artifactRelease.getType());
 	}
 
 	private void addUrlJava(Map<String, String> attributes) {
diff --git a/buildSrc/src/test/java/org/springframework/boot/build/antora/AntoraAsciidocAttributesTests.java b/buildSrc/src/test/java/org/springframework/boot/build/antora/AntoraAsciidocAttributesTests.java
index c8c2b8d6f37c..65ccedc0c245 100644
--- a/buildSrc/src/test/java/org/springframework/boot/build/antora/AntoraAsciidocAttributesTests.java
+++ b/buildSrc/src/test/java/org/springframework/boot/build/antora/AntoraAsciidocAttributesTests.java
@@ -112,6 +112,27 @@ void urlArtifactRepositoryWhenSnapshot() {
 		assertThat(attributes.get()).containsEntry("url-artifact-repository", "https://repo.spring.io/snapshot");
 	}
 
+	@Test
+	void artifactReleaseTypeWhenRelease() {
+		AntoraAsciidocAttributes attributes = new AntoraAsciidocAttributes("1.2.3", true, null,
+				mockDependencyVersions(), null);
+		assertThat(attributes.get()).containsEntry("artifact-release-type", "release");
+	}
+
+	@Test
+	void artifactReleaseTypeWhenMilestone() {
+		AntoraAsciidocAttributes attributes = new AntoraAsciidocAttributes("1.2.3-M1", true, null,
+				mockDependencyVersions(), null);
+		assertThat(attributes.get()).containsEntry("artifact-release-type", "milestone");
+	}
+
+	@Test
+	void artifactReleaseTypeWhenSnapshot() {
+		AntoraAsciidocAttributes attributes = new AntoraAsciidocAttributes("1.2.3-SNAPSHOT", true, null,
+				mockDependencyVersions(), null);
+		assertThat(attributes.get()).containsEntry("artifact-release-type", "snapshot");
+	}
+
 	@Test
 	void urlLinksFromLibrary() {
 		Map<String, Function<LibraryVersion, String>> links = new LinkedHashMap<>();