nutshell/cashu/mint/crud.py

import time
from typing import Any, List, Optional

from ..core.base import BlindedSignature, Invoice, MintKeyset, Proof
from ..core.db import Connection, Database, table_with_schema


class LedgerCrud:
    """
    Database interface for Cashu mint.

    This class needs to be overloaded by any app that imports the Cashu mint.
    """

    async def get_keyset(*args, **kwags):
        return await get_keyset(*args, **kwags)  # type: ignore

    async def get_lightning_invoice(*args, **kwags):
        return await get_lightning_invoice(*args, **kwags)  # type: ignore

    async def get_proofs_used(*args, **kwags):
        return await get_proofs_used(*args, **kwags)  # type: ignore

    async def invalidate_proof(*args, **kwags):
        return await invalidate_proof(*args, **kwags)  # type: ignore

    async def get_proofs_pending(*args, **kwags):
        return await get_proofs_pending(*args, **kwags)  # type: ignore

    async def set_proof_pending(*args, **kwags):
        return await set_proof_pending(*args, **kwags)  # type: ignore

    async def unset_proof_pending(*args, **kwags):
        return await unset_proof_pending(*args, **kwags)  # type: ignore

    async def store_keyset(*args, **kwags):
        return await store_keyset(*args, **kwags)  # type: ignore

    async def store_lightning_invoice(*args, **kwags):
        return await store_lightning_invoice(*args, **kwags)  # type: ignore

    async def store_promise(*args, **kwags):
        return await store_promise(*args, **kwags)  # type: ignore

    async def get_promise(*args, **kwags):
        return await get_promise(*args, **kwags)  # type: ignore

    async def update_lightning_invoice(*args, **kwags):
        return await update_lightning_invoice(*args, **kwags)  # type: ignore


async def store_promise(
    db: Database,
    amount: int,
    B_: str,
    C_: str,
    id: str,
    conn: Optional[Connection] = None,
):
    await (conn or db).execute(
        f"""
        INSERT INTO {table_with_schema(db, 'promises')}
          (amount, B_b, C_b, id)
        VALUES (?, ?, ?, ?)
        """,
        (
            amount,
            str(B_),
            str(C_),
            id,
        ),
    )


async def get_promise(
    db: Database,
    B_: str,
    conn: Optional[Connection] = None,
):
    row = await (conn or db).fetchone(
        f"""
        SELECT * from {table_with_schema(db, 'promises')}
        WHERE B_b = ?
        """,
        (str(B_),),
    )
    return BlindedSignature(amount=row[0], C_=row[2], id=row[3]) if row else None


async def get_proofs_used(
    db: Database,
    conn: Optional[Connection] = None,
):
    rows = await (conn or db).fetchall(
        f"""
        SELECT secret from {table_with_schema(db, 'proofs_used')}
        """
    )
    return [row[0] for row in rows]


async def invalidate_proof(
    db: Database,
    proof: Proof,
    conn: Optional[Connection] = None,
):
    # we add the proof and secret to the used list
    await (conn or db).execute(
        f"""
        INSERT INTO {table_with_schema(db, 'proofs_used')}
          (amount, C, secret, id)
        VALUES (?, ?, ?, ?)
        """,
        (
            proof.amount,
            str(proof.C),
            str(proof.secret),
            str(proof.id),
        ),
    )


async def get_proofs_pending(
    db: Database,
    conn: Optional[Connection] = None,
):
    rows = await (conn or db).fetchall(
        f"""
        SELECT * from {table_with_schema(db, 'proofs_pending')}
        """
    )
    return [Proof(**r) for r in rows]


async def set_proof_pending(
    db: Database,
    proof: Proof,
    conn: Optional[Connection] = None,
):
    # we add the proof and secret to the used list
    await (conn or db).execute(
        f"""
        INSERT INTO {table_with_schema(db, 'proofs_pending')}
          (amount, C, secret)
        VALUES (?, ?, ?)
        """,
        (
            proof.amount,
            str(proof.C),
            str(proof.secret),
        ),
    )


async def unset_proof_pending(
    proof: Proof,
    db: Database,
    conn: Optional[Connection] = None,
):
    await (conn or db).execute(
        f"""
        DELETE FROM {table_with_schema(db, 'proofs_pending')}
        WHERE secret = ?
        """,
        (str(proof["secret"]),),
    )


async def store_lightning_invoice(
    db: Database,
    invoice: Invoice,
    conn: Optional[Connection] = None,
):
    await (conn or db).execute(
        f"""
        INSERT INTO {table_with_schema(db, 'invoices')}
          (amount, pr, hash, issued, payment_hash)
        VALUES (?, ?, ?, ?, ?)
        """,
        (
            invoice.amount,
            invoice.pr,
            invoice.hash,
            invoice.issued,
            invoice.payment_hash,
        ),
    )


async def get_lightning_invoice(
    db: Database,
    hash: str,
    conn: Optional[Connection] = None,
):
    row = await (conn or db).fetchone(
        f"""
        SELECT * from {table_with_schema(db, 'invoices')}
        WHERE hash = ?
        """,
        (hash,),
    )

    return Invoice(**row) if row else None


async def update_lightning_invoice(
    db: Database,
    hash: str,
    issued: bool,
    conn: Optional[Connection] = None,
):
    await (conn or db).execute(
        f"UPDATE {table_with_schema(db, 'invoices')} SET issued = ? WHERE hash = ?",
        (
            issued,
            hash,
        ),
    )


async def store_keyset(
    db: Database,
    keyset: MintKeyset,
    conn: Optional[Connection] = None,
):
    await (conn or db).execute(  # type: ignore
        f"""
        INSERT INTO {table_with_schema(db, 'keysets')}
          (id, derivation_path, valid_from, valid_to, first_seen, active, version)
        VALUES (?, ?, ?, ?, ?, ?, ?)
        """,
        (
            keyset.id,
            keyset.derivation_path,
            keyset.valid_from or db.timestamp_now,
            keyset.valid_to or db.timestamp_now,
            keyset.first_seen or db.timestamp_now,
            True,
            keyset.version,
        ),
    )


async def get_keyset(
    db: Database,
    id: str = "",
    derivation_path: str = "",
    conn: Optional[Connection] = None,
):
    clauses = []
    values: List[Any] = []
    clauses.append("active = ?")
    values.append(True)
    if id:
        clauses.append("id = ?")
        values.append(id)
    if derivation_path:
        clauses.append("derivation_path = ?")
        values.append(derivation_path)
    where = ""
    if clauses:
        where = f"WHERE {' AND '.join(clauses)}"

    rows = await (conn or db).fetchall(  # type: ignore
        f"""
        SELECT * from {table_with_schema(db, 'keysets')}
        {where}
        """,
        tuple(values),
    )
    return [MintKeyset(**row) for row in rows]