Shell provisioner sudo fix

2 files changed, 10 insertions, 7 deletions

diff --git a/TODO.md b/TODO.md
index 5c8723c..62e44b3 100644
--- a/TODO.md
+++ b/TODO.md
@@ -1,8 +1,6 @@
 TODO
 ====
 
-* Minimal manifest for fast provisioning.
-* Update to new nodo style (hiera and nodo::role).
 * Support for recursive clones in `bin/mrconfig`.
 * Test!
 * Puppet 3.x support:
diff --git a/bin/provision b/bin/provision
index a3e80d7..6e06320 100755
--- a/bin/provision
+++ b/bin/provision
@@ -3,8 +3,13 @@
 # Simple shell provisioner for Vagrant instances.
 #
 
+# Set sudo config
+if [ "`whoami`" != 'root' ]; then
+  sudo="sudo"
+fi
+
 # Ensure the system is updated.
-sudo apt-get update && DEBIAN_FRONTEND=noninteractive sudo apt-get dist-upgrade -y && sudo apt-get autoremove -y && sudo apt-get clean
+$sudo apt-get update && DEBIAN_FRONTEND=noninteractive $sudo apt-get dist-upgrade -y && $sudo apt-get autoremove -y && $sudo apt-get clean
 
 # Install dependencies
 source /vagrant/puppet/bin/dependencies
@@ -16,12 +21,12 @@ done
 
 # Link hiera configuration.
 if [ ! -h "/etc/puppet/hiera.yaml" ]; then
-  sudo rm -f /etc/puppet/hiera.yaml
-  sudo ln -s /vagrant/puppet/hiera/hiera.yaml /etc/puppet/hiera.yaml
+  $sudo rm -f /etc/puppet/hiera.yaml
+  $sudo ln -s /vagrant/puppet/hiera/hiera.yaml /etc/puppet/hiera.yaml
 fi
 
 # Link puppet configuration.
 if [ ! -h "/etc/puppet/puppet.conf" ]; then
-  sudo rm -f /etc/puppet/puppet.conf
-  sudo ln -s /vagrant/puppet/puppet.conf /etc/puppet/puppet.conf
+  $sudo rm -f /etc/puppet/puppet.conf
+  $sudo ln -s /vagrant/puppet/puppet.conf /etc/puppet/puppet.conf
 fi