Merge pull request #4 from lncm/leftist-flatland

Now working.. Thanks @meeDamian for contributions!
4 years ago · ec38650475
2 changed files with 80 additions and 82 deletions
--- a/14
+++ b/14
@ -1,12 +1,10 @@
-FROM alpine:3.10
+FROM alpine:3.12

-RUN apk add --no-cache curl jq bash docker
+RUN apk add --no-cache curl jq docker

-RUN mkdir /lnd/
-RUN mkdir /secrets/
-RUN mkdir /statuses/
+RUN mkdir /lnd/ /secrets/ /statuses/

-COPY switch.sh /bin/switch
-RUN chmod +x /bin/switch
+COPY switch.sh /usr/local/bin/switch
+RUN chmod +x   /usr/local/bin/switch

-ENTRYPOINT ["switch"]
+ENTRYPOINT ["/usr/local/bin/switch"]
--- a/switch.sh
+++ b/switch.sh
@ -1,4 +1,4 @@
-#!/bin/bash -e
+#!/bin/sh -e

 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 # EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
@ -15,83 +15,83 @@

 # Output: /statuses/node-status-bitcoind-ready  (when ready, where a service can pick it up)

+RPCPASS="${RPCPASS:-$(cat /secrets/rpcpass.txt)}"  # Default password location: /secrets/rpcpass.txt
+SLEEPTIME="${SLEEPTIME:-3600}"                     # Default sleep: 3600
+JSONRPCURL="${JSONRPCURL:-http://10.254.2.2:8332}" # Default RPC endpoint: http://10.254.2.2:8332
+LND_CONTAINER_NAME="${LND_CONTAINER_NAME:-lnd}"    # Default Docker container name: lnd

-# if RPCPASS doesn't exist then set it (Default to whats in /secrets/rpcpass.txt)
-if [ -z $RPCPASS ]; then
-    RPCPASS=`cat /secrets/rpcpass.txt`
-fi
+PREV_MATCH=

-# If sleeptime isn't set, set it to 3600 (1 hour)
-if [ -z $SLEEPTIME ]; then
-    SLEEPTIME=3600
-fi
+switch_on_sync_done() {
+	# Node not pruned so lets do the switching
+	echo 'Checking if synced...'
+	if [ -f /statuses/node-status-bitcoind-ready ]; then
+		echo 'LND is already switched to bitcoind!'
+		return 1
+	fi

-# If JSONRPCURL doesn't exist then set it
-if [ -z $JSONRPCURL ]; then
-    JSONRPCURL="http://10.254.2.2:8332"
-fi
+	if ! grep -q 'bitcoin.node=neutrino' /lnd/lnd.conf; then
+		echo 'Neutrino mode has been disabled'
+		echo 'Switchback is not supported in this version'
+		return 1
+	fi

-# if LND_CONTAINER_NAME doesn't exist then set it
-if [ -z $LND_CONTAINER_NAME ]; then
-    LND_CONTAINER_NAME="lnd"
-fi
+	echo 'If set to neutrino then lets check bitcoind'
+
+	if ! INFO="$(curl --silent --user "lncm:$RPCPASS" --data-binary '{"jsonrpc": "1.0", "id":"switchme", "method": "getblockchaininfo", "params": [] }' "$JSONRPCURL")"; then
+		echo "Error: 'getblockchaininfo' request to bitcoind failed"
+		return
+	fi
+
+	if [ -z "$INFO" ] || err="$(jq -ner "$INFO | .error")"; then
+		echo 'Error: from bitcoind'
+		echo "${err:-Unknown error}"
+		return
+	fi
+
+	INFO="$(jq -ne "$INFO | .result")"
+
+	# Check if pruned
+	if jq -ne "$INFO | .pruned == true"; then
+		echo 'No need to switch from neutrino in pruned mode'
+		return 1
+	fi
+	echo 'Not pruned'
+
+	if jq -ne "$INFO | .headers - .blocks > 10"; then
+		echo "Node isn't full synced yet"
+		PREV_MATCH=
+		return
+	fi
+
+	if [ -z "$PREV_MATCH" ]; then
+		PREV_MATCH="$(jq -ne "$INFO | .headers")"
+		echo 'Sync seems complete!  Will switch on next check.'
+		return
+	fi
+
+	# Skip switch, if headers number didn't change since last check
+	#	(possible network issue).
+	if jq -ne "$INFO | .headers == $PREV_MATCH"; then
+		echo 'Skipping switch for now: headers seem stale'
+		return
+	fi
+
+	echo 'Bitcoind has been switched across to neutrino'
+	touch /statuses/node-status-bitcoind-ready
+	sed -Ei 's|(bitcoin.node)=neutrino|\1=bitcoind|g' /lnd/lnd.conf
+  
+	echo "Restarting LND"
+	docker stop  "$LND_CONTAINER_NAME"
+	docker start "$LND_CONTAINER_NAME"
+}

 while true; do
-  IS_NEUTRINO=`grep -c 'bitcoin.node=neutrino' /lnd/lnd.conf`
-  if [ $IS_NEUTRINO -eq 1 ]; then
-    echo "If set to neutrino then lets check bitcoind"
-
-    INFO=`curl --user lncm:$RPCPASS --data-binary '{"jsonrpc": "1.0", "id":"switchme", "method": "getblockchaininfo", "params": [] }' $JSONRPCURL 2>/dev/null`
-    # check for errors
-    ERROR=`echo $INFO | jq .error`
-    if [ ! -z $ERROR ]; then
-      # if no errors
-      # Check prune mode
-      PRUNE_MODE=`echo $INFO | jq .result.pruned`
-      # check headers
-      HEADERS=`echo $INFO | jq .result.headers`
-      # check blocks
-      BLOCKS=`echo $INFO | jq .result.blocks`
-
-      if [ $PRUNE_MODE != "true" ]; then
-        echo "Not pruned"
-        # Node pruned so lets do the switching
-        echo "Checking if synced...."
-        if [ ! -f /statuses/node-status-bitcoind-ready ]; then
-          if [ $HEADERS -eq $BLOCKS ]; then
-              echo "Bitcoind has been switched across to neutrino"
-              touch /statuses/node-status-bitcoind-ready
-              sed -i 's/bitcoin.node\=neutrino/bitcoin.node\=bitcoind/g; ' /lnd/lnd.conf
-              echo "Attempting to Restarting LND"
-              if command -v docker &> /dev/null; then
-                # restart docker
-                docker stop $LND_CONTAINER_NAME
-                docker start $LND_CONTAINER_NAME
-              else
-                echo "Docker command doesn't exist, reverting config"
-                rm /statuses/node-status-bitcoind-ready
-                sed -i 's/bitcoin.node\=bitcoind/bitcoin.node\=neutrino/g; ' /lnd/lnd.conf
-              fi
-          else
-              echo "Node isn't full synched yet"
-          fi
-        else
-            echo "LND is already switched to bitcoind!"
-        fi
-      else
-        echo "No need to switch from neutrino in pruned mode"
-      fi
-    else
-      # if bitcoind error
-      echo "Error from bitcoind"
-      echo $ERROR
-    fi
-  else
-    echo "Neutrino mode has been disabled"
-    echo "Switchback is not supported in this version"
-
-    #TODO: Lets maybe try to switch back
-  fi
-  # Run every every 1 hour by default or as per configurable
-  sleep $SLEEPTIME
+	if ! switch_on_sync_done; then
+		echo 'Checking not necessary. Exiting.'
+		break
+	fi
+
+	# Run every every 1 hour by default or as per configurable
+	sleep "$SLEEPTIME"
 done