Added error handling for association registeration

zunnu · Oct 21, 2022 · ab59046 · ab59046
1 parent 7160eba
commit ab59046
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/Gate.php b/src/Gate.php
@@ -163,10 +163,19 @@ private function _associations($model, $plugin = null) {
  if($associations = $setModel->associations()) {
  foreach ($associations->normalizeKeys($associations->getIterator()) as $key => $association) {
  $source = $association->getSource();
- $target = $association->getTarget();
  $sourceRegistery = 'App';
  $targetRegistery = 'App';
 
+ // error handling for association registeration
+ try {
+ $target = $association->getTarget();
+ } catch (\Exception $e) {
+ $association->getTarget()->setRegistryAlias(ucfirst($association->getProperty()) . $association->getName());
+ $association->getTarget()->setAlias(ucfirst($association->getProperty()) . $association->getName());
+ $association->setName(ucfirst($association->getProperty()) . $association->getName());
+ $target = $association->getTarget();
+ }
+
  if(strpos($source->getRegistryAlias(), '.') !== false) {
  $sourceRegistery = strtok($source->getRegistryAlias(), '.');