| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117 | <?php declare(strict_types=1);namespace PhpParser\Builder;use PhpParser;use PhpParser\BuilderHelpers;use PhpParser\Node;use PhpParser\Node\Identifier;use PhpParser\Node\Name;use PhpParser\Node\Stmt;class Enum_ extends Declaration{    protected $name;    protected $scalarType = null;    protected $implements = [];    protected $uses = [];    protected $enumCases = [];    protected $constants = [];    protected $methods = [];    /** @var Node\AttributeGroup[] */    protected $attributeGroups = [];    /**     * Creates an enum builder.     *     * @param string $name Name of the enum     */    public function __construct(string $name) {        $this->name = $name;    }    /**     * Sets the scalar type.     *     * @param string|Identifier $type     *     * @return $this     */    public function setScalarType($scalarType) {        $this->scalarType = BuilderHelpers::normalizeType($scalarType);        return $this;    }    /**     * Implements one or more interfaces.     *     * @param Name|string ...$interfaces Names of interfaces to implement     *     * @return $this The builder instance (for fluid interface)     */    public function implement(...$interfaces) {        foreach ($interfaces as $interface) {            $this->implements[] = BuilderHelpers::normalizeName($interface);        }        return $this;    }    /**     * Adds a statement.     *     * @param Stmt|PhpParser\Builder $stmt The statement to add     *     * @return $this The builder instance (for fluid interface)     */    public function addStmt($stmt) {        $stmt = BuilderHelpers::normalizeNode($stmt);        $targets = [            Stmt\TraitUse::class    => &$this->uses,            Stmt\EnumCase::class    => &$this->enumCases,            Stmt\ClassConst::class  => &$this->constants,            Stmt\ClassMethod::class => &$this->methods,        ];        $class = \get_class($stmt);        if (!isset($targets[$class])) {            throw new \LogicException(sprintf('Unexpected node of type "%s"', $stmt->getType()));        }        $targets[$class][] = $stmt;        return $this;    }    /**     * Adds an attribute group.     *     * @param Node\Attribute|Node\AttributeGroup $attribute     *     * @return $this The builder instance (for fluid interface)     */    public function addAttribute($attribute) {        $this->attributeGroups[] = BuilderHelpers::normalizeAttribute($attribute);        return $this;    }    /**     * Returns the built class node.     *     * @return Stmt\Enum_ The built enum node     */    public function getNode() : PhpParser\Node {        return new Stmt\Enum_($this->name, [            'scalarType' => $this->scalarType,            'implements' => $this->implements,            'stmts' => array_merge($this->uses, $this->enumCases, $this->constants, $this->methods),            'attrGroups' => $this->attributeGroups,        ], $this->attributes);    }}
 |