mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-20 10:34:20 +01:00
116 lines
2.1 KiB
Python
116 lines
2.1 KiB
Python
import secrets
|
|
from typing import Optional
|
|
|
|
from core.base import Invoice, Proof
|
|
from core.db import Connection, Database
|
|
|
|
|
|
async def store_promise(
|
|
amount: int,
|
|
B_x: str,
|
|
B_y: str,
|
|
C_x: str,
|
|
C_y: str,
|
|
db: Database,
|
|
conn: Optional[Connection] = None,
|
|
):
|
|
|
|
await (conn or db).execute(
|
|
"""
|
|
INSERT INTO promises
|
|
(amount, B_x, B_y, C_x, C_y)
|
|
VALUES (?, ?, ?, ?, ?)
|
|
""",
|
|
(
|
|
amount,
|
|
str(B_x),
|
|
str(B_y),
|
|
str(C_x),
|
|
str(C_y),
|
|
),
|
|
)
|
|
|
|
|
|
async def get_proofs_used(
|
|
db: Database,
|
|
conn: Optional[Connection] = None,
|
|
):
|
|
|
|
rows = await (conn or db).fetchall(
|
|
"""
|
|
SELECT secret from proofs_used
|
|
"""
|
|
)
|
|
return [row[0] for row in rows]
|
|
|
|
|
|
async def invalidate_proof(
|
|
proof: Proof,
|
|
db: Database,
|
|
conn: Optional[Connection] = None,
|
|
):
|
|
|
|
# we add the proof and secret to the used list
|
|
await (conn or db).execute(
|
|
"""
|
|
INSERT INTO proofs_used
|
|
(amount, C_x, C_y, secret)
|
|
VALUES (?, ?, ?, ?)
|
|
""",
|
|
(
|
|
proof.amount,
|
|
str(proof.C.x),
|
|
str(proof.C.y),
|
|
str(proof.secret),
|
|
),
|
|
)
|
|
|
|
|
|
async def store_lightning_invoice(
|
|
invoice: Invoice,
|
|
db: Database,
|
|
conn: Optional[Connection] = None,
|
|
):
|
|
|
|
await (conn or db).execute(
|
|
"""
|
|
INSERT INTO invoices
|
|
(amount, pr, hash, issued)
|
|
VALUES (?, ?, ?, ?)
|
|
""",
|
|
(
|
|
invoice.amount,
|
|
invoice.pr,
|
|
invoice.hash,
|
|
invoice.issued,
|
|
),
|
|
)
|
|
|
|
|
|
async def get_lightning_invoice(
|
|
hash: str,
|
|
db: Database,
|
|
conn: Optional[Connection] = None,
|
|
):
|
|
|
|
row = await (conn or db).fetchone(
|
|
"""
|
|
SELECT * from invoices
|
|
WHERE hash = ?
|
|
""",
|
|
hash,
|
|
)
|
|
return Invoice.from_row(row)
|
|
|
|
|
|
async def update_lightning_invoice(
|
|
hash: str,
|
|
issued: bool,
|
|
db: Database,
|
|
conn: Optional[Connection] = None,
|
|
):
|
|
await (conn or db).execute(
|
|
"UPDATE invoices SET issued = ? WHERE hash = ?",
|
|
(issued, hash),
|
|
)
|