From cc99f2343021cd9817fe8c71812a0f932d1276ac Mon Sep 17 00:00:00 2001
From: Yan <yans@yancomm.net>
Date: Thu, 7 May 2015 19:27:54 -0700
Subject: [PATCH] added a 'setup' action to manage-tools

---
 README.md        | 11 +++++++++--
 bin/manage-tools | 27 ++++++++++++++++++++++++++-
 2 files changed, 35 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 5342fe9..fd29ebc 100644
--- a/README.md
+++ b/README.md
@@ -5,11 +5,18 @@ Of course, this isn't a hard problem, but it's really nice to have them in one p
 To use, do:
 
 ```bash
+# set up the path
+manage-tools setup
+source ~/.bashrc
+
 # list the available tools
 manage-tools list
 
-# install gdb
-manage-tools install gdb
+# install gdb, allowing it to try to sudo install dependencies
+manage-tools -s install gdb
+
+# install pwntools, but don't let it sudo install dependencies
+manage-tools install pwntools
 
 # uninstall gdb
 manage-tools uninstall gdb
diff --git a/bin/manage-tools b/bin/manage-tools
index 929d2e5..5ee8887 100755
--- a/bin/manage-tools
+++ b/bin/manage-tools
@@ -1,11 +1,36 @@
 #!/bin/bash -e
 
+ALLOW_SUDO=0
+while [[ $1 == -* ]]
+do
+	case $1 in
+		-h)
+			cat <<END
+Usage: $0 [-s] (list|setup|install|uninstall|bin) tool
+Where:
+	-s		allow running things with sudo (i.e., to install debs)
+	tool		the name of the tool
+END
+			exit
+			;;
+		-s)
+			ALLOW_SUDO=1
+			;;
+		*)
+			;;
+	esac
+	shift
+done
+
 ACTION=$1
 TOOL=$2
 
 cd $(dirname "${BASH_SOURCE[0]}")/..
 
 case $ACTION in
+	setup)
+		echo "PATH=\"$PWD/bin:\$PATH\"" >> ~/.bashrc
+		;;
 	list)
 		for t in *
 		do
@@ -32,7 +57,7 @@ case $ACTION in
 		cd $TOOL
 
 		echo "TOOLS | $TOOL | starting install"
-		[ -x ./install-root ] && ./install-root
+		[ -x ./install-root -a "$ALLOW_SUDO" -eq 1 ] && sudo ./install-root
 		./install
 		echo "TOOLS | $TOOL | install finished"