Add support for Mercurial SSH-based clones.

This commit is contained in:
Dan Cryer 2015-02-23 15:58:14 +00:00
parent 893deada33
commit c3bf9e65c4

View file

@ -13,36 +13,87 @@ use PHPCI\Model\Build;
use PHPCI\Builder; use PHPCI\Builder;
/** /**
* Mercurial Build Model * Mercurial Build Model
* @author Pavel Gopanenko <pavelgopanenko@gmail.com> * @author Pavel Gopanenko <pavelgopanenko@gmail.com>
* @package PHPCI * @package PHPCI
* @subpackage Core * @subpackage Core
*/ */
class MercurialBuild extends Build class MercurialBuild extends Build
{ {
/** /**
* Get the URL to be used to clone this remote repository. * Get the URL to be used to clone this remote repository.
*/ */
protected function getCloneUrl() protected function getCloneUrl()
{ {
return $this->getProject()->getReference(); return $this->getProject()->getReference();
} }
/** /**
* Create a working copy by cloning, copying, or similar. * Create a working copy by cloning, copying, or similar.
*/ */
public function createWorkingCopy(Builder $builder, $buildPath) public function createWorkingCopy(Builder $builder, $buildPath)
{ {
$this->cloneByHttp($builder, $buildPath); $key = trim($this->getProject()->getSshPublicKey());
if (!empty($key) && strpos($this->getProject()->getReference(), 'ssh') > -1) {
$success = $this->cloneBySsh($builder, $buildPath);
} else {
$success = $this->cloneByHttp($builder, $buildPath);
}
if (!$success) {
$builder->logFailure('Failed to clone remote git repository.');
return false;
}
return $this->handleConfig($builder, $buildPath); return $this->handleConfig($builder, $buildPath);
} }
/** /**
* Use an mercurial clone. * Use a HTTP-based Mercurial clone.
*/ */
protected function cloneByHttp(Builder $builder, $cloneTo) protected function cloneByHttp(Builder $builder, $cloneTo)
{ {
return $builder->executeCommand('hg clone %s "%s" -r %s', $this->getCloneUrl(), $cloneTo, $this->getBranch()); return $builder->executeCommand('hg clone %s "%s" -r %s', $this->getCloneUrl(), $cloneTo, $this->getBranch());
} }
/**
* Use an SSH-based Mercurial clone.
*/
protected function cloneBySsh(Builder $builder, $cloneTo)
{
$keyFile = $this->writeSshKey();
// Do the git clone:
$cmd = 'hg clone --ssh "ssh -i '.$keyFile.'" %s "%s"';
$success = $builder->executeCommand($cmd, $this->getCloneUrl(), $cloneTo);
if ($success) {
$success = $this->postCloneSetup($builder, $cloneTo);
}
// Remove the key file:
unlink($keyFile);
return $success;
}
/**
* Handle post-clone tasks (switching branch, etc.)
* @param Builder $builder
* @param $cloneTo
* @return bool
*/
protected function postCloneSetup(Builder $builder, $cloneTo)
{
$success = true;
$commit = $this->getCommitId();
// Allow switching to a specific branch:
if (!empty($commit) && $commit != 'Manual') {
$cmd = 'cd "%s" && hg checkout %s';
$success = $builder->executeCommand($cmd, $cloneTo, $this->getBranch());
}
return $success;
}
} }