LBRY-Vault/electrum/submarine_swaps.py

import asyncio
import json
import os
from .crypto import sha256, hash_160
from .ecc import ECPrivkey
from .bitcoin import address_to_script, script_to_p2wsh, redeem_script_to_address, opcodes, p2wsh_nested_script, push_script, is_segwit_address
from .transaction import TxOutpoint, PartialTxInput, PartialTxOutput, PartialTransaction, construct_witness
from .transaction import script_GetOp, match_script_against_template, OPPushDataGeneric, OPPushDataPubkey
from .util import log_exceptions
from .bitcoin import dust_threshold
from typing import TYPE_CHECKING
from .logging import Logger

if TYPE_CHECKING:
    from .network import Network
    from .wallet import Abstract_Wallet

API_URL = 'https://lightning.electrum.org/api'


WITNESS_TEMPLATE_SWAP = [
    opcodes.OP_HASH160,
    OPPushDataGeneric(lambda x: x == 20),
    opcodes.OP_EQUAL,
    opcodes.OP_IF,
    OPPushDataPubkey,
    opcodes.OP_ELSE,
    OPPushDataGeneric(None),
    opcodes.OP_CHECKLOCKTIMEVERIFY,
    opcodes.OP_DROP,
    OPPushDataPubkey,
    opcodes.OP_ENDIF,
    opcodes.OP_CHECKSIG
]


WITNESS_TEMPLATE_REVERSE_SWAP = [
    opcodes.OP_SIZE,
    OPPushDataGeneric(None),
    opcodes.OP_EQUAL,
    opcodes.OP_IF,
    opcodes.OP_HASH160,
    OPPushDataGeneric(lambda x: x == 20),
    opcodes.OP_EQUALVERIFY,
    OPPushDataPubkey,
    opcodes.OP_ELSE,
    opcodes.OP_DROP,
    OPPushDataGeneric(None),
    opcodes.OP_CHECKLOCKTIMEVERIFY,
    opcodes.OP_DROP,
    OPPushDataPubkey,
    opcodes.OP_ENDIF,
    opcodes.OP_CHECKSIG
]


def create_claim_tx(txin, witness_script, preimage, privkey:bytes, address, amount_sat, locktime):
    pubkey = ECPrivkey(privkey).get_public_key_bytes(compressed=True)
    if is_segwit_address(txin.address):
        txin.script_type = 'p2wsh'
        txin.script_sig = b''
    else:
        txin.script_type = 'p2wsh-p2sh'
        txin.redeem_script = bytes.fromhex(p2wsh_nested_script(witness_script.hex()))
        txin.script_sig = bytes.fromhex(push_script(txin.redeem_script.hex()))
    txin.witness_script = witness_script
    txout = PartialTxOutput(scriptpubkey=bytes.fromhex(address_to_script(address)), value=amount_sat)
    tx = PartialTransaction.from_io([txin], [txout], version=2, locktime=locktime)
    tx.set_rbf(True)
    sig = bytes.fromhex(tx.sign_txin(0, privkey))
    witness = [sig, preimage, witness_script]
    txin.witness = bytes.fromhex(construct_witness(witness))
    return tx



class SwapManager(Logger):

    @log_exceptions
    async def _claim_swap(self, lockup_address, redeem_script, preimage, privkey, locktime):
        utxos = self.lnwatcher.get_addr_utxo(lockup_address)
        if not utxos:
            return
        if not preimage:
            delta = self.network.get_local_height() - locktime
            if delta < 0:
                self.logger.info(f'height not reached for refund {lockup_address} {delta}, {locktime}')
                return
        for txin in list(utxos.values()):
            fee = self.lnwatcher.config.estimate_fee(136, allow_fallback_to_static_rates=True)
            amount_sat = txin._trusted_value_sats - fee
            if amount_sat < dust_threshold():
                self.logger.info('utxo value below dust threshold')
                continue
            address = self.wallet.get_unused_address()
            tx = create_claim_tx(txin, redeem_script, preimage, privkey, address, amount_sat, locktime)
            await self.network.broadcast_transaction(tx)

    def __init__(self, wallet: 'Abstract_Wallet', network:'Network'):
        Logger.__init__(self)
        self.network = network
        self.wallet = wallet
        self.lnworker = wallet.lnworker
        self.lnwatcher = self.wallet.lnworker.lnwatcher
        swaps = self.wallet.db.get_dict('submarine_swaps')
        for key, data in swaps.items():
            redeem_script = bytes.fromhex(data['redeemScript'])
            locktime = data['timeoutBlockHeight']
            privkey = bytes.fromhex(data['privkey'])
            if data.get('invoice'):
                lockup_address = data['lockupAddress']
                preimage = bytes.fromhex(data['preimage'])
            else:
                lockup_address = data['address']
                preimage = 0
            self.add_lnwatcher_callback(lockup_address, redeem_script, preimage, privkey, locktime)

    def add_lnwatcher_callback(self, lockup_address, redeem_script, preimage, privkey, locktime):
        callback = lambda: self._claim_swap(lockup_address, redeem_script, preimage, privkey, locktime)
        self.lnwatcher.add_callback(lockup_address, callback)

    @log_exceptions
    async def normal_swap(self, amount_sat, password):
        privkey = os.urandom(32)
        pubkey = ECPrivkey(privkey).get_public_key_bytes(compressed=True)
        key = await self.lnworker._add_request_coro(amount_sat, 'swap', expiry=3600*24)
        request = self.wallet.get_request(key)
        invoice = request['invoice']
        lnaddr = self.lnworker._check_invoice(invoice, amount_sat)
        payment_hash = lnaddr.paymenthash
        preimage = self.lnworker.get_preimage(payment_hash)
        request_data = {
            "type": "submarine",
            "pairId": "BTC/BTC",
            "orderSide": "sell",
            "invoice": invoice,
            "refundPublicKey": pubkey.hex()
        }
        response = await self.network._send_http_on_proxy(
            'post',
            API_URL + '/createswap',
            json=request_data,
            timeout=30)
        data = json.loads(response)
        response_id = data["id"]
        zeroconf = data["acceptZeroConf"]
        onchain_amount = data["expectedAmount"]
        locktime = data["timeoutBlockHeight"]
        lockup_address = data["address"]
        redeem_script = data["redeemScript"]
        # verify redeem_script is built with our pubkey and preimage
        redeem_script = bytes.fromhex(redeem_script)
        parsed_script = [x for x in script_GetOp(redeem_script)]
        assert match_script_against_template(redeem_script, WITNESS_TEMPLATE_SWAP)
        assert script_to_p2wsh(redeem_script.hex()) == lockup_address
        assert hash_160(preimage) == parsed_script[1][1]
        assert pubkey == parsed_script[9][1]
        # verify that they are not locking up funds for more than a day
        assert locktime == int.from_bytes(parsed_script[6][1], byteorder='little')
        assert locktime - self.network.get_local_height() < 144
        # save swap data in wallet in case we need a refund
        data['privkey'] = privkey.hex()
        data['preimage'] = preimage.hex()
        swaps = self.wallet.db.get_dict('submarine_swaps')
        swaps[response_id] = data
        self.add_lnwatcher_callback(lockup_address, redeem_script, 0, privkey, locktime)
        outputs = [PartialTxOutput.from_address_and_value(lockup_address, onchain_amount)]
        tx = self.wallet.create_transaction(outputs=outputs, rbf=False, password=password)
        await self.network.broadcast_transaction(tx)
        #
        attempt = await self.lnworker.await_payment(payment_hash)
        return {
            'id':response_id,
            'success':attempt.success,
        }

    @log_exceptions
    async def reverse_swap(self, amount_sat):
        privkey = os.urandom(32)
        pubkey = ECPrivkey(privkey).get_public_key_bytes(compressed=True)
        preimage = os.urandom(32)
        preimage_hash = sha256(preimage)
        request_data = {
            "type": "reversesubmarine",
            "pairId": "BTC/BTC",
            "orderSide": "buy",
            "invoiceAmount": amount_sat,
            "preimageHash": preimage_hash.hex(),
            "claimPublicKey": pubkey.hex()
        }
        response = await self.network._send_http_on_proxy(
            'post',
            API_URL + '/createswap',
            json=request_data,
            timeout=30)
        data = json.loads(response)
        invoice = data['invoice']
        lockup_address = data['lockupAddress']
        redeem_script = data['redeemScript']
        locktime = data['timeoutBlockHeight']
        onchain_amount = data["onchainAmount"]
        response_id = data['id']
        # verify redeem_script is built with our pubkey and preimage
        redeem_script = bytes.fromhex(redeem_script)
        parsed_script = [x for x in script_GetOp(redeem_script)]
        assert match_script_against_template(redeem_script, WITNESS_TEMPLATE_REVERSE_SWAP)
        assert script_to_p2wsh(redeem_script.hex()) == lockup_address
        assert hash_160(preimage) == parsed_script[5][1]
        assert pubkey == parsed_script[7][1]
        # verify that we will have enought time to get our tx confirmed
        assert locktime == int.from_bytes(parsed_script[10][1], byteorder='little')
        assert locktime - self.network.get_local_height() > 10
        # verify invoice preimage_hash
        lnaddr = self.lnworker._check_invoice(invoice, amount_sat)
        assert lnaddr.paymenthash == preimage_hash
        # save swap data in wallet in case payment fails
        data['privkey'] = privkey.hex()
        data['preimage'] = preimage.hex()
        # save data to wallet file
        swaps = self.wallet.db.get_dict('submarine_swaps')
        swaps[response_id] = data
        # add callback to lnwatcher
        self.add_lnwatcher_callback(lockup_address, redeem_script, preimage, privkey, locktime)
        # initiate payment.
        success, log = await self.lnworker._pay(invoice, attempts=5)
        # discard data; this should be done by lnwatcher
        if success:
            swaps.pop(response_id)
        return {
            'id':response_id,
            'success':success,
        }

    @log_exceptions
    async def get_pairs(self):
        response = await self.network._send_http_on_proxy(
            'get',
            API_URL + '/getpairs',
            timeout=30)
        data = json.loads(response)
        return data