MAGE-848 Implement virtual replica disable for all stores

Author: cammonro

Console/Command/ReplicaDisableVirtualCommand.php

@@ -3,16 +3,44 @@
 namespace Algolia\AlgoliaSearch\Console\Command;
 
 use Algolia\AlgoliaSearch\Api\Console\ReplicaSyncCommandInterface;
+use Algolia\AlgoliaSearch\Api\Product\ReplicaManagerInterface;
 use Algolia\AlgoliaSearch\Console\Traits\ReplicaSyncCommandTrait;
+use Algolia\AlgoliaSearch\Helper\ConfigHelper;
+use Algolia\AlgoliaSearch\Helper\Configuration\ConfigChecker;
+use Algolia\AlgoliaSearch\Helper\Entity\ProductHelper;
+use Algolia\AlgoliaSearch\Service\StoreNameFetcher;
+use Magento\Framework\App\Cache\Manager as CacheManager;
+use Magento\Framework\App\Config\ScopeConfigInterface;
+use Magento\Framework\App\Config\Storage\WriterInterface;
+use Magento\Framework\App\State;
 use Magento\Framework\Console\Cli;
+use Magento\Framework\Serialize\SerializerInterface;
+use Magento\Store\Model\ScopeInterface;
+use Magento\Store\Model\StoreManagerInterface;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Output\OutputInterface;
-use Symfony\Component\Console\Question\ConfirmationQuestion;
 
-class ReplicaDisableVirtualCommand extends AbstractReplicaCommand implements ReplicaSyncCommandInterface
-{
+class ReplicaDisableVirtualCommand extends AbstractReplicaCommand implements ReplicaSyncCommandInterface {
     use ReplicaSyncCommandTrait;
 
+    public function __construct(
+        protected WriterInterface         $configWriter,
+        protected ConfigChecker           $configChecker,
+        protected ScopeConfigInterface    $scopeConfig,
+        protected SerializerInterface     $serializer,
+        protected ConfigHelper            $configHelper,
+        protected CacheManager            $cacheManager,
+        protected ReplicaManagerInterface $replicaManager,
+        protected StoreManagerInterface   $storeManager,
+        protected ProductHelper           $productHelper,
+        State                             $state,
+        StoreNameFetcher                  $storeNameFetcher,
+        ?string                           $name = null
+    )
+    {
+        parent::__construct($state, $storeNameFetcher, $name);
+    }
+
     protected function getReplicaCommandName(): string
     {
         return 'disable-virtual-replicas';
@@ -48,7 +76,69 @@ protected function execute(InputInterface $input, OutputInterface $output): int
             return CLI::RETURN_SUCCESS;
         }
 
+        $this->disableVirtualReplicas($storeIds);
+
         return Cli::RETURN_SUCCESS;
     }
 
+    /**
+     * @param int[] $storeIds
+     * @return void
+     */
+    protected function disableVirtualReplicas(array $storeIds = []): void
+    {
+        if (count($storeIds)) {
+            foreach ($storeIds as $storeId) {
+                $this->disableVirtualReplicasForStore($storeId);
+            }
+        } else {
+            $this->disableVirtualReplicasForAllStores();
+        }
+    }
+
+    protected function disableVirtualReplicasForStore(int $storeId): void
+    {
+        $storeName = $this->storeNameFetcher->getStoreName($storeId);
+        $isStoreScoped = false;
+
+    }
+
+    protected function disableVirtualReplicasForAllStores(): void
+    {
+        $this->configChecker->checkAndApplyAllScopes(ConfigHelper::USE_VIRTUAL_REPLICA_ENABLED, [$this, 'removeLegacyVirtualReplicaConfig']);
+
+        $this->configChecker->checkAndApplyAllScopes(ConfigHelper::SORTING_INDICES, [$this, 'disableVirtualReplicaSortConfig']);
+
+        $this->cacheManager->clean(['config']);
+
+        $this->syncReplicasForAllStores();
+    }
+
+    public function removeLegacyVirtualReplicaConfig(string $scope = ScopeConfigInterface::SCOPE_TYPE_DEFAULT, int $scopeId = 0): void
+    {
+        $value = $this->scopeConfig->getValue(ConfigHelper::USE_VIRTUAL_REPLICA_ENABLED, $scope, $scopeId);
+        if (is_null($value)) {
+            return;
+        }
+        $this->output->writeln("<info>Removing legacy configuration " . ConfigHelper::USE_VIRTUAL_REPLICA_ENABLED . " for $scope scope" . ($scope != ScopeConfigInterface::SCOPE_TYPE_DEFAULT ? " (ID=$scopeId)" : "") . "</info>");
+        $this->configWriter->delete(ConfigHelper::USE_VIRTUAL_REPLICA_ENABLED, $scope, $scopeId);
+    }
+
+    public function disableVirtualReplicaSortConfig(string $scope = ScopeConfigInterface::SCOPE_TYPE_DEFAULT, int $scopeId = 0): void
+    {
+        $raw = $this->scopeConfig->getValue(ConfigHelper::SORTING_INDICES, $scope, $scopeId);
+        if (!$raw) {
+            return;
+        }
+        $sorting = array_map(
+            function($sort) {
+                $sort[ReplicaManagerInterface::SORT_KEY_VIRTUAL_REPLICA] = 0;
+                return $sort;
+            },
+            $this->serializer->unserialize($raw)
+        );
+        $this->output->writeln("<info>Disabling all virtual replicas in " . ConfigHelper::SORTING_INDICES . " for $scope scope" . ($scope != ScopeConfigInterface::SCOPE_TYPE_DEFAULT ? " (ID=$scopeId)" : "") . "</info>");
+        $this->configHelper->setSorting($sorting, $scope, $scopeId);
+    }
+
 }