From 2d0c7a34f83eb464b73cb6b4e3d7ecb6f43e4ee9 Mon Sep 17 00:00:00 2001
From: Nicolas Delsaux <nicolas.delsaux@gmx.fr>
Date: Tue, 13 Feb 2024 13:30:10 +0100
Subject: [PATCH] Fixes #415

---
 .../base/enhancers/graph/ViewUpdater.java     | 43 +++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/base/src/main/java/org/ndx/aadarchi/base/enhancers/graph/ViewUpdater.java b/base/src/main/java/org/ndx/aadarchi/base/enhancers/graph/ViewUpdater.java
index 0321f662..d0312891 100644
--- a/base/src/main/java/org/ndx/aadarchi/base/enhancers/graph/ViewUpdater.java
+++ b/base/src/main/java/org/ndx/aadarchi/base/enhancers/graph/ViewUpdater.java
@@ -2,6 +2,10 @@
 
 import com.structurizr.view.ComponentView;
 import com.structurizr.view.ContainerView;
+import com.structurizr.view.DeploymentView;
+import com.structurizr.view.SystemContextView;
+import com.structurizr.view.SystemLandscapeView;
+
 import org.ndx.aadarchi.base.OutputBuilder;
 import org.ndx.aadarchi.base.enhancers.ModelElementKeys;
 import org.ndx.aadarchi.base.enhancers.ViewEnhancerAdapter;
@@ -34,6 +38,45 @@ protected boolean startVisit(ComponentView c) {
 		return true;
 	}
 	
+	@Override
+	protected boolean startVisit(DeploymentView c) {
+		// TODO Auto-generated method stub
+		return super.startVisit(c);
+	}
+	
+	@Override
+	protected boolean startVisit(SystemContextView c) {
+		// TODO Auto-generated method stub
+		return super.startVisit(c);
+	}
+	@Override
+	protected boolean startVisit(SystemLandscapeView c) {
+		// TODO Auto-generated method stub
+		return super.startVisit(c);
+	}
+	@Override
+	protected void endVisit(DeploymentView c, OutputBuilder builder) {
+		if(c.getProperties().containsKey(NAME)) {
+			c.addAllDeploymentNodes();
+		}
+		super.endVisit(c, builder);
+	}
+	@Override
+	protected void endVisit(SystemContextView c, OutputBuilder builder) {
+		if(c.getProperties().containsKey(NAME)) {
+			c.addAllSoftwareSystems();
+			c.addAllPeople();
+		}
+		super.endVisit(c, builder);
+	}
+	@Override
+	protected void endVisit(SystemLandscapeView c, OutputBuilder builder) {
+		if(c.getProperties().containsKey(NAME)) {
+			c.addAllSoftwareSystems();
+			c.addAllPeople();
+		}
+	}
+	
 	@Override
 	protected void endVisit(ComponentView c, OutputBuilder builder) {
 		if(c.getProperties().containsKey(NAME)) {