discountentities.php

<?php
namespace Bitrix\Sale\Internals;
 
use Bitrix\Main;
use Bitrix\Main\Application;
use Bitrix\Main\Localization\Loc;
Loc::loadMessages(__FILE__);
 
class DiscountEntitiesTable extends Main\Entity\DataManager
{
    public static function getTableName()
    {
        return 'b_sale_discount_entities';
    }
 
    public static function getMap()
    {
        return array(
            'ID' => new Main\Entity\IntegerField('ID', array(
                'primary' => true,
                'autocomplete' => true,
                'title' => Loc::getMessage('DISCOUNT_ENTITIES_ENTITY_ID_FIELD')
            )),
            'DISCOUNT_ID' => new Main\Entity\IntegerField('DISCOUNT_ID', array(
                'required' => true,
                'title' => Loc::getMessage('DISCOUNT_ENTITIES_ENTITY_DISCOUNT_ID_FIELD')
            )),
            'MODULE_ID' => new Main\Entity\StringField('MODULE_ID', array(
                'required' => true,
                'validation' => array(__CLASS__, 'validateModuleId'),
                'title' => Loc::getMessage('DISCOUNT_ENTITIES_ENTITY_MODULE_ID_FIELD')
            )),
            'ENTITY' => new Main\Entity\StringField('ENTITY', array(
                'required' => true,
                'validation' => array(__CLASS__, 'validateEntity'),
                'title' => Loc::getMessage('DISCOUNT_ENTITIES_ENTITY_ENTITY_FIELD')
            )),
            'FIELD_ENTITY' => new Main\Entity\StringField('FIELD_ENTITY', array(
                'required' => true,
                'validation' => array(__CLASS__, 'validateFieldEntity'),
                'title' => Loc::getMessage('DISCOUNT_ENTITIES_ENTITY_FIELD_ENTITY_FIELD')
            )),
            'FIELD_TABLE' => new Main\Entity\StringField('FIELD_TABLE', array(
                'required' => true,
                'validation' => array(__CLASS__, 'validateFieldTable'),
                'title' => Loc::getMessage('DISCOUNT_ENTITIES_ENTITY_FIELD_TABLE_FIELD'),
            )),
            'DISCOUNT' => new Main\Entity\ReferenceField(
                'DISCOUNT',
                'Bitrix\Sale\Internals\Discount',
                array('=this.DISCOUNT_ID' => 'ref.ID'),
                array('join_type' => 'LEFT')
            )
        );
    }
    public static function validateModuleId()
    {
        return array(
            new Main\Entity\Validator\Length(null, 50),
        );
    }
    public static function validateEntity()
    {
        return array(
            new Main\Entity\Validator\Length(null, 255),
        );
    }
    public static function validateFieldEntity()
    {
        return array(
            new Main\Entity\Validator\Length(null, 255),
        );
    }
    public static function validateFieldTable()
    {
        return array(
            new Main\Entity\Validator\Length(null, 255),
        );
    }
 
    public static function deleteByDiscount($discount)
    {
        $discount = (int)$discount;
        if ($discount <= 0)
            return;
        $conn = Application::getConnection();
        $helper = $conn->getSqlHelper();
        $conn->queryExecute(
            'delete from '.$helper->quote(self::getTableName()).' where '.$helper->quote('DISCOUNT_ID').' = '.$discount
        );
    }
 
    public static function updateByDiscount($discount, $entityList, $clear)
    {
        $discount = (int)$discount;
        if ($discount <= 0)
            return false;
        $clear = ($clear === true);
        if ($clear)
        {
            self::deleteByDiscount($discount);
        }
        if (!empty($entityList) && is_array($entityList))
        {
            foreach ($entityList as &$entity)
            {
                $fields = array(
                    'DISCOUNT_ID' => $discount,
                    'MODULE_ID' => $entity['MODULE'],
                    'ENTITY' => $entity['ENTITY'],
                    'FIELD_ENTITY' => $entity['FIELD_ENTITY'],
                );
                if (is_array($fields['FIELD_ENTITY']))
                    $fields['FIELD_ENTITY'] = implode('-', $fields['FIELD_ENTITY']);
                if (isset($entity['FIELD_TABLE']) && is_array($entity['FIELD_TABLE']))
                {
                    foreach ($entity['FIELD_TABLE'] as $oneField)
                    {
                        if (empty($oneField))
                            continue;
                        $fields['FIELD_TABLE'] = $oneField;
                        $result = self::add($fields);
                    }
                    unset($oneField);
                }
                else
                {
                    $fields['FIELD_TABLE'] = (isset($entity['FIELD_TABLE']) ? $entity['FIELD_TABLE'] : $entity['FIELD_ENTITY']);
                    $result = self::add($fields);
                }
            }
            unset($entity);
        }
        return true;
    }
 
    public static function getByDiscount($discountList, $filter = array(), $groupModule = true)
    {
        $groupModule = ($groupModule === true);
        $result = array();
        if (!empty($discountList) && is_array($discountList))
        {
            Main\Type\Collection::normalizeArrayValuesByInt($discountList);
            if (!empty($discountList))
            {
                if (!is_array($filter))
                    $filter = array();
 
                $discountRows = array_chunk($discountList, 500);
                foreach ($discountRows as &$row)
                {
                    $filter['@DISCOUNT_ID'] = $row;
 
                    $entityIterator = self::getList(array(
                        'select' => array('DISCOUNT_ID', 'MODULE_ID', 'ENTITY', 'FIELD_ENTITY', 'FIELD_TABLE'),
                        'filter' => $filter
                    ));
                    if ($groupModule)
                    {
                        while ($entity = $entityIterator->fetch())
                        {
                            unset($entity['DISCOUNT_ID']);
                            if (!isset($result[$entity['MODULE_ID']]))
                                $result[$entity['MODULE_ID']] = array();
                            if (!isset($result[$entity['MODULE_ID']][$entity['ENTITY']]))
                                $result[$entity['MODULE_ID']][$entity['ENTITY']] = array();
                            $result[$entity['MODULE_ID']][$entity['ENTITY']][$entity['FIELD_ENTITY']] = $entity;
                        }
                    }
                    else
                    {
                        while ($entity = $entityIterator->fetch())
                        {
                            $entity['DISCOUNT_ID'] = (int)$entity['DISCOUNT_ID'];
                            if (!isset($result[$entity['DISCOUNT_ID']]))
                                $result[$entity['DISCOUNT_ID']] = array();
                            $result[$entity['DISCOUNT_ID']][] = $entity;
                        }
                    }
                    unset($entity, $entityIterator);
                }
                unset($row, $discountRows);
            }
        }
        return $result;
    }
}