Add proposition utility method

feenkcom · Nov 4, 2024 · c7a3099 · c7a3099
1 parent 28e836e
commit c7a3099
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 0 deletions.
diff --git a/src/GToolkit-WardleyMap/GtWardleyMapAddProposition.class.st b/src/GToolkit-WardleyMap/GtWardleyMapAddProposition.class.st
@@ -74,6 +74,8 @@ GtWardleyMapAddProposition >> initialize [
 	rejectAction icon: BrGlamorousVectorIcons cancel.
 	rejectAction action: [ self onRejectAction ].
 	rejectAction tooltip: 'Remove'.
+
+	isReadOnly := false
 ]
 
 { #category : #testing }

diff --git a/src/GToolkit-WardleyMap/GtWardleyMapUtility.class.st b/src/GToolkit-WardleyMap/GtWardleyMapUtility.class.st
@@ -4,6 +4,59 @@ Class {
 	#category : #'GToolkit-WardleyMap-Helpers'
 }
 
+{ #category : #'api - diffing' }
+GtWardleyMapUtility class >> addModelAsPropositionsFrom: aModel to: propositionModel [
+	^ self
+		addModelAsPropositionsFrom: aModel
+		to: propositionModel
+		withPropositionBlock: #yourself
+]
+
+{ #category : #'api - diffing' }
+GtWardleyMapUtility class >> addModelAsPropositionsFrom: aModel to: propositionModel withPropositionBlock: aBlock [
+	| unsetNodes nodes unsetEdges |
+	unsetNodes := aModel nodes copy.
+	nodes := Dictionary new.
+
+	propositionModel nodes
+		do: [ :aNode | 
+			| aName |
+			aName := aNode label.
+			unsetNodes
+				detect: [ :anUnsetNode | aName = anUnsetNode label ]
+				ifFound: [ :anUnsetNode | unsetNodes remove: anUnsetNode ]
+				ifNone: [ aNode removeParent
+						addWardleyMapDecoratorModel: (aBlock value: GtWardleyMapAddProposition new).
+					aModel addNode: aNode.
+					nodes at: aName put: aNode ] ].
+
+	unsetNodes
+		do: [ :aNode | 
+			aNode
+				addWardleyMapDecoratorModel: (aBlock value: GtWardleyMapRemoveProposition new) ].
+
+	unsetEdges := aModel edges copy.
+
+	propositionModel edges
+		do: [ :anEdge | 
+			| aKey |
+			aKey := anEdge fromNode label -> anEdge toNode label.
+			unsetEdges
+				detect: [ :anUnsetEdge | aKey = (anUnsetEdge fromNode label -> anUnsetEdge toNode label) ]
+				ifFound: [ :anUnsetEdge | unsetEdges remove: anUnsetEdge ]
+				ifNone: [ anEdge removeParent
+						addWardleyMapDecoratorModel: (aBlock value: GtWardleyMapAddProposition new);
+						fromNode: (nodes at: aKey key);
+						toNode: (nodes at: aKey value).
+					aModel addEdge: anEdge ] ].
+
+	unsetEdges
+		do: [ :anEdge | 
+			anEdge
+				addWardleyMapDecoratorModel: (aBlock value: GtWardleyMapRemoveProposition new) ].
+	^ aModel
+]
+
 { #category : #'api - views' }
 GtWardleyMapUtility class >> buttonLabelFromString: aString [
 	aString size <= GtWardleyMapConstants maxButtonLabelCharacters