Merge pull request #250 from Inchoo/feature/#215-snippets-removal

Feature/#215 snippets removal

Author: vvuksan

Diff for: Controller/Adminhtml/FastlyCdn/Manifest/ToggleModules.php

@@ -11,6 +11,10 @@
 use Magento\Framework\App\Action\Context;
 use Magento\Framework\App\Request\Http;
 use Fastly\Cdn\Model\ResourceModel\Manifest\CollectionFactory;
+use Fastly\Cdn\Model\Api;
+use Fastly\Cdn\Helper\Vcl;
+use Fastly\Cdn\Model\Config;
+use Magento\Framework\Exception\LocalizedException;
 
 /**
  * Class Create
@@ -23,30 +27,38 @@ class ToggleModules extends Action
      * @var ManifestFactory
      */
     private $manifestFactory;
-
     /**
      * @var Manifest
      */
     private $manifest;
-
     /**
      * @var Modly
      */
     private $modly;
-
     /**
      * @var ManifestResource
      */
     private $manifestResource;
-
     /**
      * @var JsonFactory
      */
     private $resultJson;
-
+    /**
+     * @var Http
+     */
     private $request;
-
+    /**
+     * @var CollectionFactory
+     */
     private $collectionFactory;
+    /**
+     * @var Api
+     */
+    private $api;
+
+    private $vcl;
+
+    private $enabledModules = [];
 
     public function __construct(
         Context $context,
@@ -56,7 +68,9 @@ public function __construct(
         Modly $modly,
         JsonFactory $resultJsonFactory,
         Http $request,
-        CollectionFactory $collectionFactory
+        CollectionFactory $collectionFactory,
+        Api $api,
+        Vcl $vcl
     ) {
         $this->manifestFactory = $manifestFactory;
         $this->manifestResource = $manifestResource;
@@ -65,6 +79,8 @@ public function __construct(
         $this->resultJson = $resultJsonFactory;
         $this->request = $request;
         $this->collectionFactory = $collectionFactory;
+        $this->api = $api;
+        $this->vcl = $vcl;
         parent::__construct($context);
     }
 
@@ -85,12 +101,22 @@ public function execute()
                     $manifest->setManifestId($moduleId);
                     $manifest->setManifestStatus(1);
                 } else {
+                    if ($module['manifest_status'] == 1) {
+                        $this->enabledModules[$moduleId] = $module['manifest_vcl'];
+                    }
                     $manifest->setManifestId($moduleId);
                     $manifest->setManifestStatus(0);
                 }
                 $this->saveManifest($manifest);
             }
 
+            if ($this->enabledModules) {
+                $removeStatus = $this->removeManifests($this->enabledModules);
+                if ($removeStatus != false) {
+                    throw new LocalizedException($removeStatus);
+                }
+            }
+
             return $result->setData([
                 'status' => true
             ]);
@@ -110,4 +136,44 @@ private function saveManifest($manifest)
     {
         $this->manifestResource->save($manifest);
     }
+
+    /**
+     * @param $enabledModules
+     * @return bool|\Exception
+     */
+    private function removeManifests($enabledModules)
+    {
+        try {
+            $service = $this->api->checkServiceDetails();
+            $activeVersion = $this->vcl->getCurrentVersion($service->versions);
+            $existingSnippets = [];
+
+            foreach ($enabledModules as $key => $value) {
+                $moduleVcl = json_decode($value);
+                foreach ($moduleVcl as $vcl) {
+                    $type = $vcl->type;
+                    $reqName = Config::FASTLY_MODLY_MODULE . '_' . $key . '_' . $type;
+                    $checkIfSnippetExist = $this->api->hasSnippet($activeVersion, $reqName);
+                    if ($checkIfSnippetExist) {
+                        $existingSnippets[] = $reqName;
+                    }
+                }
+            }
+
+            if ($existingSnippets) {
+                $clone = $this->api->cloneVersion($activeVersion);
+                foreach ($existingSnippets as $snippet) {
+                    $this->api->removeSnippet($clone->number, $snippet);
+                }
+                $this->api->validateServiceVersion($clone->number);
+                $this->api->activateVersion($clone->number);
+
+                $comment = ['comment' => 'Magento Module deleted Fastly Edge Module snippets.'];
+                $this->api->addComment($clone->number, $comment);
+            }
+            return false;
+        } catch (\Exception $e) {
+            return $e;
+        }
+    }
 }