add support for salt minions

1 files changed, 55 insertions, 0 deletions

diff --git a/cloudinit/CloudConfig/cc_salt_minion.py b/cloudinit/CloudConfig/cc_salt_minion.py
new file mode 100644
index 00000000..9a710e72
--- /dev/null
+++ b/cloudinit/CloudConfig/cc_salt_minion.py
@@ -0,0 +1,55 @@
+# vi: ts=4 expandtab
+#
+#    Author: Jeff Bauer <jbauer@rubic.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 os.path
+import subprocess
+import cloudinit.CloudConfig as cc
+import yaml
+
+def handle(_name, cfg, cloud, log, _args):
+    # If there isn't a salt key in the configuration don't do anything
+    if 'salt_minion' not in cfg:
+        return
+    salt_cfg = cfg['salt_minion']
+    # Start by installing the salt package ...
+    cc.install_packages(("salt",))
+    config_dir = '/etc/salt'
+    if not os.path.isdir(config_dir):
+        os.makedirs(config_dir)
+    # ... and then update the salt configuration
+    if 'conf' in salt_cfg:
+        # Add all sections from the conf object to /etc/salt/minion
+        minion_config = os.path.join(config_dir, 'minion')
+        yaml.dump(salt_cfg['conf'],
+                  file(minion_config, 'w'),
+                  default_flow_style=False)
+    # ... copy the key pair if specified
+    if 'public_key' in salt_cfg and 'private_key' in salt_cfg:
+        pki_dir = '/etc/salt/pki'
+        cumask = os.umask(077)
+        if not os.path.isdir(pki_dir):
+            os.makedirs(pki_dir)
+        pub_name = os.path.join(pki_dir, 'minion.pub')
+        pem_name = os.path.join(pki_dir, 'minion.pem')
+        with open(pub_name, 'w') as f:
+            f.write(salt_cfg['public_key'])
+        with open(pem_name, 'w') as f:
+            f.write(salt_cfg['private_key'])
+        os.umask(cumask)
+
+    # Start salt-minion
+    subprocess.check_call(['service', 'salt-minion', 'start'])