added RelationshipReturnTypeExtension

extension.neon

@@ -3,6 +3,10 @@ services:
 		class: Nextras\OrmPhpStan\Types\CollectionReturnTypeExtension
 		tags:
 			- phpstan.broker.dynamicMethodReturnTypeExtension
+	-
+		class: Nextras\OrmPhpStan\Types\RelationshipReturnTypeExtension
+		tags:
+			- phpstan.broker.dynamicMethodReturnTypeExtension
 	-
 		class: Nextras\OrmPhpStan\Types\RepositoryReturnTypeExtension
 		tags:

src/Types/RelationshipReturnTypeExtension.php

@@ -0,0 +1,52 @@
+<?php declare(strict_types = 1);
+
+namespace Nextras\OrmPhpStan\Types;
+
+use Nextras\Orm\Collection\ICollection;
+use Nextras\Orm\Relationships\HasMany;
+use PhpParser\Node\Expr\MethodCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Reflection\ParametersAcceptorSelector;
+use PHPStan\Type\DynamicMethodReturnTypeExtension;
+use PHPStan\Type\IntersectionType;
+use PHPStan\Type\IterableType;
+use PHPStan\Type\ObjectType;
+use PHPStan\Type\Type;
+use PHPStan\Type\TypeCombinator;
+
+
+class RelationshipReturnTypeExtension implements DynamicMethodReturnTypeExtension
+{
+	public function getClass(): string
+	{
+		return HasMany::class;
+	}
+
+
+	public function isMethodSupported(MethodReflection $methodReflection): bool
+	{
+		return $methodReflection->getName() === 'get';
+	}
+
+
+	public function getTypeFromMethodCall(
+		MethodReflection $methodReflection,
+		MethodCall $methodCall,
+		Scope $scope
+	): Type
+	{
+		$varType = $scope->getType($methodCall->var);
+
+		if ($varType instanceof IntersectionType) {
+			foreach ($varType->getTypes() as $type) {
+				if ($type instanceof IterableType) {
+					$collectionType = new ObjectType(ICollection::class);
+					return TypeCombinator::intersect($type, $collectionType);
+				}
+			}
+		}
+
+		return ParametersAcceptorSelector::selectSingle($methodReflection->getVariants())->getReturnType();
+	}
+}

tests/expected.txt

@@ -6,5 +6,6 @@
 /testbox/Rules/Test.php:26:Entity NextrasTests\OrmPhpStan\Rules\Entity: property $age (int) does not accept string.
 /testbox/Types/CollectionTypesTest.php:15:Parameter #1 $author of method NextrasTests\OrmPhpStan\Types\CollectionTypesTest::takeAuthor() expects NextrasTests\OrmPhpStan\Types\Author, NextrasTests\OrmPhpStan\Types\Author|null given.
 /testbox/Types/CollectionTypesTest.php:16:Parameter #1 $author of method NextrasTests\OrmPhpStan\Types\CollectionTypesTest::takeAuthor() expects NextrasTests\OrmPhpStan\Types\Author, NextrasTests\OrmPhpStan\Types\Author|null given.
+/testbox/Types/RelationshipTypesTest.php:10:Parameter #1 $book of method NextrasTests\OrmPhpStan\Types\RelationshipTypesTest::takeBook() expects NextrasTests\OrmPhpStan\Types\Book, NextrasTests\OrmPhpStan\Types\Book|null given.
 /testbox/Types/RepositoryTypesTest.php:9:Parameter #1 $author of method NextrasTests\OrmPhpStan\Types\RepositoryTypesTest::takeAuthor() expects NextrasTests\OrmPhpStan\Types\Author, NextrasTests\OrmPhpStan\Types\Author|null given.
 /testbox/Types/RepositoryTypesTest.php:10:Parameter #1 $author of method NextrasTests\OrmPhpStan\Types\RepositoryTypesTest::takeAuthor() expects NextrasTests\OrmPhpStan\Types\Author, NextrasTests\OrmPhpStan\Types\Author|null given.

tests/testbox/Types/RelationshipTypesTest.php

@@ -0,0 +1,29 @@
+<?php declare(strict_types = 1);
+
+namespace NextrasTests\OrmPhpStan\Types;
+
+
+class RelationshipTypesTest
+{
+	public function testError(Author $author)
+	{
+		$this->takeBook($author->books->get()->fetch());
+	}
+
+
+	public function testOk(Author $author)
+	{
+		$this->takeBookNullable($author->books->get()->fetch());
+		$this->takeBookNullable($author->books->get()->findBy([])->fetch());
+	}
+
+
+	private function takeBook(Book $book)
+	{
+	}
+
+
+	private function takeBookNullable(?Book $book)
+	{
+	}
+}

tests/testbox/Types/fixtures/Author.php

@@ -2,7 +2,11 @@
 
 namespace NextrasTests\OrmPhpStan\Types;
 
+use Nextras\Orm\Relationships\OneHasMany;
+
+
 /**
+ * @property Book[]|OneHasMany $books {1:m Book::$author}
  */
 class Author extends \Nextras\Orm\Entity\Entity
 {

tests/testbox/Types/fixtures/Book.php

@@ -0,0 +1,9 @@
+<?php declare(strict_types = 1);
+
+namespace NextrasTests\OrmPhpStan\Types;
+
+/**
+ */
+class Book extends \Nextras\Orm\Entity\Entity
+{
+}