Fixed namespaces (PHPCI -> PHPCensor)

src/PHPCensor/Model/Build/MercurialBuild.php

@@ -0,0 +1,99 @@
+<?php
+/**
+ * PHPCI - Continuous Integration for PHP
+ *
+ * @copyright    Copyright 2014, Block 8 Limited.
+ * @license      https://github.com/Block8/PHPCI/blob/master/LICENSE.md
+ * @link         https://www.phptesting.org/
+ */
+
+namespace PHPCensor\Model\Build;
+
+use PHPCensor\Model\Build;
+use PHPCensor\Builder;
+
+/**
+ * Mercurial Build Model
+ * @author       Pavel Gopanenko <pavelgopanenko@gmail.com>
+ * @package      PHPCI
+ * @subpackage   Core
+ */
+class MercurialBuild extends Build
+{
+    /**
+     * Get the URL to be used to clone this remote repository.
+     */
+    protected function getCloneUrl()
+    {
+        return $this->getProject()->getReference();
+    }
+
+    /**
+     * Create a working copy by cloning, copying, or similar.
+     */
+    public function createWorkingCopy(Builder $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);
+    }
+
+    /**
+     * Use a HTTP-based Mercurial clone.
+     */
+    protected function cloneByHttp(Builder $builder, $cloneTo)
+    {
+        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;
+    }
+}