Rollback awareness.

Tasks are aware if they are in rollbacks; they will be invoked only if
they implement the new interface RollbackAware.

Mage/Task/BuiltIn/Releases/Rollback.php

@@ -72,24 +72,28 @@ class Mage_Task_BuiltIn_Releases_Rollback
                     
                     } 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);
+                            $task = Mage_Task_Factory::get($taskName, $this->_config, true);
+                            $task->init();
+                            Mage_Console::output('Running <purple>' . $task->getName() . '</purple> ... ', 2, false);
+                            
+                            if ($task instanceOf Mage_Task_Releases_RollbackAware) {
+                                $tasks++;
+                                $result = $task->run();
+                                
+                                if ($result == true) {
+                                    Mage_Console::output('<green>OK</green>', 0);
+                                    $completedTasks++;
+                                } else {
+                                    Mage_Console::output('<red>FAIL</red>', 0);
+                                }
+                            } else {
+                                Mage_Console::output('<yellow>SKIPPED</yellow>', 0);                                
                             }
                         }
                     }
                     
                     // Changing Release
-                    Mage_Console::output('Releasing to <dark_gray>' . $releaseId . '</dark_gray> ... ', 2, false);
+                    Mage_Console::output('Running <purple>Rollback Release [id=' . $releaseId . ']</purple> ... ', 2, false);
                     
                     $userGroup = '';
                     $resultFetch = $this->_runRemoteCommand('ls -ld ' . $rollbackTo . ' | awk \'{print \$3\":\"\$4}\'', $userGroup);

Mage/Task/Factory.php

@@ -5,21 +5,22 @@ class Mage_Task_Factory
      * 
      * 
      * @param string $taskName
+     * @param boolean $inRollback
      * @return Mage_Task_TaskAbstract
      */
-    public static function get($taskName, Mage_Config $taskConfig)
+    public static function get($taskName, Mage_Config $taskConfig, $inRollback = false)
     {
         $instance = null;
         
         if (strpos($taskName, '/') === false) {
             Mage_Autoload::loadUserTask($taskName);
             $className = 'Task_' . ucfirst($taskName);
-            $instance = new $className($taskConfig);
+            $instance = new $className($taskConfig, $inRollback);
 
         } else {
             $taskName = str_replace(' ', '_', ucwords(str_replace('/', ' ', $taskName)));
             $className = 'Mage_Task_BuiltIn_' . $taskName;
-            $instance = new $className($taskConfig);
+            $instance = new $className($taskConfig, $inRollback);
         }
 
         assert($instance instanceOf Mage_Task_TaskAbstract);

Mage/Task/Releases.php

@@ -57,10 +57,4 @@ class Mage_Task_Releases
             }
         }
     }
-    
-    private function _listReleases()
-    {
-        
-    }
-    
 }

Mage/Task/Releases/RollbackAware.php

@@ -0,0 +1,4 @@
+<?php
+interface Mage_Task_Releases_RollbackAware
+{
+}

Mage/Task/TaskAbstract.php

@@ -2,14 +2,21 @@
 abstract class Mage_Task_TaskAbstract
 {
     protected $_config = null;
+    protected $_inRollback = false;
     
     public abstract function getName();
     
     public abstract function run();
     
-    public final function __construct(Mage_Config $config)
+    public final function __construct(Mage_Config $config, $inRollback = false)
     {
         $this->_config = $config;
+        $this->_inRollback = $inRollback;
+    }
+    
+    public function inRollback()
+    {
+        return $this->_inRollback;
     }
     
     public function init()

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

@@ -20,4 +20,6 @@ tasks:
     - scm/update
   on-deploy:
     - privileges
+    - sampleTask
+    - sampleTaskRollbackAware
   #post-deploy:

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

@@ -10,4 +10,6 @@ tasks:
     - scm/update
   on-deploy:
     - privileges
+    - sampleTask
+    - sampleTaskRollbackAware
   #post-deploy:

docs/example-config/.mage/tasks/SampleTask.php

@@ -0,0 +1,14 @@
+<?php
+class Task_SampleTask
+    extends Mage_Task_TaskAbstract
+{
+    public function getName()
+    {
+        return 'A Sample Task';
+    }
+
+    public function run()
+    {       
+        return true;
+    }
+}

docs/example-config/.mage/tasks/SampleTaskRollbackAware.php

@@ -0,0 +1,19 @@
+<?php
+class Task_SampleTaskRollbackAware
+    extends Mage_Task_TaskAbstract
+    implements Mage_Task_Releases_RollbackAware
+{
+    public function getName()
+    {
+        if ($this->inRollback()) {
+            return 'A Sample Task aware of rollbacks [in rollback]';
+        } else {
+            return 'A Sample Task aware of rollbacks [not in rollback]';
+        }
+    }
+
+    public function run()
+    {
+        return true;
+    }
+}