*/
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);
}
}