CustomerGroup.php

Go to the documentation of this file.

<?php
namespace Magento\Catalog\Model\Indexer\Product\Price\Plugin;

use Magento\Catalog\Model\Indexer\Product\Price\DimensionModeConfiguration;
use Magento\Customer\Api\GroupRepositoryInterface;
use Magento\Customer\Api\Data\GroupInterface;
use Magento\Catalog\Model\Indexer\Product\Price\UpdateIndexInterface;
use Magento\Catalog\Model\Indexer\Product\Price\TableMaintainer;
use Magento\Framework\Indexer\Dimension;
use Magento\Framework\Indexer\DimensionFactory;
use Magento\Customer\Model\Indexer\CustomerGroupDimensionProvider;
use Magento\Store\Model\Indexer\WebsiteDimensionProvider;

class CustomerGroup
{
    private $updateIndex;

    private $tableMaintainer;

    private $dimensionFactory;

    private $dimensionModeConfiguration;

    private $websiteDimensionProvider;

    public function __construct(
        UpdateIndexInterface $updateIndex,
        TableMaintainer $tableMaintainer,
        DimensionFactory $dimensionFactory,
        DimensionModeConfiguration $dimensionModeConfiguration,
        WebsiteDimensionProvider $websiteDimensionProvider
    ) {
        $this->updateIndex = $updateIndex;
        $this->tableMaintainer = $tableMaintainer;
        $this->dimensionFactory = $dimensionFactory;
        $this->dimensionModeConfiguration = $dimensionModeConfiguration;
        $this->websiteDimensionProvider = $websiteDimensionProvider;
    }

    public function aroundSave(
        GroupRepositoryInterface $subject,
        \Closure $proceed,
        GroupInterface $group
    ) {
        $isGroupNew = !$group->getId();
        $group = $proceed($group);
        if ($isGroupNew) {
            foreach ($this->getAffectedDimensions((string)$group->getId()) as $dimensions) {
                $this->tableMaintainer->createTablesForDimensions($dimensions);
            }
        }
        $this->updateIndex->update($group, $isGroupNew);
        return $group;
    }

    public function afterDeleteById(GroupRepositoryInterface $subject, bool $result, string $groupId)
    {
        foreach ($this->getAffectedDimensions($groupId) as $dimensions) {
            $this->tableMaintainer->dropTablesForDimensions($dimensions);
        }

        return $result;
    }

    private function getAffectedDimensions(string $groupId): array
    {
        $currentDimensions = $this->dimensionModeConfiguration->getDimensionConfiguration();
        // do not return dimensions if Customer Group dimension is not present in configuration
        if (!in_array(CustomerGroupDimensionProvider::DIMENSION_NAME, $currentDimensions, true)) {
            return [];
        }
        $customerGroupDimension = $this->dimensionFactory->create(
            CustomerGroupDimensionProvider::DIMENSION_NAME,
            $groupId
        );

        $dimensions = [];
        if (in_array(WebsiteDimensionProvider::DIMENSION_NAME, $currentDimensions, true)) {
            foreach ($this->websiteDimensionProvider as $websiteDimension) {
                $dimensions[] = [
                    $customerGroupDimension,
                    $websiteDimension
                ];
            }
        } else {
            $dimensions[] = [$customerGroupDimension];
        }

        return $dimensions;
    }
}