From bd2fe21cec8abc3fb9cb0f61aa7a03944d13a98a Mon Sep 17 00:00:00 2001
From: Sol Jerome <solj@ices.utexas.edu>
Date: Sat, 22 May 2010 17:27:01 +0000
Subject: Hg: New Mercurial plugin from Fabian Affolter

Signed-off-by: Sol Jerome <solj@ices.utexas.edu>

git-svn-id: https://svn.mcs.anl.gov/repos/bcfg/trunk/bcfg2@5862 ce84e21b-d406-0410-9b95-82705330c041
---
 src/lib/Server/Plugins/Hg.py       | 46 ++++++++++++++++++++++++++++++++++++++
 src/lib/Server/Plugins/__init__.py |  1 +
 2 files changed, 47 insertions(+)
 create mode 100644 src/lib/Server/Plugins/Hg.py

(limited to 'src/lib/Server/Plugins')

diff --git a/src/lib/Server/Plugins/Hg.py b/src/lib/Server/Plugins/Hg.py
new file mode 100644
index 000000000..ef45a5af8
--- /dev/null
+++ b/src/lib/Server/Plugins/Hg.py
@@ -0,0 +1,46 @@
+import os
+from mercurial import ui, hg
+from subprocess import Popen, PIPE
+import Bcfg2.Server.Plugin
+
+# for debugging output only
+import logging
+logger = logging.getLogger('Bcfg2.Plugins.Mercurial')
+
+class Hg(Bcfg2.Server.Plugin.Plugin,
+             Bcfg2.Server.Plugin.Version):
+    """"Mercurial is a version plugin for dealing with bcfg2 repos"""
+    name = 'Mercurial'
+    __version__ = '$Id$'
+    __author__ = 'bcfg-dev@mcs.anl.gov'
+
+    def __init__(self, core, datastore):
+        Bcfg2.Server.Plugin.Plugin.__init__(self, core, datastore)
+        Bcfg2.Server.Plugin.Version.__init__(self)
+        self.core = core
+        self.datastore = datastore
+
+        # path to hg directory for bcfg2 repo
+        hg_dir = "%s/.hg" % datastore
+
+        # Read changeset from bcfg2 repo
+        if os.path.isdir(hg_dir):
+            self.get_revision()
+        else:
+            logger.error("%s is not present." % hg_dir)
+            raise Bcfg2.Server.Plugin.PluginInitError
+
+        logger.debug("Initialized hg plugin with hg directory = %s" % hg_dir)
+
+    def get_revision(self):
+        '''Read hg revision information for the bcfg2 repository'''
+        try:
+            repo_path = "%s/" % self.datastore
+            repo = hg.repository(ui.ui(), repo_path)
+            tip = repo.changelog.tip()
+            revision = repo.changelog.rev(tip)
+        except:
+            logger.error("Failed to read hg repository; disabling mercurial support")
+            raise Bcfg2.Server.Plugin.PluginInitError
+        return revision
+
diff --git a/src/lib/Server/Plugins/__init__.py b/src/lib/Server/Plugins/__init__.py
index 8db59ecc0..11803ace8 100644
--- a/src/lib/Server/Plugins/__init__.py
+++ b/src/lib/Server/Plugins/__init__.py
@@ -11,6 +11,7 @@ __all__ = [
            'Fossil',
            'Git',
            'GroupPatterns',
+           'Hg',
            'Hostbase',
            'Metadata',
            'NagiosGen',
-- 
cgit v1.2.3-1-g7c22