flipbox\craft\ember\records\ElementAttributeTrait

Complexity

Total Complexity

11

Size/Duplication

Total Lines	89
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	5

Test Coverage

Coverage

0%

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	elementAttributes()	0	6	1
A	elementAttributeLabels()	0	6	1
A	internalSetElementId()	0	5	1
A	internalGetElementId()	0	7	2
A	resolveElement()	0	8	2
A	resolveElementFromRelation()	0	14	3
A	getElementRecord()	0	7	1

<?php

/**
 * @copyright  Copyright (c) Flipbox Digital Limited
 * @license    https://github.com/flipboxfactory/craft-ember/blob/master/LICENSE
 * @link       https://github.com/flipboxfactory/craft-ember/
 */

namespace flipbox\craft\ember\records;

use Craft;
use craft\base\ElementInterface;
use craft\records\Element as ElementRecord;
use flipbox\craft\ember\models\ElementRulesTrait;
use flipbox\craft\ember\objects\ElementMutatorTrait;
use yii\db\ActiveQueryInterface;

/**
 * Intended to be used on an ActiveRecord, this class provides `$this->elementId` attribute along with 'getters'
 * and 'setters' to ensure continuity between the Id and Object.  An element object is lazy loaded when called.
 * In addition, ActiveRecord rules are available.
 *
 * @author Flipbox Factory <hello@flipboxfactory.com>
 * @since 2.0.0
 *
 * @property ElementRecord $elementRecord
 */
trait ElementAttributeTrait
{
    use ActiveRecordTrait,
        ElementRulesTrait,
        ElementMutatorTrait;

    /**
     * @inheritdoc
     */
    public function elementAttributes(): array
    {
        return [
            'elementId'
        ];
    }

    /**
     * @inheritdoc
     */
    public function elementAttributeLabels(): array
    {
        return [
            'elementId' => Craft::t('app', 'Element Id')
        ];
    }

    /**
     * @inheritDoc
     */
    protected function internalSetElementId(int $id = null)
    {
        $this->setAttribute('elementId', $id);
        return $this;
    }

    /**
     * @inheritDoc
     */
    protected function internalGetElementId()
    {
        if (null === ($id = $this->getAttribute('elementId'))) {
            return null;
        }
        return (int)$id;
    }

    /**
     * @return ElementInterface|null
     */
    protected function resolveElement()
    {
        if ($model = $this->resolveElementFromRelation()) {
            return $model;
        }

        return $this->resolveElementFromId();
    }

    /**
     * @return ElementInterface|null
     */
    private function resolveElementFromRelation()
    {
        if (false === $this->isRelationPopulated('elementRecord')) {
            return null;
        }

        if (null === ($record = $this->getRelation('elementRecord'))) {
            return null;
        }

        /** @var ElementRecord $record */

        return Craft::$app->getElements()->getElementById($record->id);
    }

    /**
     * Get the associated Element
     *
     * @return ActiveQueryInterface
     */
    public function getElementRecord(): ActiveQueryInterface
    {
        return $this->hasOne(
            ElementRecord::class,
            ['id' => 'elementId']
        );
    }
}