Bugfix/crash on morph relations (#12) (#13)

cybex-gmbh · Jun 7, 2024 · c08c844 · c08c844
2 parents 4691b83 + 46db78d
commit c08c844
Showing 1 changed file with 15 additions and 9 deletions.
diff --git a/src/ModelReflector.php b/src/ModelReflector.php
@@ -100,17 +100,23 @@ public function getModelRelations($model, bool $forceRefresh = false): Collectio
                 $methodName   = $reflectionMethod->getName();
                 $relation     = $modelInstance->{$methodName}();
                 $relatedModel = $relation->getRelated();
-
                 $relations->put($methodName, [
-                    'relation'                => $methodName,
-                    'returnType'              => $reflectionMethod->getReturnType()->getName(),
-                    'relatedClass'            => get_class($relatedModel),
-                    'relatedModel'            => $relatedModel,
-                    'relatedTable'            => $relatedModel->getTable(),
-                    'foreignKeyName'          => $relation->getForeignKeyName(),
-                    'qualifiedForeignKeyName' => $relation->getQualifiedForeignKeyName(),
-                    'isRelationParent'        => sprintf('%s.%s', $modelInstance->getTable(), $relation->getForeignKeyName()) !== $relation->getQualifiedForeignKeyName(),
+                    'relation'   => $methodName,
+                    'returnType' => $reflectionMethod->getReturnType()->getName(),
                 ]);
+
+                // Not available for some polymorph relations.
+                if (method_exists($relation, 'getForeignKeyName')) {
+                    $relations->put($methodName, [
+                        ...$relations->get($methodName),
+                        'relatedClass'            => get_class($relatedModel),
+                        'relatedModel'            => $relatedModel,
+                        'relatedTable'            => $relatedModel->getTable(),
+                        'foreignKeyName'          => $relation->getForeignKeyName(),
+                        'qualifiedForeignKeyName' => $relation->getQualifiedForeignKeyName(),
+                        'isRelationParent'        => sprintf('%s.%s', $modelInstance->getTable(), $relation->getForeignKeyName()) !== $relation->getQualifiedForeignKeyName(),
+                    ]);
+                }
             }
         }