import asyncio

from loguru import logger

from lnbits.core.models import Payment
from lnbits.core.services import create_invoice, pay_invoice, websocketUpdater
from lnbits.helpers import get_current_extension_name
from lnbits.tasks import register_invoice_listener

from .crud import get_temp


#######################################
########## RUN YOU TASKS HERE #########
#######################################


# the usual task is to listen to invoices related to this extension

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)


# do somethhing when an invoice related top this extension is paid

async def on_invoice_paid(payment: Payment) -> None:
    if payment.extra.get("tag") != "temp":
        return

    temp_id = payment.extra.get("tempId")
    assert temp_id

    temp = await get_temp(temp_id)
    assert temp

    # update something
    
    data_to_update = {
        "total": temp.total + payment.amount
    }

    await update_temp(temp_id=temp_id, **data_to_update.dict())


    # here we could send some data to a websocket on wss://<your-lnbits>/api/v1/ws/<temp_id>

    some_payment_data = {
        "name": temp.name,
        "amount": payment.amount,
        "fee": payment.fee,
        "checking_id": payment.checking_id
    }

    await websocketUpdater(temp_id, str(some_payment_data))