Fixes #5

structurizr · Jun 20, 2023 · dd29dfe · dd29dfe
1 parent fed5165
commit dd29dfe
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/dsl/plantuml-and-mermaid/dsl/plugins/plugin-1.0-SNAPSHOT.jar b/dsl/plantuml-and-mermaid/dsl/plugins/plugin-1.0-SNAPSHOT.jar
diff --git a/dsl/plantuml-and-mermaid/plugin/src/main/java/mermaid/MermaidEncoderPlugin.java b/dsl/plantuml-and-mermaid/plugin/src/main/java/mermaid/MermaidEncoderPlugin.java
@@ -7,6 +7,8 @@
 import com.structurizr.dsl.StructurizrDslPlugin;
 import com.structurizr.dsl.StructurizrDslPluginContext;
 import com.structurizr.model.SoftwareSystem;
+import com.structurizr.model.Container;
+import com.structurizr.model.Component;
 
 public class MermaidEncoderPlugin implements StructurizrDslPlugin {
 
@@ -27,6 +29,16 @@ public void run(StructurizrDslPluginContext context) {
                 for (Section section : softwareSystem.getDocumentation().getSections()) {
                     section.setContent(encodeMermaid(context, section.getContent(), section.getFormat()));
                 }
+                for (Container container : softwareSystem.getContainers()) {
+                    for (Section section : container.getDocumentation().getSections()) {
+                        section.setContent(encodeMermaid(context, section.getContent(), section.getFormat()));
+                    }
+                    for (Component component : container.getComponents()) {
+                        for (Section section : component.getDocumentation().getSections()) {
+                            section.setContent(encodeMermaid(context, section.getContent(), section.getFormat()));
+                        }
+                    }
+                }
             }
 
             for (Decision decision : workspace.getDocumentation().getDecisions()) {

diff --git a/dsl/plantuml-and-mermaid/plugin/src/main/java/plantuml/PlantUMLEncoderPlugin.java b/dsl/plantuml-and-mermaid/plugin/src/main/java/plantuml/PlantUMLEncoderPlugin.java
@@ -7,6 +7,8 @@
 import com.structurizr.dsl.StructurizrDslPlugin;
 import com.structurizr.dsl.StructurizrDslPluginContext;
 import com.structurizr.model.SoftwareSystem;
+import com.structurizr.model.Container;
+import com.structurizr.model.Component;
 
 public class PlantUMLEncoderPlugin implements StructurizrDslPlugin {
 
@@ -27,6 +29,16 @@ public void run(StructurizrDslPluginContext context) {
                 for (Section section : softwareSystem.getDocumentation().getSections()) {
                     section.setContent(encodePlantUML(context, section.getContent(), section.getFormat()));
                 }
+                for (Container container : softwareSystem.getContainers()) {
+                    for (Section section : container.getDocumentation().getSections()) {
+                        section.setContent(encodePlantUML(context, section.getContent(), section.getFormat()));
+                    }
+                    for (Component component : container.getComponents()) {
+                        for (Section section : component.getDocumentation().getSections()) {
+                            section.setContent(encodePlantUML(context, section.getContent(), section.getFormat()));
+                        }
+                    }
+                }
             }
 
             for (Decision decision : workspace.getDocumentation().getDecisions()) {