Find BelongsToThrough relationships by return type (#114)

staudenmeir · Feb 20, 2025 · 451496e · 451496e
1 parent 8a0405a
commit 451496e
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/IdeHelper/BelongsToThroughRelationsHook.php b/src/IdeHelper/BelongsToThroughRelationsHook.php
@@ -8,7 +8,7 @@
 use Illuminate\Database\Eloquent\Relations\Relation;
 use ReflectionClass;
 use ReflectionMethod;
-use Throwable;
+use ReflectionNamedType;
 use Znck\Eloquent\Relations\BelongsToThrough as BelongsToThroughRelation;
 use Znck\Eloquent\Traits\BelongsToThrough as BelongsToThroughTrait;
 
@@ -30,13 +30,11 @@ public function run(ModelsCommand $command, Model $model): void
                 continue;
             }
 
-            try {
+            if ($method->getReturnType() instanceof ReflectionNamedType
+                && $method->getReturnType()->getName() === BelongsToThroughRelation::class) {
+                /** @var \Illuminate\Database\Eloquent\Relations\Relation<*, *, *> $relationship */
                 $relationship = $method->invoke($model);
-            } catch (Throwable) { // @codeCoverageIgnore
-                continue; // @codeCoverageIgnore
-            }
 
-            if ($relationship instanceof BelongsToThroughRelation) {
                 $this->addRelationship($command, $method, $relationship);
             }
         }