From c1d4ef8a381132c7f039210872611e4f82e22dae Mon Sep 17 00:00:00 2001
From: Liam <liam@opentensor.dev>
Date: Tue, 14 May 2024 10:51:24 +0500
Subject: [PATCH] feat: allow invoking localnet.sh from anywhere

---
 scripts/localnet.sh | 68 ++++++++++++++++++++++++++-------------------
 1 file changed, 39 insertions(+), 29 deletions(-)

diff --git a/scripts/localnet.sh b/scripts/localnet.sh
index 35bf12528..66525baca 100755
--- a/scripts/localnet.sh
+++ b/scripts/localnet.sh
@@ -1,56 +1,66 @@
 #!/bin/bash
 
+# Determine the directory this script resides in. This allows invoking it from any location.
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" &>/dev/null && pwd)"
+
+# The base directory of the subtensor project
+BASE_DIR="$SCRIPT_DIR/.."
+
 : "${CHAIN:=local}"
 : "${BUILD_BINARY:=1}"
-: "${SPEC_PATH:=specs/}"
 : "${FEATURES:=pow-faucet}"
 
+SPEC_PATH="${SCRIPT_DIR}/specs/"
 FULL_PATH="$SPEC_PATH$CHAIN.json"
 
 if [ ! -d "$SPEC_PATH" ]; then
-	echo "*** Creating directory ${SPEC_PATH}..."
-	mkdir $SPEC_PATH
+  echo "*** Creating directory ${SPEC_PATH}..."
+  mkdir $SPEC_PATH
 fi
 
 if [[ $BUILD_BINARY == "1" ]]; then
-	echo "*** Building substrate binary..."
-	cargo build --release --features "$FEATURES"
-	echo "*** Binary compiled"
+  echo "*** Building substrate binary..."
+  cargo build --release --features "$FEATURES" --manifest-path "$BASE_DIR/node/Cargo.toml"
+  echo "*** Binary compiled"
 fi
 
 echo "*** Building chainspec..."
-./target/release/node-subtensor build-spec --disable-default-bootnode --raw --chain $CHAIN > $FULL_PATH
+"$BASE_DIR/target/release/node-subtensor" build-spec --disable-default-bootnode --raw --chain $CHAIN >$FULL_PATH
 echo "*** Chainspec built and output to file"
 
 echo "*** Purging previous state..."
-./target/release/node-subtensor purge-chain -y --base-path /tmp/bob --chain="$FULL_PATH" >/dev/null 2>&1
-./target/release/node-subtensor purge-chain -y --base-path /tmp/alice --chain="$FULL_PATH" >/dev/null 2>&1
+"$BASE_DIR/target/release/node-subtensor" purge-chain -y --base-path /tmp/bob --chain="$FULL_PATH" >/dev/null 2>&1
+"$BASE_DIR/target/release/node-subtensor" purge-chain -y --base-path /tmp/alice --chain="$FULL_PATH" >/dev/null 2>&1
 echo "*** Previous chainstate purged"
 
 echo "*** Starting localnet nodes..."
 alice_start=(
-	./target/release/node-subtensor
-	--base-path /tmp/alice
-	--chain="$FULL_PATH"
-	--alice
-	--port 30334
-	--rpc-port 9946
-	--validator
-	--rpc-cors=all
-	--allow-private-ipv4
-	--discover-local
+  "$BASE_DIR/target/release/node-subtensor"
+  --base-path /tmp/alice
+  --chain="$FULL_PATH"
+  --alice
+  --port 30334
+  --rpc-port 9946
+  --validator
+  --rpc-cors=all
+  --allow-private-ipv4
+  --discover-local
 )
 
 bob_start=(
-	./target/release/node-subtensor
-	--base-path /tmp/bob
-	--chain="$FULL_PATH"
-	--bob
-	--port 30335
-	--rpc-port 9945
-	--validator
-	--allow-private-ipv4
-	--discover-local
+  "$BASE_DIR"/target/release/node-subtensor
+  --base-path /tmp/bob
+  --chain="$FULL_PATH"
+  --bob
+  --port 30335
+  --rpc-port 9945
+  --validator
+  --allow-private-ipv4
+  --discover-local
 )
 
-(trap 'kill 0' SIGINT; ("${alice_start[@]}" 2>&1) & ("${bob_start[@]}" 2>&1))
+(
+  trap 'kill 0' SIGINT
+  ("${alice_start[@]}" 2>&1) &
+  ("${bob_start[@]}" 2>&1)
+)