feat: added support for running rebuild task with specific version mode

bimthebam · Oct 20, 2023 · f5f6a0a · f5f6a0a
1 parent e65f8d9
commit f5f6a0a
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/src/Dev/Task/RebuildAllIndexesTask.php b/src/Dev/Task/RebuildAllIndexesTask.php
@@ -8,6 +8,7 @@
 use SilverStripe\Core\ClassInfo;
 use SilverStripe\Core\Injector\Injector;
 use SilverStripe\Dev\BuildTask;
+use SilverStripe\Versioned\Versioned;
 use Throwable;
 
 /**
@@ -40,11 +41,19 @@ class RebuildAllIndexesTask extends BuildTask
      */
     public function run($request): void
     {
-        foreach (ClassInfo::subclassesFor(Index::class, false) as $indexClass) {
-            /** @var Index $index */
-            $index = Injector::inst()->create($indexClass);
+        $stage = $request->getVar('stage');
 
-            $index->rebuild();
-        }
+        Versioned::withVersionedMode(function () use ($stage) {
+            if ($stage) {
+                Versioned::set_stage($stage);
+            }
+
+            foreach (ClassInfo::subclassesFor(Index::class, false) as $indexClass) {
+                /** @var Index $index */
+                $index = Injector::inst()->create($indexClass);
+
+                $index->rebuild();
+            }
+        });
     }
 }