enumfield.php

<?php
namespace Bitrix\Main\ORM\Fields;
 
use Bitrix\Main\DB\SqlExpression;
use Bitrix\Main\SystemException;
 
class EnumField extends ScalarField
{
    protected $values;
 
    function __construct($name, $parameters = array())
    {
        parent::__construct($name, $parameters);
 
        if (isset($parameters['values']))
        {
            $this->values = $parameters['values'];
        }
    }
 
    public function postInitialize()
    {
        if (!is_array($this->values))
        {
            throw new SystemException(sprintf(
                'Parameter "values" for %s field in `%s` entity should be an array',
                $this->name, $this->entity->getDataClass()
            ));
        }
 
        if (empty($this->values))
        {
            throw new SystemException(sprintf(
                'Required parameter "values" for %s field in `%s` entity is not found',
                $this->name, $this->entity->getDataClass()
            ));
        }
    }
 
    public function configureValues($values)
    {
        $this->values = $values;
        return $this;
    }
 
    public function getValidators()
    {
        $validators = parent::getValidators();
 
        if ($this->validation === null)
        {
            $validators[] = new Validators\EnumValidator;
        }
 
        return $validators;
    }
 
    public function getValues()
    {
        return $this->values;
    }
 
    public function cast($value)
    {
        if ($this->is_nullable && $value === null)
        {
            return $value;
        }
 
        if ($value instanceof SqlExpression)
        {
            return $value;
        }
 
        return $value;
    }
 
    public function convertValueFromDb($value)
    {
        return $value;
    }
 
    public function convertValueToDb($value)
    {
        if ($value instanceof SqlExpression)
        {
            return $value;
        }
 
        return $value === null && $this->is_nullable
            ? $value
            : $this->getConnection()->getSqlHelper()->convertToDbString($value);
    }
}