From 463779ca3d297ade2568893372fb6e982fb5ec74 Mon Sep 17 00:00:00 2001
From: Bruce Williams <bruce@codefluency.com>
Date: Thu, 18 Mar 2010 14:46:12 -0700
Subject: Support different create and revision= operations depending on the
 revision type in Git (commit, tag, or local/remote branch)

---
 lib/puppet/provider/vcsrepo/git.rb | 53 +++++++++++++++++++++++++++++++++++---
 1 file changed, 50 insertions(+), 3 deletions(-)

(limited to 'lib')

diff --git a/lib/puppet/provider/vcsrepo/git.rb b/lib/puppet/provider/vcsrepo/git.rb
index e511a10..952126c 100644
--- a/lib/puppet/provider/vcsrepo/git.rb
+++ b/lib/puppet/provider/vcsrepo/git.rb
@@ -16,7 +16,7 @@ Puppet::Type.type(:vcsrepo).provide(:git, :parent => Puppet::Provider::Vcsrepo)
         if @resource.value(:ensure) == :bare
           notice "Ignoring revision for bare repository"
         else
-          reset(@resource.value(:revision))
+          checkout_branch_or_reset
         end
       end
       if @resource.value(:ensure) != :bare
@@ -41,9 +41,24 @@ Puppet::Type.type(:vcsrepo).provide(:git, :parent => Puppet::Provider::Vcsrepo)
 
   def revision=(desired)
     fetch
-    reset(desired)
-    unless @resource.value(:ensure) == :bare
+    if local_revision_branch?(desired)
+      at_path do
+        git('checkout', desired)
+        git('pull', 'origin')
+      end
+      update_submodules
+    elsif remote_revision_branch?(desired)
+      at_path do
+        git('checkout',
+            '-b', @resource.value(:revision),
+            '--track', "origin/#{@resource.value(:revision)}")
+      end
       update_submodules
+    else
+      reset(desired)
+      if @resource.value(:ensure) != :bare
+        update_submodules
+      end
     end
   end
 
@@ -90,6 +105,12 @@ Puppet::Type.type(:vcsrepo).provide(:git, :parent => Puppet::Provider::Vcsrepo)
     end
   end
 
+  def pull
+    at_path do
+      git('pull', 'origin')
+    end
+  end
+  
   def init_repository(path)
     if @resource.value(:ensure) == :bare && working_copy_exists?
       convert_working_copy_to_bare
@@ -150,6 +171,16 @@ Puppet::Type.type(:vcsrepo).provide(:git, :parent => Puppet::Provider::Vcsrepo)
     false
   end
 
+  def checkout_branch_or_reset
+    if remote_revision_branch?
+      at_path do
+        git('checkout', '-b', @resource.value(:revision), '--track', "origin/#{@resource.value(:revision)}")
+      end
+    else
+      reset(@resource.value(:revision))
+    end
+  end
+
   def reset(desired)
     at_path do
       git('reset', '--hard', desired)
@@ -163,4 +194,20 @@ Puppet::Type.type(:vcsrepo).provide(:git, :parent => Puppet::Provider::Vcsrepo)
     end
   end
 
+  def remote_branch_revision?(revision = @resource.value(:revision))
+    at_path do
+      branches.include?("origin/#{revision}")
+    end
+  end
+
+  def local_branch_revision?(revision = @resource.value(:revision))
+    at_path do
+      branches.include?(revision)
+    end
+  end
+
+  def branches
+    at_path { git('branch', '-a') }.gsub('*', ' ').split(/\n/).map { |line| line.strip }
+  end
+
 end
-- 
cgit v1.2.3