app\behaviors\PivotSaveBehavior

Complexity

Total Complexity

15

Size/Duplication

Total Lines	62
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	canSetProperty()	0	3	1
A	setPivots()	0	13	5
A	__set()	0	6	2
A	getStoredPivots()	0	7	2
A	savePivots()	0	8	4
A	getModelClass()	0	3	1

<?php


namespace app\behaviors;


use carono\yii2file\FileUploadTrait;
use carono\yii2migrate\traits\PivotTrait;
use yii\db\ActiveRecord;

/**
 * Class PivotSaveBehavior
 *
 * @package app\behaviors
 * @property PivotTrait|ActiveRecord $owner
 */
class PivotSaveBehavior extends \yii\base\Behavior
{
    protected $_pivots;
    public $attribute;
    public $field;
    public $pivotClass;
    public $fileClass;

    public function canSetProperty($name, $checkVars = true)
    {
        return $name == $this->attribute;
    }

    public function __set($name, $value)
    {
        if ($this->canSetProperty($name)) {
            $this->setPivots($value);
        } else {
            parent::__set($name, $value);
        }
    }

    public function getStoredPivots($attribute)
    {
        if ($attribute == $this->attribute) {
            return $this->_pivots;
        }

        return null;
    }

    /**
     * @return ActiveRecord|FileUploadTrait
     */
    protected function getModelClass()
    {
        return $this->fileClass;
    }

    protected function setPivots($values)
    {
        $class = $this->getModelClass();
        $this->_pivots = [];
        foreach ((array)$values as $value) {
            if (is_numeric($value)) {
                $this->_pivots[] = $class::findOne($value);
            } elseif ($value instanceof $class) {
                $this->_pivots[] = $value;
            }
        }
        $eventName = $this->owner->isNewRecord ? ActiveRecord::EVENT_AFTER_INSERT : ActiveRecord::EVENT_AFTER_UPDATE;
        $this->owner->on($eventName, [$this, 'savePivots']);

It seems like on() must be provided by classes using this trait. How about adding it as abstract method to this trait?

    }

    public function savePivots()
    {
        if ($this->_pivots !== null) {
            $this->owner->deletePivots($this->pivotClass);
        }
        if ($this->_pivots) {
            foreach ($this->_pivots as $pv) {
                $this->owner->addPivot($pv, $this->pivotClass);
            }
        }
    }
}