From c714947908b69c538ea9d050d4375e2b3f19e7ac Mon Sep 17 00:00:00 2001
From: Bruce Williams <bruce@codefluency.com>
Date: Sat, 13 Mar 2010 17:32:17 -0800
Subject: Support Mercurial tags, restructure examples

---
 lib/puppet/provider/vcsrepo/hg.rb | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

(limited to 'lib/puppet/provider/vcsrepo/hg.rb')

diff --git a/lib/puppet/provider/vcsrepo/hg.rb b/lib/puppet/provider/vcsrepo/hg.rb
index 7263b4f..928049f 100644
--- a/lib/puppet/provider/vcsrepo/hg.rb
+++ b/lib/puppet/provider/vcsrepo/hg.rb
@@ -24,7 +24,24 @@ Puppet::Type.type(:vcsrepo).provide(:hg, :parent => Puppet::Provider::Vcsrepo) d
   
   def revision
     at_path do
-      hg('parents')[/^changeset:\s+(?:-?\d+):(\S+)/m, 1]
+      current = hg('parents')[/^changeset:\s+(?:-?\d+):(\S+)/m, 1]
+      desired = @resource.value(:revision)
+      if current == desired
+        current
+      else
+        mapped = hg('tags')[/^#{Regexp.quote(desired)}\s+\d+:(\S+)/m, 1]
+        if mapped
+          # A tag, return that tag if it maps to the current nodeid
+          if current == mapped
+            desired
+          else
+            current
+          end
+        else
+          # Use the current nodeid
+          current
+        end
+      end
     end
   end
 
-- 
cgit v1.2.3