From acb1171422e8cc0ade4cc8cb0889b20cb52d9151 Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Thu, 6 Sep 2018 09:29:56 -0700
Subject: [PATCH] Add -e option

---
 net/init-metrics.sh | 55 +++++++++++++++++++++++++++------------------
 1 file changed, 33 insertions(+), 22 deletions(-)

diff --git a/net/init-metrics.sh b/net/init-metrics.sh
index 04da893de..621659363 100755
--- a/net/init-metrics.sh
+++ b/net/init-metrics.sh
@@ -11,12 +11,14 @@ usage() {
     echo "Error: $*"
   fi
   cat <<EOF
-usage: $0 [-d] [username]
+usage: $0 [-e] [-d] [username]
 
 Creates a testnet dev metrics database
 
   username        InfluxDB user with access to create a new database
   -d              Delete the database instead of creating it
+  -e              Assume database already exists and SOLANA_METRICS_CONFIG is
+                  defined in the environment already
 
 EOF
   exit $exitcode
@@ -24,8 +26,9 @@ EOF
 
 loadConfigFile
 
+useEnv=false
 delete=false
-while getopts "hd" opt; do
+while getopts "hde" opt; do
   case $opt in
   h|\?)
     usage
@@ -34,6 +37,9 @@ while getopts "hd" opt; do
   d)
     delete=true
     ;;
+  e)
+    useEnv=true
+    ;;
   *)
     usage "Error: unhandled option: $opt"
     ;;
@@ -41,29 +47,34 @@ while getopts "hd" opt; do
 done
 shift $((OPTIND - 1))
 
-username=$1
-[[ -n "$username" ]] || usage "username not specified"
+if $useEnv; then
+  [[ -n $SOLANA_METRICS_CONFIG ]] ||
+    usage "Error: SOLANA_METRICS_CONFIG is not defined in the environment"
+else
+  username=$1
+  [[ -n "$username" ]] || usage "username not specified"
 
-read -rs -p "InfluxDB password for $username: " password
-[[ -n $password ]] || { echo "Password not specified"; exit 1; }
-echo
+  read -rs -p "InfluxDB password for $username: " password
+  [[ -n $password ]] || { echo "Password not specified"; exit 1; }
+  echo
 
-query() {
-  echo "$*"
-  curl -XPOST \
-    "https://metrics.solana.com:8086/query?u=${username}&p=${password}" \
-    --data-urlencode "q=$*"
-}
+  query() {
+    echo "$*"
+    curl -XPOST \
+      "https://metrics.solana.com:8086/query?u=${username}&p=${password}" \
+      --data-urlencode "q=$*"
+  }
 
-query "DROP DATABASE \"$netBasename\""
-! $delete || exit 0
-query "CREATE DATABASE \"$netBasename\""
-query "ALTER RETENTION POLICY autogen ON \"$netBasename\" DURATION 7d"
-query "GRANT READ ON \"$netBasename\" TO \"ro\""
-query "GRANT WRITE ON \"$netBasename\" TO \"scratch_writer\""
+  query "DROP DATABASE \"$netBasename\""
+  ! $delete || exit 0
+  query "CREATE DATABASE \"$netBasename\""
+  query "ALTER RETENTION POLICY autogen ON \"$netBasename\" DURATION 7d"
+  query "GRANT READ ON \"$netBasename\" TO \"ro\""
+  query "GRANT WRITE ON \"$netBasename\" TO \"scratch_writer\""
 
-echo "export \
-  SOLANA_METRICS_CONFIG=\"db=$netBasename,u=scratch_writer,p=topsecret\" \
-" >> "$configFile"
+  SOLANA_METRICS_CONFIG="db=$netBasename,u=scratch_writer,p=topsecret"
+fi
+
+echo "export SOLANA_METRICS_CONFIG=\"$SOLANA_METRICS_CONFIG\"" >> "$configFile"
 
 exit 0