carono\yii2migrate\ForeignKeyColumn

Complexity

Total Complexity

25

Size/Duplication

Total Lines	214
Duplicated Lines	0 %

Importance

Changes

0

21 Methods

Rating	Name	Duplication	Size	Complexity
A	onDeleteDefault()	0	3	1
A	refColumn()	0	4	1
A	refTable()	0	4	1
A	onDeleteCascade()	0	3	1
A	getRefTable()	0	3	1
A	sourceColumn()	0	4	1
A	removeSchema()	0	7	2
A	getRefColumn()	0	7	3
A	setName()	0	4	1
A	getOnUpdate()	0	3	1
A	onDelete()	0	4	1
A	remove()	0	3	1
A	getName()	0	9	2
A	getSourceColumn()	0	3	1
A	apply()	0	5	1
A	getOnDelete()	0	3	1
A	onDeleteNull()	0	3	1
A	getSourceTable()	0	3	1
A	setMigrate()	0	4	1
A	formName()	0	5	1
A	sourceTable()	0	4	1

<?php

namespace carono\yii2migrate;


use yii\db\ColumnSchemaBuilder;

/**
 * Class ForeignKeyColumn
 *
 * @package carono\yii2installer
 */
class ForeignKeyColumn extends ColumnSchemaBuilder
{
    const FK_CASCADE = 'CASCADE';
    const FK_DEFAULT = 'SET DEFAULT';
    const FK_NULL = 'SET NULL';
    public $_onDelete = self::FK_CASCADE;
    public $_onUpdate = null;
    protected $_refTable = null;
    protected $_refColumn = null;
    protected $_sourceTable = null;
    protected $_sourceColumn = null;
    protected $_name;
    /**
     * @var Migration
     */
    public $migrate;

    /**
     * @param $migrate
     * @return $this
     */
    public function setMigrate($migrate)
    {
        $this->migrate = $migrate;
        return $this;
    }

    /**
     * @param $value
     * @return $this
     */
    public function setName($value)
    {
        $this->_name = $value;
        return $this;
    }

    /**
     * @return mixed
     */
    public function getName()
    {
        if ($this->_name) {
            $name = $this->_name;
        } else {
            $name = self::formName($this->getSourceTable(), $this->getSourceColumn(), $this->getRefTable(), $this->getRefColumn());
        }
        $name = $this->migrate->expandTablePrefix($name);
        return $name;
    }

    public function apply()
    {
        return $this->migrate->addForeignKey(
            $this->getName(), $this->getSourceTable(), $this->getSourceColumn(), $this->getRefTable(),
            $this->getRefColumn(), $this->getOnDelete(), $this->getOnUpdate()

It seems like $this->getOnUpdate() can also be of type string; however, parameter $update of carono\yii2migrate\Migration::addForeignKey() does only seem to accept null, maybe add an additional type check?

        );
    }

    public function remove()
    {
        $this->migrate->dropForeignKeyByColumn($this->getSourceTable(), $this->getSourceColumn());
    }

    /**
     * @return null|string
     */
    public function getRefTable()
    {
        return $this->_refTable;
    }

    /**
     * @return null|string
     */
    public function getRefColumn()
    {
        if (!$this->_refColumn && $this->migrate) {
            $pk = $this->migrate->db->getTableSchema($this->getRefTable())->primaryKey;
            $this->refColumn(current($pk));
        }
        return $this->_refColumn;
    }

    /**
     * @return null|string
     */
    public function getSourceTable()
    {
        return $this->_sourceTable;
    }

    /**
     * @return null|string
     */
    public function getSourceColumn()
    {
        return $this->_sourceColumn;
    }

    /**
     * @return string
     */
    public function getOnDelete()
    {
        return $this->_onDelete;
    }

    /**
     * @return null|string
     */
    public function getOnUpdate()
    {
        return $this->_onUpdate;
    }

    /**
     * @param $string
     * @return $this
     */
    public function onDelete($string)
    {
        $this->_onDelete = $string;
        return $this;
    }

    /**
     * @return ForeignKeyColumn
     */
    public function onDeleteCascade()
    {
        return $this->onDelete(self::FK_CASCADE);
    }

    /**
     * @return ForeignKeyColumn
     */
    public function onDeleteNull()
    {
        return $this->onDelete(self::FK_NULL);
    }

    public function onDeleteDefault()
    {
        $this->_onDelete = self::FK_DEFAULT;
    }

    /**
     * @param $name
     * @return $this
     */
    public function refTable($name)
    {
        $this->_refTable = $name;
        return $this;
    }

    /**
     * @param $name
     * @return $this
     */
    public function sourceColumn($name)
    {
        $this->_sourceColumn = $name;
        return $this;
    }

    /**
     * @param $str
     * @return mixed
     */
    private static function removeSchema($str)
    {
        if (strpos($str, '.') !== false) {
            $arr = explode('.', $str);
            return $arr[1];
        } else {
            return $str;
        }
    }

    /**
     * @param $table
     * @param $column
     * @param $refTable
     * @param $refColumn
     * @return string
     */
    public static function formName($table, $column, $refTable, $refColumn)
    {
        $table = self::removeSchema($table);
        $refTable = self::removeSchema($refTable);
        return "{$table}[{$column}]_{$refTable}[{$refColumn}]_fk";
    }

    /**
     * @param $name
     *
     * @return $this
     */
    public function sourceTable($name)
    {
        $this->_sourceTable = $name;
        return $this;
    }

    /**
     * @param $name
     * @return $this
     */
    public function refColumn($name)
    {
        $this->_refColumn = $name;
        return $this;
    }
}