delivery.php

<?php
namespace Bitrix\Sale\Services\PaySystem\Restrictions;
 
use Bitrix\Main\Localization\Loc;
use Bitrix\Sale\Delivery\Services;
use Bitrix\Sale\Internals\CollectableEntity;
use Bitrix\Sale\Internals\DeliveryPaySystemTable;
use Bitrix\Sale\Internals\Entity;
use Bitrix\Sale\Order;
use Bitrix\Sale\Payment;
use Bitrix\Sale\PaymentCollection;
use Bitrix\Sale\Services\Base\Restriction;
use Bitrix\Sale\Services\Base\RestrictionManager;
use Bitrix\Sale\ShipmentCollection;
 
Loc::loadMessages(__FILE__);
 
class Delivery extends Restriction
{
    public static $easeSort = 200;
    protected static $preparedData = array();
 
    public static function getClassTitle()
    {
        return Loc::getMessage("SALE_SRV_RSTR_BY_DELIVERY_NAME");
    }
 
    public static function getClassDescription()
    {
        return Loc::getMessage("SALE_SRV_RSTR_BY_DELIVERY_DESC");
    }
 
    public static function getOnApplyErrorMessage(): string
    {
        return Loc::getMessage('SALE_SRV_RSTR_BY_DELIVERY_ON_APPLY_ERROR_MSG');
    }
 
    public static function check($params, array $restrictionParams, $serviceId = 0)
    {
        if(intval($serviceId) <= 0)
            return true;
 
        if(empty($params))
            return true;
 
        $deliveries = self::getDeliveryByPaySystemsId($serviceId);
 
        if(empty($deliveries))
            return true;
 
        $diff = array_diff($params, $deliveries);
 
        return empty($diff);
    }
 
    protected static function extractParams(Entity $entity)
    {
        $shipmentCollection = null;
        $result = array();
 
        if ($entity instanceof Payment)
        {
            $paymentCollection = $entity->getCollection();
 
            $order = $paymentCollection->getOrder();
 
            $shipmentCollection = $order->getShipmentCollection();
 
        }
        elseif ($entity instanceof Order)
        {
            $shipmentCollection = $entity->getShipmentCollection();
        }
 
        if ($shipmentCollection)
        {
            foreach ($shipmentCollection as $shipment)
            {
                if (!$shipment->isSystem())
                {
                    $deliveryId = $shipment->getDeliveryId();
                    if ($deliveryId)
                        $result[] = $deliveryId;
                }
            }
        }
 
        return $result;
    }
 
    protected static function getDeliveryServiceList()
    {
        static $result = null;
 
        if ($result !== null)
            return $result;
 
        $serviceList = array();
        $dbRes = Services\Table::getList(array('select' => array('ID', 'NAME', 'PARENT_ID', 'CLASS_NAME')));
        while ($service = $dbRes->fetch())
            $serviceList[$service['ID']] = $service;
 
        foreach ($serviceList as $service)
        {
            if (is_callable($service['CLASS_NAME'].'::canHasChildren') && $service['CLASS_NAME']::canHasChildren())
                continue;
 
            if ((int)$service['PARENT_ID'] > 0 && array_key_exists($service['PARENT_ID'], $serviceList))
            {
                $parentService = $serviceList[$service['PARENT_ID']];
 
                if (is_callable($parentService['CLASS_NAME'].'::canHasChildren') && $parentService['CLASS_NAME']::canHasChildren())
                    $name = $service['NAME'].' ['.$service['ID'].']';
                else
                    $name = $parentService['NAME'].': '.$service['NAME'].' ['.$service['ID'].']';
            }
            else
            {
                $name = $service['NAME'].' ['.$service['ID'].']';
            }
 
            $result[$service['ID']] = $name;
        }
 
        return $result;
    }
 
    public static function getParamsStructure($entityId = 0)
    {
        $result =  array(
            "DELIVERY" => array(
                "TYPE" => "ENUM",
                'MULTIPLE' => 'Y',
                "LABEL" => Loc::getMessage("SALE_SRV_RSTR_BY_DELIVERY_PRM_PS"),
                "OPTIONS" => self::getDeliveryServiceList()
            )
        );
 
        return $result;
    }
 
    protected static function getDeliveryByPaySystemsId($paySystemId = 0)
    {
        if ($paySystemId == 0)
            return array();
 
        $result = DeliveryPaySystemTable::getLinks($paySystemId, DeliveryPaySystemTable::ENTITY_TYPE_PAYSYSTEM, self::$preparedData);
        return $result;
    }
 
    protected static function prepareParamsForSaving(array $params = array(), $paySystemId = 0)
    {
        if(intval($paySystemId) <= 0)
            return $params;
 
        if(isset($params["DELIVERY"]) && is_array($params["DELIVERY"]))
        {
            DeliveryPaySystemTable::setLinks(
                $paySystemId,
                DeliveryPaySystemTable::ENTITY_TYPE_PAYSYSTEM,
                $params["DELIVERY"],
                true
            );
 
            unset($params["DELIVERY"]);
        }
 
        return $params;
    }
 
    public static function save(array $fields, $restrictionId = 0)
    {
        $params = $fields["PARAMS"];
        $fields["PARAMS"] = array();
 
        $result = parent::save($fields, $restrictionId);
 
        self::prepareParamsForSaving($params, $fields["SERVICE_ID"]);
        return $result;
    }
 
    public static function prepareParamsValues(array $paramsValues, $entityId = 0)
    {
        return array("DELIVERY" => self::getDeliveryByPaySystemsId($entityId));
    }
 
    public static function delete($restrictionId, $entityId = 0)
    {
        DeliveryPaySystemTable::setLinks(
            $entityId,
            DeliveryPaySystemTable::ENTITY_TYPE_PAYSYSTEM,
            array(),
            true
        );
 
        return parent::delete($restrictionId);
    }
 
    public static function prepareData(array $servicesIds)
    {
        if(empty($servicesIds))
            return;
 
        self::$preparedData = DeliveryPaySystemTable::prepareData($servicesIds, DeliveryPaySystemTable::ENTITY_TYPE_PAYSYSTEM);
    }
}