138 lines
4.8 KiB
Bash
Executable File
138 lines
4.8 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# shellcheck disable=SC1090
|
|
# shellcheck disable=SC1091
|
|
source "$(dirname "$0")"/get_program_accounts.sh
|
|
|
|
usage() {
|
|
exitcode=0
|
|
if [[ -n "$1" ]]; then
|
|
exitcode=1
|
|
echo "Error: $*"
|
|
fi
|
|
cat <<EOF
|
|
usage: $0 [cluster_rpc_url] [identity_pubkey]
|
|
|
|
Report the account addresses, balances and lockups of all stake accounts
|
|
for which a given key is the authorized staker.
|
|
Also report the system account balance for that same key.
|
|
|
|
Required arguments:
|
|
cluster_rpc_url - RPC URL and port for a running Solana cluster (ex: http://34.83.146.144:8899)
|
|
identity_pubkey - Base58 pubkey that is an authorized staker for at least one stake account on the cluster.
|
|
EOF
|
|
exit $exitcode
|
|
}
|
|
|
|
function parse_stake_account_data_to_file {
|
|
account_key=$(echo "$1" | tr -d '"')
|
|
filter_key="$2"
|
|
csvfile="$3"
|
|
|
|
account_data="$(solana --url "$url" show-stake-account "$account_key")"
|
|
staker="$(echo "$account_data" | grep -i 'authorized staker' | cut -f3 -d " ")"
|
|
lockup_epoch="$(echo "$account_data" | grep -i 'lockup epoch' | cut -f3 -d " ")"
|
|
if [[ "$staker" == "$filter_key" ]] ; then
|
|
echo STAKE,"$account_key","$lamports","$lockup_epoch" >> "$csvfile"
|
|
fi
|
|
}
|
|
|
|
function display_results_summary {
|
|
stake_account_balance_total=0
|
|
num_stake_accounts=0
|
|
{
|
|
read -r
|
|
while IFS=, read -r program account_pubkey lamports lockup_epoch; do
|
|
case $program in
|
|
SYSTEM)
|
|
system_account_balance=$lamports
|
|
;;
|
|
STAKE)
|
|
stake_account_balance_total=$((stake_account_balance_total + lamports))
|
|
num_stake_accounts=$((num_stake_accounts + 1))
|
|
;;
|
|
*)
|
|
echo "Unknown program: $program"
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
} < "$results_file"
|
|
|
|
stake_account_balance_total_sol="$(bc <<< "scale=3; $stake_account_balance_total/$LAMPORTS_PER_SOL")"
|
|
system_account_balance_sol="$(bc <<< "scale=3; $system_account_balance/$LAMPORTS_PER_SOL")"
|
|
|
|
all_account_total_balance="$(bc <<< "scale=3; $system_account_balance+$stake_account_balance_total")"
|
|
all_account_total_balance_sol="$(bc <<< "scale=3; ($system_account_balance+$stake_account_balance_total)/$LAMPORTS_PER_SOL")"
|
|
|
|
echo "--------------------------------------------------------------------------------------"
|
|
echo "Results written to: $results_file"
|
|
echo "--------------------------------------------------------------------------------------"
|
|
echo "Summary of accounts owned by $filter_pubkey"
|
|
echo ""
|
|
printf "Number of STAKE accounts: %'d\n" "$num_stake_accounts"
|
|
printf "Balance of all STAKE accounts: %'d lamports\n" "$stake_account_balance_total"
|
|
printf "Balance of all STAKE accounts: %'.3f SOL\n" "$stake_account_balance_total_sol"
|
|
printf "\n"
|
|
printf "Balance of SYSTEM account: %'d lamports\n" "$system_account_balance"
|
|
printf "Balance of SYSTEM account: %'.3f SOL\n" "$system_account_balance_sol"
|
|
printf "\n"
|
|
printf "Total Balance of ALL accounts: %'d lamports\n" "$all_account_total_balance"
|
|
printf "Total Balance of ALL accounts: %'.3f SOL\n" "$all_account_total_balance_sol"
|
|
echo "--------------------------------------------------------------------------------------"
|
|
}
|
|
|
|
function display_results_details {
|
|
# shellcheck disable=SC2002
|
|
cat "$results_file" | column -t -s,
|
|
}
|
|
|
|
LAMPORTS_PER_SOL=1000000000 # 1 billion
|
|
all_stake_accounts_json_file=all_stake_accounts_data.json
|
|
all_stake_accounts_csv_file=all_stake_accounts_data.csv
|
|
|
|
url=$1
|
|
[[ -n $url ]] || usage "Missing required RPC URL"
|
|
shift
|
|
filter_pubkey=$1
|
|
[[ -n $filter_pubkey ]] || usage "Missing required pubkey"
|
|
shift
|
|
|
|
results_file=accounts_owned_by_${filter_pubkey}.csv
|
|
system_account_json_file=system_account_${filter_pubkey}.json
|
|
|
|
echo "Program,Account_Pubkey,Lamports,Lockup_Epoch" > "$results_file"
|
|
|
|
echo "Getting system account data"
|
|
get_account_info "$filter_pubkey" "$url" "$system_account_json_file"
|
|
# shellcheck disable=SC2002
|
|
system_account_balance="$(cat "$system_account_json_file" | jq -r '(.result | .value | .lamports)')"
|
|
if [[ "$system_account_balance" == "null" ]]; then
|
|
echo "The provided pubkey is not found in the system program: $filter_pubkey"
|
|
exit 1
|
|
fi
|
|
echo SYSTEM,"$filter_pubkey","$system_account_balance",N/A >> "$results_file"
|
|
|
|
echo "Getting all stake program accounts"
|
|
get_program_accounts STAKE "$STAKE_PROGRAM_PUBKEY" "$url" "$all_stake_accounts_json_file"
|
|
write_program_account_data_csv STAKE "$all_stake_accounts_json_file" "$all_stake_accounts_csv_file"
|
|
|
|
echo "Querying cluster at $url for stake accounts with authorized staker: $filter_pubkey"
|
|
last_tick=$SECONDS
|
|
{
|
|
read -r
|
|
while IFS=, read -r account_pubkey lamports; do
|
|
parse_stake_account_data_to_file "$account_pubkey" "$filter_pubkey" "$results_file" &
|
|
sleep 0.01
|
|
if [[ $((SECONDS - last_tick)) == 1 ]]; then
|
|
last_tick=$SECONDS
|
|
printf "."
|
|
fi
|
|
done
|
|
} < "$all_stake_accounts_csv_file"
|
|
wait
|
|
printf "\n"
|
|
|
|
display_results_details
|
|
display_results_summary
|