wallet_reset.sh

#!/bin/bash

##
## Decker (c) 2018
##

# How to use?

# This is small script for wallet reset automation. Before use you should send all mined funds
# from node to an other address or cold wallet using ./sendawaynn.sh or ./sendawaynn_auto.sh
# script to leave only iguana outputs (this is not mandatory, but recommended). Also, you should
# have normal utxo count, "normal" mean - to be able to send it in one single transaction. For example,
# if you have more than 1000 utxos - tx will be oversized. Optimal count of uxos to don't break
# max single tx. size limit (100000 bytes) is 100-200 utxos.

# After you start this script it will automatically do following steps:

# 1. Wait for daemon start if it's not started.
# 2. Get pubkey and privkey for given NN_ADDRESS below from daemon.
# 3. Send all your balance in one tx to NN_ADDRESS. If daemon can't send all balance (for example, 
#    in case of oversized tx - script will just exit on this step, nothing hurt)
# 4. Wait for 1 confirmation and save height of block in which this tx is included.
# 5. Create new z-address and grab privkey from it (needed to the trick with rescan).
# 6. Stop daemon and make sure it really stopped.
# 7. Copy current wallet.dat in $HOME/.komodo as backup_%date%_%time%.dat and remove wallet.dat after.
# 8. Start daemon with -gen -notary -pubkey="$NN_PUBKEY" & args .
# 9. Wait for daemon start .
# 10. Imports your NN privkey and rescan from height from step 4.
# 
# Done. 

# Currently script works only with KMD, assets no supported. But if you wish to continue work on this,
# PRs in repo are welcome.

komodo_cli="$HOME/komodo/src/komodo-cli"
komodo_daemon="$HOME/komodo/src/komodod"

NN_ADDRESS=RFVYf9XKRmhR2eoxR9d46v7gKYACMucXs8 
# you'll need only to set NN_ADDRESS, other needed info such as pubkey and privkey
# script will get automatically from daemon

function init_colors () 
{
RESET="\033[0m"
BLACK="\033[30m"    
RED="\033[31m"      
GREEN="\033[32m"    
YELLOW="\033[33m"   
BLUE="\033[34m"     
MAGENTA="\033[35m"  
CYAN="\033[36m"     
WHITE="\033[37m"    
}

function log_print()
{
   datetime=$(date '+%Y-%m-%d %H:%M:%S')
   echo [$datetime] $1
}

function wait_for_daemon()
{

#if [[ ! -z $1 && $1 != "KMD" ]]
if [ ! -z $1 ] && [ $1 != "KMD" ]
then
    coin=$1
    asset=" -ac_name=$1"
else
    coin="KMD"
    asset=""
fi

i=0
while ! $komodo_cli $asset getinfo >/dev/null 2>&1
do 
   i=$((i+1))
   log_print "Waiting for daemon start $coin ($i)"
   sleep 1
   # TODO: in case if daemon start too long, for example, more than 5-7 mins. we should exit from script
done
}

function stop_daemon()
{

#if [[ ! -z $1 && $1 != "KMD" ]]
if [ ! -z $1 ] && [ $1 != "KMD" ]
then
    coin=$1
    asset=" -ac_name=$1"
else
    coin="KMD"
    asset=""
fi

i=0
$komodo_cli $asset stop
# TODO: fix .pid file check for assets
while [ -f $HOME/.komodo/komodod.pid ]
do 
   i=$((i+1))
   log_print "Waiting for daemon $coin stop ($i)"
   sleep 1
done
}


function send_balance()
{
    #if [[ ! -z $1 && $1 != "KMD" ]]
    if [ ! -z $1 ] && [ $1 != "KMD" ]
    then
        coin=$1
        asset=" -ac_name=$1"
    else
    coin="KMD"
        asset=""
    fi

    #echo $komodo_cli $asset getbalance
    BALANCE=$($komodo_cli $asset getbalance 2>/dev/null)
    ERRORLEVEL=$?
    
    if [ "$ERRORLEVEL" -eq "0" ] && [ "$BALANCE" != "0.00000000" ]; then
        message=$(echo -e "[${GREEN}$coin${RESET}] $BALANCE")
        log_print "$message"
    else
        BALANCE="0.00000000"
	    message=$(echo -e "[${RED}$coin${RESET}] $BALANCE")
        log_print "$message"
        exit
    fi

    # sendtoaddress
    #$komodo_cli $asset sendtoaddress $NN_ADDRESS $BALANCE "" "" true

    # redirected stderr to stdout
    RESULT=$($komodo_cli $asset sendtoaddress $NN_ADDRESS $BALANCE "" "" true 2>&1)
    ERRORLEVEL=$?
    if [ "$ERRORLEVEL" -ne "0" ]; then
	log_print "tx $RESULT"
    	exit
    fi
    log_print "txid: $RESULT"

    i=0
    confirmations=0
    while [ "$confirmations" -eq "0" ]
    do
    confirmations=$($komodo_cli $asset gettransaction $RESULT | jq .confirmations)
    i=$((i+1))
    log_print "Waiting for confirmations ($i).$confirmations"
    sleep 10
    done
    blockhash=$($komodo_cli $asset gettransaction $RESULT | jq -r .blockhash)
    height=$($komodo_cli $asset getblock $blockhash | jq .height)
}

# Main

curdir=$(pwd)

#rpcpassword=$(cat $HOME/.komodo/komodo.conf | grep -P "^rpcpassword" | cut -d '=' -f2)
#rpcuser=$(cat $HOME/.komodo/komodo.conf | grep -P "^rpcuser" | cut -d '=' -f2)
#rpcport=$(cat $HOME/.komodo/komodo.conf | grep -P "^rpcport" | cut -d '=' -f2)

#if [ -z $rpcport ]
#then
#rpcport=7771
#fi

init_colors
log_print "Start ..."

log_print "Current dir: $curdir"

wait_for_daemon

log_print "Gathering pubkey ..."
NN_PUBKEY=$($komodo_cli validateaddress $NN_ADDRESS | jq -r .pubkey)
if [ -z $NN_PUBKEY ]
then
    log_print "Failed to obtain pubkey. Exit"
    exit
else
    log_print "Pubkey is $NN_PUBKEY"
fi

log_print "Gathering privkey ..."
NN_PRIVKEY=$($komodo_cli dumpprivkey $NN_ADDRESS)
if [ -z $NN_PRIVKEY ]
then
    log_print "Failed to obtain privkey. Exit"
    exit
else
    log_print "Privkey is obtained"
fi

# disable generate to avoid daemon crash during multiple "error adding notary vin" messages
$komodo_cli setgenerate false

send_balance
log_print "ht.$height ($blockhash)"

NN_ZADDRESS=$($komodo_cli z_getnewaddress)
NN_ZKEY=$($komodo_cli z_exportkey $NN_ZADDRESS)
log_print "New z-address $NN_ZADDRESS"

log_print "Stopping daemon ... "
stop_daemon
log_print "Removing old wallet ... "

wallet_file=backup_$(date '+%Y_%m_%d_%H%M%S').dat
cp $HOME/.komodo/wallet.dat $HOME/.komodo/$wallet_file
rm $HOME/.komodo/wallet.dat
sleep 5
log_print "Starting daemon ... "
# *** STARTING DAEMON ***
# check args 
$komodo_daemon -gen -notary -pubkey="$NN_PUBKEY" &
wait_for_daemon
log_print "Importing private key ... "
$komodo_cli importprivkey $NN_PRIVKEY "" false
log_print "Rescanning from ht.$height ... "
$komodo_cli z_importkey "$NN_ZKEY" \"yes\" $height
log_print "Done"