cycle / proxy-factory

use PhpParser\Node;

use PhpParser\NodeVisitorAbstract;

final class AddMagicIssetMethod extends NodeVisitorAbstract

{

    /** @var string */

    private $resolverProperty;

    /** @var string */

    private $resolveMethod;

    /** @var string */

    private $unsetPropertiesProperty;

    /**

     * @param string $resolverProperty

     * @param string $resolveMethod

     * @param string $unsetPropertiesProperty

     */

    public function __construct(string $resolverProperty, string $resolveMethod, string $unsetPropertiesProperty)

    {

        $this->unsetPropertiesProperty = $unsetPropertiesProperty;

        $this->resolveMethod = $resolveMethod;

        $this->resolverProperty = $resolverProperty;

    }

    /**

     * @param Node $node

     * @return int|Node|Node[]|null

     */

    public function leaveNode(Node $node)

    {

        if ($node instanceof Node\Stmt\Class_) {

            $method = new Builder\Method('__isset');

            $method->makePublic();

            $method->addParam(new Builder\Param('name'));

            $method->addStmt($this->buildIssetExpression());

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

        }

        return null;

    }

    /**

     * @return Node\Stmt\If_

     */

    private function buildIssetExpression(): Node\Stmt\If_

    {

        $if = new Node\Stmt\If_(Expressions::inConstArrayFunc('name', 'self', $this->unsetPropertiesProperty));

        $if->stmts[] = Expressions::resolveIntoVar('entity', 'this', $this->resolverProperty, $this->resolveMethod);

        $if->stmts[] = Expressions::throwExceptionOnNull(

            new Node\Expr\Variable('entity'),

            new Node\Stmt\Return_(Expressions::issetFunc('entity', '{$name}'))

        );

        $if->else = new Node\Stmt\Else_([

            new Node\Stmt\Return_(Expressions::issetFunc('this', '{$name}'))

        ]);

        return $if;

    }

}

use PhpParser\Node;

use PhpParser\NodeVisitorAbstract;

final class AddMagicUnset extends NodeVisitorAbstract

{

    /** @var string */

    private $resolverProperty;

    /** @var string */

    private $resolveMethod;

    /** @var string */

    private $unsetPropertiesConst;

    /**

     * @param string $resolverProperty

     * @param string $resolveMethod

     * @param string $unsetPropertiesConst

     */

    public function __construct(string $resolverProperty, string $resolveMethod, string $unsetPropertiesConst)

    {

        $this->resolveMethod = $resolveMethod;

        $this->resolverProperty = $resolverProperty;

        $this->unsetPropertiesConst = $unsetPropertiesConst;

    }

    /**

     * @param Node $node

     * @return int|Node|Node[]|null

     */

    public function leaveNode(Node $node)

    {

        if ($node instanceof Node\Stmt\Class_) {

            $method = new Builder\Method('__unset');

            $method->makePublic();

            $method->addParam(new Builder\Param('name'));

            $method->addStmt($this->buildUnsetExpression());

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

        }

        return null;

    }

    /**

     * @return Node\Stmt\If_

     */

    private function buildUnsetExpression(): Node\Stmt\If_

    {

        $if = new Node\Stmt\If_(Expressions::inConstArrayFunc('name', 'self', $this->unsetPropertiesConst));

        $if->stmts[] = Expressions::resolveIntoVar('entity', 'this', $this->resolverProperty, $this->resolveMethod);

        $if->stmts[] = Expressions::throwExceptionOnNull(

            new Node\Expr\Variable('entity'),

            new Node\Stmt\Expression(Expressions::unsetFunc('entity', '{$name}'))

        );

        $if->else = new Node\Stmt\Else_([

            new Node\Stmt\Expression(Expressions::unsetFunc('this', '{$name}'))

        ]);

        return $if;

    }

}