henzeb/enumhancer

<?php

namespace Henzeb\Enumhancer\Laravel\Concerns;

use BackedEnum;
use Henzeb\Enumhancer\Helpers\EnumGetters;
use Henzeb\Enumhancer\Helpers\EnumValue;
use Illuminate\Database\Eloquent\Model;
use UnitEnum;
use ValueError;
use function Henzeb\Enumhancer\Functions\backing;

/**
 * @mixin Model
 * @property bool $keepEnumCase Change public to private/protected if needed.
 */
trait CastsBasicEnumerations
{
    protected function getEnumCastableAttributeValue($key, $value)
    {
        if (is_null($value)) {
            return null;
        }

        $castType = $this->getCasts()[$key];

        if (!$value instanceof $castType) {
            $value = EnumGetters::get($castType, $value);
        }

        if ($this->shouldUseBasicEnumWorkaround($castType)) {
            $keepEnumCase = property_exists($this, 'keepEnumCase') ? $this->keepEnumCase : true;

            return backing($value, $keepEnumCase);
        }

        return $value;
    }

    protected function setEnumCastableAttribute($key, $value)
    {
        $enumClass = $this->getCasts()[$key];

        $keepEnumCase = property_exists($this, 'keepEnumCase') ? $this->keepEnumCase : true;

        if (!isset($value)) {
            $this->attributes[$key] = null;
            return;
        }

        if ($value instanceof $enumClass) {
            $value = EnumValue::value($value, $keepEnumCase);
        }

        if ($value instanceof UnitEnum && !$value instanceof $enumClass) {
            throw new ValueError(
                sprintf('Enum of `%s` expected, got `%s`', $enumClass, $value::class)
            );
        }

        $this->attributes[$key] = EnumValue::value(EnumGetters::get($enumClass, $value), $keepEnumCase);
    }

    private function shouldUseBasicEnumWorkaround(string $enumClass): bool
    {
        return (!is_subclass_of($enumClass, BackedEnum::class, true))
            && 'toArray' === (debug_backtrace(2)[5]['function'] ?? null);
    }

    protected function getStorableEnumValue($expectedEnumValue, $value = null)
    {
        $value ??= $expectedEnumValue;
        if ($value instanceof UnitEnum) {
            $keepEnumCase = property_exists($this, 'keepEnumCase') ? $this->keepEnumCase : true;

            return EnumValue::value($value, $keepEnumCase);
        }

        return $value;
    }
}