Releases Rollback. Tweaks.

Mage/Console.php

@@ -111,7 +111,12 @@ class Mage_Console
                 switch ($this->_args[1]) {
                     case 'list':
                         $task->setAction($this->_args[1]);
-                        break;                        
+                        break;
+
+                    case 'rollback':
+                        $task->setAction($this->_args[1]);
+                        $task->setRelease($this->_args[2]);
+                        break;
                 }
                 $task->run($config);
                 break;

Mage/Task/BuiltIn/Deployment/Releases.php

@@ -15,7 +15,14 @@ class Mage_Task_BuiltIn_Deployment_Releases
 
             $currentCopy = $releasesDirectory . '/' . $this->_config->getReleaseId();
 
-            $result = $this->_runRemoteCommand('ln -sf ' . $currentCopy . ' ' . $symlink);
+            $userGroup = '';
+            $resultFetch = $this->_runRemoteCommand('ls -ld ' . $currentCopy . ' | awk \'{print \$3\":\"\$4}\'', $userGroup);
+            $command = 'rm -f ' . $symlink
+                     . ' && '
+                     . 'ln -sf ' . $currentCopy . ' ' . $symlink
+                     . ' && '
+                     . 'chown -h ' . $userGroup . ' ' . $symlink;
+            $result = $this->_runRemoteCommand($command);
             return $result;
 
         } else {

Mage/Task/BuiltIn/Releases/List.php

@@ -46,8 +46,11 @@ class Mage_Task_BuiltIn_Releases_List
                 }
             }
 
+            Mage_Console::output('');
             return $result;
+
         } else {
+            Mage_Console::output('');
             return false;
         }
     }

Mage/Task/BuiltIn/Releases/Rollback.php

@@ -0,0 +1,126 @@
+<?php
+class Mage_Task_BuiltIn_Releases_Rollback
+    extends Mage_Task_TaskAbstract
+{
+    private $_release = null;
+    
+    public function getName()
+    {
+        return 'Rollback release [built-in]';
+    }
+
+    public function setRelease($releaseId)
+    {
+        $this->_release = $releaseId;
+        return $this;
+    }
+    
+    public function getRelease()
+    {
+        return $this->_release;
+    }
+    
+    public function run()
+    {
+        if ($this->_config->release('enabled', false) == true) {
+            $releasesDirectory = $this->_config->release('directory', 'releases');
+            $symlink = $this->_config->release('symlink', 'current');
+            
+            $output = '';
+            $result = $this->_runRemoteCommand('ls -1 ' . $releasesDirectory, $output);
+            $releases = ($output == '') ? array() : explode(PHP_EOL, $output);
+                        
+            if (count($releases) == 0) {
+                Mage_Console::output('Release are not available for <dark_gray>' . $this->_config->getHost() . '</dark_gray> ... <red>FAIL</red>');
+
+            } else {
+                rsort($releases);
+                
+                $releaseIsAvailable = false;
+                if ($this->getRelease() == '') {
+                    $releaseId = $releases[0];
+
+                } else if ($this->getRelease() <= 0) {
+                    $index = $this->getRelease() * -1;
+                    if (isset($releases[$index])) {
+                        $releaseId = $releases[$index];
+                        $releaseIsAvailable = true;
+                    }
+                } else {
+                    if (in_array($this->getRelease(), $releases)) {
+                        $releaseId = $this->getRelease();
+                        $releaseIsAvailable = true;
+                    }
+                }
+                
+                if (!$releaseIsAvailable) {
+                    Mage_Console::output('Release <dark_gray>' . $this->getRelease() . '</dark_gray> is invalid or unavailable for <dark_gray>' . $this->_config->getHost() . '</dark_gray> ... <red>FAIL</red>');                    
+
+                } else {
+                    Mage_Console::output('Rollback release on <dark_gray>' . $this->_config->getHost() . '</dark_gray>');
+                    $rollbackTo = $releasesDirectory . '/' . $releaseId;
+                    
+                    // Tasks
+                    $tasks = 1;
+                    $completedTasks = 0;
+                    $tasksToRun = $this->_config->getTasks();
+                    $this->_config->setReleaseId($releaseId);
+                    
+                    if (count($tasksToRun) == 0) {
+                        Mage_Console::output('<light_purple>Warning!</light_purple> <dark_gray>No </dark_gray><light_cyan>Deployment</light_cyan> <dark_gray>tasks defined.</dark_gray>', 2);
+                        Mage_Console::output('Deployment to <dark_gray>' . $this->_config->getHost() . '</dark_gray> skipped!', 1, 3);
+                    
+                    } else {
+                        foreach ($tasksToRun as $taskName) {
+                            $tasks++;
+                            $task = Mage_Task_Factory::get($taskName, $this->_config);
+                            $task->init();
+                    
+                            Mage_Console::output('Running <purple>' . $task->getName() . '</purple> ... ', 2, false);
+                            $result = $task->run();
+                    
+                            if ($result == true) {
+                                Mage_Console::output('<green>OK</green>', 0);
+                                $completedTasks++;
+                            } else {
+                                Mage_Console::output('<red>FAIL</red>', 0);
+                            }
+                        }
+                    }
+                    
+                    // Changing Release
+                    Mage_Console::output('Releasing to <dark_gray>' . $releaseId . '</dark_gray> ... ', 2, false);
+                    
+                    $userGroup = '';
+                    $resultFetch = $this->_runRemoteCommand('ls -ld ' . $rollbackTo . ' | awk \'{print \$3\":\"\$4}\'', $userGroup);
+                    $command = 'rm -f ' . $symlink
+                             . ' && '
+                             . 'ln -sf ' . $rollbackTo . ' ' . $symlink
+                             . ' && '
+                             . 'chown -h ' . $userGroup . ' ' . $symlink;
+                    $result = $this->_runRemoteCommand($command);
+
+                    if ($result) {
+                        Mage_Console::output('<green>OK</green>', 0);
+                        $completedTasks++;
+                    } else {
+                        Mage_Console::output('<red>FAIL</red>', 0);
+                    }
+                    
+                    if ($completedTasks == $tasks) {
+                        $tasksColor = 'green';
+                    } else {
+                        $tasksColor = 'red';
+                    }
+                    
+                    Mage_Console::output('Release rollback on <dark_gray>' . $this->_config->getHost() . '</dark_gray> compted: <' . $tasksColor . '>' . $completedTasks . '/' . $tasks . '</' . $tasksColor . '> tasks done.', 1, 3);
+                }
+            }
+
+            return $result;
+        } else {
+            return false;
+        }
+    }
+
+}

Mage/Task/Deploy.php

@@ -31,6 +31,8 @@ class Mage_Task_Deploy
                 Mage_Console::output('Deploying to <dark_gray>' . $host . '</dark_gray>');
                 
                 $tasksToRun = $config->getTasks();
+                array_unshift($tasksToRun, 'deployment/rsync');
+
                 if ($config->release('enabled', false) == true) {
                     $config->setReleaseId($this->_releaseId);
                     array_push($tasksToRun, 'deployment/releases');                    
@@ -57,8 +59,6 @@ class Mage_Task_Deploy
                         }
                     }
                     
-                    // Run Post-Deployment Tasks
-                    
                     if ($completedTasks == $tasks) {
                         $tasksColor = 'green';                
                     } else {

Mage/Task/Releases.php

@@ -3,6 +3,7 @@ class Mage_Task_Releases
 {
     private $_config = null;
     private $_action = null;
+    private $_release = null;
     
     public function setAction($action)
     {
@@ -15,6 +16,17 @@ class Mage_Task_Releases
         return $this->_action;
     }
     
+    public function setRelease($releaseId)
+    {
+        $this->_release = $releaseId;
+        return $this;
+    }
+    
+    public function getRelease()
+    {
+        return $this->_release;
+    }
+    
     public function run(Mage_Config $config)
     {
         $this->_config = $config;
@@ -34,11 +46,16 @@ class Mage_Task_Releases
                         $task->init();
                         $result = $task->run();
                         break;
+                        
+                    case 'rollback':
+                        $task = Mage_Task_Factory::get('releases/rollback', $config);
+                        $task->init();
+                        $task->setRelease($this->getRelease());
+                        $result = $task->run();
+                        break;
                 }
-                Mage_Console::output('');
             }
         }
-
     }
     
     private function _listReleases()

bin/mage.php

@@ -7,10 +7,11 @@
 
 # mage releases list to:production
 # mage releases rollback to:production
-# mage releases rollback:-1 to:production
+# mage releases rollback -1 to:production
-# mage releases rollback:-2 to:production
+# mage releases rollback -2 to:production
-# mage releases rollback:-3 to:production
+# mage releases rollback -3 to:production
-# mage releases rollback:0 to:production
+# mage releases rollback 0 to:production
+# mage releases rollback 20120101172148 to:production
 # mage add environment production --width-releases
 
 # mage init

docs/example-config/.mage/config/environment/production.yaml

@@ -19,6 +19,5 @@ tasks:
   pre-deploy:
     - scm/update
   on-deploy:
-    - deployment/rsync
     - privileges
   #post-deploy:

docs/example-config/.mage/config/environment/staging.yaml

@@ -9,6 +9,5 @@ tasks:
   pre-deploy:
     - scm/update
   on-deploy:
-    - deployment/rsync
     - privileges
   #post-deploy: