*/ final class TraitTypeResolver implements NodeTypeResolverInterface { /** * @readonly * @var \PHPStan\Reflection\ReflectionProvider */ private $reflectionProvider; public function __construct(ReflectionProvider $reflectionProvider) { $this->reflectionProvider = $reflectionProvider; } /** * @return array> */ public function getNodeClasses() : array { return [Trait_::class]; } /** * @param Trait_ $node */ public function resolve(Node $node) : Type { $traitName = (string) $node->namespacedName; if (!$this->reflectionProvider->hasClass($traitName)) { return new MixedType(); } $classReflection = $this->reflectionProvider->getClass($traitName); $types = []; $types[] = new ObjectType($traitName); foreach ($classReflection->getTraits() as $usedTraitReflection) { $types[] = new ObjectType($usedTraitReflection->getName()); } if (\count($types) === 1) { return $types[0]; } return new UnionType($types); } }