accountnumber.php

<?php
 
namespace Bitrix\Sale\Internals;
 
 
use Bitrix\Main;
use Bitrix\Sale;
 
class AccountNumberGenerator
{
    const ACCOUNT_NUMBER_SEPARATOR = "/";
 
    public static function generateForOrder(Sale\OrderBase $order)
    {
        $id = (int)$order->getId();
        if ($id <= 0)
        {
            return false;
        }
 
        $accountNumber = static::generateCustom($order);
        if ($accountNumber)
        {
            $dbRes = $order::getList([
                'select' => ['ID'],
                'filter' => ['=ACCOUNT_NUMBER' => $accountNumber]
            ]);
            if ($dbRes->fetch())
            {
                $accountNumber = null;
            }
        }
        else
        {
            $accountNumber = static::generateBySettings($order);
        }
 
        if (!$accountNumber) // if no special template is used or error occured
        {
            $accountNumber = static::generateById($order);
        }
 
        $dbRes = $order::getList([
            'select' => ['ID'],
            'filter' => ['=ACCOUNT_NUMBER' => $accountNumber]
        ]);
        if ($dbRes->fetch())
        {
            $accountNumber = static::generateForOrder($order);
        }
 
        return $accountNumber;
    }
 
    private static function generateBySettings(Sale\OrderBase $order)
    {
        $accountNumber = static::getNextNumber($order);
        if ($accountNumber)
        {
            $dbRes = $order::getList(['filter' => ["=ACCOUNT_NUMBER" => $accountNumber]]);
            if ($dbRes->fetch())
            {
                return null;
            }
        }
 
        return $accountNumber;
    }
 
    private static function generateById(Sale\OrderBase $order)
    {
        $accountNumber = $order->getId();
        for ($i = 1; $i <= 10; $i++)
        {
            $dbRes = $order::getList([
                'select' => ['ID'],
                'filter' => ['=ACCOUNT_NUMBER' => $accountNumber]
            ]);
            if ($dbRes->fetch())
            {
                $accountNumber = $order->getId()."-".$i;
            }
            else
            {
                break;
            }
        }
 
        return $accountNumber;
    }
 
    private static function generateCustom(Sale\OrderBase $order)
    {
        $type = Main\Config\Option::get("sale", "account_number_template", "");
 
        foreach(GetModuleEvents("sale", "OnBeforeOrderAccountNumberSet", true) as $arEvent)
        {
            $tmpRes = ExecuteModuleEventEx($arEvent, array($order->getId(), $type));
            if ($tmpRes !== false)
            {
                return $tmpRes;
            }
        }
 
        return null;
    }
 
    public static function generateForPayment(CollectableEntity $item)
    {
        return static::generate($item);
    }
 
    public static function generateForShipment(CollectableEntity $item)
    {
        return static::generate($item);
    }
 
    private static function generate(CollectableEntity $item)
    {
        $accountNumber = null;
        if (!$collection = $item->getCollection())
        {
            throw new Main\ObjectNotFoundException('Entity "Collection" not found');
        }
 
        if (!method_exists($collection, "getOrder"))
        {
            throw new Main\NotSupportedException();
        }
 
        if (!$order = $collection->getOrder())
        {
            throw new Main\ObjectNotFoundException('Entity "Order" not found');
        }
 
        $accountNumber = $order->getField('ACCOUNT_NUMBER').static::ACCOUNT_NUMBER_SEPARATOR;
 
        $count = 1;
        foreach ($collection as $itemCollection)
        {
            if (strval($itemCollection->getField("ACCOUNT_NUMBER")) != "")
            {
                $accountNumberIdList = explode(static::ACCOUNT_NUMBER_SEPARATOR, $itemCollection->getField("ACCOUNT_NUMBER"));
 
                $itemAccountNumber = trim(end($accountNumberIdList));
 
                if ($count <= $itemAccountNumber)
                    $count = $itemAccountNumber + 1;
            }
        }
 
        return $accountNumber.$count;
    }
 
    private static function getNextNumber(Sale\OrderBase $order)
    {
        $accountNumber = null;
 
        $numeratorForOrderSettings = Main\Numerator\Numerator::getOneByType($order::getRegistryType());
        $numerator = null;
        if ($numeratorForOrderSettings)
        {
            $numerator = Main\Numerator\Numerator::load(
                $numeratorForOrderSettings['id'],
                [
                    'ORDER_ID' => $order->getId()
                ]);
        }
        if ($numerator)
        {
            $accountNumber = $numerator->getNext();
        }
 
        return $accountNumber;
    }
}