mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2024-11-20 10:39:59 +01:00
99 lines
2.4 KiB
Python
99 lines
2.4 KiB
Python
import trio # type: ignore
|
|
from http import HTTPStatus
|
|
from typing import Optional, List, Callable
|
|
from quart import Request, g
|
|
from quart_trio import QuartTrio
|
|
from werkzeug.datastructures import Headers
|
|
|
|
from lnbits.db import open_db
|
|
from lnbits.settings import WALLET
|
|
|
|
from lnbits.core.crud import get_standalone_payment
|
|
|
|
main_app: Optional[QuartTrio] = None
|
|
|
|
|
|
def grab_app_for_later(app: QuartTrio):
|
|
global main_app
|
|
main_app = app
|
|
|
|
|
|
deferred_async: List[Callable] = []
|
|
|
|
|
|
def record_async(func: Callable) -> Callable:
|
|
def recorder(state):
|
|
deferred_async.append(func)
|
|
|
|
return recorder
|
|
|
|
|
|
def run_deferred_async(nursery):
|
|
for func in deferred_async:
|
|
nursery.start_soon(func)
|
|
|
|
|
|
async def send_push_promise(a, b) -> None:
|
|
pass
|
|
|
|
|
|
async def run_on_pseudo_request(func: Callable, *args):
|
|
fk = Request(
|
|
"GET",
|
|
"http",
|
|
"/background/pseudo",
|
|
b"",
|
|
Headers([("host", "lnbits.background")]),
|
|
"",
|
|
"1.1",
|
|
send_push_promise=send_push_promise,
|
|
)
|
|
assert main_app
|
|
|
|
async def run():
|
|
async with main_app.request_context(fk):
|
|
with open_db() as g.db: # type: ignore
|
|
await func(*args)
|
|
|
|
async with trio.open_nursery() as nursery:
|
|
nursery.start_soon(run)
|
|
|
|
|
|
invoice_listeners: List[trio.MemorySendChannel] = []
|
|
|
|
|
|
def register_invoice_listener(send_chan: trio.MemorySendChannel):
|
|
"""
|
|
A method intended for extensions to call when they want to be notified about
|
|
new invoice payments incoming.
|
|
"""
|
|
invoice_listeners.append(send_chan)
|
|
|
|
|
|
async def webhook_handler():
|
|
handler = getattr(WALLET, "webhook_listener", None)
|
|
if handler:
|
|
return await handler()
|
|
return "", HTTPStatus.NO_CONTENT
|
|
|
|
|
|
internal_invoice_paid, internal_invoice_received = trio.open_memory_channel(0)
|
|
|
|
|
|
async def internal_invoice_listener():
|
|
async for checking_id in internal_invoice_received:
|
|
await run_on_pseudo_request(invoice_callback_dispatcher, checking_id)
|
|
|
|
|
|
async def invoice_listener():
|
|
async for checking_id in WALLET.paid_invoices_stream():
|
|
await run_on_pseudo_request(invoice_callback_dispatcher, checking_id)
|
|
|
|
|
|
async def invoice_callback_dispatcher(checking_id: str):
|
|
payment = get_standalone_payment(checking_id)
|
|
if payment and payment.is_in:
|
|
payment.set_pending(False)
|
|
for send_chan in invoice_listeners:
|
|
await send_chan.send(payment)
|