Fixes #415

Riduidel · Feb 13, 2024 · 2d0c7a3 · 2d0c7a3
1 parent bfb4be7
commit 2d0c7a3
Showing 1 changed file with 43 additions and 0 deletions.
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
@@ -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)) {