ForeignKey.php

<?php
class ZendDbSchema_Db_Schema_Table_ForeignKey
    extends ZendDbSchema_Db_Schema_Table_AbstractDefinition
{
    const COLUMNS_KEY   = 'columns';
    const REFERENCE_KEY = 'references';
    const ONDELETE_KEY  = 'delete';
    const ONUPDATE_KEY  = 'update';

    protected function _getSchemaKey()
    {
        return ZendDbSchema_Db_Schema_Table::FOREIGNS_KEY;
    }

    protected function _getDropSql()
    {
        return $this->getTable()->getGenerator()->dropForeignKey($this);
    }

    protected function _getAlterSql()
    {
        return $this->getTable()->getGenerator()->alterForeignKey($this);
    }

    protected function _getCreateSql()
    {
        return $this->getTable()->getGenerator()->addForeignKey($this);
    }

    public function setReferences($table)
    {
        if (!$table instanceof ZendDbSchema_Db_Schema_Table) {
            $table = new ZendDbSchema_Db_Schema_Table($table);
        }

        if (!$table->isExist()) {
            throw new ZendDbSchema_Db_Schema_Exception('Cannot set reference to not existing table');
        }

        $this->__set(self::REFERENCE_KEY, $table->getOriginName());
        return $this;
    }

    public function getReferences()
    {
        return $this->__get(self::REFERENCE_KEY);
    }

    public function setColumns(array $columns)
    {
        $this->__set(self::COLUMNS_KEY, $columns);
        return $this;
    }

    public function hasColumn($columnName)
    {
        return array_key_exists($columnName, $this->getColumns());
    }

    public function hasRefColumn($refColumnName)
    {
        $columns = $this->getColumns();
        return in_array($refColumnName, $columns);
    }

    public function setColumn($columnName, $refColumnName)
    {
        $columns = $this->getColumns();
        $columns[$columnName] = $refColumnName;
        $this->setColumns($columns);

        return $this;
    }

    public function getColumns()
    {
        $columns = $this->__get(self::COLUMNS_KEY);

        return is_array($columns) ? $columns : array();
    }

    public function removeColumn($columnName)
    {
        if ($this->hasColumn($columnName)) {
            $columns = $this->getColumns();
            unset($columns[$columnName]);
            $this->setColumns($columns);
        }
        return $this;
    }

    public function setOnDelete($onDelete)
    {
        $this->__set(self::ONDELETE_KEY, $onDelete);
        return $this;
    }

    public function getOnDelete()
    {
        return $this->__get(self::ONDELETE_KEY);
    }


    public function setOnUpdate($onUpdate)
    {
        $this->__set(self::ONUPDATE_KEY, $onUpdate);
        return $this;
    }

    public function getOnUpdate()
    {
        return $this->__get(self::ONUPDATE_KEY);
    }
}