refactor

cashu/wallet/cli_helpers.py

@@ -2,10 +2,14 @@ import os
 
 import click
 
-from cashu.core.base import Proof, WalletKeyset
+from cashu.core.base import Proof, WalletKeyset, TokenJson, TokenMintJson
 from cashu.core.settings import CASHU_DIR
 from cashu.wallet.crud import get_keyset
 from cashu.wallet.wallet import Wallet as Wallet
+import urllib.parse
+from cashu.core.settings import (
+    MINT_URL,
+)
 
 
 async def verify_mints(ctx, dtoken):
@@ -97,19 +101,19 @@ async def get_mint_wallet(ctx):
     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)
+    if len(mint_balances) > 1:
+        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_nr_str = (
+            input(f"Select mint [1-{len(mint_balances)}, press enter for default 1]: ")
+            or "1"
+        )
+        if not mint_nr_str.isdigit():
+            raise Exception("invalid input.")
+        mint_nr = int(mint_nr_str)
+    else:
+        mint_nr = 1
 
     mint_url = list(mint_balances.keys())[mint_nr - 1]
 
@@ -121,3 +125,51 @@ async def get_mint_wallet(ctx):
     await mint_wallet.load_mint(keyset_id=mint_keysets.id)
 
     return mint_wallet
+
+
+# LNbits token link parsing
+# can extract minut URL from LNbits token links like:
+# https://lnbits.server/cashu/wallet?mint_id=aMintId&recv_token=W3siaWQiOiJHY2...
+def token_from_lnbits_link(link):
+    url, token = "", ""
+    if len(link.split("&recv_token=")) == 2:
+        # extract URL params
+        params = urllib.parse.parse_qs(link.split("?")[1])
+        # extract URL
+        if "mint_id" in params:
+            url = (
+                link.split("?")[0].split("/wallet")[0]
+                + "/api/v1/"
+                + params["mint_id"][0]
+            )
+        # extract token
+        token = params["recv_token"][0]
+        return token, url
+    else:
+        return link, ""
+
+
+async def proofs_to_token(wallet, proofs, url: str):
+    """
+    Ingests proofs and
+    """
+    # and add url and keyset id to token
+    token: TokenJson = await wallet._make_token(proofs, include_mints=False)
+    token.mints = {}
+
+    # get keysets of proofs
+    keysets = list(set([p.id for p in proofs]))
+    assert keysets is not None, "no keysets"
+
+    # check whether we know the mint urls for these proofs
+    for k in keysets:
+        ks = await get_keyset(id=k, db=wallet.db)
+        url = ks.mint_url if ks is not None else None
+
+    url = url or (
+        input(f"Enter mint URL (press enter for default {MINT_URL}): ") or MINT_URL
+    )
+
+    token.mints[url] = TokenMintJson(url=url, ks=keysets)  # type: ignore
+    token_serialized = await wallet._serialize_token_base64(token)
+    return token_serialized