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()
    {
        switch ($this->getConfig()->general('scm')) {
            case 'git':
            	if ($this->getParameter('_changeBranchRevert', false)) {
            		$command = 'git checkout ' . self::$startingBranch;
            		$result = $this->runCommandLocal($command);

            	} else {
            		$command = 'git branch | grep \'*\' | cut -d\' \' -f 2';
            		$currentBranch = 'master';
            		$result = $this->runCommandLocal($command, $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;

        				self::$startingBranch = $currentBranch;
            		} else {
            			throw new SkipException;
            		}
            	}
                break;

            default:
                throw new SkipException;
                break;
        }

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

        return $result;
    }
}