Changes to puppet module + mcollective module

1 files changed, 63 insertions, 0 deletions

diff --git a/cloudinit/CloudConfig/cc_mcollective.py b/cloudinit/CloudConfig/cc_mcollective.py
new file mode 100644
index 00000000..ce86a2aa
--- /dev/null
+++ b/cloudinit/CloudConfig/cc_mcollective.py
@@ -0,0 +1,63 @@
+# vi: ts=4 expandtab
+#
+#    Copyright (C) 2009-2011 Canonical Ltd.
+#
+#    Author: Marc Cluet <marc.cluet@canonical.com>
+#    Based on code by Scott Moser <scott.moser@canonical.com>
+#
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License version 3, as
+#    published by the Free Software Foundation.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+import os
+import pwd
+import socket
+import subprocess
+import StringIO
+import ConfigParser
+import cloudinit.CloudConfig as cc
+
+# Our fake header section
+class FakeSecHead(object):
+   def __init__(self, fp):
+     self.fp = fp
+     self.sechead = '[nullsection]\n'
+   def readline(self):
+     if self.sechead:
+       try: return self.sechead
+       finally: self.sechead = None
+     else: return self.fp.readline()
+
+def handle(name,cfg,cloud,log,args):
+    # If there isn't a mcollective key in the configuration don't do anything
+    if not cfg.has_key('mcollective'): return
+    mcollective_cfg = cfg['mcollective']
+    # Start by installing the mcollective package ...
+    cc.install_packages(("mcollective",))
+
+    # ... and then update the mcollective configuration
+    if mcollective_cfg.has_key('conf'):
+        # Create object for reading server.cfg values
+        mcollective_config = ConfigParser.ConfigParser()
+        # Read server.cfg values from original file in order to be able to mix the rest up
+        mcollective_config.readfp(FakeSecHead(open('/etc/mcollective/server.cfg')))
+        for cfg_name, cfg in mcollective_cfg['conf'].iteritems():
+            # Iterate throug the config items, we'll use ConfigParser.set
+            # to overwrite or create new items as needed
+            for o, v in cfg.iteritems():
+                mcollective_config.set(cfg_name,o,v)
+            # We got all our config as wanted we'll rename
+            # the previous server.cfg and create our new one
+            os.rename('/etc/mcollective/server.cfg','/etc/mcollective/server.cfg.old')
+            with open('/etc/mcollective/server.cfg', 'wb') as configfile:
+                mcollective_config.write(configfile)
+    # Start mcollective
+    subprocess.check_call(['service', 'mcollective', 'start'])
+