paysystem.php

<?php
namespace Bitrix\Sale\Services\Company\Restrictions;
 
use Bitrix\Main\Localization\Loc;
use Bitrix\Sale\Internals\CollectableEntity;
use Bitrix\Sale\Internals\CompanyServiceTable;
use Bitrix\Sale;
use Bitrix\Sale\Services\Base;
 
Loc::loadMessages(__FILE__);
 
class PaySystem extends Base\Restriction
{
    public static $easeSort = 200;
 
    public static function getClassTitle()
    {
        return Loc::getMessage("SALE_COMPANY_RULES_BY_PS_TITLE");
    }
 
    public static function getClassDescription()
    {
        return Loc::getMessage("SALE_COMPANY_RULES_BY_PS_DESC");
    }
 
    public static function check($params, array $restrictionParams, $serviceId = 0)
    {
        if ((int)$serviceId <= 0)
            return true;
 
        if (!$params)
            return true;
 
        $paySystemIds = self::getPaySystemsByCompanyId($serviceId);
 
        if (empty($paySystemIds))
            return true;
 
        $diff = array_diff($params, $paySystemIds);
 
        return empty($diff);
    }
 
    protected static function extractParams(Sale\Internals\Entity $entity)
    {
        $result = array();
 
        $paymentCollection = null;
 
        if ($entity instanceof Sale\Payment)
        {
            $paymentCollection = $entity->getCollection();
        }
        elseif ($entity instanceof Sale\Shipment)
        {
            $shipmentCollection = $entity->getCollection();
            if ($shipmentCollection)
            {
                $order = $shipmentCollection->getOrder();
                if ($order)
                    $paymentCollection = $order->getPaymentCollection();
            }
        }
        elseif ($entity instanceOf Sale\Order)
        {
            $paymentCollection = $entity->getPaymentCollection();
        }
 
        if ($paymentCollection !== null)
        {
            foreach ($paymentCollection as $payment)
            {
                $paySystemId = $payment->getPaymentSystemId();
                if ($paySystemId)
                    $result[] = $paySystemId;
            }
        }
 
        return $result;
    }
 
    protected static function getPaySystemList()
    {
        $result = array();
 
        $dbRes = Sale\PaySystem\Manager::getList(array('select' => array('ID', 'NAME')));
        while ($paySystem = $dbRes->fetch())
            $result[$paySystem['ID']] = $paySystem['NAME'].' ['.$paySystem['ID'].']';
 
        return $result;
    }
 
    public static function getParamsStructure($entityId = 0)
    {
        $result =  array(
            "PAYSYSTEM" => array(
                "TYPE" => "ENUM",
                'MULTIPLE' => 'Y',
                "LABEL" => Loc::getMessage("SALE_COMPANY_RULES_BY_PS"),
                "OPTIONS" => self::getPaySystemList()
            )
        );
 
        return $result;
    }
 
    protected static function getPaySystemsByCompanyId($companyId = 0)
    {
        $result = array();
        if ($companyId == 0)
            return $result;
 
        $dbRes = CompanyServiceTable::getList(
            array(
                'select' => array('SERVICE_ID'),
                'filter' => array(
                    'COMPANY_ID' => $companyId,
                    'SERVICE_TYPE' => Sale\Services\Company\Restrictions\Manager::SERVICE_TYPE_PAYMENT)
            )
        );
 
        while ($data = $dbRes->fetch())
            $result[] = $data['SERVICE_ID'];
 
        return $result;
    }
 
    public static function save(array $fields, $restrictionId = 0)
    {
        $serviceIds = $fields["PARAMS"];
        $fields["PARAMS"] = array();
 
        if ($restrictionId > 0)
        {
            $dbRes = CompanyServiceTable::getList(
                array(
                    'select' => array('SERVICE_ID'),
                    'filter' => array(
                        'SERVICE_TYPE' => Sale\Services\Company\Restrictions\Manager::SERVICE_TYPE_PAYMENT,
                        'COMPANY_ID' => $fields['SERVICE_ID']
                    )
                )
            );
 
            while($data = $dbRes->fetch())
            {
                $key = array_search($data['SERVICE_ID'], $serviceIds['PAYSYSTEM']);
                if (!$key)
                {
                    CompanyServiceTable::delete(array('COMPANY_ID' => $fields['SERVICE_ID'], 'SERVICE_ID' => $data['SERVICE_ID'], 'SERVICE_TYPE' => Sale\Services\Company\Restrictions\Manager::SERVICE_TYPE_PAYMENT));
                }
                else
                {
                    unset($serviceIds['PAYSYSTEM'][$key]);
                }
            }
        }
 
        $result = parent::save($fields, $restrictionId);
 
        $addFields = array('COMPANY_ID' => $fields['SERVICE_ID'], 'SERVICE_TYPE' => Sale\Services\Company\Restrictions\Manager::SERVICE_TYPE_PAYMENT);
        foreach ($serviceIds['PAYSYSTEM'] as $id)
        {
            $addFields['SERVICE_ID'] = $id;
            CompanyServiceTable::add($addFields);
        }
 
        return $result;
    }
 
    public static function prepareParamsValues(array $paramsValues, $entityId = 0)
    {
        return array("PAYSYSTEM" => self::getPaySystemsByCompanyId($entityId));
    }
 
    public static function delete($restrictionId, $entityId = 0)
    {
        $dbRes = CompanyServiceTable::getList(
            array(
                'select' => array('SERVICE_ID'),
                'filter' => array(
                    'SERVICE_TYPE' => Sale\Services\Company\Restrictions\Manager::SERVICE_TYPE_PAYMENT,
                    'COMPANY_ID' => $entityId
                )
            )
        );
 
        while ($data = $dbRes->fetch())
        {
            CompanyServiceTable::delete(array('COMPANY_ID' => $entityId, 'SERVICE_ID' => $data['SERVICE_ID'], 'SERVICE_TYPE' => Sale\Services\Company\Restrictions\Manager::SERVICE_TYPE_PAYMENT));
        }
 
        return parent::delete($restrictionId);
    }
}