2019-10-13 05:59:52 +00:00
|
|
|
<?php
|
|
|
|
|
2021-05-09 20:15:43 +00:00
|
|
|
declare (strict_types=1);
|
2021-02-09 17:25:16 +00:00
|
|
|
namespace Rector\Privatization\NodeManipulator;
|
2018-07-31 08:54:00 +00:00
|
|
|
|
2021-07-22 23:37:17 +00:00
|
|
|
use PhpParser\Node\Param;
|
2018-07-31 08:54:00 +00:00
|
|
|
use PhpParser\Node\Stmt\Class_;
|
2018-07-31 09:15:27 +00:00
|
|
|
use PhpParser\Node\Stmt\ClassConst;
|
2018-07-31 08:54:00 +00:00
|
|
|
use PhpParser\Node\Stmt\ClassMethod;
|
|
|
|
use PhpParser\Node\Stmt\Property;
|
2020-12-24 22:01:30 +00:00
|
|
|
use Rector\Core\ValueObject\Visibility;
|
2021-11-18 07:58:51 +00:00
|
|
|
use RectorPrefix20211118\Webmozart\Assert\Assert;
|
2019-02-27 21:54:39 +00:00
|
|
|
final class VisibilityManipulator
|
2018-07-31 08:54:00 +00:00
|
|
|
{
|
2018-07-31 09:15:27 +00:00
|
|
|
/**
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Param|\PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $node
|
2018-07-31 09:15:27 +00:00
|
|
|
*/
|
2021-07-22 23:37:17 +00:00
|
|
|
public function hasVisibility($node, int $visibility) : bool
|
|
|
|
{
|
|
|
|
return (bool) ($node->flags & $visibility);
|
|
|
|
}
|
2019-02-09 13:22:20 +00:00
|
|
|
/**
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $node
|
2019-02-09 13:22:20 +00:00
|
|
|
*/
|
2021-06-29 13:37:16 +00:00
|
|
|
public function makeStatic($node) : void
|
2019-02-09 13:22:20 +00:00
|
|
|
{
|
2021-05-10 22:23:08 +00:00
|
|
|
$this->addVisibilityFlag($node, \Rector\Core\ValueObject\Visibility::STATIC);
|
2019-02-09 13:22:20 +00:00
|
|
|
}
|
2019-05-26 11:47:23 +00:00
|
|
|
/**
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\Class_|\PhpParser\Node\Stmt\ClassMethod $node
|
2019-05-26 11:47:23 +00:00
|
|
|
*/
|
2021-06-29 13:37:16 +00:00
|
|
|
public function makeAbstract($node) : void
|
2019-05-26 11:47:23 +00:00
|
|
|
{
|
2021-05-10 22:23:08 +00:00
|
|
|
$this->addVisibilityFlag($node, \Rector\Core\ValueObject\Visibility::ABSTRACT);
|
2019-05-26 11:47:23 +00:00
|
|
|
}
|
2020-09-10 07:27:03 +00:00
|
|
|
/**
|
2021-08-23 00:20:32 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $node
|
2020-09-10 07:27:03 +00:00
|
|
|
*/
|
2021-06-29 13:37:16 +00:00
|
|
|
public function makeNonStatic($node) : void
|
2020-09-10 07:27:03 +00:00
|
|
|
{
|
2021-05-09 20:15:43 +00:00
|
|
|
if (!$node->isStatic()) {
|
2020-09-17 07:34:27 +00:00
|
|
|
return;
|
|
|
|
}
|
2021-05-10 22:23:08 +00:00
|
|
|
$node->flags -= \PhpParser\Node\Stmt\Class_::MODIFIER_STATIC;
|
2020-09-10 07:27:03 +00:00
|
|
|
}
|
2019-08-25 10:29:15 +00:00
|
|
|
/**
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\Class_|\PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod $node
|
2019-08-25 10:29:15 +00:00
|
|
|
*/
|
2021-06-29 14:24:45 +00:00
|
|
|
public function makeFinal($node) : void
|
2019-08-25 10:29:15 +00:00
|
|
|
{
|
2021-05-10 22:23:08 +00:00
|
|
|
$this->addVisibilityFlag($node, \Rector\Core\ValueObject\Visibility::FINAL);
|
2019-08-25 10:29:15 +00:00
|
|
|
}
|
2020-12-05 21:33:30 +00:00
|
|
|
/**
|
2021-08-23 00:20:32 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\Class_|\PhpParser\Node\Stmt\ClassMethod $node
|
2020-12-05 21:33:30 +00:00
|
|
|
*/
|
2021-06-29 14:24:45 +00:00
|
|
|
public function makeNonFinal($node) : void
|
2020-12-05 21:33:30 +00:00
|
|
|
{
|
2021-05-09 20:15:43 +00:00
|
|
|
if (!$node->isFinal()) {
|
2020-12-05 21:33:30 +00:00
|
|
|
return;
|
|
|
|
}
|
2021-05-10 22:23:08 +00:00
|
|
|
$node->flags -= \PhpParser\Node\Stmt\Class_::MODIFIER_FINAL;
|
2020-12-05 21:33:30 +00:00
|
|
|
}
|
2020-11-17 10:45:31 +00:00
|
|
|
/**
|
|
|
|
* This way "abstract", "static", "final" are kept
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $node
|
2020-11-17 10:45:31 +00:00
|
|
|
*/
|
2021-06-29 14:24:45 +00:00
|
|
|
public function removeVisibility($node) : void
|
2020-11-17 10:45:31 +00:00
|
|
|
{
|
|
|
|
// no modifier
|
|
|
|
if ($node->flags === 0) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
if ($node->isPublic()) {
|
2021-05-10 22:23:08 +00:00
|
|
|
$node->flags -= \PhpParser\Node\Stmt\Class_::MODIFIER_PUBLIC;
|
2020-11-17 10:45:31 +00:00
|
|
|
}
|
|
|
|
if ($node->isProtected()) {
|
2021-05-10 22:23:08 +00:00
|
|
|
$node->flags -= \PhpParser\Node\Stmt\Class_::MODIFIER_PROTECTED;
|
2020-11-17 10:45:31 +00:00
|
|
|
}
|
|
|
|
if ($node->isPrivate()) {
|
2021-05-10 22:23:08 +00:00
|
|
|
$node->flags -= \PhpParser\Node\Stmt\Class_::MODIFIER_PRIVATE;
|
2020-11-17 10:45:31 +00:00
|
|
|
}
|
|
|
|
}
|
2020-11-27 21:29:40 +00:00
|
|
|
/**
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $node
|
2020-11-27 21:29:40 +00:00
|
|
|
*/
|
2021-06-29 14:24:45 +00:00
|
|
|
public function changeNodeVisibility($node, int $visibility) : void
|
2020-11-27 21:29:40 +00:00
|
|
|
{
|
2021-11-18 07:58:51 +00:00
|
|
|
\RectorPrefix20211118\Webmozart\Assert\Assert::oneOf($visibility, [\Rector\Core\ValueObject\Visibility::PUBLIC, \Rector\Core\ValueObject\Visibility::PROTECTED, \Rector\Core\ValueObject\Visibility::PRIVATE, \Rector\Core\ValueObject\Visibility::STATIC, \Rector\Core\ValueObject\Visibility::ABSTRACT, \Rector\Core\ValueObject\Visibility::FINAL]);
|
2020-12-24 23:18:02 +00:00
|
|
|
$this->replaceVisibilityFlag($node, $visibility);
|
2020-11-27 21:29:40 +00:00
|
|
|
}
|
|
|
|
/**
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $node
|
2020-11-27 21:29:40 +00:00
|
|
|
*/
|
2021-06-29 14:24:45 +00:00
|
|
|
public function makePublic($node) : void
|
2020-11-27 21:29:40 +00:00
|
|
|
{
|
2021-05-10 22:23:08 +00:00
|
|
|
$this->replaceVisibilityFlag($node, \Rector\Core\ValueObject\Visibility::PUBLIC);
|
2020-11-27 21:29:40 +00:00
|
|
|
}
|
|
|
|
/**
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $node
|
2020-11-27 21:29:40 +00:00
|
|
|
*/
|
2021-06-29 14:24:45 +00:00
|
|
|
public function makeProtected($node) : void
|
2020-11-27 21:29:40 +00:00
|
|
|
{
|
2021-05-10 22:23:08 +00:00
|
|
|
$this->replaceVisibilityFlag($node, \Rector\Core\ValueObject\Visibility::PROTECTED);
|
2020-11-27 21:29:40 +00:00
|
|
|
}
|
|
|
|
/**
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $node
|
2020-11-27 21:29:40 +00:00
|
|
|
*/
|
2021-06-29 14:24:45 +00:00
|
|
|
public function makePrivate($node) : void
|
2020-11-27 21:29:40 +00:00
|
|
|
{
|
2021-05-10 22:23:08 +00:00
|
|
|
$this->replaceVisibilityFlag($node, \Rector\Core\ValueObject\Visibility::PRIVATE);
|
2020-11-27 21:29:40 +00:00
|
|
|
}
|
2021-07-22 23:37:17 +00:00
|
|
|
/**
|
2021-08-23 00:20:32 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\Class_|\PhpParser\Node\Stmt\ClassConst $node
|
2021-07-22 23:37:17 +00:00
|
|
|
*/
|
|
|
|
public function removeFinal($node) : void
|
2021-02-08 21:00:45 +00:00
|
|
|
{
|
2021-07-22 23:37:17 +00:00
|
|
|
$node->flags -= \PhpParser\Node\Stmt\Class_::MODIFIER_FINAL;
|
2021-02-08 21:00:45 +00:00
|
|
|
}
|
2021-08-08 19:52:23 +00:00
|
|
|
public function removeAbstract(\PhpParser\Node\Stmt\ClassMethod $classMethod) : void
|
2021-08-08 16:58:46 +00:00
|
|
|
{
|
2021-08-08 19:52:23 +00:00
|
|
|
$classMethod->flags -= \PhpParser\Node\Stmt\Class_::MODIFIER_ABSTRACT;
|
2021-08-08 16:58:46 +00:00
|
|
|
}
|
2018-07-31 09:15:27 +00:00
|
|
|
/**
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Param|\PhpParser\Node\Stmt\Property $node
|
2018-07-31 09:15:27 +00:00
|
|
|
*/
|
2021-07-22 23:37:17 +00:00
|
|
|
public function makeReadonly($node) : void
|
2018-07-31 09:15:27 +00:00
|
|
|
{
|
2021-07-22 23:37:17 +00:00
|
|
|
$this->addVisibilityFlag($node, \PhpParser\Node\Stmt\Class_::MODIFIER_READONLY);
|
2018-07-31 09:15:27 +00:00
|
|
|
}
|
2021-07-22 23:37:17 +00:00
|
|
|
/**
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Param|\PhpParser\Node\Stmt\Class_|\PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $node
|
2021-07-22 23:37:17 +00:00
|
|
|
*/
|
|
|
|
private function addVisibilityFlag($node, int $visibility) : void
|
2018-07-31 08:54:00 +00:00
|
|
|
{
|
2021-07-22 23:37:17 +00:00
|
|
|
$node->flags |= $visibility;
|
2018-07-31 08:54:00 +00:00
|
|
|
}
|
2020-11-30 09:51:11 +00:00
|
|
|
/**
|
2021-10-30 14:18:31 +00:00
|
|
|
* @param \PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $node
|
2020-11-30 09:51:11 +00:00
|
|
|
*/
|
2021-06-29 14:24:45 +00:00
|
|
|
private function replaceVisibilityFlag($node, int $visibility) : void
|
2020-11-30 09:51:11 +00:00
|
|
|
{
|
2021-05-10 22:23:08 +00:00
|
|
|
$isStatic = $node instanceof \PhpParser\Node\Stmt\ClassMethod && $node->isStatic();
|
2020-12-23 14:28:30 +00:00
|
|
|
if ($isStatic) {
|
2021-01-30 21:41:25 +00:00
|
|
|
$this->removeVisibility($node);
|
2020-12-23 14:28:30 +00:00
|
|
|
}
|
2021-05-10 22:23:08 +00:00
|
|
|
if ($visibility !== \Rector\Core\ValueObject\Visibility::STATIC && $visibility !== \Rector\Core\ValueObject\Visibility::ABSTRACT && $visibility !== \Rector\Core\ValueObject\Visibility::FINAL) {
|
2021-01-30 21:41:25 +00:00
|
|
|
$this->removeVisibility($node);
|
2020-11-30 09:51:11 +00:00
|
|
|
}
|
|
|
|
$this->addVisibilityFlag($node, $visibility);
|
2020-12-23 14:28:30 +00:00
|
|
|
if ($isStatic) {
|
|
|
|
$this->makeStatic($node);
|
|
|
|
}
|
2020-11-30 09:51:11 +00:00
|
|
|
}
|
2018-07-31 08:54:00 +00:00
|
|
|
}
|