Allow dynamic list of stub files thanks to StubFilesExtension

Author: ondrejmirtes

src/PhpDoc/StubFilesExtension.php

@@ -0,0 +1,13 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\PhpDoc;
+
+interface StubFilesExtension
+{
+
+	public const EXTENSION_TAG = 'phpstan.stubFilesExtension';
+
+	/** @return string[] */
+	public function getFiles(): array;
+
+}

src/PhpDoc/StubPhpDocProviderFactory.php

@@ -2,6 +2,7 @@
 
 namespace PHPStan\PhpDoc;
 
+use PHPStan\DependencyInjection\Container;
 use PHPStan\Parser\Parser;
 use PHPStan\Type\FileTypeMapper;
 
@@ -12,6 +13,8 @@ class StubPhpDocProviderFactory
 
 	private \PHPStan\Type\FileTypeMapper $fileTypeMapper;
 
+	private Container $container;
+
 	/** @var string[] */
 	private array $stubFiles;
 
@@ -22,20 +25,31 @@ class StubPhpDocProviderFactory
 	public function __construct(
 		Parser $parser,
 		FileTypeMapper $fileTypeMapper,
+		Container $container,
 		array $stubFiles
 	)
 	{
 		$this->parser = $parser;
 		$this->fileTypeMapper = $fileTypeMapper;
+		$this->container = $container;
 		$this->stubFiles = $stubFiles;
 	}
 
 	public function create(): StubPhpDocProvider
 	{
+		$stubFiles = $this->stubFiles;
+		$extensions = $this->container->getServicesByTag(StubFilesExtension::EXTENSION_TAG);
+		foreach ($extensions as $extension) {
+			$extensionFiles = $extension->getFiles();
+			foreach ($extensionFiles as $extensionFile) {
+				$stubFiles[] = $extensionFile;
+			}
+		}
+
 		return new StubPhpDocProvider(
 			$this->parser,
 			$this->fileTypeMapper,
-			$this->stubFiles
+			$stubFiles
 		);
 	}