From 490d7875ddb5a0a294b6f34dbce60fb48a2f16fe Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Fri, 13 Jul 2018 23:37:56 -0700
Subject: [PATCH] Snap client-demo fixes

---
 multinode-demo/client.sh | 9 ++++++++-
 multinode-demo/common.sh | 7 +++++++
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/multinode-demo/client.sh b/multinode-demo/client.sh
index 5e11b455a..6a4c8caca 100755
--- a/multinode-demo/client.sh
+++ b/multinode-demo/client.sh
@@ -7,7 +7,14 @@ here=$(dirname "$0")
 # shellcheck source=multinode-demo/common.sh
 source "$here"/common.sh
 
-leader=${1:-${here}/..}  # Default to local solana repo
+leader=$1
+if [[ -z $leader ]]; then
+  if [[ -d "$SNAP" ]]; then
+    leader=testnet.solana.com # Default to testnet when running as a Snap
+  else
+    leader=$here/.. # Default to local solana repo
+  fi
+fi
 count=${2:-1}
 
 rsync_leader_url=$(rsync_url "$leader")
diff --git a/multinode-demo/common.sh b/multinode-demo/common.sh
index 84edfde7d..8255997ec 100644
--- a/multinode-demo/common.sh
+++ b/multinode-demo/common.sh
@@ -14,6 +14,13 @@ if [[ -d "$SNAP" ]]; then # Running inside a Linux Snap?
     else
       printf "%s/command-%s.wrapper" "$SNAP" "$program"
     fi
+    if [[ "$program" = client-demo ]]; then
+      # TODO: Merge client.sh functionality into solana-client-demo proper and
+      #       remove this special case
+      printf "%s/bin/solana-%s" "$SNAP" "$program"
+    else
+      printf "%s/command-%s.wrapper" "$SNAP" "$program"
+    fi
   }
   rsync="$SNAP"/bin/rsync
   SOLANA_METRICS_CONFIG="$(snapctl get metrics-config)"