Your IP : 216.73.216.43
<?php
/**
* @author Amasty Team
* @copyright Copyright (c) 2023 Amasty (https://www.amasty.com)
* @package Reward Points Base for Magento 2
*/
namespace Amasty\Rewards\Model\ResourceModel\Rule;
use Amasty\Rewards\Model\ResourceModel\Rule;
use Magento\Framework\EntityManager\MetadataPool;
use Magento\Framework\EntityManager\Operation\AttributeInterface;
/**
* Save Handler Data
*/
class SaveHandler implements AttributeInterface
{
/**
* @var Rule
*/
protected $ruleResource;
/**
* @var MetadataPool
*/
protected $metadataPool;
/**
* @param Rule $ruleResource
* @param MetadataPool $metadataPool
*/
public function __construct(
Rule $ruleResource,
MetadataPool $metadataPool
) {
$this->ruleResource = $ruleResource;
$this->metadataPool = $metadataPool;
}
/**
* @param string $entityType
* @param array $entityData
* @param array $arguments
* @return array
* @throws \Exception
* @SuppressWarnings(PHPMD.UnusedFormalParameter)
*/
public function execute($entityType, $entityData, $arguments = [])
{
$linkField = $this->metadataPool->getMetadata($entityType)->getLinkField();
if (isset($entityData['website_ids'])) {
$storeIds = $entityData['website_ids'];
if (!is_array($storeIds)) {
$storeIds = explode(',', (string)$storeIds);
}
$this->ruleResource->bindRuleToEntity($entityData[$linkField], $storeIds, 'website');
}
if (isset($entityData['customer_group_ids'])) {
$customerGroupIds = $entityData['customer_group_ids'];
if (!is_array($customerGroupIds)) {
$customerGroupIds = explode(',', (string)$customerGroupIds);
}
$this->ruleResource->bindRuleToEntity($entityData[$linkField], $customerGroupIds, 'customer_group');
}
return $entityData;
}
}