Add @before hook

src/Compiler.php

@@ -36,6 +36,8 @@ class Compiler
         'MacroStop',
         'TaskStart',
         'TaskStop',
+        'Before',
+        'BeforeStop',
         'After',
         'AfterStop',
         'Finished',
@@ -334,6 +336,30 @@ protected function compileTaskStop($value)
         return preg_replace($pattern, '$1<?php $__container->endTask(); ?>$2', $value);
     }
 
+    /**
+     * Compile Envoy before statements into valid PHP.
+     *
+     * @param  string  $value
+     * @return string
+     */
+    protected function compileBefore($value)
+    {
+        $pattern = $this->createPlainMatcher('before');
+
+        return preg_replace($pattern, '$1<?php $_vars = get_defined_vars(); $__container->before(function($task) use ($_vars) { extract($_vars, EXTR_SKIP)  ; $2', $value);
+    }
+
+    /**
+     * Compile Envoy before stop statements into valid PHP.
+     *
+     * @param  string  $value
+     * @return string
+     */
+    protected function compileBeforeStop($value)
+    {
+        return preg_replace($this->createPlainMatcher('endbefore'), '$1}); ?>$2', $value);
+    }
+
     /**
      * Compile Envoy after statements into valid PHP.
      *

src/Console/RunCommand.php

@@ -131,6 +131,10 @@ protected function runTask($container, $task)
             return;
         }
 
+        foreach ($container->getBeforeCallbacks() as $callback) {
+            call_user_func($callback, $task);
+        }
+
         if (($exitCode = $this->runTaskOverSSH($container->getTask($task, $macroOptions))) > 0) {
             foreach ($container->getErrorCallbacks() as $callback) {
                 call_user_func($callback, $task);

src/TaskContainer.php

@@ -51,6 +51,13 @@ class TaskContainer
      */
     protected $error = [];
 
+    /**
+     * All of the "before" callbacks.
+     *
+     * @var array
+     */
+    protected $before = [];
+
     /**
      * All of the "after" callbacks.
      *
@@ -446,6 +453,27 @@ public function endTask()
         }
     }
 
+    /**
+     * Register a before-task callback.
+     *
+     * @param  \Closure  $callback
+     * @return void
+     */
+    public function before(Closure $callback)
+    {
+        $this->before[] = $callback;
+    }
+
+    /**
+     * Get all of the before-task callbacks.
+     *
+     * @return array
+     */
+    public function getBeforeCallbacks()
+    {
+        return $this->before;
+    }
+
     /**
      * Register an after-task callback.
      *

tests/CompilerTest.php

@@ -19,4 +19,17 @@ public function test_it_compiles_finished_statement()
 
         $this->assertSame(1, preg_match('/\$__container->finished\(.*?\}\);/s', $result, $matches));
     }
+
+    public function test_compile_before_statement()
+    {
+        $str = <<<'EOL'
+@before
+    echo "Running {{ $task }} task.";
+@endbefore
+EOL;
+        $compiler = new Compiler();
+        $result = $compiler->compile($str);
+
+        $this->assertSame(1, preg_match('/\$__container->before\(.*?\}\);/s', $result, $matches));
+    }
 }