Update to allow multiple branchesHEAD0.0.5origin/masterorigin/HEADmaster

1 files changed, 74 insertions, 25 deletions

diff --git a/src/GitDeploy/GitDeploy.php b/src/GitDeploy/GitDeploy.php
index d1c2cf6..a8c420c 100644
--- a/src/GitDeploy/GitDeploy.php
+++ b/src/GitDeploy/GitDeploy.php
@@ -7,11 +7,55 @@ use GitWrapper\GitWorkingCopy;
 
 class GitDeploy
 {
-    protected $repositories;
+    protected $projects = array();
 
-    public function __construct($repositories = array())
+    public function __construct(array $projects = array())
     {
-        $this->repositories = $repositories;
+        if (!empty($projects)) {
+            $this->setProjects($projects);
+        }
+    }
+
+    /**
+     * Parses the given projects configuration array into valid configuration.
+     */
+    public function setProjects(array $projects = array())
+    {
+        $output = array();
+
+        foreach ($projects as $name => $project) {
+            if (is_string($project)) {
+                $output[$name]['repo'] = $project;
+            }
+            elseif (is_array($project)) {
+                $output[$name] = $project;
+            }
+
+            // Construct the default branches array if one is not given.
+            if (!isset($output[$name]['branches']) || !is_array($output[$name]['branches'])) {
+                $output[$name]['branches'] = array(
+                    'gh-pages' => $name,
+                );
+            }
+            else {
+                $branches = array();
+                foreach ($output[$name]['branches'] as $source => $destination) {
+                    if (is_numeric($source)) {
+                        $source = $destination;
+                        $destination = $name . '-' . $source;
+                    }
+                    $branches[$source] = $destination;
+                }
+                $output[$name]['branches'] = $branches;
+            }
+        }
+
+        $this->projects = $output;
+    }
+
+    public function getProjects()
+    {
+        return $this->projects;
     }
 
     public static function fromFile($file = 'git-deploy.json')
@@ -20,41 +64,46 @@ class GitDeploy
 
         if (is_file($file)) {
             $contents = file_get_contents($file);
-            $results = json_decode($contents);
+            $results = json_decode($contents, true);
+            if ($results === NULL) {
+                throw new \UnexpectedValueException("The given JSON file could not be parsed.");
+            }
         }
 
         return new GitDeploy($results);
     }
 
-    public function update()
+    public function deploy()
     {
         // Create the wrapper.
         $wrapper = new GitWrapper();
         $wrapper->streamOutput();
 
-        foreach ($this->repositories as $dir => $repo) {
-            // Build our git interface.
-            $git = null;
-            if (!is_dir($dir)) {
-                $git = $wrapper->cloneRepository($repo, $dir);
-            }
-            else {
-                $git = new GitWorkingCopy($wrapper, $dir);
-            }
+        // Iterate through each project.
+        foreach ($this->projects as $name => $project) {
+            // Check out all branches.
+            foreach ($project['branches'] as $branch => $destination) {
+                // Build our git interface.
+                $git = null;
+                if (!is_dir($destination)) {
+                    $git = $wrapper->cloneRepository($project['repo'], $destination);
+                }
+                else {
+                    $git = new GitWorkingCopy($wrapper, $destination);
+                }
 
-            // Fetch all the latest.
-            $git->fetch('--all');
+                // Fetch the latest.
+                $git->fetch('origin');
 
-            // Reset over to the gh-pages branch.
-            $git->reset('origin/gh-pages', array('hard' => true));
+                // Checkout the desired branch.
+                $git->checkout($branch, array('force' => true));
 
-            // Remove any extra files.
-            $git->clean('-d', '-f', '-x');
-        }
-    }
+                // Reset any local changes.
+                $git->reset(array('hard' => true));
 
-    public function getRepositories()
-    {
-        return $this->repositories;
+                // Pull the latest from the branch.
+                $git->pull('origin', $branch, array('force' => true));
+            }
+        }
     }
 }