[phpstan-extensions] Provide phpstan container to bootstrap file from extensions (rectorphp#3401)

Co-authored-by: Abdul Malik Ikhsan <[email protected]>

Author: TomasVotruba

packages/NodeTypeResolver/DependencyInjection/PHPStanServicesFactory.php

@@ -57,6 +57,11 @@ public function __construct(
         $filesystem->remove($purifiedConfigFiles);
     }
 
+    public function provideContainer(): Container
+    {
+        return $this->container;
+    }
+
     /**
      * @api
      */

phpstan.neon

@@ -812,3 +812,11 @@ parameters:
         - '#Call to deprecated method getDirectClassNames\(\) of class PHPStan\\Type\\TypeUtils.*#'
         - '#Parameter 3 should use "PHPStan\\Reflection\\ParameterReflectionWithPhpDocs" type as the only type passed to this method#'
         - '#Parameters should use "PHPStan\\Reflection\\ParameterReflectionWithPhpDocs\|PhpParser\\Node\\Param" types as the only types passed to this method#'
+
+        # various usage
+        - '#Parameters should use "PHPStan\\DependencyInjection\\Container" types as the only types passed to this method#'
+
+        # actually used in global scope
+        -
+            message: '#Anonymous function has an unused use \$container#'
+            path: src/Autoloading/BootstrapFilesIncluder.php

src/Autoloading/BootstrapFilesIncluder.php

@@ -5,6 +5,7 @@
 namespace Rector\Core\Autoloading;
 
 use Nette\Neon\Neon;
+use PHPStan\DependencyInjection\Container;
 use Rector\Core\Configuration\Option;
 use Rector\Core\Configuration\Parameter\ParameterProvider;
 use Rector\Core\Exception\ShouldNotHappenException;
@@ -26,14 +27,14 @@ public function __construct(
     ) {
     }
 
-    public function includePHPStanExtensionsBoostrapFiles(): void
+    public function includePHPStanExtensionsBoostrapFiles(?Container $container = null): void
     {
         $extensionConfigFiles = $this->phpStanExtensionsConfigResolver->resolve();
 
         $absoluteBootstrapFilePaths = $this->resolveAbsoluteBootstrapFilePaths($extensionConfigFiles);
 
         foreach ($absoluteBootstrapFilePaths as $absoluteBootstrapFilePath) {
-            $this->tryRequireFile($absoluteBootstrapFilePath);
+            $this->tryRequireFile($absoluteBootstrapFilePath, $container);
         }
     }
 
@@ -87,10 +88,16 @@ private function resolveAbsoluteBootstrapFilePaths(array $extensionConfigFiles):
         return $absoluteBootstrapFilePaths;
     }
 
-    private function tryRequireFile(string $bootstrapFile): void
+    /**
+     * PHPStan container mimics:
+     * https://github.com/phpstan/phpstan-src/blob/34881e682e36e30917dcfa8dc69c70e857143436/src/Command/CommandHelper.php#L513-L515
+     */
+    private function tryRequireFile(string $bootstrapFile, ?Container $container = null): void
     {
         try {
-            require_once $bootstrapFile;
+            (static function (string $bootstrapFile) use ($container): void {
+                require_once $bootstrapFile;
+            })($bootstrapFile);
         } catch (Throwable $throwable) {
             $errorMessage = sprintf(
                 '"%s" thrown in "%s" on line %d while loading bootstrap file %s: %s',

src/DependencyInjection/RectorContainerFactory.php

@@ -9,6 +9,7 @@
 use Rector\Core\Autoloading\BootstrapFilesIncluder;
 use Rector\Core\Kernel\RectorKernel;
 use Rector\Core\ValueObject\Bootstrap\BootstrapConfigs;
+use Rector\NodeTypeResolver\DependencyInjection\PHPStanServicesFactory;
 
 final class RectorContainerFactory
 {
@@ -27,7 +28,12 @@ public function createFromBootstrapConfigs(BootstrapConfigs $bootstrapConfigs):
         /** @var BootstrapFilesIncluder $bootstrapFilesIncluder */
         $bootstrapFilesIncluder = $container->get(BootstrapFilesIncluder::class);
         $bootstrapFilesIncluder->includeBootstrapFiles();
-        $bootstrapFilesIncluder->includePHPStanExtensionsBoostrapFiles();
+
+        $phpStanServicesFactory = $container->get(PHPStanServicesFactory::class);
+
+        /** @var PHPStanServicesFactory $phpStanServicesFactory */
+        $phpStanContainer = $phpStanServicesFactory->provideContainer();
+        $bootstrapFilesIncluder->includePHPStanExtensionsBoostrapFiles($phpStanContainer);
 
         return $container;
     }