3 files changed, 56 insertions, 1 deletions
diff --git a/hooks/charmhelpers/payload/__init__.py b/hooks/charmhelpers/payload/__init__.py
new file mode 100644
index 0000000..fc9fbc0
--- /dev/null
+++ b/hooks/charmhelpers/payload/__init__.py
@@ -0,0 +1 @@
+"Tools for working with files injected into a charm just before deployment."
diff --git a/hooks/charmhelpers/payload/execd.py b/hooks/charmhelpers/payload/execd.py
new file mode 100644
index 0000000..6476a75
--- /dev/null
+++ b/hooks/charmhelpers/payload/execd.py
@@ -0,0 +1,50 @@
+#!/usr/bin/env python
+
+import os
+import sys
+import subprocess
+from charmhelpers.core import hookenv
+
+
+def default_execd_dir():
+ return os.path.join(os.environ['CHARM_DIR'], 'exec.d')
+
+
+def execd_module_paths(execd_dir=None):
+ """Generate a list of full paths to modules within execd_dir."""
+ if not execd_dir:
+ execd_dir = default_execd_dir()
+
+ if not os.path.exists(execd_dir):
+ return
+
+ for subpath in os.listdir(execd_dir):
+ module = os.path.join(execd_dir, subpath)
+ if os.path.isdir(module):
+ yield module
+
+
+def execd_submodule_paths(command, execd_dir=None):
+ """Generate a list of full paths to the specified command within exec_dir.
+ """
+ for module_path in execd_module_paths(execd_dir):
+ path = os.path.join(module_path, command)
+ if os.access(path, os.X_OK) and os.path.isfile(path):
+ yield path
+
+
+def execd_run(command, execd_dir=None, die_on_error=False, stderr=None):
+ """Run command for each module within execd_dir which defines it."""
+ for submodule_path in execd_submodule_paths(command, execd_dir):
+ try:
+ subprocess.check_call(submodule_path, shell=True, stderr=stderr)
+ except subprocess.CalledProcessError as e:
+ hookenv.log("Error ({}) running {}. Output: {}".format(
+ e.returncode, e.cmd, e.output))
+ if die_on_error:
+ sys.exit(e.returncode)
+
+
+def execd_preinstall(execd_dir=None):
+ """Run charm-pre-install for each module within execd_dir."""
+ execd_run('charm-pre-install', execd_dir=execd_dir)
diff --git a/hooks/hooks.py b/hooks/hooks.py
index bcdc76f..70aeae6 100755
--- a/hooks/hooks.py
+++ b/hooks/hooks.py
@@ -43,6 +43,8 @@ from charmhelpers.core.hookenv import (
 
 from charmhelpers.core.hookenv import log as juju_log
 
+from charmhelpers.payload.execd import execd_preinstall
+
 from charmhelpers.core.host import (
 service,
 )
@@ -724,6 +726,8 @@ PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin
 ###############################################################################
 @hooks.hook('install')
 def install_hook():
+ log('Begin install hook.')
+ execd_preinstall()
 juju_log("Installing mongodb")
 add_source(config('source'), config('key'))
 apt_update(fatal=True)
@@ -1369,4 +1373,4 @@ def write_logrotate_config(config_data,
 ###############################################################################
 if __name__ == "__main__":
 # execute a hook based on the name the program is called by
- hooks.execute(sys.argv) \ No newline at end of file
+ hooks.execute(sys.argv)