Merge pull request #1 from cloudflightio/KL/workspace-export

use generic export method
cloudflightio · Jul 5, 2022 · d6c6e71 · d6c6e71
2 parents efdfbe9 + 4003e9a
commit d6c6e71
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 10 deletions.
diff --git a/build.gradle.kts b/build.gradle.kts
@@ -7,7 +7,7 @@ plugins {
 
 description = "Spring Boot AutoConfigure Support for the Structurizr Client"
 group = "io.cloudflight.structurizr"
-version = "1.0.0"
+version = "1.0.1"
 
 autoConfigure {
     java {

diff --git a/.../kotlin/io/cloudflight/architecture/structurizr/service/export/C4PlantUmlExportService.kt b/.../kotlin/io/cloudflight/architecture/structurizr/service/export/C4PlantUmlExportService.kt
@@ -2,7 +2,7 @@ package io.cloudflight.architecture.structurizr.service.export
 
 import com.structurizr.Workspace
 import com.structurizr.export.Diagram
-import com.structurizr.view.*
+import com.structurizr.view.ThemeUtils
 import io.cloudflight.architecture.structurizr.WorkspaceExportService
 import io.cloudflight.architecture.structurizr.autoconfigure.StructurizrProperties
 import io.cloudflight.structurizr.plantuml.ExtendedC4PlantUmlExporter
@@ -21,14 +21,8 @@ internal class C4PlantUmlExportService(private val properties: StructurizrProper
         ThemeUtils.loadThemes(workspace)
 
         if (properties.export.c4PlantUml.enabled) {
-            workspace.views.views.forEach {
-                when (it) {
-                    is ContainerView -> storeDiagram(c4exporter.export(it))
-                    is DeploymentView -> storeDiagram(c4exporter.export(it))
-                    is SystemContextView -> storeDiagram(c4exporter.export(it))
-                    is ComponentView -> storeDiagram(c4exporter.export(it))
-                    is SystemLandscapeView -> storeDiagram(c4exporter.export(it))
-                }
+            c4exporter.export(workspace).forEach {
+                storeDiagram(it)
             }
         }
     }