Skip to content

Instantly share code, notes, and snippets.

@aemon

aemon/InitialDbMigrationCommand.php

Forked from bmarston/InitialDbMigrationCommand.php
Last active August 29, 2015 14:22
Download ZIP

Revisions

  1. @bmarston bmarston created this gist May 8, 2013.
    75 changes: 75 additions & 0 deletions InitialDbMigrationCommand.php
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,75 @@
    <?php
    class InitialDbMigrationCommand extends CConsoleCommand
    {

    public function run($args) {
    $schema = $args[0];
    $tables = Yii::app()->db->schema->getTables($schema);

    $addForeignKeys = '';
    $dropForeignKeys = '';

    $result = "public function up()\n{\n";
    foreach ($tables as $table) {
    $compositePrimaryKeyCols = array();

    // Create table
    $result .= ' $this->createTable(\'' . $table->name . '\', array(' . "\n";
    foreach ($table->columns as $col) {
    $result .= ' \'' . $col->name . '\'=>\'' . $this->getColType($col) . '\',' . "\n";

    if ($col->isPrimaryKey && !$col->autoIncrement) {
    // Add column to composite primary key array
    $compositePrimaryKeyCols[] = $col->name;
    }
    }
    $result .= ' ), \'\');' . "\n\n";

    // Add foreign key(s) and create indexes
    foreach ($table->foreignKeys as $col => $fk) {
    // Foreign key naming convention: fk_table_foreignTable_col (max 64 characters)
    $fkName = substr('fk_' . $table->name . '_' . $fk[0] . '_' . $col, 0 , 64);
    $addForeignKeys .= ' $this->addForeignKey(' . "'$fkName', '$table->name', '$col', '$fk[0]', '$fk[1]', 'NO ACTION', 'NO ACTION');\n\n";
    $dropForeignKeys .= ' $this->dropForeignKey(' . "'$fkName', '$table->name');\n\n";

    // Index naming convention: idx_col
    $result .= ' $this->createIndex(\'idx_' . $col . "', '$table->name', '$col', FALSE);\n\n";
    }

    // Add composite primary key for join tables
    if ($compositePrimaryKeyCols) {
    $result .= ' $this->addPrimaryKey(\'pk_' . $table->name . "', '$table->name', '" . implode(',', $compositePrimaryKeyCols) . "');\n\n";

    }

    }
    $result .= $addForeignKeys; // This needs to come after all of the tables have been created.
    $result .= "}\n\n\n";

    $result .= "public function down()\n{\n";
    $result .= $dropForeignKeys; // This needs to come before the tables are dropped.
    foreach ($tables as $table) {
    $result .= ' $this->dropTable(\'' . $table->name . '\');' . "\n";
    }
    $result .= "}\n";


    echo $result;
    }

    public function getColType($col) {
    if ($col->isPrimaryKey && $col->autoIncrement) {
    return "pk";
    }
    $result = $col->dbType;
    if (!$col->allowNull) {
    $result .= ' NOT NULL';
    }
    if ($col->defaultValue != null) {
    $result .= " DEFAULT '{$col->defaultValue}'";
    } elseif ($col->allowNull) {
    $result .= ' DEFAULT NULL';
    }
    return $result;
    }
    }