1 files changed, 21 insertions, 17 deletions

diff --git a/bin/ak-get-balances b/bin/ak-get-balances
index 5b13999..a8ee44e 100755
--- a/bin/ak-get-balances
+++ b/bin/ak-get-balances
@@ -28,14 +28,15 @@
 fullprogrampath="$(realpath $0)"
 PROGRAM="$(basename $0)"
 descriptionString="Export balances from schain and zchains"
-source $AK_LIBDIR/_ak_log
-source $AK_LIBDIR/_ak_script
-source $AK_LIBDIR/_ak_ipfs
-source $AK_LIBDIR/_ak_gpg
-source $AK_LIBDIR/_ak_zchain
-source $AK_LIBDIR/_ak_schain
-source $AK_LIBDIR/_ak_sblock
-source $AK_LIBDIR/_ak_coin
+source $AK_LIBDIR/_ak_lib_load
+_ak_lib_load _ak_log
+_ak_lib_load _ak_script
+_ak_lib_load _ak_ipfs
+_ak_lib_load _ak_gpg
+_ak_lib_load _ak_zchain
+_ak_lib_load _ak_schain
+_ak_lib_load _ak_sblock
+_ak_lib_load _ak_coin
 #set -xe
 AK_DB="$AK_WORKDIR/db"
 
@@ -73,7 +74,7 @@ fi
 TEMP="$(_ak_make_temp_directory)"
 cd $TEMP
 
-_ak_get_zblocks_from_sblock(){
+function _ak_get_zblocks_from_sblock(){
     if [ "$(_ak_sblock_show $1 | jq 'has("zblocks")')" == "true" ]
     then
         mkdir $1 && cd $1
@@ -85,7 +86,7 @@ _ak_get_zblocks_from_sblock(){
     fi
 }
 
-_ak_verify_zblocks_found(){
+function _ak_verify_zblocks_found(){
     if [ -d $1 ]
     then
         cd $1
@@ -119,7 +120,7 @@ _ak_verify_zblocks_found(){
     fi
 }
 
-_ak_balances_from_sblock(){
+function _ak_balances_from_sblock(){
     if [ -z $1 ] || [ ! -n "$1" ]
     then
         _ak_log_error "No sblock provided $1"
@@ -155,7 +156,7 @@ _ak_balances_from_sblock(){
     fi
 }
 
-_ak_rewards_from_sblock(){
+function _ak_rewards_from_sblock(){
     if [ ! -z $1 ] && [ -n "$1" ]
     then
         CUR_TARGET="$1"
@@ -191,7 +192,7 @@ _ak_rewards_from_sblock(){
     fi
 }
 
-_ak_sblock_get_previous(){
+function _ak_sblock_get_previous(){
     if [ -z $1 ] || [ ! -n "$1" ]
     then
         _ak_log_error "No sblock provided !!"
@@ -207,7 +208,7 @@ _ak_sblock_get_previous(){
     echo -n $PREVIOUS
 }
 
-_ak_balances_calculate(){
+function _ak_balances_calculate(){
     if [ ! -z $1 ] && [ -n "$1" ]
     then
         _ak_log_info "Calculating balance for $1"
@@ -237,7 +238,7 @@ _ak_balances_calculate(){
     fi
 }
 
-_ak_balances_print(){
+function _ak_balances_print(){
     if [ ! -z $1 ] && [ -n "$1" ]
     then
         CUR_TARGET="$1"
@@ -255,7 +256,7 @@ _ak_balances_print(){
     fi
 }
 
-_ak_schain_counting_balances(){
+function _ak_schain_counting_balances(){
     if [ ! -z $1 ] && [ -n "$1" ]
     then
         CUR_TARGET="$1"
@@ -285,9 +286,12 @@ _ak_schain_counting_balances(){
     fi
 }
 
-if [ ! -z $1 ] && [ -n "$1" ]
+if [ ! -z $1 ] && [ -n "$1" ] && [ [ "$1" == "--help" ] || [ "$1" == "-h" ] ]
 then
     _ak_usage
+elif [ ! -z $1 ] && [ -n "$1" ]
+then
+    _ak_usage err
 else
     CUR_TARGET="$LATEST"
     _ak_log_info "$CUR_TARGET $LATEST"