diff --git a/Mage/Task/AbstractTask.php b/Mage/Task/AbstractTask.php index 0188e6f..53c26b9 100644 --- a/Mage/Task/AbstractTask.php +++ b/Mage/Task/AbstractTask.php @@ -131,7 +131,7 @@ abstract class AbstractTask } /** - * Runs a Shell Command Locally + * Runs a Shell Command Localy * @param string $command * @param string $output * @return boolean @@ -172,4 +172,20 @@ abstract class AbstractTask return $this->runCommandLocal($localCommand, $output); } + + /** + * Runs a Shell Command Localy or in the Remote Host based on the Task Stage. + * If the stage is "deploy" then it will be executed in the remote host. + * @param string $command + * @param string $output + * @return boolean + */ + protected final function runCommand($command, &$output = null) + { + if ($this->getStage() == 'deploy') { + return $this->runCommandRemote($command, $output); + } else { + return $this->runCommandLocal($command, $output); + } + } } \ No newline at end of file