support changing of comments

Author: shmax

src/Printer/Printer.php

@@ -535,8 +535,15 @@ private function printArrayFormatPreserving(array $nodes, array $originalNodes,
 					throw new LogicException();
 				}
 
-				//              $comments = $arrItem->getComments();
-				//              $origComments = $origArrayItem->getComments();
+				$comments = $arrItem->getAttribute(Attribute::COMMENTS) ?? [];
+				$origComments = $origArrayItem->getAttribute(Attribute::COMMENTS) ?? [];
+
+				if(count($comments) && !count($origComments)) {
+					$idfdf= 5;
+				}
+
+				$commentStartPos = count($origComments) ? $origComments[0]->startIndex : $itemStartPos;
+				\assert($commentStartPos >= 0);
 
 				$result .= $originalTokens->getContentBetween($tokenIndex, $itemStartPos);
 
@@ -570,6 +577,13 @@ private function printArrayFormatPreserving(array $nodes, array $originalNodes,
 					$result .= '(';
 				}
 
+				if ($comments !== $origComments) {
+					if ($comments) {
+						$result .= $this->pComments($comments, $beforeAsteriskIndent, $afterAsteriskIndent);// . "\n";
+						$result .= sprintf('%s%s*%s', $originalTokens->getDetectedNewline() ?? "\n", $beforeAsteriskIndent, $afterAsteriskIndent);
+					}
+				}
+
 				$result .= $this->printNodeFormatPreserving($arrItem, $originalTokens);
 				if ($addParentheses) {
 					$result .= ')';

tests/PHPStan/Printer/PrinterTest.php

@@ -664,7 +664,7 @@ public function enterNode(Node $node)
 
 		};
 
-		$addCommentsToMultilineArrayShape = new class extends AbstractNodeVisitor {
+		$addItemsWithCommentsToMultilineArrayShape = new class extends AbstractNodeVisitor {
 
 			public function enterNode(Node $node)
 			{
@@ -725,7 +725,35 @@ public function enterNode(Node $node)
  *  e: string
  * }
  */',
-			$addCommentsToMultilineArrayShape,
+			$addItemsWithCommentsToMultilineArrayShape,
+		];
+
+		$changeCommentOnArrayShapeItem = new class extends AbstractNodeVisitor {
+
+			public function enterNode(Node $node)
+			{
+				if ($node instanceof ArrayShapeItemNode) {
+					$node->setAttribute(Attribute::COMMENTS, [new Comment('// puppies')]);
+				}
+
+				return $node;
+			}
+
+		};
+
+		yield [
+			'/**
+ * @return array{
+ *   a: int,
+ * }
+ */',
+			'/**
+ * @return array{
+ *   // puppies
+ *   a: int,
+ * }
+ */',
+			$changeCommentOnArrayShapeItem,
 		];