usageentity.php

<?php
namespace Bitrix\Rest;
 
use Bitrix\Main;
use \Bitrix\Main\DB\SqlQueryException;
 
class UsageEntityTable extends Main\Entity\DataManager
{
 
    const ENTITY_TYPE_APPLICATION = 'A';
    const ENTITY_TYPE_WEBHOOK = 'W';
 
    const SUB_ENTITY_TYPE_METHOD = 'M';
    const SUB_ENTITY_TYPE_EVENT = 'E';
    const SUB_ENTITY_TYPE_PLACEMENT = 'P';
    const SUB_ENTITY_TYPE_ROBOT = 'R';
    const SUB_ENTITY_TYPE_BIZ_PROC = 'B';
    const SUB_ENTITY_TYPE_ACTIVITY = 'A';
    const SUB_ENTITY_TYPE_CONFIGURATION = 'C';
    const SUB_ENTITY_TYPE_SEND_MESSAGE = 'S';
    const SUB_ENTITY_TYPE_LANDING = 'L';
    const SUB_ENTITY_TYPE_LANDING_KNOWLEDGE = 'K';
    const SUB_ENTITY_TYPE_UI = 'U';
    const SUB_ENTITY_TYPE_BI_SUPERSET = 'I';
 
    protected static $info = array();
 
    public static function getTableName()
    {
        return 'b_rest_usage_entity';
    }
 
    public static function getMap()
    {
        return array(
            'ID' => array(
                'data_type' => 'integer',
                'primary' => true,
                'autocomplete' => true
            ),
            'ENTITY_TYPE' => array(
                'data_type' => 'string',
                'required' => true,
                'values' => array(
                    self::ENTITY_TYPE_APPLICATION,
                    self::ENTITY_TYPE_WEBHOOK
                ),
                'validation' => array(
                    __CLASS__,
                    'validateEntityType'
                )
            ),
            'ENTITY_ID' => array(
                'data_type' => 'integer',
                'required' => true
            ),
            'ENTITY_CODE' => array(
                'data_type' => 'string',
                'validation' => array(
                    __CLASS__,
                    'validateEntityCode'
                )
            ),
            'SUB_ENTITY_TYPE' => array(
                'data_type' => 'string',
                'values' => array(
                    self::SUB_ENTITY_TYPE_METHOD,
                    self::SUB_ENTITY_TYPE_EVENT,
                    self::SUB_ENTITY_TYPE_PLACEMENT,
                    self::SUB_ENTITY_TYPE_ROBOT,
                    self::SUB_ENTITY_TYPE_BIZ_PROC,
                    self::SUB_ENTITY_TYPE_ACTIVITY,
                    self::SUB_ENTITY_TYPE_CONFIGURATION,
                    self::SUB_ENTITY_TYPE_SEND_MESSAGE,
                    self::SUB_ENTITY_TYPE_LANDING,
                    self::SUB_ENTITY_TYPE_LANDING_KNOWLEDGE,
                    self::SUB_ENTITY_TYPE_BI_SUPERSET,
                ),
                'validation' => array(
                    __CLASS__,
                    'validateSubEntityType'
                )
            ),
            'SUB_ENTITY_NAME' => array(
                'data_type' => 'string',
                'validation' => array(
                    __CLASS__,
                    'validateSubEntityName'
                )
            ),
        );
    }
 
    protected static function getEntityInfo($entityType, $entityId)
    {
        $key = $entityType.'|'.$entityId;
        if (!isset(static::$info[$key]))
        {
            if ($entityType == self::ENTITY_TYPE_APPLICATION)
            {
                $appInfo = AppTable::getByClientId($entityId);
                static::$info[$key] = array(
                    'ENTITY_ID' => $appInfo['ID'],
                    'ENTITY_CODE' => $appInfo['CLIENT_ID'],
                );
            }
            else
            {
                static::$info[$key] = array(
                    'ENTITY_ID' => $entityId,
                    'ENTITY_CODE' => '',
                );
            }
        }
        return static::$info[$key];
    }
 
    public static function register($entityType, $entityId, $subEntityType, $subEntityName)
    {
        $entity = static::getEntityInfo($entityType, $entityId);
 
        $getListParameters = [
            'filter' => [
                '=ENTITY_TYPE' => $entityType,
                '=ENTITY_ID' => $entity['ENTITY_ID'],
                '=SUB_ENTITY_TYPE' => $subEntityType,
                '=SUB_ENTITY_NAME' => $subEntityName,
            ],
            'select' => [
                'ID',
            ],
            'limit' => 1,
        ];
 
        $res = static::getList($getListParameters);
        $element = $res->fetch();
        if ($element)
        {
            return $element['ID'];
        }
 
        $newEntity = array(
            'ENTITY_TYPE' => $entityType,
            'ENTITY_ID' => $entity['ENTITY_ID'],
            'ENTITY_CODE' => $entity['ENTITY_CODE'],
            'SUB_ENTITY_TYPE' => $subEntityType,
            'SUB_ENTITY_NAME' => $subEntityName
        );
 
        try
        {
            $res = static::add($newEntity);
            return $res->isSuccess() ? $res->getId() : false;
        }
        catch (SqlQueryException $e)
        {
            if (mb_strpos($e->getMessage(), 'Duplicate entry') !== false)
            {
                //Try to get one more time
                $res = static::getList($getListParameters);
                $element = $res->fetch();
                if ($element)
                {
                    return $element['ID'];
                }
            }
            throw $e;
        }
    }
 
    public static function validateEntityType()
    {
        return array(new Main\Entity\Validator\Length(null, 1),);
    }
 
    public static function validateEntityCode()
    {
        return array(new Main\Entity\Validator\Length(null, 255),);
    }
 
    public static function validateSubEntityType()
    {
        return array(new Main\Entity\Validator\Length(null, 1),);
    }
 
    public static function validateSubEntityName()
    {
        return array(new Main\Entity\Validator\Length(null, 255),);
    }
}