From 232cb3c9ec991ed645c1ef2dfccee094fdb8e517 Mon Sep 17 00:00:00 2001
From: Oliver Drotbohm <oliver.drotbohm@broadcom.com>
Date: Sun, 21 Jan 2024 12:23:29 +0100
Subject: [PATCH] GH-476 - Move off legacy Structurizr PlantUML generation.

---
 spring-modulith-docs/pom.xml                  |  6 ---
 .../modulith/docs/Documenter.java             | 40 ++++++++++---------
 2 files changed, 22 insertions(+), 24 deletions(-)

diff --git a/spring-modulith-docs/pom.xml b/spring-modulith-docs/pom.xml
index 607962a6..ef373796 100644
--- a/spring-modulith-docs/pom.xml
+++ b/spring-modulith-docs/pom.xml
@@ -29,12 +29,6 @@
 			<version>1.29.0</version>
 		</dependency>
 
-		<dependency>
-			<groupId>com.structurizr</groupId>
-			<artifactId>structurizr-plantuml</artifactId>
-			<version>1.6.3</version>
-		</dependency>
-
 		<dependency>
 			<groupId>com.structurizr</groupId>
 			<artifactId>structurizr-export</artifactId>
diff --git a/spring-modulith-docs/src/main/java/org/springframework/modulith/docs/Documenter.java b/spring-modulith-docs/src/main/java/org/springframework/modulith/docs/Documenter.java
index 08595f78..f54d0711 100644
--- a/spring-modulith-docs/src/main/java/org/springframework/modulith/docs/Documenter.java
+++ b/spring-modulith-docs/src/main/java/org/springframework/modulith/docs/Documenter.java
@@ -20,7 +20,6 @@ import static org.springframework.modulith.docs.Asciidoctor.*;
 import java.io.File;
 import java.io.FileWriter;
 import java.io.IOException;
-import java.io.StringWriter;
 import java.io.Writer;
 import java.lang.annotation.Annotation;
 import java.nio.file.Files;
@@ -50,10 +49,9 @@ import org.springframework.util.LinkedMultiValueMap;
 import org.springframework.util.MultiValueMap;
 
 import com.structurizr.Workspace;
-import com.structurizr.export.Diagram;
+import com.structurizr.export.IndentingWriter;
 import com.structurizr.export.plantuml.C4PlantUMLExporter;
-import com.structurizr.io.plantuml.BasicPlantUMLWriter;
-import com.structurizr.io.plantuml.PlantUMLWriter;
+import com.structurizr.export.plantuml.StructurizrPlantUMLExporter;
 import com.structurizr.model.Component;
 import com.structurizr.model.Container;
 import com.structurizr.model.Element;
@@ -487,26 +485,18 @@ public class Documenter {
 
 			case C4:
 
-				C4PlantUMLExporter exporter = new C4PlantUMLExporter();
-				Diagram diagram = exporter.export(view);
+				var c4PlantUmlExporter = new C4PlantUMLExporter();
+				var diagram = c4PlantUmlExporter.export(view);
+
 				return diagram.getDefinition();
 
 			case UML:
 			default:
 
-				Writer writer = new StringWriter();
-				PlantUMLWriter umlWriter = new BasicPlantUMLWriter() {
+				var plantUmlExporter = new CustomizedPlantUmlExporter();
+				plantUmlExporter.addSkinParam("componentStyle", "uml1");
 
-					@Override
-					protected void writeContainerForContainer(ComponentView view, Writer writer,
-							BiConsumer<ComponentView, Writer> packageContentWriter) throws IOException {
-						packageContentWriter.accept(view, writer);
-					}
-				};
-				umlWriter.addSkinParam("componentStyle", "uml1");
-				umlWriter.write(view, writer);
-
-				return writer.toString();
+				return plantUmlExporter.export(view).getDefinition();
 		}
 	}
 
@@ -1173,4 +1163,18 @@ public class Documenter {
 			}
 		}
 	}
+
+	private static class CustomizedPlantUmlExporter extends StructurizrPlantUMLExporter {
+
+		@Override
+		protected boolean includeTitle(ModelView view) {
+			return false;
+		};
+
+		@Override
+		protected void startContainerBoundary(ModelView view, Container container, IndentingWriter writer) {}
+
+		@Override
+		protected void endContainerBoundary(ModelView view, IndentingWriter writer) {};
+	};
 }