minor updates

jellygnite · Apr 13, 2021 · 83ab1a5 · 83ab1a5
1 parent 5c874e5
commit 83ab1a5
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/src/Model/BaseElementObject.php b/src/Model/BaseElementObject.php
@@ -198,7 +198,18 @@ public function getPage()
         // because $page can be a SiteTree or Controller
         return $page instanceof SiteTree ? $page : null;
     }
-
+
+  // method that can be overridden by indivudal element objects
+    public function getOwnerElement()
+    {
+        // You can't find the owner element of an object that hasn't been saved yet
+        if (!$this->isInDB()) {
+            return null;
+        }
+
+	    return null;
+    }
+
     public function getLinkURL()
     {
 		if($link = $this->ElementLink){

diff --git a/src/Model/GalleryObject.php b/src/Model/GalleryObject.php
@@ -162,6 +162,16 @@ public function getCMSFields()
 		return $fields;
     }
 
+    public function getOwnerElement()
+    {
+        // You can't find the owner element of an object that hasn't been saved yet
+        if (!$this->isInDB()) {
+            return null;
+        }
+
+	    return $this->ElementGallery()->First();
+    }
+
     /**
      * @return mixed
      */