74 lines
2 KiB
Python
74 lines
2 KiB
Python
import asyncio
|
|
|
|
from loguru import logger
|
|
|
|
from lnbits.core.models import Payment
|
|
from lnbits.helpers import get_current_extension_name
|
|
from lnbits.tasks import register_invoice_listener
|
|
|
|
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, get_current_extension_name())
|
|
|
|
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")
|
|
|
|
if payment.extra.get("action") == "join":
|
|
await invoice_paid_to_join(relay_id, pubkey)
|
|
return
|
|
|
|
if payment.extra.get("action") == "storage":
|
|
storage_to_buy = payment.extra.get("storage_to_buy")
|
|
await invoice_paid_for_storage(relay_id, pubkey, storage_to_buy)
|
|
return
|
|
|
|
|
|
async def invoice_paid_to_join(relay_id: str, pubkey: str):
|
|
try:
|
|
account = await get_account(relay_id, pubkey)
|
|
if not account:
|
|
await create_account(
|
|
relay_id, NostrAccount(pubkey=pubkey, paid_to_join=True)
|
|
)
|
|
return
|
|
|
|
if account.blocked or account.paid_to_join:
|
|
return
|
|
|
|
account.paid_to_join = True
|
|
await update_account(relay_id, account)
|
|
|
|
except Exception as ex:
|
|
logger.warning(ex)
|
|
|
|
|
|
async def invoice_paid_for_storage(relay_id: str, pubkey: str, storage_to_buy: int):
|
|
try:
|
|
account = await get_account(relay_id, pubkey)
|
|
if not account:
|
|
await create_account(
|
|
relay_id, NostrAccount(pubkey=pubkey, storage=storage_to_buy)
|
|
)
|
|
return
|
|
|
|
if account.blocked:
|
|
return
|
|
|
|
account.storage = storage_to_buy
|
|
await update_account(relay_id, account)
|
|
|
|
except Exception as ex:
|
|
logger.warning(ex)
|