Add withSetProviders method (#6515)

* Add `withSetProviders` method Signed-off-by: Nathanael Esayeas <[email protected]> * Prevent duplicate set providers Signed-off-by: Nathanael Esayeas <[email protected]> --------- Signed-off-by: Nathanael Esayeas <[email protected]>
rectorphp · Jan 11, 2025 · 0a6eb12 · 0a6eb12
1 parent 9d81eba
commit 0a6eb12
Showing 1 changed file with 39 additions and 2 deletions.
diff --git a/src/Configuration/RectorConfigBuilder.php b/src/Configuration/RectorConfigBuilder.php
@@ -23,6 +23,7 @@
 use Rector\Exception\Configuration\InvalidConfigurationException;
 use Rector\Php\PhpVersionResolver\ComposerJsonPhpVersionResolver;
 use Rector\PHPUnit\Set\PHPUnitSetList;
+use Rector\Set\Contract\SetProviderInterface;
 use Rector\Set\Enum\SetGroup;
 use Rector\Set\SetManager;
 use Rector\Set\ValueObject\DowngradeLevelSetList;
@@ -162,10 +163,22 @@ final class RectorConfigBuilder
 
     private ?bool $isWithPhpLevelUsed = null;
 
+    /**
+     * @var array<class-string<SetProviderInterface>,bool>
+     */
+    private array $setProviders = [];
+
     public function __invoke(RectorConfig $rectorConfig): void
     {
-        if ($this->setGroups !== []) {
-            $setManager = new SetManager(new SetProviderCollector(), new InstalledPackageResolver(getcwd()));
+        if ($this->setGroups !== [] || $this->setProviders !== []) {
+            $setProviderCollector = new SetProviderCollector(array_map(
+                static fn (string $setProvider): SetProviderInterface =>
+                $rectorConfig->make($setProvider),
+                \array_keys($this->setProviders)
+            ));
+
+            $setManager = new SetManager($setProviderCollector, new InstalledPackageResolver(getcwd()));
+
             $this->groupLoadedSets = $setManager->matchBySetGroups($this->setGroups);
 
             SimpleParameterProvider::addParameter(Option::COMPOSER_BASED_SETS, $this->groupLoadedSets);
@@ -1112,4 +1125,28 @@ public function withEditorUrl(string $editorUrl): self
 
         return $this;
     }
+
+    /**
+     * @param class-string<SetProviderInterface> ...$setProviders
+     */
+    public function withSetProviders(string ...$setProviders): self
+    {
+        foreach ($setProviders as $setProvider) {
+            if (\array_key_exists($setProvider, $this->setProviders)) {
+                continue;
+            }
+
+            if (! is_a($setProvider, SetProviderInterface::class, true)) {
+                throw new InvalidConfigurationException(sprintf(
+                    'Set provider "%s" must implement "%s"',
+                    $setProvider,
+                    SetProviderInterface::class
+                ));
+            }
+
+            $this->setProviders[$setProvider] = true;
+        }
+
+        return $this;
+    }
 }