FNSR ExpressionResultStorage

conf/config.neon

@@ -238,6 +238,7 @@ extensions:
 	autowiredAttributeServices: PHPStan\DependencyInjection\AutowiredAttributeServicesExtension
 	validateServiceTags: PHPStan\DependencyInjection\ValidateServiceTagsExtension
 	gnsr: PHPStan\DependencyInjection\GnsrExtension
+	fnsr: PHPStan\DependencyInjection\FnsrExtension
 
 autowiredAttributeServices:
 	level: null

issue-bot/src/Console/RunCommand.php

@@ -111,6 +111,7 @@ private function analyseHash(OutputInterface $output, int $phpVersion, Playgroun
 		$output->writeln(sprintf('Starting analysis of %s', $hash));
 
 		$startTime = microtime(true);
+		putenv("PHPSTAN_FNSR=1");
 		exec(implode(' ', $commandArray), $outputLines, $exitCode);
 		$elapsedTime = microtime(true) - $startTime;
 		$output->writeln(sprintf('Analysis of %s took %.2f s', $hash, $elapsedTime));

src/Analyser/DirectInternalScopeFactory.php

@@ -3,6 +3,7 @@
 namespace PHPStan\Analyser;
 
 use PhpParser\Node;
+use PHPStan\Analyser\Fiber\FiberScope;
 use PHPStan\DependencyInjection\Type\DynamicReturnTypeExtensionRegistryProvider;
 use PHPStan\DependencyInjection\Type\ExpressionTypeResolverExtensionRegistryProvider;
 use PHPStan\Node\Printer\ExprPrinter;
@@ -38,6 +39,7 @@ public function __construct(
 		private int|array|null $configPhpVersion,
 		private $nodeCallback,
 		private ConstantResolver $constantResolver,
+		private bool $fiber = false,
 	)
 	{
 	}
@@ -61,7 +63,12 @@ public function create(
 		bool $nativeTypesPromoted = false,
 	): MutatingScope
 	{
-		return new MutatingScope(
+		$className = MutatingScope::class;
+		if ($this->fiber) {
+			$className = FiberScope::class;
+		}
+
+		return new $className(
 			$this,
 			$this->reflectionProvider,
 			$this->initializerExprTypeResolver,
@@ -97,4 +104,48 @@ public function create(
 		);
 	}
 
+	public function toFiberFactory(): InternalScopeFactory
+	{
+		return new self(
+			$this->reflectionProvider,
+			$this->initializerExprTypeResolver,
+			$this->dynamicReturnTypeExtensionRegistryProvider,
+			$this->expressionTypeResolverExtensionRegistryProvider,
+			$this->exprPrinter,
+			$this->typeSpecifier,
+			$this->propertyReflectionFinder,
+			$this->parser,
+			$this->nodeScopeResolver,
+			$this->richerScopeGetTypeHelper,
+			$this->phpVersion,
+			$this->attributeReflectionFactory,
+			$this->configPhpVersion,
+			$this->nodeCallback,
+			$this->constantResolver,
+			true,
+		);
+	}
+
+	public function toMutatingFactory(): InternalScopeFactory
+	{
+		return new self(
+			$this->reflectionProvider,
+			$this->initializerExprTypeResolver,
+			$this->dynamicReturnTypeExtensionRegistryProvider,
+			$this->expressionTypeResolverExtensionRegistryProvider,
+			$this->exprPrinter,
+			$this->typeSpecifier,
+			$this->propertyReflectionFinder,
+			$this->parser,
+			$this->nodeScopeResolver,
+			$this->richerScopeGetTypeHelper,
+			$this->phpVersion,
+			$this->attributeReflectionFactory,
+			$this->configPhpVersion,
+			$this->nodeCallback,
+			$this->constantResolver,
+			false,
+		);
+	}
+
 }

src/Analyser/ExpressionResult.php

@@ -23,6 +23,7 @@ final class ExpressionResult
 	 */
 	public function __construct(
 		private MutatingScope $scope,
+		private MutatingScope $beforeScope,
 		private bool $hasYield,
 		private bool $isAlwaysTerminating,
 		private array $throwPoints,
@@ -40,6 +41,11 @@ public function getScope(): MutatingScope
 		return $this->scope;
 	}
 
+	public function getBeforeScope(): MutatingScope
+	{
+		return $this->beforeScope;
+	}
+
 	public function hasYield(): bool
 	{
 		return $this->hasYield;

src/Analyser/ExpressionResultStorage.php

@@ -0,0 +1,40 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Analyser;
+
+use PhpParser\Node\Expr;
+use PHPStan\Analyser\Fiber\ExpressionAnalysisRequest;
+use SplObjectStorage;
+
+final class ExpressionResultStorage
+{
+
+	/** @var SplObjectStorage<Expr, ExpressionResult> */
+	private SplObjectStorage $results;
+
+	/** @var array<array{fiber: Fiber<mixed, ExpressionResult|null, null, ExpressionAnalysisRequest|NodeCallbackRequest>, request: ExpressionAnalysisRequest}> */
+	public array $pendingFibers = [];
+
+	public function __construct()
+	{
+		$this->results = new SplObjectStorage();
+	}
+
+	public function duplicate(): self
+	{
+		$new = new self();
+		$new->results->addAll($this->results);
+		return $new;
+	}
+
+	public function storeResult(Expr $expr, ExpressionResult $result): void
+	{
+		$this->results[$expr] = $result;
+	}
+
+	public function findResult(Expr $expr): ?ExpressionResult
+	{
+		return $this->results[$expr] ?? null;
+	}
+
+}

src/Analyser/Fiber/ExpressionAnalysisRequest.php

@@ -0,0 +1,15 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Analyser\Fiber;
+
+use PhpParser\Node\Expr;
+use PHPStan\Analyser\MutatingScope;
+
+final class ExpressionAnalysisRequest
+{
+
+	public function __construct(public readonly Expr $expr, public readonly MutatingScope $scope)
+	{
+	}
+
+}

src/Analyser/Fiber/FiberNodeScopeResolver.php

@@ -0,0 +1,125 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Analyser\Fiber;
+
+use Fiber;
+use PhpParser\Node;
+use PhpParser\Node\Expr;
+use PHPStan\Analyser\ExpressionContext;
+use PHPStan\Analyser\ExpressionResult;
+use PHPStan\Analyser\ExpressionResultStorage;
+use PHPStan\Analyser\MutatingScope;
+use PHPStan\Analyser\NodeScopeResolver;
+use PHPStan\Analyser\NoopNodeCallback;
+use PHPStan\DependencyInjection\AutowiredService;
+use PHPStan\ShouldNotHappenException;
+use function get_class;
+use function get_debug_type;
+use function sprintf;
+
+#[AutowiredService(as: FiberNodeScopeResolver::class)]
+final class FiberNodeScopeResolver extends NodeScopeResolver
+{
+
+	protected function callNodeCallback(
+		callable $nodeCallback,
+		Node $node,
+		MutatingScope $scope,
+		ExpressionResultStorage $storage,
+	): void
+	{
+		$fiber = new Fiber(static function () use ($node, $scope, $nodeCallback) {
+			$nodeCallback($node, $scope->toFiberScope());
+		});
+		$request = $fiber->start();
+		$this->runFiberForNodeCallback($storage, $fiber, $request);
+	}
+
+	private function runFiberForNodeCallback(
+		ExpressionResultStorage $storage,
+		Fiber $fiber,
+		?ExpressionAnalysisRequest $request,
+	): void
+	{
+		while (!$fiber->isTerminated()) {
+			if ($request instanceof ExpressionAnalysisRequest) {
+				$result = $storage->findResult($request->expr);
+				if ($result !== null) {
+					$request = $fiber->resume($result);
+					continue;
+				}
+
+				$storage->pendingFibers[] = [
+					'fiber' => $fiber,
+					'request' => $request,
+				];
+				return;
+			}
+
+			throw new ShouldNotHappenException(
+				'Unknown fiber suspension: ' . get_debug_type($request),
+			);
+		}
+
+		if ($request !== null) {
+			throw new ShouldNotHappenException(
+				'Fiber terminated but we did not handle its request ' . get_debug_type($request),
+			);
+		}
+	}
+
+	protected function processPendingFibers(ExpressionResultStorage $storage): void
+	{
+		foreach ($storage->pendingFibers as $pending) {
+			$request = $pending['request'];
+			$result = $storage->findResult($request->expr);
+
+			if ($result !== null) {
+				throw new ShouldNotHappenException('Pending fibers at the end should be about synthetic nodes');
+			}
+
+			$this->processExprNode(
+				new Node\Stmt\Expression($request->expr),
+				$request->expr,
+				$request->scope->toMutatingScope(),
+				$storage,
+				new NoopNodeCallback(),
+				ExpressionContext::createTopLevel(),
+			);
+			if ($storage->findResult($request->expr) === null) {
+				throw new ShouldNotHappenException(sprintf('processExprNode should have stored the result of %s on line %s', get_class($request->expr), $request->expr->getStartLine()));
+			}
+			$this->processPendingFibers($storage);
+
+			// Break and restart the loop since the array may have been modified
+			return;
+		}
+	}
+
+	protected function processPendingFibersForRequestedExpr(ExpressionResultStorage $storage, Expr $expr, ExpressionResult $result): void
+	{
+		$restartLoop = true;
+
+		while ($restartLoop) {
+			$restartLoop = false;
+
+			foreach ($storage->pendingFibers as $key => $pending) {
+				$request = $pending['request'];
+				if ($request->expr !== $expr) {
+					continue;
+				}
+
+				unset($storage->pendingFibers[$key]);
+				$restartLoop = true;
+
+				$fiber = $pending['fiber'];
+				$request = $fiber->resume($result);
+				$this->runFiberForNodeCallback($storage, $fiber, $request);
+
+				// Break and restart the loop since the array may have been modified
+				break;
+			}
+		}
+	}
+
+}

src/Analyser/Fiber/FiberScope.php

@@ -0,0 +1,46 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Analyser\Fiber;
+
+use Fiber;
+use PhpParser\Node\Expr;
+use PHPStan\Analyser\ExpressionResult;
+use PHPStan\Analyser\MutatingScope;
+use PHPStan\Type\Type;
+
+final class FiberScope extends MutatingScope
+{
+
+	/** @api */
+	public function getType(Expr $node): Type
+	{
+		/** @var ExpressionResult $exprResult */
+		$exprResult = Fiber::suspend(
+			new ExpressionAnalysisRequest($node, $this),
+		);
+
+		return $exprResult->getBeforeScope()->toMutatingScope()->getType($node);
+	}
+
+	/** @api */
+	public function getNativeType(Expr $expr): Type
+	{
+		/** @var ExpressionResult $exprResult */
+		$exprResult = Fiber::suspend(
+			new ExpressionAnalysisRequest($expr, $this),
+		);
+
+		return $exprResult->getBeforeScope()->toMutatingScope()->getNativeType($expr);
+	}
+
+	public function getKeepVoidType(Expr $node): Type
+	{
+		/** @var ExpressionResult $exprResult */
+		$exprResult = Fiber::suspend(
+			new ExpressionAnalysisRequest($node, $this),
+		);
+
+		return $exprResult->getBeforeScope()->toMutatingScope()->getKeepVoidType($node);
+	}
+
+}

src/Analyser/InternalScopeFactory.php

@@ -39,4 +39,8 @@ public function create(
 		bool $nativeTypesPromoted = false,
 	): MutatingScope;
 
+	public function toFiberFactory(): self;
+
+	public function toMutatingFactory(): self;
+
 }

src/Analyser/LazyInternalScopeFactory.php

@@ -3,6 +3,7 @@
 namespace PHPStan\Analyser;
 
 use PhpParser\Node;
+use PHPStan\Analyser\Fiber\FiberScope;
 use PHPStan\DependencyInjection\Container;
 use PHPStan\DependencyInjection\GenerateFactory;
 use PHPStan\DependencyInjection\Type\DynamicReturnTypeExtensionRegistryProvider;
@@ -29,6 +30,7 @@ final class LazyInternalScopeFactory implements InternalScopeFactory
 	public function __construct(
 		private Container $container,
 		private $nodeCallback,
+		private bool $fiber = false,
 	)
 	{
 		$this->phpVersion = $this->container->getParameter('phpVersion');
@@ -53,7 +55,12 @@ public function create(
 		bool $nativeTypesPromoted = false,
 	): MutatingScope
 	{
-		return new MutatingScope(
+		$className = MutatingScope::class;
+		if ($this->fiber) {
+			$className = FiberScope::class;
+		}
+
+		return new $className(
 			$this,
 			$this->container->getByType(ReflectionProvider::class),
 			$this->container->getByType(InitializerExprTypeResolver::class),
@@ -89,4 +96,14 @@ public function create(
 		);
 	}
 
+	public function toFiberFactory(): InternalScopeFactory
+	{
+		return new self($this->container, $this->nodeCallback, true);
+	}
+
+	public function toMutatingFactory(): InternalScopeFactory
+	{
+		return new self($this->container, $this->nodeCallback, false);
+	}
+
 }