[PostRector] Keep first comment after declare(strict_types=1) on UnusedImportRemovingPostRector (#7042)

Author: samsonasik

src/PostRector/Rector/UnusedImportRemovingPostRector.php

@@ -11,6 +11,7 @@
 use PhpParser\Node\Identifier;
 use PhpParser\Node\Name;
 use PhpParser\Node\Name\FullyQualified;
+use PhpParser\Node\Stmt\Declare_;
 use PhpParser\Node\Stmt\Namespace_;
 use PhpParser\Node\Stmt\Nop;
 use PhpParser\Node\Stmt\Use_;
@@ -42,7 +43,12 @@ public function enterNode(Node $node): ?Node
         $namesInOriginalCase = $this->resolveUsedPhpAndDocNames($node);
         $namesInLowerCase = array_map(strtolower(...), $namesInOriginalCase);
 
+        $firstStmtKey = 0;
         foreach ($node->stmts as $key => $stmt) {
+            if ($stmt instanceof Declare_ && $key === 0) {
+                $firstStmtKey += 1;
+            }
+
             if (! $stmt instanceof Use_) {
                 continue;
             }
@@ -74,7 +80,7 @@ public function enterNode(Node $node): ?Node
             if ($stmt->uses === []) {
                 $comments = $node->stmts[$key]->getComments();
 
-                if ($key === 0 && $comments !== []) {
+                if ($key === $firstStmtKey && $comments !== []) {
                     $node->stmts[$key] = new Nop();
                     $node->stmts[$key]->setAttribute(AttributeKey::COMMENTS, $comments);
                 } else {

tests/Issues/NoNamespaced/Fixture/keep_first_comment_after_declare.php.inc

@@ -0,0 +1,40 @@
+<?php
+
+declare(strict_types=1);
+
+// some comment about the file
+// not about the use statement below it.
+/* comment
+   block */
+######
+
+use Exception;
+use stdClass;
+use SomeOtherClass;
+
+final class KeepFirstCommentAfterDeclare
+{
+    public ?stdClass $b;
+}
+
+?>
+-----
+<?php
+
+declare(strict_types=1);
+
+// some comment about the file
+// not about the use statement below it.
+/* comment
+   block */
+######
+
+
+use stdClass;
+
+final class KeepFirstCommentAfterDeclare
+{
+    public ?stdClass $b;
+}
+
+?>