lnbits-legend/lnbits/extensions/boltcards/tasks.py
2022-09-23 17:11:19 +02:00

67 lines
2 KiB
Python

import asyncio
import json
import httpx
from loguru import logger
from lnbits.core import db as core_db
from lnbits.core.models import Payment
from lnbits.tasks import register_invoice_listener
from .crud import create_refund, get_card, get_hit
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 on_invoice_paid(payment: Payment) -> None:
if not payment.extra.get("refund"):
return
if payment.extra.get("wh_status"):
# this webhook has already been sent
return
hit = await get_hit(payment.extra.get("refund"))
if hit:
refund = await create_refund(
hit_id=hit.id, refund_amount=(payment.amount / 1000)
)
await mark_webhook_sent(payment, 1)
card = await get_card(hit.card_id)
if card.webhook_url:
async with httpx.AsyncClient() as client:
try:
r = await client.post(
card.webhook_url,
json={
"notification": "card_refund",
"payment_hash": payment.payment_hash,
"payment_request": payment.bolt11,
"card_external_id": card.external_id,
"card_name": card.card_name,
"amount": int(payment.amount / 1000),
},
timeout=40,
)
except Exception as exc:
logger.error("Caught exception when dispatching webhook url:", exc)
async def mark_webhook_sent(payment: Payment, status: int) -> None:
payment.extra["wh_status"] = status
await core_db.execute(
"""
UPDATE apipayments SET extra = ?
WHERE hash = ?
""",
(json.dumps(payment.extra), payment.payment_hash),
)