check enable module

Author: ninhkx

Model/Resolver/LabelDataProvider.php

@@ -29,6 +29,7 @@
 use Magento\Framework\GraphQl\Query\ResolverInterface;
 use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
 use Mageplaza\ProductLabels\Block\Label;
+use Mageplaza\ProductLabels\Helper\Data;
 use Mageplaza\ProductLabels\Model\LabelRepository;
 use Mageplaza\ProductLabels\Model\Rule;
 
@@ -48,31 +49,43 @@ class LabelDataProvider implements ResolverInterface
      */
     protected $label;
 
+    /**
+     * @var Data
+     */
+    protected $helperData;
+
     /**
      * LabelDataProvider constructor.
      *
      * @param LabelRepository $labelRepository
      * @param Label $label
+     * @param Data $helperData
      */
     public function __construct(
         LabelRepository $labelRepository,
-        Label $label
+        Label $label,
+        Data $helperData
     ) {
         $this->labelRepository = $labelRepository;
         $this->label           = $label;
+        $this->helperData      = $helperData;
     }
 
     /**
      * @inheritdoc
      */
     public function resolve(Field $field, $context, ResolveInfo $info, array $value = null, array $args = null)
     {
+        if (!$this->helperData->isEnabled()) {
+            return [];
+        }
+
         if (!array_key_exists('model', $value) || !$value['model'] instanceof ProductInterface) {
             throw new LocalizedException(__('"model" value should be specified'));
         }
 
         /* @var $product ProductInterface */
-        $product = $value['model'];
+        $product   = $value['model'];
         $labelData = [];
 
         /** @var Rule $rule */

Model/Resolver/LabelRules.php

@@ -29,13 +29,19 @@
 use Magento\Framework\GraphQl\Query\ResolverInterface;
 use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
 use Mageplaza\ProductLabels\Api\LabelRepositoryInterface;
+use Mageplaza\ProductLabels\Helper\Data;
 
 /**
  * Class LabelRules
  * @package Mageplaza\ProductLabelsGraphQl\Model\Resolver
  */
 class LabelRules implements ResolverInterface
 {
+    /**
+     * @var Data
+     */
+    protected $helperData;
+
     /**
      * @var SearchCriteriaBuilder
      */
@@ -47,15 +53,18 @@ class LabelRules implements ResolverInterface
     protected $labelRuleRepository;
 
     /**
-     * Post constructor.
+     * LabelRules constructor.
      *
+     * @param Data $helperData
      * @param SearchCriteriaBuilder $searchCriteriaBuilder
      * @param LabelRepositoryInterface $labelRuleRepository
      */
     public function __construct(
+        Data $helperData,
         SearchCriteriaBuilder $searchCriteriaBuilder,
         LabelRepositoryInterface $labelRuleRepository
     ) {
+        $this->helperData      = $helperData;
         $this->searchCriteriaBuilder = $searchCriteriaBuilder;
         $this->labelRuleRepository   = $labelRuleRepository;
     }
@@ -65,6 +74,10 @@ public function __construct(
      */
     public function resolve(Field $field, $context, ResolveInfo $info, array $value = null, array $args = null)
     {
+        if (!$this->helperData->isEnabled()) {
+            return [];
+        }
+
         $this->vailidateArgs($args);
         $searchCriteria = $this->searchCriteriaBuilder->build('label_rules', $args);
         $searchCriteria->setCurrentPage($args['currentPage']);