mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-25 07:07:48 +01:00
79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
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 payment.extra.get("tag") == "splitpayments":
|
|
# 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"checking if tagged for {len(targets)} targets")
|
|
tagged = False
|
|
for target in targets:
|
|
if target.tag in payment.extra:
|
|
tagged = True
|
|
payment_hash, payment_request = await create_invoice(
|
|
wallet_id=target.wallet,
|
|
amount=int(payment.amount / 1000), # sats
|
|
internal=True,
|
|
memo=f"Pushed tagged payment to {target.alias}",
|
|
extra={"tag": "splitpayments"},
|
|
)
|
|
logger.debug(f"created split invoice: {payment_hash}")
|
|
|
|
checking_id = await pay_invoice(
|
|
payment_request=payment_request,
|
|
wallet_id=payment.wallet_id,
|
|
extra={"tag": "splitpayments"},
|
|
)
|
|
logger.debug(f"paid split invoice: {checking_id}")
|
|
|
|
logger.debug(f"performing split to {len(targets)} targets")
|
|
|
|
if tagged == False:
|
|
for target in targets:
|
|
if target.percent > 0:
|
|
amount = int(payment.amount * target.percent / 100) # msats
|
|
payment_hash, payment_request = await create_invoice(
|
|
wallet_id=target.wallet,
|
|
amount=int(amount / 1000), # sats
|
|
internal=True,
|
|
memo=f"split payment: {target.percent}% for {target.alias or target.wallet}",
|
|
extra={"tag": "splitpayments"},
|
|
)
|
|
logger.debug(f"created split invoice: {payment_hash}")
|
|
|
|
checking_id = await pay_invoice(
|
|
payment_request=payment_request,
|
|
wallet_id=payment.wallet_id,
|
|
extra={"tag": "splitpayments"},
|
|
)
|
|
logger.debug(f"paid split invoice: {checking_id}")
|