lnbits-legend/lnbits/extensions/splitpayments/tasks.py

import asyncio

from loguru import logger

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

from .crud import get_targets


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) -> None:
    if not payment.extra:
        return

    if payment.extra.get("tag") == "splitpayments" or payment.extra.get("splitted"):
        # already a splitted payment, ignore
        return

    targets = await get_targets(payment.wallet_id)
    logger.debug(targets)
    if not targets:
        return

    total_percent = sum([target.percent for target in targets])

    if total_percent > 100:
        logger.error("splitpayment failure: total percent adds up to more than 100%")
        return

    logger.debug(f"performing split payments to {len(targets)} targets")

    amount_to_split = payment.amount

    if payment.extra.get("amount"):
        amount_to_split = (payment.extra.get("amount") or 0) * 1000

    for target in targets:
        if target.percent > 0:
            tagged = target.tag in payment.extra

            amount = int(amount_to_split * target.percent / 100)  # msats
            memo = (
                f"split payment: {target.percent}% for {target.alias or target.wallet}"
            )
            if tagged:
                memo = f"Pushed tagged payment to {target.alias}"
                amount = int(amount_to_split)

            payment_hash, payment_request = await create_invoice(
                wallet_id=target.wallet,
                amount=int(amount / 1000),  # sats
                internal=True,
                memo=memo,
            )

            logger.debug(f"created split invoice: {payment_hash}")

            extra = {**payment.extra, "splitted": True}

            checking_id = await pay_invoice(
                payment_request=payment_request,
                wallet_id=payment.wallet_id,
                extra=extra,
            )
            logger.debug(f"paid split invoice: {checking_id}")

    logger.debug(f"performing split to {len(targets)} targets")