withdraw.sh

#!/bin/bash

# Function to display the formatted wallet addresses with indexed options
function display_wallets() {
    echo "Available wallets:"
    echo "---------------------------------"
    printf "%-5s %-15s %-45s\n" "No" "Name" "Address"
    echo "---------------------------------"
    local index=1
    for row in $(jq -c '.[]' wallets.json); do
        name=$(echo "$row" | jq -r '.name')
        address=$(echo "$row" | jq -r '.address')
        printf "%-5s %-15s %-45s\n" "$index" "$name" "$address"
        index=$((index + 1))
    done
    echo "---------------------------------"
}

# Function to display available stake accounts
function display_stake_accounts() {
    echo "Available Stake Accounts:"
    echo "---------------------------------"
    printf "%-5s %-9s %-45s %-20s\n" "No" "Name" "Address" "Unstaked Balance"
    echo "---------------------------------"
    
    # Display stake accounts without sorting
    local index=1
    for row in $(jq -c '.[]' allstakes.json); do
        name=$(echo "$row" | jq -r '.name')
        address=$(echo "$row" | jq -r '.address')
        
        # Get balance and active stake
        output=$(solana stake-account "$address")
        balance=$(echo "$output" | grep "Balance:" | awk '{print $2}')
        active_stake=$(echo "$output" | grep "Active Stake:" | awk '{print $3}')

        # Format the balance and active stake to avoid leading zeros
        balance=$(printf "%.8f" "$balance")
        active_stake=$(printf "%.8f" "$active_stake")

        # Calculate unstaked balance
        if [[ -z "$active_stake" || "$active_stake" == "0" ]]; then
            unstaked_balance="$balance"
        else
            unstaked_balance=$(echo "$balance - $active_stake" | bc)
        fi

        # Ensure that unstaked balance is formatted properly
        if [[ "$unstaked_balance" == .* ]]; then
            unstaked_balance="0$unstaked_balance"
        fi

        # Print the output row
        printf "%-5s %-9s %-45s %-20s\n" "$index" "$name" "$address" "$unstaked_balance"
        index=$((index + 1))
    done
    echo "---------------------------------"
}

# Main menu loop
while true; do
    # Prompt the user for an action
    echo "What would you like to do?"
    echo "1. Withdraw Stake"
    echo "2. Withdraw from Vote account"
    echo "3. Exit Withdrawal"
    read -p "Please select an option (1, 2, or 3): " option

    if [[ "$option" -eq 1 ]]; then
        # Display available stake accounts
        display_stake_accounts

        # Ask the user which stake account to withdraw from
        read -p "Choose the stake account number to withdraw from (1-5): " choice
        
        # Get the address and name of the selected stake account
        stake_address=$(jq -r ".[$((choice - 1))].address" allstakes.json)
        stake_name=$(jq -r ".[$((choice - 1))].name" allstakes.json)

        # Display wallets to withdraw to
        display_wallets

        # Ask the user which wallet they want to withdraw to
        read -p "Choose the wallet number to withdraw to (1-4): " wallet_choice
        withdraw_to_address=$(jq -r ".[$((wallet_choice - 1))].address" wallets.json)

        # Now retrieving the stake account details for the selected account
        echo "Retrieving details for the selected stake account \"$stake_name\"..."
        output=$(solana stake-account "$stake_address")
        
        # Ensure the output retrieval is correct
        if [[ "$output" == *"Error"* ]]; then
            echo "Failed to retrieve details for the stake account. Please check the address."
            continue
        fi
        
        balance=$(echo "$output" | grep "Balance:" | awk '{print $2}')
        active_stake=$(echo "$output" | grep "Active Stake:" | awk '{print $3}')

        # Format balance and active stake
        balance=$(printf "%.8f" "$balance")
        active_stake=$(printf "%.8f" "$active_stake")

        # Calculate unstaked balance
        if [[ -z "$active_stake" || "$active_stake" == "0" ]]; then
            unstaked_balance="$balance"
        else
            unstaked_balance=$(echo "$balance - $active_stake" | bc)
        fi
        
        # Ensure unstaked balance is properly formatted
        if [[ "$unstaked_balance" == .* ]]; then
            unstaked_balance="0$unstaked_balance"
        fi

        # Print the outputs in the requested format
        echo "---------------------------------"
        echo "Stake Account Name: $stake_name"
        echo "Address: $stake_address"
        echo "Balance: $balance"
        echo "Active Stake: $active_stake"
        echo "Unstaked Balance: $unstaked_balance"
        echo "---------------------------------"

        # Loop for withdrawal amount
        while true; do
            read -p "How much unstaked balance would you like to withdraw (0 - $unstaked_balance)? " withdraw_amount
            
            # Check if the user wants to cancel
            if [ -z "$withdraw_amount" ]; then
                echo "Withdrawal canceled."
                break
            fi
            
            # Validate the amount (must be a number)
            if ! [[ "$withdraw_amount" =~ ^[0-9]+([.][0-9]+)?$ ]]; then
                echo "Incorrect value entered, returning to menu."
                break
            fi
            
            # Validate the amount is within the unstaked balance range
            if (( $(echo "$withdraw_amount <= $unstaked_balance" | bc -l) && $(echo "$withdraw_amount >= 0" | bc -l) )); then
                # Withdraw funds
                solana withdraw-stake "$stake_address" "$withdraw_to_address" "$withdraw_amount"
                echo "Withdrawal of $withdraw_amount SOL to $withdraw_to_address initiated."
                break
            else
                echo "Invalid withdrawal amount. Please try again."
                echo "---------------------------------"
            fi
        done

    elif [[ "$option" -eq 2 ]]; then
        # Display wallets for Vote withdrawal
        display_wallets

        # Ask the user which address they want to withdraw Vote funds to
        read -p "Choose the wallet number to withdraw Vote funds to (1-4): " choice
        withdraw_to_address=$(jq -r ".[$((choice - 1))].address" wallets.json)

        # Extract the vote address
        vote_address=$(jq -r '.[] | select(.name=="Vote") | .address' wallets.json)

        # Get the balance of the Vote account
        vote_balance_output=$(solana balance "$vote_address")
        vote_balance=$(echo "$vote_balance_output" | awk '{print $1}')
        echo "---------------------------------"
        echo "Vote Account Balance: $vote_balance SOL"
        echo "---------------------------------"

        # Prompt for withdrawal amount
        while true; do
            read -p "How much funds would you like to withdraw from the Vote account (0 - $vote_balance)? " withdraw_amount
            
            # Check if the user wants to cancel
            if [ -z "$withdraw_amount" ]; then
                echo "Withdrawal canceled."
                break
            fi
            
            # Validate the amount (must be a number)
            if ! [[ "$withdraw_amount" =~ ^[0-9]+([.][0-9]+)?$ ]]; then
                echo "Incorrect value entered, returning to menu."
                break
            fi
            
            # Validate the amount is within the vote balance range
            if (( $(echo "$withdraw_amount <= $vote_balance" | bc -l) && $(echo "$withdraw_amount >= 0" | bc -l) )); then
                # Withdraw funds from Vote account
                solana withdraw-from-vote-account "$vote_address" "$withdraw_to_address" "$withdraw_amount"
                echo "Withdrawal of $withdraw_amount SOL from Vote account to $withdraw_to_address initiated."
                break
            else
                echo "Invalid withdrawal amount. Please try again."
                echo "---------------------------------"
            fi
        done

    elif [[ "$option" -eq 3 ]]; then
        echo "Exiting withdrawal process."
        exit 0
    else
        echo "Invalid option selected."
    fi
done