From 2ba9332d29c1acdde0f85a2781308d2e8972f00e Mon Sep 17 00:00:00 2001
From: Ondrej Mirtes <ondrej@mirtes.cz>
Date: Thu, 2 Sep 2021 13:42:38 +0200
Subject: [PATCH] Allow dynamic list of stub files thanks to StubFilesExtension

---
 src/PhpDoc/StubFilesExtension.php        | 13 +++++++++++++
 src/PhpDoc/StubPhpDocProviderFactory.php | 16 +++++++++++++++-
 2 files changed, 28 insertions(+), 1 deletion(-)
 create mode 100644 src/PhpDoc/StubFilesExtension.php

diff --git a/src/PhpDoc/StubFilesExtension.php b/src/PhpDoc/StubFilesExtension.php
new file mode 100644
index 0000000000..961b2f4b89
--- /dev/null
+++ b/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;
+
+}
diff --git a/src/PhpDoc/StubPhpDocProviderFactory.php b/src/PhpDoc/StubPhpDocProviderFactory.php
index 7b83eae369..382891ab4c 100644
--- a/src/PhpDoc/StubPhpDocProviderFactory.php
+++ b/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
 		);
 	}