Update CMSEditLinkAPI.php

sunnysideup · Jul 8, 2020 · beb13ed · beb13ed
1 parent 90f4982
commit beb13ed
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/Api/CMSEditLinkAPI.php b/src/Api/CMSEditLinkAPI.php
@@ -68,8 +68,10 @@ public static function find_edit_link_for_object($objectOrClassName, $action = n
         if ($modelAdminURLOverwrite) {
             $classFound = true;
         } else {
-            $modelNameToEdit = self::getModelAdmin($modelNameToEdit);
-            if($modelNameToEdit) {
+            $modelAdminResults = self::getModelAdmin($modelNameToEdit);
+            if(count($modelAdminResults)) {
+                $modelNameToEdit = $modelAdminResults['ModelNameToEdit'];
+                $myModelAdminclassObject = $modelAdminResults['MyModelAdminclassObject'];
                 $classFound = true;
             }
         }
@@ -100,7 +102,7 @@ protected static function getModelAdmin($modelNameToEdit) : string
     {
         $originalModelNameToEdit = $modelNameToEdit;
         if(! isset(self::$_cache[$originalModelNameToEdit])) {
-            self::$_cache[$originalModelNameToEdit] = '';
+            self::$_cache[$originalModelNameToEdit] = [];
             $classFound = false;
             foreach (ClassInfo::subclassesFor(ModelAdmin::class) as $i => $myAdminClassName) {
                 for ($includeChildren = 0; $includeChildren < 2; $includeChildren++) {
@@ -140,7 +142,10 @@ protected static function getModelAdmin($modelNameToEdit) : string
             }
         }
         if($classFound) {
-            self::$_cache[$originalModelNameToEdit] = $modelNameToEdit;
+            self::$_cache[$originalModelNameToEdit] = [
+                'ModelNameToEdit' => $modelNameToEdit,
+                'MyModelAdminclassObject' => $myModelAdminclassObject,
+            ];
         }
         return self::$_cache[$originalModelNameToEdit];
     }