[CodeQuality] Add ForRepeatedCountToOwnVariableRector

config/set/code-quality/code-quality.yaml

@@ -49,3 +49,4 @@ services:
     Rector\CodeQuality\Rector\Ternary\ArrayKeyExistsTernaryThenValueToCoalescingRector: ~
     Rector\CodeQuality\Rector\Include_\AbsolutizeRequireAndIncludePathRector: ~
     Rector\CodeQuality\Rector\FuncCall\ChangeArrayPushToArrayAssignRector: ~
+    Rector\CodeQuality\Rector\For_\ForRepeatedCountToOwnVariableRector: ~

packages/CodeQuality/src/Rector/For_/ForRepeatedCountToOwnVariableRector.php

@@ -0,0 +1,106 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\CodeQuality\Rector\For_;
+
+use PhpParser\Node;
+use PhpParser\Node\Expr\Assign;
+use PhpParser\Node\Expr\FuncCall;
+use PhpParser\Node\Expr\Variable;
+use PhpParser\Node\Stmt\For_;
+use Rector\Rector\AbstractRector;
+use Rector\RectorDefinition\CodeSample;
+use Rector\RectorDefinition\RectorDefinition;
+
+/**
+ * @see \Rector\CodeQuality\Tests\Rector\For_\ForRepeatedCountToOwnVariableRector\ForRepeatedCountToOwnVariableRectorTest
+ */
+final class ForRepeatedCountToOwnVariableRector extends AbstractRector
+{
+    /**
+     * @var string
+     */
+    private const DEFAULT_VARIABLE_COUNT_NAME = 'itemsCount';
+
+    public function getDefinition(): RectorDefinition
+    {
+        return new RectorDefinition('Change count() in for function to own variable', [
+            new CodeSample(
+                <<<'PHP'
+class SomeClass
+{
+    public function run($items)
+    {
+        for ($i = 5; $i <= count($items); $i++) {
+            echo $items[$i];
+        }
+    }
+}
+PHP
+,
+                <<<'PHP'
+class SomeClass
+{
+    public function run($items)
+    {
+        $itemsCount = count($items);
+        for ($i = 5; $i <= $itemsCount; $i++) {
+            echo $items[$i];
+        }
+    }
+}
+PHP
+
+            ),
+        ]);
+    }
+
+    /**
+     * @return string[]
+     */
+    public function getNodeTypes(): array
+    {
+        return [For_::class];
+    }
+
+    /**
+     * @param For_ $node
+     */
+    public function refactor(Node $node): ?Node
+    {
+        $countInCond = null;
+        $countedValueName = null;
+
+        $this->traverseNodesWithCallable($node->cond, function (Node $node) use (&$countInCond, &$countedValueName) {
+            if (! $node instanceof FuncCall) {
+                return null;
+            }
+
+            if (! $this->isName($node, 'count')) {
+                return null;
+            }
+
+            $countInCond = $node;
+            $valueName = $this->getName($node->args[0]->value);
+
+            if ($valueName === null) {
+                $countedValueName = self::DEFAULT_VARIABLE_COUNT_NAME;
+            } else {
+                $countedValueName = $valueName . 'Count';
+            }
+
+            return new Variable($countedValueName);
+        });
+
+        if ($countInCond === null || $countedValueName === null) {
+            return null;
+        }
+
+        $countAssign = new Assign(new Variable($countedValueName), $countInCond);
+
+        $this->addNodeBeforeNode($countAssign, $node);
+
+        return $node;
+    }
+}

packages/CodeQuality/tests/Rector/For_/ForRepeatedCountToOwnVariableRector/Fixture/fallback_for_complex.php.inc

@@ -0,0 +1,32 @@
+<?php
+
+namespace Rector\CodeQuality\Tests\Rector\For_\ForRepeatedCountToOwnVariableRector\Fixture;
+
+class FallbackForComplex
+{
+    public function run($items, \stdClass $someObject)
+    {
+        for ($i = 5; $i <= count($someObject->getItems() + 10); $i++) {
+            echo $items[$i];
+        }
+    }
+}
+
+?>
+-----
+<?php
+
+namespace Rector\CodeQuality\Tests\Rector\For_\ForRepeatedCountToOwnVariableRector\Fixture;
+
+class FallbackForComplex
+{
+    public function run($items, \stdClass $someObject)
+    {
+        $itemsCount = count($someObject->getItems() + 10);
+        for ($i = 5; $i <= $itemsCount; $i++) {
+            echo $items[$i];
+        }
+    }
+}
+
+?>

packages/CodeQuality/tests/Rector/For_/ForRepeatedCountToOwnVariableRector/Fixture/fixture.php.inc

@@ -0,0 +1,32 @@
+<?php
+
+namespace Rector\CodeQuality\Tests\Rector\For_\ForRepeatedCountToOwnVariableRector\Fixture;
+
+class SomeClass
+{
+    public function run($items)
+    {
+        for ($i = 5; $i <= count($items); $i++) {
+            echo $items[$i];
+        }
+    }
+}
+
+?>
+-----
+<?php
+
+namespace Rector\CodeQuality\Tests\Rector\For_\ForRepeatedCountToOwnVariableRector\Fixture;
+
+class SomeClass
+{
+    public function run($items)
+    {
+        $itemsCount = count($items);
+        for ($i = 5; $i <= $itemsCount; $i++) {
+            echo $items[$i];
+        }
+    }
+}
+
+?>

packages/CodeQuality/tests/Rector/For_/ForRepeatedCountToOwnVariableRector/Fixture/method_call_count.php.inc

@@ -0,0 +1,32 @@
+<?php
+
+namespace Rector\CodeQuality\Tests\Rector\For_\ForRepeatedCountToOwnVariableRector\Fixture;
+
+class MethodCallCount
+{
+    public function run($items, \stdClass $someObject)
+    {
+        for ($i = 5; $i <= count($someObject->getItems()); $i++) {
+            echo $items[$i];
+        }
+    }
+}
+
+?>
+-----
+<?php
+
+namespace Rector\CodeQuality\Tests\Rector\For_\ForRepeatedCountToOwnVariableRector\Fixture;
+
+class MethodCallCount
+{
+    public function run($items, \stdClass $someObject)
+    {
+        $getItemsCount = count($someObject->getItems());
+        for ($i = 5; $i <= $getItemsCount; $i++) {
+            echo $items[$i];
+        }
+    }
+}
+
+?>

packages/CodeQuality/tests/Rector/For_/ForRepeatedCountToOwnVariableRector/ForRepeatedCountToOwnVariableRectorTest.php

@@ -0,0 +1,30 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\CodeQuality\Tests\Rector\For_\ForRepeatedCountToOwnVariableRector;
+
+use Iterator;
+use Rector\CodeQuality\Rector\For_\ForRepeatedCountToOwnVariableRector;
+use Rector\Testing\PHPUnit\AbstractRectorTestCase;
+
+final class ForRepeatedCountToOwnVariableRectorTest extends AbstractRectorTestCase
+{
+    /**
+     * @dataProvider provideDataForTest()
+     */
+    public function test(string $file): void
+    {
+        $this->doTestFile($file);
+    }
+
+    public function provideDataForTest(): Iterator
+    {
+        return $this->yieldFilesFromDirectory(__DIR__ . '/Fixture');
+    }
+
+    protected function getRectorClass(): string
+    {
+        return ForRepeatedCountToOwnVariableRector::class;
+    }
+}