Cycle\ORM\Promise\Visitor\AddProxiedMethods

Complexity

Total Complexity

20

Size/Duplication

Total Lines	122
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	9

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	leaveNode()	0	20	5
A	hasReturnStmt()	0	16	5
A	findReturnStmt()	0	18	6
A	modifyMethodMethod()	0	10	1
A	packMethodArgs()	0	10	2

<?php

/**
 * Spiral Framework.
 *
 * @license MIT
 * @author  Valentin V (Vvval)
 */
declare(strict_types=1);

namespace Cycle\ORM\Promise\Visitor;

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

/**
 * Add parent call via proxy resolver
 */
final class AddProxiedMethods extends NodeVisitorAbstract
{
    /** @var string */
    private $resolverProperty;

    /** @var Node\Stmt\ClassMethod[] */
    private $methods;

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

    /**
     * @param string $property
     * @param array  $methods
     * @param string $resolveMethod
     */
    public function __construct(string $property, array $methods, string $resolveMethod)
    {
        $this->resolverProperty = $property;
        $this->methods = $methods;
        $this->resolveMethod = $resolveMethod;
    }

    /**
     * {@inheritdoc}
     */
    public function leaveNode(Node $node)
    {
        if (!$node instanceof Node\Stmt\Class_) {
            return null;
        }

        foreach ($this->methods as $method) {
            if ($method->name->name === '__clone') {
                $method->stmts = [Expressions::buildCloneExpression($this->resolverProperty)];
                $node->stmts[] = $method;
            } else {
                $node->stmts[] = $this->modifyMethodMethod(
                    $method,
                    $this->hasReturnStmt($method) ? Node\Stmt\Return_::class : Node\Stmt\Expression::class
                );
            }
        }

        return $node;
    }

    /**
     * @param Node\Stmt\ClassMethod $method
     * @return bool
     */
    private function hasReturnStmt(Node\Stmt\ClassMethod $method): bool
    {
        if ($method->returnType === 'void') {
            return false;
        }

        if ($method->returnType === null) {
            return $this->findReturnStmt($method);
        }

        if ($method->returnType instanceof Node\NullableType) {
            return true;
        }

        return $method->returnType instanceof Node\Identifier && $method->returnType->name !== 'void';
    }

    /**
     * @param Node\Stmt|Node\Stmt\ClassMethod $node
     * @return bool
     */
    private function findReturnStmt(Node\Stmt $node): bool
    {
        if (!property_exists($node, 'stmts') || !is_array($node->stmts)) {
            return false;
        }

        foreach ($node->stmts as $stmt) {
            if ($stmt instanceof Node\Stmt\Return_) {
                return true;
            }

            if ($this->findReturnStmt($stmt) === true) {
                return true;
            }
        }

        return false;
    }

    /**
     * @param Node\Stmt\ClassMethod $method
     * @param string                $stmtWrapper
     * @return Node\Stmt\ClassMethod
     */
    private function modifyMethodMethod(Node\Stmt\ClassMethod $method, string $stmtWrapper): Node\Stmt\ClassMethod
    {
        $resolved = Expressions::resolveMethodCall('this', $this->resolverProperty, $this->resolveMethod);
        $methodCall = new Node\Expr\MethodCall($resolved, $method->name->name, $this->packMethodArgs($method));

        $method->setDocComment(PHPDoc::writeInheritdoc());
        $method->stmts = [Expressions::throwExceptionOnNull($resolved, new $stmtWrapper($methodCall))];

        return $method;
    }

    /**
     * @param Node\Stmt\ClassMethod $method
     * @return array
     */
    private function packMethodArgs(Node\Stmt\ClassMethod $method): array
    {
        $args = [];
        /** @var \PhpParser\Node\Param $param */
        foreach ($method->getParams() as $param) {
            $args[] = (new Param($param->var->name))->getNode();
        }

        return $args;
    }
}