[CodeQuality] Handle on assign on FlipTypeControlToUseExclusiveTypeRe…

…ctor (#6653)

* [CodeQuality] Handle on assign on FlipTypeControlToUseExclusiveTypeRector

* fix

* skip on assign

* clean

* fix

* fix

* fix

rules-tests/CodeQuality/Rector/Identical/FlipTypeControlToUseExclusiveTypeRector/Fixture/on_assign.php.inc

@@ -0,0 +1,35 @@
+<?php
+
+namespace Rector\Tests\CodeQuality\Rector\Identical\FlipTypeControlToUseExclusiveTypeRector\Fixture;
+
+class OnAssign
+{
+    public function run(?\DateTimeImmutable $param)
+    {
+        if (null === ($paramValue = $param)) {
+            return 'no';
+        }
+
+        return $paramValue;
+    }
+}
+
+?>
+-----
+<?php
+
+namespace Rector\Tests\CodeQuality\Rector\Identical\FlipTypeControlToUseExclusiveTypeRector\Fixture;
+
+class OnAssign
+{
+    public function run(?\DateTimeImmutable $param)
+    {
+        if (!($paramValue = $param) instanceof \DateTimeImmutable) {
+            return 'no';
+        }
+
+        return $paramValue;
+    }
+}
+
+?>

rules/CodeQuality/Rector/Identical/FlipTypeControlToUseExclusiveTypeRector.php

@@ -6,12 +6,14 @@
 
 use PhpParser\Node;
 use PhpParser\Node\Expr;
+use PhpParser\Node\Expr\Assign;
 use PhpParser\Node\Expr\BinaryOp\Identical;
 use PhpParser\Node\Expr\BinaryOp\NotIdentical;
 use PhpParser\Node\Expr\BooleanNot;
 use PhpParser\Node\Expr\Instanceof_;
 use PhpParser\Node\Name\FullyQualified;
 use PHPStan\Type\ObjectType;
+use Rector\NodeTypeResolver\Node\AttributeKey;
 use Rector\PhpParser\Node\Value\ValueResolver;
 use Rector\Rector\AbstractRector;
 use Rector\StaticTypeMapper\ValueObject\Type\AliasedObjectType;
@@ -91,6 +93,10 @@ private function processConvertToExclusiveType(
             ? $objectType->getFullyQualifiedName()
             : $objectType->getClassName();
 
+        if ($expr instanceof Assign) {
+            $expr->setAttribute(AttributeKey::WRAPPED_IN_PARENTHESES, true);
+        }
+
         $instanceof = new Instanceof_($expr, new FullyQualified($fullyQualifiedType));
         if ($binaryOp instanceof NotIdentical) {
             return $instanceof;

rules/DeadCode/Rector/If_/RemoveDeadInstanceOfRector.php

@@ -22,6 +22,7 @@
 use PHPStan\Type\MixedType;
 use PHPStan\Type\ObjectType;
 use Rector\NodeManipulator\IfManipulator;
+use Rector\NodeTypeResolver\Node\AttributeKey;
 use Rector\Rector\AbstractRector;
 use Rector\Reflection\ReflectionResolver;
 use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
@@ -110,6 +111,7 @@ private function refactorStmtAndInstanceof(If_ $if, Instanceof_ $instanceof): nu
         }
 
         if ($instanceof->expr instanceof Assign) {
+            $instanceof->expr->setAttribute(AttributeKey::WRAPPED_IN_PARENTHESES, false);
             $assignExpression = new Expression($instanceof->expr);
             return array_merge([$assignExpression], $if->stmts);
         }