rector-src/rules/CodingStyle/Rector/ClassConst/SplitGroupedClassConstantsRector.php at ae1a848b3239dde935d6493ae0a4a95c56a8290c · rectorphp/rector-src · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
<?php

declare(strict_types=1);

namespace Rector\CodingStyle\Rector\ClassConst;

use PhpParser\Node;
use PhpParser\Node\Const_;
use PhpParser\Node\Stmt\ClassConst;
use Rector\Rector\AbstractRector;
use Rector\RuleDoc\CodeSample\CodeSample;
use Rector\RuleDoc\RuleDefinition;

/**
 * @see \Rector\Tests\CodingStyle\Rector\ClassConst\SplitGroupedClassConstantsRector\SplitGroupedClassConstantsRectorTest
 */
final class SplitGroupedClassConstantsRector extends AbstractRector
{
    public function getRuleDefinition(): RuleDefinition
    {
        return new RuleDefinition(
            'Separate class constant to own lines',
            [
                new CodeSample(
                    <<<'CODE_SAMPLE'
class SomeClass
{
    const HI = true, HELLO = 'true';
}
CODE_SAMPLE
                    ,
                    <<<'CODE_SAMPLE'
class SomeClass
{
    const HI = true;
    const HELLO = 'true';
}
CODE_SAMPLE
                ),
            ]
        );
    }

    /**
     * @return array<class-string<Node>>
     */
    public function getNodeTypes(): array
    {
        return [ClassConst::class];
    }

    /**
     * @param ClassConst $node
     * @return ClassConst[]|null
     */
    public function refactor(Node $node): ?array
    {
        if (count($node->consts) < 2) {
            return null;
        }

        /** @var Const_[] $allConsts */
        $allConsts = $node->consts;

        /** @var Const_ $firstConst */
        $firstConst = array_shift($allConsts);
        $node->consts = [$firstConst];

        $nextClassConsts = $this->createNextClassConsts($allConsts, $node);

        return [$node, ...$nextClassConsts];
    }

    /**
     * @param Const_[] $consts
     * @return ClassConst[]
     */
    private function createNextClassConsts(array $consts, ClassConst $classConst): array
    {
        $decoratedConsts = [];

        foreach ($consts as $const) {
            $decoratedConsts[] = new ClassConst(
                [$const],
                $classConst->flags,
                $classConst->getAttributes(),
                array_values($classConst->attrGroups),
                $classConst->type
            );
        }

        return $decoratedConsts;
    }
}