Merge pull request #1107 from sukhwinder-somar/fix-elemental-form-sub…

…mission/issue-27 fix form submission, when form block is inside another block
silverstripe · Oct 13, 2023 · 1e9f05c · 1e9f05c
2 parents 6e3180c + a2edb5e
commit 1e9f05c
Showing 1 changed file with 40 additions and 11 deletions.
diff --git a/src/Extensions/ElementalContentControllerExtension.php b/src/Extensions/ElementalContentControllerExtension.php
@@ -2,9 +2,10 @@
 
 namespace DNADesign\Elemental\Extensions;
 
+use SilverStripe\Core\Extension;
+use SilverStripe\ORM\DataObject;
 use DNADesign\Elemental\Models\ElementalArea;
 use DNADesign\Elemental\Extensions\ElementalAreasExtension;
-use SilverStripe\Core\Extension;
 
 class ElementalContentControllerExtension extends Extension
 {
@@ -24,27 +25,55 @@ public function handleElement()
             return false;
         }
 
-        /** @var SiteTree $elementOwner */
+        /** @var DataObject $elementOwner */
         $elementOwner = $this->owner->data();
 
-        $elementalAreaRelations = $this->owner->getElementalRelations();
+        if (!$elementOwner->hasExtension(ElementalAreasExtension::class)) {
+            user_error(get_class($elementOwner) . ' doesnt have the ElementalAreasExtension applied', E_USER_ERROR);
+            return false;
+        }
 
+        $elementalAreaRelations = $elementOwner->getElementalRelations();
         if (!$elementalAreaRelations) {
-            user_error(get_class($this->owner) . ' has no ElementalArea relationships', E_USER_ERROR);
+            user_error(get_class($elementOwner) . ' has no ElementalArea relationships', E_USER_ERROR);
             return false;
         }
 
-        foreach ($elementalAreaRelations as $elementalAreaRelation) {
-            $element = $elementOwner->$elementalAreaRelation()->Elements()
-                ->filter('ID', $id)
-                ->First();
+        $element = $this->findElement($elementalAreaRelations, $elementOwner, $id);
 
-            if ($element) {
-                return $element->getController();
-            }
+        if ($element) {
+            return $element->getController();
         }
 
         user_error('Element $id not found for this page', E_USER_ERROR);
         return false;
     }
+
+    private function findElement(iterable $elementalAreaRelations, DataObject $owner, $id): ?DataObject
+    {
+        foreach ($elementalAreaRelations as $elementalAreaRelation) {
+            $elements = $owner->$elementalAreaRelation()->Elements();
+            $found = $elements->filter('ID', $id)->First();
+
+            if ($found) {
+                return $found;
+            }
+
+            /** @var BaseElement $element */
+            foreach ($elements as $element) {
+                if (!$element->hasExtension(ElementalAreasExtension::class)) {
+                    continue;
+                }
+
+                /** @var BaseElement&ElementalAreasExtension $element */
+                $found = $this->findElement($element->getElementalRelations(), $element, $id);
+
+                if ($found) {
+                    return $found;
+                }
+            }
+        }
+
+        return null;
+    }
 }