Unfuck class constant visibilities as well ...

jaem3l · Feb 26, 2022 · 21e9a46 · 21e9a46
1 parent fd5204c
commit 21e9a46
Show file tree

Hide file tree

Showing 4 changed files with 49 additions and 44 deletions.
diff --git a/composer.lock b/composer.lock
diff --git a/src/Unfucker/VisibilityUnfucker.php b/src/Unfucker/VisibilityUnfucker.php
@@ -6,6 +6,7 @@
 
 use PhpParser\Node;
 use PhpParser\Node\Stmt\Class_;
+use PhpParser\Node\Stmt\ClassConst;
 use PhpParser\Node\Stmt\ClassMethod;
 use PhpParser\Node\Stmt\Property;
 use PhpParser\NodeVisitorAbstract;
@@ -14,7 +15,7 @@ class VisibilityUnfucker extends NodeVisitorAbstract
 {
     public function leaveNode(Node $node): void
     {
-        if (!$node instanceof ClassMethod && !$node instanceof Property) {
+        if (!$node instanceof ClassMethod && !$node instanceof Property && !$node instanceof ClassConst) {
             return;
         }
 

diff --git a/tests/samples/class-visibility-input.php b/tests/samples/class-visibility-input.php
@@ -2,6 +2,8 @@
 
 class Example
 {
+    private const TEST = 123;
+    protected const ONETWOTHREE = 'test';
     private $foo;
     protected $bar;
     private function foo()

diff --git a/tests/samples/class-visibility-unfucked.php b/tests/samples/class-visibility-unfucked.php
@@ -2,6 +2,8 @@
 
 class Example
 {
+    public const TEST = 123;
+    public const ONETWOTHREE = 'test';
     public $foo;
     public $bar;
     public function foo()