99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
import asyncio
|
|
import json
|
|
|
|
from lnbits.core.models import Payment
|
|
from lnbits.core.services import websocket_updater
|
|
from lnbits.tasks import register_invoice_listener
|
|
from loguru import logger
|
|
|
|
from .crud import create_account, get_account, update_account
|
|
from .models import NostrAccount
|
|
|
|
|
|
async def wait_for_paid_invoices():
|
|
invoice_queue = asyncio.Queue()
|
|
register_invoice_listener(invoice_queue, "ext_nostrrelay")
|
|
|
|
while True:
|
|
payment = await invoice_queue.get()
|
|
await on_invoice_paid(payment)
|
|
|
|
|
|
async def on_invoice_paid(payment: Payment):
|
|
if payment.extra.get("tag") != "nostrrely":
|
|
return
|
|
|
|
relay_id = payment.extra.get("relay_id")
|
|
pubkey = payment.extra.get("pubkey")
|
|
payment_hash = payment.payment_hash
|
|
|
|
if not relay_id or not pubkey:
|
|
message = (
|
|
"Invoice extra data missing for 'relay_id' and 'pubkey'. "
|
|
f"Payment hash: {payment_hash}"
|
|
)
|
|
logger.warning(message)
|
|
await websocket_updater(
|
|
payment_hash, json.dumps({"success": False, "message": message})
|
|
)
|
|
return
|
|
|
|
action = payment.extra.get("action")
|
|
if action == "join":
|
|
await invoice_paid_to_join(relay_id, pubkey, payment.amount)
|
|
await websocket_updater(payment_hash, json.dumps({"success": True}))
|
|
return
|
|
|
|
if action == "storage":
|
|
storage_to_buy = payment.extra.get("storage_to_buy")
|
|
if not storage_to_buy:
|
|
message = (
|
|
"Invoice extra data missing for 'storage_to_buy'. "
|
|
f"Payment hash: {payment_hash}"
|
|
)
|
|
logger.warning(message)
|
|
return
|
|
await invoice_paid_for_storage(relay_id, pubkey, storage_to_buy, payment.amount)
|
|
await websocket_updater(payment_hash, json.dumps({"success": True}))
|
|
return
|
|
|
|
await websocket_updater(
|
|
payment_hash,
|
|
json.dumps({"success": False, "message": f"Bad action name: '{action}'"}),
|
|
)
|
|
|
|
|
|
async def invoice_paid_to_join(relay_id: str, pubkey: str, amount: int):
|
|
account = await get_account(relay_id, pubkey)
|
|
if not account:
|
|
account = NostrAccount(
|
|
relay_id=relay_id, pubkey=pubkey, paid_to_join=True, sats=amount
|
|
)
|
|
await create_account(account)
|
|
return
|
|
|
|
if account.blocked or account.paid_to_join:
|
|
return
|
|
|
|
account.paid_to_join = True
|
|
account.sats += amount
|
|
await update_account(account)
|
|
|
|
|
|
async def invoice_paid_for_storage(
|
|
relay_id: str, pubkey: str, storage_to_buy: int, amount: int
|
|
):
|
|
account = await get_account(relay_id, pubkey)
|
|
if not account:
|
|
account = NostrAccount(
|
|
relay_id=relay_id, pubkey=pubkey, storage=storage_to_buy, sats=amount
|
|
)
|
|
await create_account(account)
|
|
return
|
|
|
|
if account.blocked:
|
|
return
|
|
|
|
account.storage = storage_to_buy
|
|
account.sats += amount
|
|
await update_account(account)
|