cycle / proxy-factory

src/Visitor/AddUnsetPropertiesConst.php

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

namespace Cycle\ORM\Promise\Visitor;

use Cycle\ORM\Promise\Utils;
use PhpParser\Node;
use PhpParser\NodeVisitorAbstract;

/**
 * Add "unset properties" property
 */
final class AddUnsetPropertiesConst extends NodeVisitorAbstract
{
    /** @var string */
    private $unsetPropertyConst;

    /** @var array */
    private $unsetPropertiesValues;

    /**
     * @param string $unsetPropertiesConst
     * @param array  $unsetPropertiesValues
     */
    public function __construct(string $unsetPropertiesConst, array $unsetPropertiesValues)
    {
        $this->unsetPropertyConst = $unsetPropertiesConst;
        $this->unsetPropertiesValues = $unsetPropertiesValues;
    }

    /**
     * {@inheritdoc}
     */
    public function leaveNode(Node $node)

This method seems to be duplicated in your project.

    {
        if ($node instanceof Node\Stmt\Class_) {
            $node->stmts = Utils::injectValues($node->stmts, $this->definePlacementID($node), [$this->buildProperty()]);
        }

        return null;
    }

    /**
     * @param Node\Stmt\Class_ $node
     * @return int
     */
    private function definePlacementID(Node\Stmt\Class_ $node): int
    {
        foreach ($node->stmts as $index => $child) {
            if ($child instanceof Node\Stmt\ClassMethod) {
                return $index;
            }
        }

        return 0;
    }

    /**
     * @return Node\Stmt\ClassConst
     */
    private function buildProperty(): Node\Stmt\ClassConst
    {
        $array = [];
        foreach ($this->unsetPropertiesValues as $value) {
            $array[] = new Node\Expr\ArrayItem(new Node\Scalar\String_($value));
        }

        $const = new Node\Stmt\ClassConst([
            new Node\Const_(
                $this->unsetPropertyConst,
                new Node\Expr\Array_($array, ['kind' => Node\Expr\Array_::KIND_SHORT])
            )
        ], Node\Stmt\Class_::MODIFIER_PRIVATE);

        return $const;
    }
}