@ -10,6 +10,7 @@
namespace yii\console\controllers;
namespace yii\console\controllers;
use Yii;
use yii\console\Controller;
use yii\console\Controller;
/**
/**
@ -67,7 +68,7 @@ class MigrateController extends Controller
* in terms of a path alias, and the corresponding directory must exist.
* in terms of a path alias, and the corresponding directory must exist.
* Defaults to 'application.migrations' (meaning 'protected/migrations').
* Defaults to 'application.migrations' (meaning 'protected/migrations').
*/
*/
public $migrationPath='application. migrations';
public $migrationPath = '@application/ migrations';
/**
/**
* @var string the name of the table for keeping applied migration information.
* @var string the name of the table for keeping applied migration information.
* This table will be automatically created if not exists. Defaults to 'tbl_migration'.
* This table will be automatically created if not exists. Defaults to 'tbl_migration'.
@ -95,16 +96,17 @@ class MigrateController extends Controller
*/
*/
public $interactive = true;
public $interactive = true;
public function beforeAction($action)
public function beforeAction($action)
{
{
$path = \ Yii::getAlias($this->migrationPath);
$path = Yii::getAlias($this->migrationPath);
if ($path === false || !is_dir($path)) {
if ($path === false || !is_dir($path)) {
echo 'Error: The migration directory does not exist: ' . $this->migrationPath . "\n";
echo 'Error: the migration directory does not exist "' . $this->migrationPath . "\ "\n";
\Yii::$application->end(1) ;
return false ;
}
}
$this->migrationPath = $path;
$this->migrationPath = $path;
$yiiVersion = \ Yii::getVersion();
$yiiVersion = Yii::getVersion();
echo "\nYii Migration Tool v2.0 (based on Yii v{$yiiVersion})\n\n";
echo "\nYii Migration Tool v2.0 (based on Yii v{$yiiVersion})\n\n";
return parent::beforeAction($action);
return parent::beforeAction($action);
@ -115,10 +117,9 @@ class MigrateController extends Controller
*/
*/
public function actionUp($args)
public function actionUp($args)
{
{
if(($migrations = $this->getNewMigrations())===array())
if (($migrations = $this->getNewMigrations()) === array()) {
{
echo "No new migration found. Your system is up-to-date.\n";
echo "No new migration found. Your system is up-to-date.\n";
\ Yii::$application->end();
Yii::$application->end();
}
}
$total = count($migrations);
$total = count($migrations);
@ -131,18 +132,17 @@ class MigrateController extends Controller
if ($n === $total)
if ($n === $total)
echo "Total $n new " . ($n === 1 ? 'migration' : 'migrations') . " to be applied:\n";
echo "Total $n new " . ($n === 1 ? 'migration' : 'migrations') . " to be applied:\n";
else
else
{
echo "Total $n out of $total new " . ($total === 1 ? 'migration' : 'migrations') . " to be applied:\n";
echo "Total $n out of $total new " . ($total === 1 ? 'migration' : 'migrations') . " to be applied:\n";
}
foreach ($migrations as $migration)
foreach ($migrations as $migration)
echo " $migration\n";
echo " $migration\n";
echo "\n";
echo "\n";
if($this->confirm('Apply the above '.($n===1 ? 'migration':'migrations')."?"))
if ($this->confirm('Apply the above ' . ($n === 1 ? 'migration' : 'migrations') . "?")) {
{
foreach ($migrations as $migration) {
foreach($migrations as $migration)
if ($this->migrateUp($migration) === false) {
{
if($this->migrateUp($migration)===false)
{
echo "\nMigration failed. All later migrations are canceled.\n";
echo "\nMigration failed. All later migrations are canceled.\n";
return;
return;
}
}
@ -157,8 +157,7 @@ class MigrateController extends Controller
if ($step < 1 )
if ($step < 1 )
die("Error: The step parameter must be greater than 0.\n");
die("Error: The step parameter must be greater than 0.\n");
if(($migrations=$this->getMigrationHistory($step))===array())
if (($migrations = $this->getMigrationHistory($step)) === array()) {
{
echo "No migration has been done before.\n";
echo "No migration has been done before.\n";
return;
return;
}
}
@ -170,12 +169,9 @@ class MigrateController extends Controller
echo " $migration\n";
echo " $migration\n";
echo "\n";
echo "\n";
if($this->confirm('Revert the above '.($n===1 ? 'migration':'migrations')."?"))
if ($this->confirm('Revert the above ' . ($n === 1 ? 'migration' : 'migrations') . "?")) {
{
foreach ($migrations as $migration) {
foreach($migrations as $migration)
if ($this->migrateDown($migration) === false) {
{
if($this->migrateDown($migration)===false)
{
echo "\nMigration failed. All later migrations are canceled.\n";
echo "\nMigration failed. All later migrations are canceled.\n";
return;
return;
}
}
@ -190,8 +186,7 @@ class MigrateController extends Controller
if ($step < 1 )
if ($step < 1 )
die("Error: The step parameter must be greater than 0.\n");
die("Error: The step parameter must be greater than 0.\n");
if(($migrations=$this->getMigrationHistory($step))===array())
if (($migrations = $this->getMigrationHistory($step)) === array()) {
{
echo "No migration has been done before.\n";
echo "No migration has been done before.\n";
return;
return;
}
}
@ -203,20 +198,15 @@ class MigrateController extends Controller
echo " $migration\n";
echo " $migration\n";
echo "\n";
echo "\n";
if($this->confirm('Redo the above '.($n===1 ? 'migration':'migrations')."?"))
if ($this->confirm('Redo the above ' . ($n === 1 ? 'migration' : 'migrations') . "?")) {
{
foreach ($migrations as $migration) {
foreach($migrations as $migration)
if ($this->migrateDown($migration) === false) {
{
if($this->migrateDown($migration)===false)
{
echo "\nMigration failed. All later migrations are canceled.\n";
echo "\nMigration failed. All later migrations are canceled.\n";
return;
return;
}
}
}
}
foreach(array_reverse($migrations) as $migration)
foreach (array_reverse($migrations) as $migration) {
{
if ($this->migrateUp($migration) === false) {
if($this->migrateUp($migration)===false)
{
echo "\nMigration failed. All later migrations are canceled.\n";
echo "\nMigration failed. All later migrations are canceled.\n";
return;
return;
}
}
@ -240,10 +230,8 @@ class MigrateController extends Controller
// try migrate up
// try migrate up
$migrations = $this->getNewMigrations();
$migrations = $this->getNewMigrations();
foreach($migrations as $i=>$migration)
foreach ($migrations as $i => $migration) {
{
if (strpos($migration, $version . '_') === 0) {
if(strpos($migration,$version.'_')===0)
{
$this->actionUp(array($i + 1));
$this->actionUp(array($i + 1));
return;
return;
}
}
@ -251,10 +239,8 @@ class MigrateController extends Controller
// try migrate down
// try migrate down
$migrations = array_keys($this->getMigrationHistory(-1));
$migrations = array_keys($this->getMigrationHistory(-1));
foreach($migrations as $i=>$migration)
foreach ($migrations as $i => $migration) {
{
if (strpos($migration, $version . '_') === 0) {
if(strpos($migration,$version.'_')===0)
{
if ($i === 0)
if ($i === 0)
echo "Already at '$originalVersion'. Nothing needs to be done.\n";
echo "Already at '$originalVersion'. Nothing needs to be done.\n";
else
else
@ -282,15 +268,11 @@ class MigrateController extends Controller
// try mark up
// try mark up
$migrations = $this->getNewMigrations();
$migrations = $this->getNewMigrations();
foreach($migrations as $i=>$migration)
foreach ($migrations as $i => $migration) {
{
if (strpos($migration, $version . '_') === 0) {
if(strpos($migration,$version.'_')===0)
if ($this->confirm("Set migration history at $originalVersion?")) {
{
if($this->confirm("Set migration history at $originalVersion?"))
{
$command = $db->createCommand();
$command = $db->createCommand();
for($j=0;$j< =$i;++$j)
for ($j = 0; $j < = $i; ++$j) {
{
$command->insert($this->migrationTable, array(
$command->insert($this->migrationTable, array(
'version' => $migrations[$j],
'version' => $migrations[$j],
'apply_time' => time(),
'apply_time' => time(),
@ -304,16 +286,12 @@ class MigrateController extends Controller
// try mark down
// try mark down
$migrations = array_keys($this->getMigrationHistory(-1));
$migrations = array_keys($this->getMigrationHistory(-1));
foreach($migrations as $i=>$migration)
foreach ($migrations as $i => $migration) {
{
if (strpos($migration, $version . '_') === 0) {
if(strpos($migration,$version.'_')===0)
{
if ($i === 0)
if ($i === 0)
echo "Already at '$originalVersion'. Nothing needs to be done.\n";
echo "Already at '$originalVersion'. Nothing needs to be done.\n";
else
else {
{
if ($this->confirm("Set migration history at $originalVersion?")) {
if($this->confirm("Set migration history at $originalVersion?"))
{
$command = $db->createCommand();
$command = $db->createCommand();
for ($j = 0; $j < $i; ++$j)
for ($j = 0; $j < $i; ++$j)
$command->delete($this->migrationTable, $db->quoteColumnName('version') . '=:version', array(':version' => $migrations[$j]));
$command->delete($this->migrationTable, $db->quoteColumnName('version') . '=:version', array(':version' => $migrations[$j]));
@ -333,8 +311,7 @@ class MigrateController extends Controller
$migrations = $this->getMigrationHistory($limit);
$migrations = $this->getMigrationHistory($limit);
if ($migrations === array())
if ($migrations === array())
echo "No migration has been done before.\n";
echo "No migration has been done before.\n";
else
else {
{
$n = count($migrations);
$n = count($migrations);
if ($limit > 0)
if ($limit > 0)
echo "Showing the last $n applied " . ($n === 1 ? 'migration' : 'migrations') . ":\n";
echo "Showing the last $n applied " . ($n === 1 ? 'migration' : 'migrations') . ":\n";
@ -351,15 +328,12 @@ class MigrateController extends Controller
$migrations = $this->getNewMigrations();
$migrations = $this->getNewMigrations();
if ($migrations === array())
if ($migrations === array())
echo "No new migrations found. Your system is up-to-date.\n";
echo "No new migrations found. Your system is up-to-date.\n";
else
else {
{
$n = count($migrations);
$n = count($migrations);
if($limit>0 & & $n>$limit)
if ($limit > 0 & & $n > $limit) {
{
$migrations = array_slice($migrations, 0, $limit);
$migrations = array_slice($migrations, 0, $limit);
echo "Showing $limit out of $n new " . ($n === 1 ? 'migration' : 'migrations') . ":\n";
echo "Showing $limit out of $n new " . ($n === 1 ? 'migration' : 'migrations') . ":\n";
}
} else
else
echo "Found $n new " . ($n === 1 ? 'migration' : 'migrations') . ":\n";
echo "Found $n new " . ($n === 1 ? 'migration' : 'migrations') . ":\n";
foreach ($migrations as $migration)
foreach ($migrations as $migration)
@ -381,20 +355,12 @@ class MigrateController extends Controller
$content = strtr($this->getTemplate(), array('{ClassName}' => $name));
$content = strtr($this->getTemplate(), array('{ClassName}' => $name));
$file = $this->migrationPath . DIRECTORY_SEPARATOR . $name . '.php';
$file = $this->migrationPath . DIRECTORY_SEPARATOR . $name . '.php';
if($this->confirm("Create new migration '$file'?"))
if ($this->confirm("Create new migration '$file'?")) {
{
file_put_contents($file, $content);
file_put_contents($file, $content);
echo "New migration created successfully.\n";
echo "New migration created successfully.\n";
}
}
}
}
public function confirm($message)
{
if(!$this->interactive)
return true;
return parent::confirm($message);
}
protected function migrateUp($class)
protected function migrateUp($class)
{
{
if ($class === self::BASE_MIGRATION)
if ($class === self::BASE_MIGRATION)
@ -403,17 +369,14 @@ class MigrateController extends Controller
echo "*** applying $class\n";
echo "*** applying $class\n";
$start = microtime(true);
$start = microtime(true);
$migration = $this->instantiateMigration($class);
$migration = $this->instantiateMigration($class);
if($migration->up()!==false)
if ($migration->up() !== false) {
{
$this->getDb()->createCommand()->insert($this->migrationTable, array(
$this->getDb()->createCommand()->insert($this->migrationTable, array(
'version' => $class,
'version' => $class,
'apply_time' => time(),
'apply_time' => time(),
));
));
$time = microtime(true) - $start;
$time = microtime(true) - $start;
echo "*** applied $class (time: " . sprintf("%.3f", $time) . "s)\n\n";
echo "*** applied $class (time: " . sprintf("%.3f", $time) . "s)\n\n";
}
} else {
else
{
$time = microtime(true) - $start;
$time = microtime(true) - $start;
echo "*** failed to apply $class (time: " . sprintf("%.3f", $time) . "s)\n\n";
echo "*** failed to apply $class (time: " . sprintf("%.3f", $time) . "s)\n\n";
return false;
return false;
@ -428,15 +391,12 @@ class MigrateController extends Controller
echo "*** reverting $class\n";
echo "*** reverting $class\n";
$start = microtime(true);
$start = microtime(true);
$migration = $this->instantiateMigration($class);
$migration = $this->instantiateMigration($class);
if($migration->down()!==false)
if ($migration->down() !== false) {
{
$db = $this->getDb();
$db = $this->getDb();
$db->createCommand()->delete($this->migrationTable, $db->quoteColumnName('version') . '=:version', array(':version' => $class));
$db->createCommand()->delete($this->migrationTable, $db->quoteColumnName('version') . '=:version', array(':version' => $class));
$time = microtime(true) - $start;
$time = microtime(true) - $start;
echo "*** reverted $class (time: " . sprintf("%.3f", $time) . "s)\n\n";
echo "*** reverted $class (time: " . sprintf("%.3f", $time) . "s)\n\n";
}
} else {
else
{
$time = microtime(true) - $start;
$time = microtime(true) - $start;
echo "*** failed to revert $class (time: " . sprintf("%.3f", $time) . "s)\n\n";
echo "*** failed to revert $class (time: " . sprintf("%.3f", $time) . "s)\n\n";
return false;
return false;
@ -456,11 +416,12 @@ class MigrateController extends Controller
* @var CDbConnection
* @var CDbConnection
*/
*/
private $_db;
private $_db;
protected function getDb()
protected function getDb()
{
{
if ($this->_db !== null)
if ($this->_db !== null)
return $this->_db;
return $this->_db;
else if(($this->_db=\ Yii::$application->getComponent($this->connectionID)) instanceof CDbConnection)
else if (($this->_db = Yii::$application->getComponent($this->connectionID)) instanceof CDbConnection)
return $this->_db;
return $this->_db;
else
else
die("Error: CMigrationCommand.connectionID '{$this->connectionID}' is invalid. Please make sure it refers to the ID of a CDbConnection application component.\n");
die("Error: CMigrationCommand.connectionID '{$this->connectionID}' is invalid. Please make sure it refers to the ID of a CDbConnection application component.\n");
@ -469,8 +430,7 @@ class MigrateController extends Controller
protected function getMigrationHistory($limit)
protected function getMigrationHistory($limit)
{
{
$db = $this->getDb();
$db = $this->getDb();
if($db->schema->getTable($this->migrationTable)===null)
if ($db->schema->getTable($this->migrationTable) === null) {
{
$this->createMigrationHistoryTable();
$this->createMigrationHistoryTable();
}
}
return CHtml::listData($db->createCommand()
return CHtml::listData($db->createCommand()
@ -504,8 +464,7 @@ class MigrateController extends Controller
$migrations = array();
$migrations = array();
$handle = opendir($this->migrationPath);
$handle = opendir($this->migrationPath);
while(($file=readdir($handle))!==false)
while (($file = readdir($handle)) !== false) {
{
if ($file === '.' || $file === '..')
if ($file === '.' || $file === '..')
continue;
continue;
$path = $this->migrationPath . DIRECTORY_SEPARATOR . $file;
$path = $this->migrationPath . DIRECTORY_SEPARATOR . $file;