lnbits-legend/lnbits/extensions/lnaddress/tasks.py
2021-11-26 05:58:20 +00:00

62 lines
1.7 KiB
Python

import asyncio
import httpx
from lnbits.core.models import Payment
from lnbits.tasks import register_invoice_listener
from .crud import get_address, get_domain, set_address_paid, set_address_renewed
async def wait_for_paid_invoices():
invoice_queue = asyncio.Queue()
register_invoice_listener(invoice_queue)
while True:
payment = await invoice_queue.get()
await on_invoice_paid(payment)
async def call_webhook_on_paid(payment_hash):
### Use webhook to notify about cloudflare registration
address = await get_address(payment_hash)
domain = await get_domain(address.domain)
if not domain.webhook:
return
async with httpx.AsyncClient() as client:
try:
r = await client.post(
domain.webhook,
json={
"domain": domain.domain,
"address": address.username,
"email": address.email,
"cost": str(address.sats) + " sats",
"duration": str(address.duration) + " days",
},
timeout=40,
)
except AssertionError:
webhook = None
async def on_invoice_paid(payment: Payment) -> None:
if "lnaddress" == payment.extra.get("tag"):
await payment.set_pending(False)
await set_address_paid(payment_hash=payment.payment_hash)
await call_webhook_on_paid(payment.payment_hash)
elif "renew lnaddress" == payment.extra.get("tag"):
await payment.set_pending(False)
await set_address_renewed(
address_id=payment.extra["id"], duration=payment.extra["duration"]
)
await call_webhook_on_paid(payment.payment_hash)
else:
return