mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-20 10:34:20 +01:00
124 lines
4.5 KiB
Python
124 lines
4.5 KiB
Python
import os
|
|
|
|
import click
|
|
|
|
from cashu.core.base import Proof, WalletKeyset
|
|
from cashu.core.settings import CASHU_DIR
|
|
from cashu.wallet.crud import get_keyset
|
|
from cashu.wallet.wallet import Wallet as Wallet
|
|
|
|
|
|
async def verify_mints(ctx, dtoken):
|
|
trust_token_mints = True
|
|
for mint_id in dtoken.get("mints"):
|
|
for keyset in set(dtoken["mints"][mint_id]["ks"]):
|
|
mint_url = dtoken["mints"][mint_id]["url"]
|
|
# init a temporary wallet object
|
|
keyset_wallet = Wallet(
|
|
mint_url, os.path.join(CASHU_DIR, ctx.obj["WALLET_NAME"])
|
|
)
|
|
# make sure that this mint supports this keyset
|
|
mint_keysets = await keyset_wallet._get_keysets(mint_url)
|
|
assert keyset in mint_keysets["keysets"], "mint does not have this keyset."
|
|
|
|
# we validate the keyset id by fetching the keys from the mint
|
|
mint_keyset = await keyset_wallet._get_keyset(mint_url, keyset)
|
|
assert keyset == mint_keyset.id, Exception("keyset not valid.")
|
|
|
|
# we check the db whether we know this mint already and ask the user if not
|
|
mint_keysets = await get_keyset(mint_url=mint_url, db=keyset_wallet.db)
|
|
if mint_keysets is None:
|
|
# we encountered a new mint and ask for a user confirmation
|
|
trust_token_mints = False
|
|
print("")
|
|
print("Warning: Tokens are from a mint you don't know yet.")
|
|
print("\n")
|
|
print(f"Mint URL: {mint_url}")
|
|
print(f"Mint keyset: {keyset}")
|
|
print("\n")
|
|
click.confirm(
|
|
f"Do you trust this mint and want to receive the tokens?",
|
|
abort=True,
|
|
default=True,
|
|
)
|
|
trust_token_mints = True
|
|
|
|
assert trust_token_mints, Exception("Aborted!")
|
|
|
|
|
|
async def redeem_multimint(ctx, dtoken, script, signature):
|
|
# we get the mint information in the token and load the keys of each mint
|
|
# we then redeem the tokens for each keyset individually
|
|
for mint_id in dtoken.get("mints"):
|
|
for keyset in set(dtoken["mints"][mint_id]["ks"]):
|
|
mint_url = dtoken["mints"][mint_id]["url"]
|
|
# init a temporary wallet object
|
|
keyset_wallet = Wallet(
|
|
mint_url, os.path.join(CASHU_DIR, ctx.obj["WALLET_NAME"])
|
|
)
|
|
|
|
# load the keys
|
|
await keyset_wallet.load_mint(keyset_id=keyset)
|
|
|
|
# redeem proofs of this keyset
|
|
redeem_proofs = [
|
|
Proof(**p) for p in dtoken["tokens"] if Proof(**p).id == keyset
|
|
]
|
|
_, _ = await keyset_wallet.redeem(
|
|
redeem_proofs, scnd_script=script, scnd_siganture=signature
|
|
)
|
|
|
|
|
|
async def print_mint_balances(ctx, wallet, show_mints=False):
|
|
# get balances per mint
|
|
mint_balances = await wallet.balance_per_minturl()
|
|
|
|
# if we have a balance on a non-default mint, we show its URL
|
|
keysets = [k for k, v in wallet.balance_per_keyset().items()]
|
|
for k in keysets:
|
|
ks = await get_keyset(id=str(k), db=wallet.db)
|
|
if ks and ks.mint_url != ctx.obj["HOST"]:
|
|
show_mints = True
|
|
|
|
# or we have a balance on more than one mint
|
|
# show balances per mint
|
|
if len(mint_balances) > 1 or show_mints:
|
|
print(f"You have balances in {len(mint_balances)} mints:")
|
|
print("")
|
|
for i, (k, v) in enumerate(mint_balances.items()):
|
|
print(
|
|
f"Mint {i+1}: Balance: {v['available']} sat (pending: {v['balance']-v['available']} sat) URL: {k}"
|
|
)
|
|
print("")
|
|
|
|
|
|
async def get_mint_wallet(ctx):
|
|
wallet: Wallet = ctx.obj["WALLET"]
|
|
await wallet.load_mint()
|
|
|
|
mint_balances = await wallet.balance_per_minturl()
|
|
# if there is only one mint, use it
|
|
if len(mint_balances) == 1:
|
|
return wallet
|
|
|
|
await print_mint_balances(ctx, wallet, show_mints=True)
|
|
|
|
mint_nr = (
|
|
input(f"Select mint [1-{len(mint_balances)}, press enter for default 1]: ")
|
|
or "1"
|
|
)
|
|
if not mint_nr.isdigit():
|
|
raise Exception("invalid input.")
|
|
mint_nr = int(mint_nr)
|
|
|
|
mint_url = list(mint_balances.keys())[mint_nr - 1]
|
|
|
|
# load this mint_url into a wallet
|
|
mint_wallet = Wallet(mint_url, os.path.join(CASHU_DIR, ctx.obj["WALLET_NAME"]))
|
|
mint_keysets: WalletKeyset = await get_keyset(mint_url=mint_url, db=mint_wallet.db) # type: ignore
|
|
|
|
# load the keys
|
|
await mint_wallet.load_mint(keyset_id=mint_keysets.id)
|
|
|
|
return mint_wallet
|