add related selection check

fykosak · Aug 16, 2023 · fd245e8 · fd245e8
1 parent aae7972
commit fd245e8
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/Model.php b/src/Model.php
@@ -24,6 +24,18 @@ final public function __construct(array $data, Selection $table)
         parent::__construct($data, $table);
     }
 
+    /**
+     * @phpstan-return TypedGroupedSelection<Model>
+     */
+    public function related(string $key, ?string $throughColumn = null): TypedGroupedSelection
+    {
+        $selection = parent::related($key, $throughColumn);
+        if ($selection instanceof TypedGroupedSelection) {
+            return $selection;
+        }
+        throw new \InvalidArgumentException();
+    }
+
     /**
      * @return Model|mixed
      * @throws MemberAccessException|\ReflectionException