mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-26 07:31:22 +01:00
d89a6a337a
* add status to statusdialog * first commits for boltz update * formatting * add latest boltz-clien package * big refactor, depending on boltz_client package, clean up, mypy issues, not tested yet * blacking, sorting and stuff * remove unused req_wrap helper * remove api docs from frontend * bug: frontend boltz limits error * clean up buttons * update to boltz-client 0.0.8 * fix tests to poetry version 1.3.1 * update requirements * formatting * recurring swap works now, need more finetuning * add exceptions for multiple auto swaps and swapping in with active auto swap * black * auto reverse swap actually works :) * remove swap status dialogs * update to boltz_client 0.0.9 * update to boltz-client 0.1.1, and fix startup * update requirement.txt for boltz-client * fixup columns in table, remove unused payment.extra, change deezy label * remove balance check for auto swap out * update boltzc-lient to 0.1.2, fix mypy issue inside boltz package * nitpicks calle tasks.py * calle nitpicks crud * calle nitpicks crud * refactor * fix formatting * circular import * black :) Co-authored-by: callebtc <93376500+callebtc@users.noreply.github.com>
180 lines
7 KiB
Python
180 lines
7 KiB
Python
import asyncio
|
|
|
|
from boltz_client.boltz import BoltzNotFoundException, BoltzSwapStatusException
|
|
from boltz_client.mempool import MempoolBlockHeightException
|
|
from loguru import logger
|
|
|
|
from lnbits.core.crud import get_wallet
|
|
from lnbits.core.models import Payment
|
|
from lnbits.core.services import check_transaction_status, fee_reserve
|
|
from lnbits.helpers import get_current_extension_name
|
|
from lnbits.tasks import register_invoice_listener
|
|
|
|
from .crud import (
|
|
create_reverse_submarine_swap,
|
|
get_all_pending_reverse_submarine_swaps,
|
|
get_all_pending_submarine_swaps,
|
|
get_auto_reverse_submarine_swap_by_wallet,
|
|
get_submarine_swap,
|
|
update_swap_status,
|
|
)
|
|
from .models import CreateReverseSubmarineSwap, ReverseSubmarineSwap, SubmarineSwap
|
|
from .utils import create_boltz_client, execute_reverse_swap
|
|
|
|
|
|
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:
|
|
|
|
await check_for_auto_swap(payment)
|
|
|
|
if payment.extra.get("tag") != "boltz":
|
|
# not a boltz invoice
|
|
return
|
|
|
|
await payment.set_pending(False)
|
|
|
|
if payment.extra:
|
|
swap_id = payment.extra.get("swap_id")
|
|
if swap_id:
|
|
swap = await get_submarine_swap(swap_id)
|
|
if swap:
|
|
await update_swap_status(swap_id, "complete")
|
|
|
|
|
|
async def check_for_auto_swap(payment: Payment) -> None:
|
|
auto_swap = await get_auto_reverse_submarine_swap_by_wallet(payment.wallet_id)
|
|
if auto_swap:
|
|
wallet = await get_wallet(payment.wallet_id)
|
|
if wallet:
|
|
reserve = fee_reserve(wallet.balance_msat) / 1000
|
|
balance = wallet.balance_msat / 1000
|
|
amount = balance - auto_swap.balance - reserve
|
|
if amount >= auto_swap.amount:
|
|
|
|
client = create_boltz_client()
|
|
claim_privkey_wif, preimage_hex, swap = client.create_reverse_swap(
|
|
amount=int(amount)
|
|
)
|
|
new_swap = await create_reverse_submarine_swap(
|
|
CreateReverseSubmarineSwap(
|
|
wallet=auto_swap.wallet,
|
|
amount=int(amount),
|
|
instant_settlement=auto_swap.instant_settlement,
|
|
onchain_address=auto_swap.onchain_address,
|
|
),
|
|
claim_privkey_wif,
|
|
preimage_hex,
|
|
swap,
|
|
)
|
|
await execute_reverse_swap(client, new_swap)
|
|
|
|
logger.info(
|
|
f"Boltz: auto reverse swap created with amount: {amount}, boltz_id: {new_swap.boltz_id}"
|
|
)
|
|
|
|
|
|
"""
|
|
testcases for boltz startup
|
|
A. normal swaps
|
|
1. test: create -> kill -> start -> startup invoice listeners -> pay onchain funds -> should complete
|
|
2. test: create -> kill -> pay onchain funds -> mine block -> start -> startup check -> should complete
|
|
3. test: create -> kill -> mine blocks and hit timeout -> start -> should go timeout/failed
|
|
4. test: create -> kill -> pay to less onchain funds -> mine blocks hit timeout -> start lnbits -> should be refunded
|
|
|
|
B. reverse swaps
|
|
1. test: create instant -> kill -> boltz does lockup -> not confirmed -> start lnbits -> should claim/complete
|
|
2. test: create -> kill -> boltz does lockup -> not confirmed -> start lnbits -> mine blocks -> should claim/complete
|
|
3. test: create -> kill -> boltz does lockup -> confirmed -> start lnbits -> should claim/complete
|
|
"""
|
|
|
|
|
|
async def check_for_pending_swaps():
|
|
try:
|
|
swaps = await get_all_pending_submarine_swaps()
|
|
reverse_swaps = await get_all_pending_reverse_submarine_swaps()
|
|
if len(swaps) > 0 or len(reverse_swaps) > 0:
|
|
logger.debug(f"Boltz - startup swap check")
|
|
except:
|
|
logger.error(
|
|
f"Boltz - startup swap check, database is not created yet, do nothing"
|
|
)
|
|
return
|
|
|
|
client = create_boltz_client()
|
|
|
|
if len(swaps) > 0:
|
|
logger.debug(f"Boltz - {len(swaps)} pending swaps")
|
|
for swap in swaps:
|
|
await check_swap(swap, client)
|
|
|
|
if len(reverse_swaps) > 0:
|
|
logger.debug(f"Boltz - {len(reverse_swaps)} pending reverse swaps")
|
|
for reverse_swap in reverse_swaps:
|
|
await check_reverse_swap(reverse_swap, client)
|
|
|
|
|
|
async def check_swap(swap: SubmarineSwap, client):
|
|
try:
|
|
payment_status = await check_transaction_status(swap.wallet, swap.payment_hash)
|
|
if payment_status.paid:
|
|
logger.debug(f"Boltz - swap: {swap.boltz_id} got paid while offline.")
|
|
await update_swap_status(swap.id, "complete")
|
|
else:
|
|
try:
|
|
_ = client.swap_status(swap.id)
|
|
except:
|
|
txs = client.mempool.get_txs_from_address(swap.address)
|
|
if len(txs) == 0:
|
|
await update_swap_status(swap.id, "timeout")
|
|
else:
|
|
await client.refund_swap(
|
|
privkey_wif=swap.refund_privkey,
|
|
lockup_address=swap.address,
|
|
receive_address=swap.refund_address,
|
|
redeem_script_hex=swap.redeem_script,
|
|
timeout_block_height=swap.timeout_block_height,
|
|
)
|
|
await update_swap_status(swap.id, "refunded")
|
|
except BoltzNotFoundException as exc:
|
|
logger.debug(f"Boltz - swap: {swap.boltz_id} does not exist.")
|
|
await update_swap_status(swap.id, "failed")
|
|
except MempoolBlockHeightException as exc:
|
|
logger.debug(
|
|
f"Boltz - tried to refund swap: {swap.id}, but has not reached the timeout."
|
|
)
|
|
except Exception as exc:
|
|
logger.error(f"Boltz - unhandled exception, swap: {swap.id} - {str(exc)}")
|
|
|
|
|
|
async def check_reverse_swap(reverse_swap: ReverseSubmarineSwap, client):
|
|
try:
|
|
_ = client.swap_status(reverse_swap.boltz_id)
|
|
await client.claim_reverse_swap(
|
|
lockup_address=reverse_swap.lockup_address,
|
|
receive_address=reverse_swap.onchain_address,
|
|
privkey_wif=reverse_swap.claim_privkey,
|
|
preimage_hex=reverse_swap.preimage,
|
|
redeem_script_hex=reverse_swap.redeem_script,
|
|
zeroconf=reverse_swap.instant_settlement,
|
|
)
|
|
await update_swap_status(reverse_swap.id, "complete")
|
|
|
|
except BoltzSwapStatusException as exc:
|
|
logger.debug(f"Boltz - swap_status: {str(exc)}")
|
|
await update_swap_status(reverse_swap.id, "failed")
|
|
# should only happen while development when regtest is reset
|
|
except BoltzNotFoundException as exc:
|
|
logger.debug(f"Boltz - reverse swap: {reverse_swap.boltz_id} does not exist.")
|
|
await update_swap_status(reverse_swap.id, "failed")
|
|
except Exception as exc:
|
|
logger.error(
|
|
f"Boltz - unhandled exception, reverse swap: {reverse_swap.id} - {str(exc)}"
|
|
)
|