provide config + add post Rectors for FileSystemRector

packages/file-system-rector/src/FileSystemFileProcessor.php

@@ -4,6 +4,7 @@
 
 namespace Rector\FileSystemRector;
 
+use Rector\Core\Logging\CurrentRectorProvider;
 use Rector\FileSystemRector\Contract\FileSystemRectorInterface;
 use Rector\NodeTypeResolver\FileSystem\CurrentFileInfoProvider;
 use Symplify\SmartFileSystem\SmartFileInfo;
@@ -20,20 +21,31 @@ final class FileSystemFileProcessor
      */
     private $fileSystemRectors = [];
 
+    /**
+     * @var CurrentRectorProvider
+     */
+    private $currentRectorProvider;
+
     /**
      * @param FileSystemRectorInterface[] $fileSystemRectors
      */
-    public function __construct(CurrentFileInfoProvider $currentFileInfoProvider, array $fileSystemRectors = [])
-    {
+    public function __construct(
+        CurrentFileInfoProvider $currentFileInfoProvider,
+        CurrentRectorProvider $currentRectorProvider,
+        array $fileSystemRectors = []
+    ) {
         $this->currentFileInfoProvider = $currentFileInfoProvider;
         $this->fileSystemRectors = $fileSystemRectors;
+        $this->currentRectorProvider = $currentRectorProvider;
     }
 
     public function processFileInfo(SmartFileInfo $smartFileInfo): void
     {
-        $this->currentFileInfoProvider ->setCurrentFileInfo($smartFileInfo);
+        $this->currentFileInfoProvider->setCurrentFileInfo($smartFileInfo);
 
         foreach ($this->fileSystemRectors as $fileSystemRector) {
+            $this->currentRectorProvider->changeCurrentRector($fileSystemRector);
+
             $fileSystemRector->refactor($smartFileInfo);
         }
     }

packages/file-system-rector/src/Rector/AbstractFileSystemRector.php

@@ -17,6 +17,7 @@
 use Rector\Core\Rector\AbstractRector\AbstractRectorTrait;
 use Rector\FileSystemRector\Contract\FileSystemRectorInterface;
 use Rector\NodeTypeResolver\NodeScopeAndMetadataDecorator;
+use Rector\PostRector\Application\PostFileProcessor;
 use Symplify\PackageBuilder\Parameter\ParameterProvider;
 use Symplify\SmartFileSystem\SmartFileInfo;
 use TypeError;
@@ -70,6 +71,11 @@ abstract class AbstractFileSystemRector implements FileSystemRectorInterface
      */
     private $parserFactory;
 
+    /**
+     * @var PostFileProcessor
+     */
+    private $postFileProcessor;
+
     /**
      * @required
      */
@@ -82,7 +88,8 @@ public function autowireAbstractFileSystemRector(
         RemovedAndAddedFilesCollector $removedAndAddedFilesCollector,
         Configuration $configuration,
         BetterStandardPrinter $betterStandardPrinter,
-        ParameterProvider $parameterProvider
+        ParameterProvider $parameterProvider,
+        PostFileProcessor $postFileProcessor
     ): void {
         $this->parser = $parser;
         $this->parserFactory = $parserFactory;
@@ -93,6 +100,7 @@ public function autowireAbstractFileSystemRector(
         $this->configuration = $configuration;
         $this->betterStandardPrinter = $betterStandardPrinter;
         $this->parameterProvider = $parameterProvider;
+        $this->postFileProcessor = $postFileProcessor;
     }
 
     /**
@@ -125,6 +133,8 @@ protected function parseFileInfoToNodesWithoutScope(SmartFileInfo $smartFileInfo
      */
     protected function printNodesToFilePath(array $nodes, string $fileDestination): void
     {
+        $nodes = $this->postFileProcessor->traverse($nodes);
+
         $fileContent = $this->formatPerservingPrinter->printToString(
             $nodes,
             $this->oldStmts,
@@ -139,6 +149,8 @@ protected function printNodesToFilePath(array $nodes, string $fileDestination):
      */
     protected function printNewNodesToFilePath(array $nodes, string $fileDestination): void
     {
+        $nodes = $this->postFileProcessor->traverse($nodes);
+
         // 1. if nodes are the same, prefer format preserving printer
         try {
             $dummyLexer = new Lexer();

src/Testing/PHPUnit/AbstractFileSystemRectorTestCase.php

@@ -109,7 +109,13 @@ private function createContainerWithProvidedRector(): void
 
         FileSystem::write($configFileTempPath, $yamlContent);
 
-        $this->bootKernelWithConfigs(RectorKernel::class, [$configFileTempPath]);
+        // for 3rd party testing with services defined in configs
+        $configFilePaths = [$configFileTempPath];
+        if ($this->provideConfig() !== '') {
+            $configFilePaths[] = $this->provideConfig();
+        }
+
+        $this->bootKernelWithConfigs(RectorKernel::class, $configFilePaths);
     }
 
     private function createConfigFileTempPath(): string

src/Testing/PHPUnit/AbstractGenericRectorTestCase.php

@@ -34,6 +34,12 @@ protected function getRectorClass(): string
         return '';
     }
 
+    protected function provideConfig(): string
+    {
+        // can be implemented
+        return '';
+    }
+
     /**
      * @return array<string, mixed[]>
      */

src/Testing/PHPUnit/AbstractRectorTestCase.php

@@ -124,12 +124,6 @@ protected function doTestFileWithoutAutoload(string $file): void
         $this->autoloadTestFixture = true;
     }
 
-    protected function provideConfig(): string
-    {
-        // can be implemented
-        return '';
-    }
-
     protected function doTestFile(string $file): void
     {
         $smartFileInfo = new SmartFileInfo($file);