56 lines
1.9 KiB
PHP
56 lines
1.9 KiB
PHP
<?php
|
|
|
|
declare(strict_types = 1);
|
|
|
|
namespace App\PHPStan;
|
|
|
|
use App\Core\DB\DB;
|
|
use PhpParser\Node\Expr\StaticCall;
|
|
use PhpParser\Node\Name;
|
|
use PhpParser\Node\Scalar\String_;
|
|
use PHPStan\Analyser\Scope;
|
|
use PHPStan\Reflection\MethodReflection;
|
|
use PHPStan\Type\DynamicStaticMethodReturnTypeExtension;
|
|
use PHPStan\Type\NullType;
|
|
use PHPStan\Type\UnionType;
|
|
|
|
class ClassFromTableNameDynamicStaticMethodReturnTypeExtension implements DynamicStaticMethodReturnTypeExtension
|
|
{
|
|
public function __construct(private ?GNUsocialProvider $provider = null)
|
|
{
|
|
}
|
|
|
|
public function getClass(): string
|
|
{
|
|
return \App\Core\DB\DB::class;
|
|
}
|
|
|
|
public function isStaticMethodSupported(MethodReflection $methodReflection): bool
|
|
{
|
|
return \in_array($methodReflection->getName(), DB::METHODS_ACCEPTING_TABLE_NAME);
|
|
}
|
|
|
|
/**
|
|
* For calls to DB::find and such, if the first argument is a
|
|
* constant string, it's a table name, so convert it to the
|
|
* corresponding entity. Only run if the environment variable
|
|
* PHPSTAN_BOOT_KERNEL is defined
|
|
*/
|
|
public function getTypeFromStaticMethodCall(
|
|
MethodReflection $methodReflection,
|
|
StaticCall $staticCall,
|
|
Scope $scope,
|
|
): \PHPStan\Type\Type {
|
|
if (isset($_ENV['PHPSTAN_BOOT_KERNEL']) && \count($staticCall->args) >= 1 && ($arg = $staticCall->args[0]->value) instanceof String_) {
|
|
// If called with the first argument as a string, it's a table name
|
|
return new UnionType([
|
|
$scope->resolveTypeByName(new Name(DB::filterTableName($staticCall->name->toString(), [$arg->value]))),
|
|
new NullType(),
|
|
]);
|
|
} else {
|
|
// Let PHPStan handle it normally
|
|
return \PHPStan\Reflection\ParametersAcceptorSelector::selectFromArgs($scope, $staticCall->args, $methodReflection->getVariants())->getReturnType();
|
|
}
|
|
}
|
|
}
|