Spice up run script.

1 files changed, 18 insertions, 2 deletions

diff --git a/run b/run
index b267ab7..c9e150c 100755
--- a/run
+++ b/run
@@ -1,4 +1,20 @@
 #!/bin/bash
 
-sudo /opt/puppetlabs/bin/puppet apply manifests \
-	--modulepath=/etc/puppetlabs/code/environments/production/modules:modules
+# Needed on ubuntu, since this path is not set for root
+# Arch installs puppet in /usr/bin/puppet.
+export PATH="/opt/puppetlabs/bin/puppet:$PATH"
+
+osid=$(awk -F= '/^ID/ { print $2 }' /etc/os-release)
+
+# This is the WRONG way to do it, but it sholud work for now
+case $osid in
+	ubuntu)
+		modpath=/etc/puppetlabs/code/environments/production/modules
+		;;
+	arch)
+		modpath=/etc/puppetlabs/code/modules/
+		;;
+esac
+
+sudo puppet apply manifests \
+	--modulepath="${modpath}:modules"