#!/bin/bash

# command info
if [ $# -eq 0 ] || [ "$1" = "-h" ] || [ "$1" = "-help" ]; then
 echo "# small config script to autounlock lnd after restart"
 echo "# lnd.autounlock.sh status"
 echo "# lnd.autounlock.sh [on|off] [?passwordC]"
 exit 1
fi

if [ "$1" = "status" ]; then
  autoUnlock=$(sudo cat /mnt/hdd/app-data/lnd/lnd.conf 2>/dev/null | grep -c "^wallet-unlock-password-file=")
  if [ ${autoUnlock} -eq 0 ]; then
    echo "autoUnlock=off"
  else
    echo "autoUnlock=on"
  fi
  exit 0
fi

# 1. parameter [on|off]
turn="off"
if [ "$1" = "1" ] || [ "$1" = "on" ]; then turn="on"; fi

# 2. parameter [?passwordC]
passwordC=$2

# run interactive if 'turn on' && no further parameters
if [ "${turn}" = "on" ] && [ ${#passwordC} -eq 0 ]; then

  dialog --backtitle "LND Auto-Unlock" --inputbox "ENTER your PASSWORD C:

For more details see chapter in GitHub README 
'Auto-unlock LND on startup'
https://github.com/rootzoll/raspiblitz

Password C will be stored on the device.
" 13 52 2>./.tmp
 
  wasCancel=$( echo $? )
  passwordC=$( cat ./.tmp )
  
  if [ ${wasCancel} -eq 1 ]; then
    echo "# CANCEL LND Auto-Unlock"
    sleep 2
    exit 1
  fi
  if [ ${#passwordC} -eq 0 ]; then
    echo "# input cannot be empty - repeat"
    sleep 3
    sudo /home/admin/config.scripts/lnd.autounlock.sh on
    exit $?
  fi

  # test if correct
  echo "# testing password .. please wait"
  sudo systemctl restart lnd
  sleep 4
  error=""
  source <(sudo /home/admin/config.scripts/lnd.unlock.sh unlock "$passwordC")
  if [ "${error}" != "" ];then
    echo "# PASSWORD C is wrong - try again or cancel"
    sleep 3
    sudo /home/admin/config.scripts/lnd.autounlock.sh on
    exit $?
  fi
  shred -u ./.tmp
fi

# lnd conf file
lndConfig="/mnt/hdd/app-data/lnd/lnd.conf"
passwordFile="/mnt/hdd/app-data/lnd/data/chain/bitcoin/mainnet/password.info"

# switch on
if [ "$1" = "1" ] || [ "$1" = "on" ]; then

  echo "# switching the Auto-Unlock ON"

  # password C needs to be stored on RaspiBlitz
  echo "# storing password on hdd ${passwordFile}"
  sudo sh -c "echo \"${passwordC}\" > ${passwordFile}"
  sudo chmod 660 "${passwordFile}"
  sudo chown bitcoin:bitcoin "${passwordFile}"

  # remove any existing active config in lnd.conf
  sudo sed -i "/^wallet-unlock-password-file=/d" /mnt/hdd/app-data/lnd/lnd.conf

  # add the config line under [Application Options] section
  sudo sed -i "/^\[Application Options\]/ { 
n
a wallet-unlock-password-file=${passwordFile}
}" /mnt/hdd/app-data/lnd/lnd.conf

  echo "# Auto-Unlock is now ON (after manual lnd restart)"
  exit 0
fi

# switch off
if [ "$1" = "0" ] || [ "$1" = "off" ]; then
  echo "# switching the Auto-Unlock OFF"

  # setting value in raspi blitz config
  /home/admin/config.scripts/blitz.conf.sh set autoUnlock "off"

  # delete password C securely
  echo "# shredding password on for RaspiBlitz Auto-Unlock"
  sudo shred -u "${passwordFile}" 2>/dev/null

  # remove any existing active config in lnd.conf
  sudo sed -i "/^wallet-unlock-password-file=/d" /mnt/hdd/app-data/lnd/lnd.conf

  echo "# Auto-Unlock is now OFF (after manual lnd restart)"
  exit 0
fi