add tpos to pr

This commit is contained in:
dni ⚡ 2022-10-21 18:15:27 +02:00
parent 2d61a8854f
commit fe879ad9b9

View File

@ -1,11 +1,11 @@
import asyncio import asyncio
import json
from lnbits.core import db as core_db from loguru import logger
from lnbits.core.crud import create_payment
from lnbits.core.models import Payment from lnbits.core.models import Payment
from lnbits.helpers import get_current_extension_name, urlsafe_short_hash from lnbits.core.services import create_invoice, pay_invoice
from lnbits.tasks import internal_invoice_queue, register_invoice_listener from lnbits.helpers import get_current_extension_name
from lnbits.tasks import register_invoice_listener
from .crud import get_tpos from .crud import get_tpos
@ -20,11 +20,9 @@ async def wait_for_paid_invoices():
async def on_invoice_paid(payment: Payment) -> None: async def on_invoice_paid(payment: Payment) -> None:
if payment.extra.get("tag") == "tpos" and payment.extra.get("tipSplitted"): if payment.extra.get("tag") != "tpos":
# already splitted, ignore
return return
# now we make some special internal transfers (from no one to the receiver)
tpos = await get_tpos(payment.extra.get("tposId")) tpos = await get_tpos(payment.extra.get("tposId"))
tipAmount = payment.extra.get("tipAmount") tipAmount = payment.extra.get("tipAmount")
@ -32,39 +30,17 @@ async def on_invoice_paid(payment: Payment) -> None:
# no tip amount # no tip amount
return return
tipAmount = tipAmount * 1000 payment_hash, payment_request = await create_invoice(
amount = payment.amount - tipAmount
# mark the original payment with one extra key, "splitted"
# (this prevents us from doing this process again and it's informative)
# and reduce it by the amount we're going to send to the producer
await core_db.execute(
"""
UPDATE apipayments
SET extra = ?, amount = ?
WHERE hash = ?
AND checking_id NOT LIKE 'internal_%'
""",
(
json.dumps(dict(**payment.extra, tipSplitted=True)),
amount,
payment.payment_hash,
),
)
# perform the internal transfer using the same payment_hash
internal_checking_id = f"internal_{urlsafe_short_hash()}"
await create_payment(
wallet_id=tpos.tip_wallet, wallet_id=tpos.tip_wallet,
checking_id=internal_checking_id, amount=int(tipAmount), # sats
payment_request="", internal=True,
payment_hash=payment.payment_hash, memo=f"tpos tip",
amount=tipAmount,
memo=f"Tip for {payment.memo}",
pending=False,
extra={"tipSplitted": True},
) )
logger.debug(f"tpos: tip invoice created: {payment_hash}")
# manually send this for now checking_id = await pay_invoice(
await internal_invoice_queue.put(internal_checking_id) payment_request=payment_request,
return wallet_id=payment.wallet_id,
extra={"tag": "tpos"},
)
logger.debug(f"tpos: tip invoice paid: {checking_id}")