Refactor component filters to "has..." helper functions

avisi-cloud · May 16, 2024 · 6a3e139 · 6a3e139
1 parent e888f41
commit 6a3e139
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/kotlin/nl/avisi/structurizr/site/generatr/site/SiteGenerator.kt b/src/main/kotlin/nl/avisi/structurizr/site/generatr/site/SiteGenerator.kt
@@ -176,8 +176,8 @@ private fun generateHtmlFiles(context: GeneratorContext, branchDir: File) {
 
             it.containers
                 .filter { container ->
-                    context.workspace.views.componentViews.any { containerView -> containerView.container == container } or
-                            context.workspace.views.imageViews.any { imageView -> imageView.elementId in container.id } }
+                    context.workspace.hasComponentDiagrams(container) or
+                            context.workspace.hasImageViews(container.id) }
                 .forEach { container ->
                     add { writeHtmlFile(branchDir, SoftwareSystemContainerComponentsPageViewModel(context, container)) } }