Feature: on failure tasks.

Author: antonmedv

recipe/common.php

@@ -520,9 +520,18 @@
     }
 });
 
+
 /**
  * Success message
  */
 task('success', function () {
     Deployer::setDefault('terminate_message', '<info>Successfully deployed!</info>');
 })->once()->setPrivate();
+
+
+/**
+ * Deploy failure
+ */
+task('deploy:failed', function () {
+})->setPrivate();
+onFailure('deploy', 'deploy:failed');

src/Console/TaskCommand.php

@@ -79,6 +79,14 @@ protected function execute(Input $input, Output $output)
             $executor->run($tasks, $servers, $environments, $input, $output);
         } catch (\Exception $exception) {
             \Deployer\logger($exception->getMessage(), Logger::ERROR);
+
+            // Check if we have tasks to execute on failure.
+            if ($this->deployer['onFailure']->has($this->getName())) {
+                $taskName = $this->deployer['onFailure']->get($this->getName());
+                $tasks = $this->deployer->getScriptManager()->getTasks($taskName, $stage);
+                $executor->run($tasks, $servers, $environments, $input, $output);
+            }
+
             throw $exception;
         }
 

src/Deployer.php

@@ -88,6 +88,9 @@ public function __construct(Application $console, Console\Input\InputInterface $
         $this['stageStrategy'] = function ($c) {
             return new StageStrategy($c['servers'], $c['environments'], $c['config']['default_stage']);
         };
+        $this['onFailure'] = function () {
+            return new Collection();
+        };
 
         /******************************
          *           Logger           *

src/functions.php

@@ -193,6 +193,18 @@ function after($it, $that)
     $afterTask->addAfter($that);
 }
 
+/**
+ * Setup which task run on failure of first.
+ *
+ * @param string $it
+ * @param string $that
+ */
+function onFailure($it, $that)
+{
+    $deployer = Deployer::get();
+    $deployer['onFailure']->set($it, $that);
+}
+
 /**
  * Add users arguments.
  *