aboutsummaryrefslogtreecommitdiff
path: root/bin/ak-get-balances
blob: 5b13999a3ef62eec52e16c27df0ddc2c1eae962a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
#!/usr/bin/env bash
###
### arching-kaos-tools
### Tools to interact and build an Arching Kaos Infochain
### Copyright (C) 2021 - 2025  kaotisk
###
### This program is free software: you can redistribute it and/or modify
### it under the terms of the GNU General Public License as published by
### the Free Software Foundation, either version 3 of the License, or
### (at your option) any later version.
###
### This program is distributed in the hope that it will be useful,
### but WITHOUT ANY WARRANTY; without even the implied warranty of
### MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
### GNU General Public License for more details.
###
### You should have received a copy of the GNU General Public License
### along with this program.  If not, see <http://www.gnu.org/licenses/>.
###
##
## Calculates the balances of the sblocks found from a chain
## Saves stuff too for later reference
##
## The program is not accepting sblocks that hold invalid zchains
## The zchains are the ones that can be crawled back from a zblock that is found
## in the sblock.
##
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
#set -xe
AK_DB="$AK_WORKDIR/db"

AK_BALANCESDIR="$AK_WORKDIR/balances"
if [ ! -d $AK_BALANCESDIR ]
then
    mkdir $AK_BALANCESDIR
    if [ $? -ne 0 ]
    then
        _ak_log_error "Could not create $AK_BALANCESDIR"
        exit 1
    fi
fi

AK_INVALIDATED_SBLOCKS="$AK_WORKDIR/invalid_sblocks"
if [ ! -d $AK_INVALIDATED_SBLOCKS ]
then
    mkdir -p $AK_INVALIDATED_SBLOCKS
    if [ $? -ne 0 ]
    then
        _ak_log_error "Could not create $AK_INVALIDATED_SBLOCKS"
        exit 1
    fi
fi

if [ ! -f $AK_WORKDIR/schain.latest ]
then
    _ak_schain_latest_cached
    LATEST="$(cat $AK_WORKDIR/schain.latest | jq -r '.latest_block')"
    #echo -n $LATEST > $AK_WORKDIR/schain.latest
else
    LATEST="$(cat $AK_WORKDIR/schain.latest | jq -r '.latest_block')"
fi

TEMP="$(_ak_make_temp_directory)"
cd $TEMP

_ak_get_zblocks_from_sblock(){
    if [ "$(_ak_sblock_show $1 | jq 'has("zblocks")')" == "true" ]
    then
        mkdir $1 && cd $1
        _ak_sblock_show $1 | jq -r '.zblocks[]' | while read zblock
        do
            touch $zblock.untested
        done
        cd ..
    fi
}

_ak_verify_zblocks_found(){
    if [ -d $1 ]
    then
        cd $1
        find . -type f | while read found_filename
        do
            filename="$(basename $found_filename)"
            zblock="$(echo -n $filename | cut -d '.' -f 1)"
            zblock_status="$(echo -n $filename | cut -d '.' -f 2)"
            if [ "$zblock_status" == "untested" ] && \
                [ ! -f $AK_DB/zblocks-tested/$zblock.success ] && \
                [ ! -f $AK_DB/zblocks-tested/$zblock.failed ]
            then
                _ak_zchain_crawl $zblock > /dev/null 2>&1
                if [ $? -ne 0 ]
                then
                    cp $filename $AK_DB/zblocks-tested/$zblock.failed
                    mv $filename $zblock.failed
                else
                    cp $filename $AK_DB/zblocks-tested/$zblock.success
                    mv $filename $zblock.success
                fi
            elif [ -f $AK_DB/zblocks-tested/$zblock.success ]
            then
                mv $filename $zblock.success
            elif [ -f $AK_DB/zblocks-tested/$zblock.failed ]
            then
                mv $filename $zblock.failed
            fi
        done
        cd ..
    fi
}

_ak_balances_from_sblock(){
    if [ -z $1 ] || [ ! -n "$1" ]
    then
        _ak_log_error "No sblock provided $1"
        exit 1
    fi
    CUR_TARGET="$1"
    _ak_log_info "Extracting balances from sblock provided $CUR_TARGET"
    if [ ! -f $AK_MINEDBLOCKSDIR/$CUR_TARGET ]
    then
        _ak_log_error "Sblock not found $CUR_TARGET"
    fi
    _ak_log_info "Checking zblocks in sblock $CUR_TARGET"
    _ak_get_zblocks_from_sblock $CUR_TARGET
    _ak_log_info "Verifying zblocks in sblock $CUR_TARGET"
    _ak_verify_zblocks_found $CUR_TARGET
    _ak_log_debug "Looking for failed ones"
    _ak_log_debug "Looking for failed ones @ `pwd` and $CUR_TARGET"
    #sleep 3
    if [ "$(_ak_sblock_show $CUR_TARGET | jq 'has("zblocks")')" == "true" ]
    then
        if [ -d $CUR_TARGET ]
        then
            find $CUR_TARGET -type f | grep failed
            if [ $? -ne 0 ]
            then
                _ak_rewards_from_sblock $CUR_TARGET
            else
                _ak_log_warning "$CUR_TARGET is invalid"
            fi
        fi
    else
        _ak_rewards_from_sblock $CUR_TARGET
    fi
}

_ak_rewards_from_sblock(){
    if [ ! -z $1 ] && [ -n "$1" ]
    then
        CUR_TARGET="$1"
    else
        _ak_log_error "No SBLOCK provided"
        exit 1
    fi
    GPG="$(_ak_sblock_show $CUR_TARGET | jq -r '.output')"
    _ak_log_debug "Got GPG : $GPG"
    if [ "$GPG" == "null" ]
    then
        GPG="$(_ak_sblock_show $CUR_TARGET | jq -r '.miner')"
    fi
    _ak_log_debug "Now got GPG : $GPG"
    AMOUNT="$(_ak_sblock_show $CUR_TARGET | jq -r '.amount')"
    if [ "$AMOUNT" == "null" ]
    then
        AMOUNT="$(_ak_sblock_show $CUR_TARGET | jq -r '.reward')"
    fi
    TIMESTAMP="$(_ak_sblock_show $CUR_TARGET | jq -r '.timestamp')"
    _ak_log_info "Found [$CUR_TARGET]($TIMESTAMP) $AMOUNT to $GPG"
    if [ -f "$AK_BALANCESDIR/$(_ak_gpg_key_get_fingerprint_from_ipfs $GPG)" ]
    then
        grep "$TIMESTAMP:$CUR_TARGET:$AMOUNT" $AK_BALANCESDIR/$(_ak_gpg_key_get_fingerprint_from_ipfs $GPG) > /dev/null 2>&1
        if [ $? -ne 0 ]
        then
            echo "$TIMESTAMP:$CUR_TARGET:$AMOUNT" >> $AK_BALANCESDIR/$(_ak_gpg_key_get_fingerprint_from_ipfs $GPG)
            _ak_log_info "Added [$CUR_TARGET]($TIMESTAMP) $AMOUNT to $GPG"
        fi
    else
        echo "$TIMESTAMP:$CUR_TARGET:$AMOUNT" >> $AK_BALANCESDIR/$(_ak_gpg_key_get_fingerprint_from_ipfs $GPG)
        _ak_log_info "Added [$CUR_TARGET]($TIMESTAMP) $AMOUNT to $GPG"
    fi
}

_ak_sblock_get_previous(){
    if [ -z $1 ] || [ ! -n "$1" ]
    then
        _ak_log_error "No sblock provided !!"
        exit 1
    fi
    CUR_TARGET="$1"
    if [ ! -f $AK_MINEDBLOCKSDIR/$CUR_TARGET ]
    then
        _ak_log_error "Sblock not found"
        exit 1
    fi
    PREVIOUS="$(_ak_sblock_show $CUR_TARGET | jq -r '.previous')"
    echo -n $PREVIOUS
}

_ak_balances_calculate(){
    if [ ! -z $1 ] && [ -n "$1" ]
    then
        _ak_log_info "Calculating balance for $1"
        CUR_TARGET="$1"
        if [ ! -f $AK_BALANCESDIR/$CUR_TARGET ]
        then
            _ak_log_warning "Address not found"
            echo -n '0' > $AK_BALANCESDIR/$CUR_TARGET.total
        else
            total=0
            # Zero cause we are now starting counting from the begging
            echo -n '0' > $AK_BALANCESDIR/$CUR_TARGET.total
            cat $AK_BALANCESDIR/$CUR_TARGET | cut -d ':' -f 3 | while read value
            do
                _ak_log_info "Found value $value for $CUR_TARGET"
                total=$(($total + $value))
                _ak_log_info "New total $total for $CUR_TARGET"
                echo -n $total > $AK_BALANCESDIR/$CUR_TARGET.total
            done
        fi
    else
        _ak_log_info "Calculating balances"
        find $AK_BALANCESDIR -type f | grep -v 'total' | while read account
        do
            _ak_balances_calculate "$(basename $account)"
        done
    fi
}

_ak_balances_print(){
    if [ ! -z $1 ] && [ -n "$1" ]
    then
        CUR_TARGET="$1"
        if [ ! -f $AK_BALANCESDIR/$CUR_TARGET.total ]
        then
            _ak_log_error "Account not found"
            exit 1
        fi
        printf '%s:%s\n' "$CUR_TARGET" "$(cat $AK_BALANCESDIR/$CUR_TARGET.total)"
    else
        find $AK_BALANCESDIR -type f | grep total | while read CUR_TARGET
        do
            printf '%s:%s\n' "$(basename $CUR_TARGET | cut -d'.' -f1)" "$(cat $CUR_TARGET)"
        done
    fi
}

_ak_schain_counting_balances(){
    if [ ! -z $1 ] && [ -n "$1" ]
    then
        CUR_TARGET="$1"
    else
        _ak_log_error "No next target found. So long for $1"
        exit 1
    fi
    _ak_log_info "Looking at $CUR_TARGET"
    counter=$(($counter + 1))
    if [ "$1" == "$GENESIS" ]
    then
        _ak_balances_calculate
        _ak_balances_print
        _ak_log_warning "Looks like genesis. Exiting with 0"
        exit 0
    fi
    if [ ! -f $AK_MINEDBLOCKSDIR/$CUR_TARGET ]
    then
        _ak_log_error "Could not find $CUR_TARGET"
        exit 1
    else
        _ak_balances_from_sblock $CUR_TARGET
        NEXT_TARGET="$(_ak_sblock_get_previous $CUR_TARGET)"
        # mv $AK_MINEDBLOCKSDIR/$CUR_TARGET $AK_INVALIDATED_SBLOCKS/$CUR_TARGET
        _ak_log_info "Found previous: $NEXT_TARGET"
        _ak_schain_counting_balances "$NEXT_TARGET"
    fi
}

if [ ! -z $1 ] && [ -n "$1" ]
then
    _ak_usage
else
    CUR_TARGET="$LATEST"
    _ak_log_info "$CUR_TARGET $LATEST"
    _ak_schain_counting_balances $CUR_TARGET
fi