Find existing diagramElement before creating one

OpenPonk · Nov 23, 2024 · e1c4602 · e1c4602
1 parent 231d511
commit e1c4602
Show file tree

Hide file tree

Showing 7 changed files with 36 additions and 4 deletions.
diff --git a/repository/OpenPonk-Core/OPAbstractDiagramControllerTest.class.st b/repository/OpenPonk-Core/OPAbstractDiagramControllerTest.class.st
@@ -17,6 +17,7 @@ OPAbstractDiagramControllerTest >> setUp [
 
 	super setUp.
 	controller model: model.
+	controller diagramElement: OPDiagram new.
 	controller canvasPresenter: (OPCanvasPresenter new
 			 canvas: RSCanvas new;
 			 yourself)

diff --git a/repository/OpenPonk-Core/OPAbstractElementControllerTest.class.st b/repository/OpenPonk-Core/OPAbstractElementControllerTest.class.st
@@ -75,6 +75,7 @@ OPAbstractElementControllerTest >> setUp [
 	diagramController canvasPresenter: (OPCanvasPresenter new
 			 canvas: RSCanvas new;
 			 yourself).
+	diagramController diagramElement: OPDiagram new.
 	self setUpCreationTarget
 ]
 

diff --git a/repository/OpenPonk-Core/OPController.class.st b/repository/OpenPonk-Core/OPController.class.st
@@ -93,6 +93,12 @@ OPController >> descriptionName [
 		  yourself
 ]
 
+{ #category : 'accessing' }
+OPController >> diagram [
+
+	^ self diagramController diagram
+]
+
 { #category : 'accessing' }
 OPController >> diagramController [
 	^ diagramController

diff --git a/repository/OpenPonk-Core/OPDiagramController.class.st b/repository/OpenPonk-Core/OPDiagramController.class.st
@@ -276,9 +276,6 @@ OPDiagramController >> deselectAll [
 { #category : 'accessing' }
 OPDiagramController >> diagram [
 
-	self
-		deprecated: 'Use diagramElement instead'
-		transformWith: '`@receiver diagram ' -> '`@receiver diagramElement '.
 	^ self diagramElement
 ]
 
@@ -534,6 +531,7 @@ OPDiagramController >> nameOfTypeForModel: aModel [
 
 { #category : 'controllers management' }
 OPDiagramController >> newControllerFor: aModel [
+
 	^ controllerFactory newControllerFor: aModel
 ]
 

diff --git a/repository/OpenPonk-Core/OPDiagramElement.class.st b/repository/OpenPonk-Core/OPDiagramElement.class.st
@@ -147,6 +147,14 @@ OPDiagramElement >> addedByMouseClickTo: aPoint [
 
 ]
 
+{ #category : 'enumerating' }
+OPDiagramElement >> allOwnedElements [
+
+	^ self ownedElements
+	  ,
+	  (self ownedElements flatCollect: [ :each | each allOwnedElements ])
+]
+
 { #category : 'enumerating' }
 OPDiagramElement >> allOwnedShapesPositionableByUser [
 
@@ -194,6 +202,18 @@ OPDiagramElement >> createOwnedElementsWithoutController [
 
 ]
 
+{ #category : 'accessing' }
+OPDiagramElement >> diagramElementOfType: aType withModelElement: anElement ifFound: aFoundBlock [
+
+	((self isKindOf: aType) and: [ modelElements includes: anElement ])
+		ifTrue: [ ^ aFoundBlock cull: self ].
+	ownedElements do: [ :each |
+		each
+			diagramElementOfType: aType
+			withModelElement: anElement
+			ifFound: [ :found | ^ aFoundBlock cull: found ] ]
+]
+
 { #category : 'testing' }
 OPDiagramElement >> exists [
 

diff --git a/repository/OpenPonk-Core/OPElementController.class.st b/repository/OpenPonk-Core/OPElementController.class.st
@@ -94,7 +94,12 @@ OPElementController >> ensureDiagramElement [
 
 	self diagramElement
 		ifExists: [ self linkDiagramElementToModel ]
-		else: [ self createDiagramElement ].
+		else: [
+			self diagram
+				diagramElementOfType: self diagramElementClass
+				withModelElement: self model
+				ifFound: [ :found | ^ self diagramElement: found ].
+			self createDiagramElement ].
 	^ self diagramElement
 ]
 

diff --git a/repository/OpenPonk-Spec/OPDiagramExplorer.class.st b/repository/OpenPonk-Spec/OPDiagramExplorer.class.st
@@ -80,6 +80,7 @@ OPDiagramExplorer >> showAllElements [
 
 { #category : 'actions' }
 OPDiagramExplorer >> showDependentFor: aModelObject [
+
 	self showElements: (self dependentFor: aModelObject)
 ]
-Original file line number
+Diff line change
@@ Expand Up / @@ -80,6 +80,7 @@ OPDiagramExplorer >> showAllElements [ @@
     { #category : 'actions' }
     OPDiagramExplorer >> showDependentFor: aModelObject [
     	self showElements: (self dependentFor: aModelObject)
     ]
@@ Expand Down @@