From 32a0bf6fd2a5c73b02bdaa39d4bc468e834e117c Mon Sep 17 00:00:00 2001
From: Girish Ramakrishnan <girish@cloudron.io>
Date: Wed, 4 May 2016 15:20:39 -0700
Subject: [PATCH] add --check arg to setup_infra.sh

---
 src/scripts/setup_infra.sh | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/scripts/setup_infra.sh b/src/scripts/setup_infra.sh
index 027d80bd7..0a4a3f472 100755
--- a/src/scripts/setup_infra.sh
+++ b/src/scripts/setup_infra.sh
@@ -2,6 +2,16 @@
 
 set -eu -o pipefail
 
+if [[ ${EUID} -ne 0 ]]; then
+    echo "This script should be run as root." > /dev/stderr
+    exit 1
+fi
+
+if [[ $# == 1 && "$1" == "--check" ]]; then
+    echo "OK"
+    exit 0
+fi
+
 readonly DATA_DIR="/home/yellowtent/data"
 
 script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"