Env variable PHPSTAN_GNSR=1 enables GNSR

Author: ondrejmirtes

conf/config.neon

@@ -237,6 +237,7 @@ extensions:
 	validateExcludePaths: PHPStan\DependencyInjection\ValidateExcludePathsExtension
 	autowiredAttributeServices: PHPStan\DependencyInjection\AutowiredAttributeServicesExtension
 	validateServiceTags: PHPStan\DependencyInjection\ValidateServiceTagsExtension
+	gnsr: PHPStan\DependencyInjection\GnsrExtension
 
 autowiredAttributeServices:
 	level: null

src/Analyser/Generator/GeneratorNodeScopeResolver.php

@@ -17,6 +17,7 @@
 use PHPStan\Analyser\Generator\NodeHandler\StmtsHandler;
 use PHPStan\Analyser\Scope;
 use PHPStan\Analyser\StatementContext;
+use PHPStan\DependencyInjection\AutowiredService;
 use PHPStan\DependencyInjection\Container;
 use PHPStan\NeverException;
 use PHPStan\Node\FunctionCallableNode;
@@ -73,6 +74,7 @@
  * @phpstan-type GeneratorTValueType = ExprAnalysisRequest|StmtAnalysisRequest|StmtsAnalysisRequest|NodeCallbackRequest|AttrGroupsAnalysisRequest|TypeExprRequest|PersistStorageRequest|RestoreStorageRequest|RunInFiberRequest<mixed>
  * @phpstan-type GeneratorTSendType = ExprAnalysisResult|StmtAnalysisResult|TypeExprResult|ExprAnalysisResultStorage|RunInFiberResult<mixed>|null
  */
+#[AutowiredService]
 final class GeneratorNodeScopeResolver
 {
 

src/DependencyInjection/GnsrExtension.php

@@ -0,0 +1,39 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\DependencyInjection;
+
+use Nette\DI\CompilerExtension;
+use Nette\DI\Definitions\ServiceDefinition;
+use Override;
+use PHPStan\Analyser\Analyser;
+use PHPStan\Analyser\FileAnalyser;
+use PHPStan\Analyser\Generator\GeneratorNodeScopeResolver;
+use PHPStan\ShouldNotHappenException;
+use function getenv;
+
+final class GnsrExtension extends CompilerExtension
+{
+
+	#[Override]
+	public function beforeCompile()
+	{
+		$enable = getenv('PHPSTAN_GNSR');
+		if ($enable !== '1') {
+			return;
+		}
+
+		$builder = $this->getContainerBuilder();
+		$analyserDef = $builder->getDefinitionByType(Analyser::class);
+		if (!$analyserDef instanceof ServiceDefinition) {
+			throw new ShouldNotHappenException();
+		}
+		$analyserDef->setArgument('nodeScopeResolver', '@' . GeneratorNodeScopeResolver::class);
+
+		$fileAnalyserDef = $builder->getDefinitionByType(FileAnalyser::class);
+		if (!$fileAnalyserDef instanceof ServiceDefinition) {
+			throw new ShouldNotHappenException();
+		}
+		$fileAnalyserDef->setArgument('nodeScopeResolver', '@' . GeneratorNodeScopeResolver::class);
+	}
+
+}

src/Testing/RuleTestCase.php

@@ -25,6 +25,7 @@
 use PHPStan\File\FileHelper;
 use PHPStan\File\FileReader;
 use PHPStan\Fixable\Patcher;
+use PHPStan\Node\Printer\ExprPrinter;
 use PHPStan\Php\PhpVersion;
 use PHPStan\PhpDoc\PhpDocInheritanceResolver;
 use PHPStan\Reflection\ClassReflectionFactory;
@@ -39,6 +40,7 @@
 use function array_map;
 use function array_merge;
 use function count;
+use function getenv;
 use function implode;
 use function sprintf;
 use function str_replace;
@@ -80,6 +82,15 @@ protected function getTypeSpecifier(): TypeSpecifier
 
 	protected function createNodeScopeResolver(): NodeScopeResolver|GeneratorNodeScopeResolver
 	{
+		$enableGnsr = getenv('PHPSTAN_GNSR');
+		if ($enableGnsr === '1') {
+			$container = self::getContainer();
+
+			return new GeneratorNodeScopeResolver(
+				$container->getByType(ExprPrinter::class),
+				$container,
+			);
+		}
 		$readWritePropertiesExtensions = $this->getReadWritePropertiesExtensions();
 		$reflectionProvider = $this->createReflectionProvider();
 		$typeSpecifier = $this->getTypeSpecifier();

src/Testing/TypeInferenceTestCase.php

@@ -18,6 +18,7 @@
 use PHPStan\File\FileHelper;
 use PHPStan\File\SystemAgnosticSimpleRelativePathHelper;
 use PHPStan\Node\InClassNode;
+use PHPStan\Node\Printer\ExprPrinter;
 use PHPStan\Php\PhpVersion;
 use PHPStan\PhpDoc\PhpDocInheritanceResolver;
 use PHPStan\PhpDoc\TypeStringResolver;
@@ -38,6 +39,7 @@
 use function fclose;
 use function fgets;
 use function fopen;
+use function getenv;
 use function in_array;
 use function is_dir;
 use function is_string;
@@ -55,8 +57,16 @@ abstract class TypeInferenceTestCase extends PHPStanTestCase
 
 	protected static function createNodeScopeResolver(): NodeScopeResolver|GeneratorNodeScopeResolver
 	{
-		$reflectionProvider = self::createReflectionProvider();
 		$container = self::getContainer();
+		$enableGnsr = getenv('PHPSTAN_GNSR');
+		if ($enableGnsr === '1') {
+			return new GeneratorNodeScopeResolver(
+				$container->getByType(ExprPrinter::class),
+				$container,
+			);
+		}
+
+		$reflectionProvider = self::createReflectionProvider();
 		$typeSpecifier = $container->getService('typeSpecifier');
 
 		return new NodeScopeResolver(