#!/usr/bin/env python3

import argparse
import os
import sys

sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
import entropy  # nopep8

parser = argparse.ArgumentParser(
    description="Shows all Wrapped SOL accounts for the wallet."
)
entropy.ContextBuilder.add_command_line_parameters(parser)
entropy.Wallet.add_command_line_parameters(parser)
args: argparse.Namespace = entropy.parse_args(parser)

with entropy.ContextBuilder.from_command_line_parameters(args) as context:
    wallet = entropy.Wallet.from_command_line_parameters_or_raise(args)
    wrapped_sol: entropy.Token = entropy.token(context, "SOL")
    token_accounts = entropy.TokenAccount.fetch_all_for_owner_and_token(
        context, wallet.address, wrapped_sol
    )

    if len(token_accounts) == 0:
        entropy.output("No wrapped SOL accounts.")
    else:
        entropy.output(f"{wrapped_sol.name}:")
        for account in token_accounts:
            entropy.output(f"    {account.address}: {account.value}")
