rhertogh\Yii2Oauth2Server\models\behaviors\DateTimeBehavior

Complexity

Total Complexity

10

Size/Duplication

Total Lines	59
Duplicated Lines	0 %

Test Coverage

Coverage

94.12%

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	events()	0	8	1
A	afterFind()	0	11	4
A	beforeSave()	0	12	5

<?php

namespace rhertogh\Yii2Oauth2Server\models\behaviors;

use DateTimeImmutable;
use yii\base\Behavior;
use yii\db\ActiveRecord;
use yii\db\BaseActiveRecord;
use yii\db\Schema;

/**
 * @property BaseActiveRecord $owner
 */
class DateTimeBehavior extends Behavior
{
    /**
     * The DateTime format to convert to before inserting the DateTime object into the database.
     * @var string
     */
    public $dateTimeFormat = 'Y-m-d H:i:s';

    /**
     * @inheritDoc
     */
    public function events()
    {
        return [
            ActiveRecord::EVENT_BEFORE_INSERT => 'beforeSave',
            ActiveRecord::EVENT_BEFORE_UPDATE => 'beforeSave',
            ActiveRecord::EVENT_AFTER_INSERT => 'afterFind',
            ActiveRecord::EVENT_AFTER_UPDATE => 'afterFind',
            ActiveRecord::EVENT_AFTER_FIND => 'afterFind',
        ];
    }

    /**
     * Convert DateTime objects to string before writing the database.
     * @throws \yii\base\InvalidConfigException
     * @since 1.0.0
     */
    public function beforeSave()
    {
        /** @var ActiveRecord $activeRecord */
        $activeRecord = $this->owner;

        foreach ($activeRecord->getTableSchema()->columns as $column) {
            $value = $this->owner[$column->name];
            if (
                $column->type === Schema::TYPE_DATETIME
                && ($value instanceof \DateTime || $value instanceof DateTimeImmutable)
            ) {
                $this->owner[$column->name] = $value->format($this->dateTimeFormat);
            }
        }
    }

    /**
     * Convert strings to DateTime objects after loading them from the database.
     * @throws \yii\base\InvalidConfigException
     * @since 1.0.0
     */
    public function afterFind()
    {
        /** @var ActiveRecord $activeRecord */
        $activeRecord = $this->owner;

        foreach ($activeRecord->getTableSchema()->columns as $column) {
            $value = $this->owner[$column->name];
            if ($column->type === Schema::TYPE_DATETIME && !empty($value)) {
                $dateTime = DateTimeImmutable::createFromFormat($this->dateTimeFormat, $value);
                $this->owner[$column->name] = $dateTime;
                $this->owner->setOldAttribute($column->name, $dateTime);
            }
        }
    }
}