mirror of
https://github.com/rectorphp/rector.git
synced 2024-06-06 03:10:50 +00:00
163 lines
4.5 KiB
PHP
163 lines
4.5 KiB
PHP
<?php declare(strict_types=1);
|
|
|
|
namespace Rector\NodeVisitor\DependencyInjection\NamedServicesToConstructor;
|
|
|
|
use PhpParser\Node;
|
|
use PhpParser\Node\Expr\Assign;
|
|
use PhpParser\Node\Expr\MethodCall;
|
|
use PhpParser\Node\Expr\PropertyFetch;
|
|
use PhpParser\Node\Expr\Variable;
|
|
use PhpParser\Node\Scalar\String_;
|
|
use PhpParser\NodeVisitorAbstract;
|
|
use Rector\Builder\Kernel\ServiceFromKernelResolver;
|
|
use Rector\Builder\Naming\NameResolver;
|
|
use Rector\Tests\NodeVisitor\DependencyInjection\NamedServicesToConstructorReconstructor\Source\LocalKernel;
|
|
|
|
/**
|
|
* Converts all:
|
|
* $this->get('some_service')
|
|
*
|
|
* into:
|
|
* $this->someService
|
|
*/
|
|
final class GetterToPropertyNodeVisitor extends NodeVisitorAbstract
|
|
{
|
|
/**
|
|
* @var NameResolver
|
|
*/
|
|
private $nameResolver;
|
|
|
|
/**
|
|
* @var ServiceFromKernelResolver
|
|
*/
|
|
private $serviceFromKernelResolver;
|
|
|
|
public function __construct(NameResolver $nameResolver, ServiceFromKernelResolver $serviceFromKernelResolver)
|
|
{
|
|
$this->nameResolver = $nameResolver;
|
|
$this->serviceFromKernelResolver = $serviceFromKernelResolver;
|
|
}
|
|
|
|
/**
|
|
* Return value semantics:
|
|
* * null
|
|
* => $node stays as-is
|
|
* * NodeTraverser::DONT_TRAVERSE_CHILDREN
|
|
* => Children of $node are not traversed. $node stays as-is
|
|
* * NodeTraverser::STOP_TRAVERSAL
|
|
* => Traversal is aborted. $node stays as-is
|
|
* * otherwise
|
|
* => $node is set to the return value
|
|
*
|
|
* @return null|int|Node
|
|
*/
|
|
public function enterNode(Node $node)
|
|
{
|
|
if ($this->isCandidate($node)) {
|
|
$this->reconstruct($node);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
private function isCandidate(Node $node): bool
|
|
{
|
|
// $var = $this->get('some_service');
|
|
// $var = $this->get('some_service')->getData();
|
|
if ($node instanceof Assign) {
|
|
if ($node->expr instanceof MethodCall || $node->var instanceof MethodCall) {
|
|
if ($this->isContainerGetCall($node->expr)) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
// ['var => $this->get('some_service')->getData()]
|
|
if ($node instanceof MethodCall && $node->var instanceof MethodCall) {
|
|
if ($this->isContainerGetCall($node->var)) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* @param Assign|MethodCall $assignOrMethodCallNode
|
|
*/
|
|
private function reconstruct(Node $assignOrMethodCallNode): void
|
|
{
|
|
if ($assignOrMethodCallNode instanceof Assign) {
|
|
$this->processAssignment($assignOrMethodCallNode);
|
|
}
|
|
|
|
if ($assignOrMethodCallNode instanceof MethodCall) {
|
|
$this->processMethodCall($assignOrMethodCallNode);
|
|
}
|
|
}
|
|
|
|
private function processAssignment(Assign $assignNode): void
|
|
{
|
|
$refactoredMethodCall = $this->processMethodCallNode($assignNode->expr);
|
|
if ($refactoredMethodCall) {
|
|
$assignNode->expr = $refactoredMethodCall;
|
|
}
|
|
}
|
|
|
|
private function processMethodCall(MethodCall $methodCallNode): void
|
|
{
|
|
$refactoredMethodCall = $this->processMethodCallNode($methodCallNode->var);
|
|
if ($refactoredMethodCall) {
|
|
$methodCallNode->var = $refactoredMethodCall;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Is "$this->get('string')" statements?
|
|
*/
|
|
private function isContainerGetCall(MethodCall $methodCall): bool
|
|
{
|
|
if ($methodCall->var->name !== 'this') {
|
|
return false;
|
|
}
|
|
|
|
if ((string) $methodCall->name !== 'get') {
|
|
return false;
|
|
}
|
|
|
|
if (! $methodCall->args[0]->value instanceof String_) {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
private function processMethodCallNode(MethodCall $methodCall): ?PropertyFetch
|
|
{
|
|
/** @var String_ $argument */
|
|
$argument = $methodCall->args[0]->value;
|
|
$serviceName = $argument->value;
|
|
|
|
$serviceType = $this->serviceFromKernelResolver->resolveServiceClassByNameFromKernel(
|
|
$serviceName, LocalKernel::class
|
|
);
|
|
|
|
$propertyName = $this->nameResolver->resolvePropertyNameFromType($serviceType);
|
|
|
|
return $this->createPropertyFetch($propertyName);
|
|
|
|
}
|
|
|
|
/**
|
|
* Creates "$this->propertyName"
|
|
*/
|
|
private function createPropertyFetch(string $propertyName): PropertyFetch
|
|
{
|
|
return new PropertyFetch(
|
|
new Variable('this', [
|
|
'name' => $propertyName
|
|
]), $propertyName
|
|
);
|
|
}
|
|
}
|