Support prep for apt things too

1 files changed, 10 insertions, 4 deletions

diff --git a/prep_for_ybd.sh b/prep_for_ybd.sh
index 96673ec..28e7ad8 100755
--- a/prep_for_ybd.sh
+++ b/prep_for_ybd.sh
@@ -1,9 +1,15 @@
 #!/bin/sh
 
-# Script to set up an aws or other yum-y machine
+# Script to set up yum-y or apt-y machine for ybd use
 
-package_manager=yum
-
-"$package_manager" install -y make automake gcc gcc-c++ kernel-devel git gawk python-pip
+if [ -x /usr/bin/apt ]; then
+    apt-get install -y make automake gcc g++ linux-headers-$(uname -r) git gawk python-pip
+elif [ -x /usr/bin/yum ]; then
+    yum install -y make automake gcc gcc-c++ kernel-devel git gawk python-pip
+else
+    echo "Unsuported package manager, try apt or yum."
+    exit 1
+fi
 
 pip install pyyaml sandboxlib jsonschema requests
+