From 47e692b57fc9de818093c79553b7ba524049c8d1 Mon Sep 17 00:00:00 2001
From: Bruno Tavares <bruno.exz@gmail.com>
Date: Fri, 21 Feb 2014 15:59:46 -0300
Subject: [PATCH] Support running with sudo

---
 README.md            | 4 ++--
 lib/kickstart-deploy | 6 +++++-
 lib/kickstart-local  | 4 +++-
 3 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index ef4ed18..7949e3e 100644
--- a/README.md
+++ b/README.md
@@ -59,7 +59,7 @@ kickstart compile
 Takes an host target and deploy the compiled code on it.
 
 ```bash
-kickstart deploy [target] [roles ...]
+kickstart deploy [--sudo] [target] [roles ...]
 ```
 
 ##### Examples
@@ -79,7 +79,7 @@ kickstart deploy "-p 2222 vagrant@localhost" nodejs redis
 Deploy localy a recipe that is not over ssh. It basicaly compile and run the install script
 
 ```bash
-kickstart local [roles ...]
+kickstart local [--sudo] [roles ...]
 ```
 
 #### infect
diff --git a/lib/kickstart-deploy b/lib/kickstart-deploy
index 1dcd696..2e8aee1 100755
--- a/lib/kickstart-deploy
+++ b/lib/kickstart-deploy
@@ -1,16 +1,20 @@
 #!/bin/bash -e
 
+[[ "$1" == "--sudo" ]] && prefix="sudo " && shift
+
 target=$1
 [ "$target" ] || echo "Please, provide an ssh target"
 shift
 
+[[ "$1" == "--sudo" ]] && prefix="sudo " && shift
+
 remote_command=$(
 cat <<CMD
   [ -d ~/kickstart ] && rm -rf ~/kickstart
   mkdir ~/kickstart
   cd ~/kickstart
   tar xz
-  bash install.sh
+  $prefix bash install.sh
 CMD
 )
 
diff --git a/lib/kickstart-local b/lib/kickstart-local
index 5706cdc..2761dac 100755
--- a/lib/kickstart-local
+++ b/lib/kickstart-local
@@ -1,7 +1,9 @@
 #!/bin/bash -e
 
+[[ "$1" == "--sudo" ]] && prefix="sudo " && shift
+
 kickstart compile $@
 (
 cd compile
-bash install.sh
+$prefix bash install.sh
 )