magallanes/Mage/Task/BuiltIn/Scm/ChangeBranchTask.php

<?php
/*
 * This file is part of the Magallanes package.
*
* (c) Andrés Montañez <andres@andresmontanez.com>
*
* For the full copyright and license information, please view the LICENSE
* file that was distributed with this source code.
*/

namespace Mage\Task\BuiltIn\Scm;

use Mage\Task\AbstractTask;
use Mage\Task\SkipException;
use Mage\Task\ErrorWithMessageException;

/**
 * Task for Changing the Branch of the SCM
 *
 * @author Andrés Montañez <andres@andresmontanez.com>
 */
class ChangeBranchTask extends AbstractTask
{
    /**
     * Branch the executiong began with
     * @var string
     */
    protected static $startingBranch = 'master';

    /**
     * Name of the Task
     * @var string
     */
    private $name = 'SCM Changing branch [built-in]';

    /**
     * (non-PHPdoc)
     * @see \Mage\Task\AbstractTask::getName()
     */
    public function getName()
    {
        return $this->name;
    }

    /**
     * (non-PHPdoc)
     * @see \Mage\Task\AbstractTask::init()
     */
    public function init()
    {
        $scmType = $this->getConfig()->general('scm');

        switch ($scmType) {
            case 'git':
                $this->name = 'SCM Changing branch (GIT) [built-in]';
                break;
        }
    }

    /**
     * Changes the Branch of the SCM
     * @see \Mage\Task\AbstractTask::run()
     */
    public function run()
    {
        $preCommand = 'cd ' . $this->getConfig()->deployment('from', './') . '; ';
        switch ($this->getConfig()->general('scm')) {
            case 'git':
                if ($this->getParameter('_changeBranchRevert', false)) {
                    $command = $preCommand . 'git checkout ' . self::$startingBranch;
                    $result = $this->runCommandLocal($command);
                } else {
                    $command = $preCommand . 'git branch | grep \'*\' | cut -d\' \' -f 2';
                    $currentBranch = 'master';
                    $result = $this->runCommandLocal($command, $currentBranch);
                    
                    self::$startingBranch = $currentBranch;

                    $scmData = $this->getConfig()->deployment('scm', false);

                    if ($result && is_array($scmData) && isset($scmData['branch']) && $scmData['branch'] != $currentBranch) {
                        $command = 'git branch | grep \'' . $scmData['branch'] . '\' | tr -s \' \' | sed \'s/^[ ]//g\'';
                        $isBranchTracked = '';
                        $result = $this->runCommandLocal($command, $isBranchTracked);

                        if ($isBranchTracked == '') {
                            throw new ErrorWithMessageException('The branch <purple>' . $scmData['branch'] . '</purple> must be tracked.');
                        }

                        $branch = $this->getParameter('branch', $scmData['branch']);
                        $command = 'git checkout ' . $branch;
                        $result = $this->runCommandLocal($command) && $result;
                    } else {
                        throw new SkipException;
                    }
                }
                break;

            default:
                throw new SkipException;
                break;
        }

        $this->getConfig()->reload();

        return $result;
    }
}