Cycle\ORM\Promise\Visitor\AddPromiseMethod

Complexity

Total Complexity

4

Size/Duplication

Total Lines	35
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	leaveNode()	0	16	3

<?php
declare(strict_types=1);

namespace Cycle\ORM\Promise\Visitor;

use Cycle\ORM\Promise\Expressions;
use Cycle\ORM\Promise\PHPDoc;
use PhpParser\Builder;
use PhpParser\Node;
use PhpParser\NodeVisitorAbstract;

class AddPromiseMethod extends NodeVisitorAbstract
{
    /** @var string */
    private $resolverProperty;

    /** @var string */
    private $name;

    /** @var string|null */
    private $returnType;

    public function __construct(string $resolverProperty, string $name, string $returnType = null)
    {
        $this->resolverProperty = $resolverProperty;
        $this->name = $name;
        $this->returnType = $returnType;
    }

    public function leaveNode(Node $node)
    {
        if ($node instanceof Node\Stmt\Class_) {
            $method = new Builder\Method($this->name);
            $method->makePublic();
            $method->addStmt(new Node\Stmt\Return_(Expressions::resolveMethodCall('this', $this->resolverProperty, $this->name)));
            if ($this->returnType !== null) {
                $method->setReturnType($this->returnType);
            }
            $method->setDocComment(PHPDoc::writeInheritdoc());

            $node->stmts[] = $method->getNode();
        }

        return null;
    }
}