import asyncio
import os
from typing import AsyncGenerator, List, Optional

import grpc
from fastapi.exceptions import HTTPException
from loguru import logger
from starlette import status

import app.bitcoind.service as btc
import app.lightning.impl.protos.lnd.lightning_pb2 as ln
import app.lightning.impl.protos.lnd.lightning_pb2_grpc as lnrpc
import app.lightning.impl.protos.lnd.router_pb2 as router
import app.lightning.impl.protos.lnd.router_pb2_grpc as routerrpc
import app.lightning.impl.protos.lnd.walletunlocker_pb2 as unlocker
import app.lightning.impl.protos.lnd.walletunlocker_pb2_grpc as unlockerrpc
from app.api.config import config as dconfig
from app.api.utils import Event, broadcast_msg, config_get_hex_str
from app.lightning.exceptions import NodeNotFoundError
from app.lightning.impl.ln_base import LightningNodeBase
from app.lightning.models import (
    Channel,
    FeeRevenue,
    ForwardSuccessEvent,
    GenericTx,
    InitLnRepoUpdate,
    Invoice,
    InvoiceState,
    LnInfo,
    LnInitState,
    NewAddressInput,
    OnchainAddressType,
    OnChainTransaction,
    Payment,
    PaymentRequest,
    SendCoinsInput,
    SendCoinsResponse,
    WalletBalance,
)
from app.lightning.utils import alias_or_empty, raise_for_pay_req_decode_error


@logger.catch(exclude=(HTTPException,))
def _check_transient_ln_error(error):
    """Map known transient LND gRPC errors to appropriate HTTP statuses.

    Returns without raising for unknown errors, leaving the caller to turn
    them into a 500.
    """
    logger.debug("logger._check_transient_ln_error()")

    details = error.details()
    if details is None:
        return

    if details.find("wallet locked") > -1:
        raise HTTPException(
            status.HTTP_423_LOCKED,
            detail="Wallet is locked. Unlock via /lightning/unlock-wallet",
        )

    if "the RPC server is in the process of starting up" in details:
        # LND is up but its RPC server isn't ready yet; signal a retryable
        # status instead of a generic 500 (blitz_api#247)
        raise HTTPException(
            status.HTTP_425_TOO_EARLY,
            detail=(
                "The Lightning RPC server is starting up and not yet ready. "
                "Please try again shortly."
            ),
        )


# Due to updated ECDSA generated tls.cert we need to let gprc know that
# we need to use that cipher suite otherwise there will be a handshake
# error when we communicate with the lnd rpc server.
os.environ["GRPC_SSL_CIPHER_SUITES"] = "HIGH+ECDSA"

# Uncomment to see full gRPC logs
# os.environ["GRPC_TRACE"] = "all"
# os.environ["GRPC_VERBOSITY"] = "DEBUG"


class LnNodeLNDgRPC(LightningNodeBase):
    _lnd_connect_error_debug_msg = """
Unable to connect to LND. Possible reasons:
* Node is not reachable (ports, network down, ...)
* Macaroon is not correct
* IP is not included in LND tls certificate
    Add tlsextraip=192.168.1.xxx to lnd.conf and restart LND.
    This will recreate the TLS certificate. The .env must be adapted accordingly.
* TLS certificate is wrong. (settings changed, ...)

To Debug gRPC problems uncomment the following line in app.lightning.impl.lnd_grpc.py
# os.environ["GRPC_VERBOSITY"] = "DEBUG"
This will show more debug information.
    """

    # Decoding the payment request take a long time,
    # hence we build a simple cache here.
    _memo_cache = {}
    _initialized = False

    def _create_stubs(self) -> None:
        if self._channel is not None:
            logger.warning("gRPC channel already created.")
            return

        opts = [("grpc.max_receive_message_length", 1024 * 1024 * 10)]

        self._channel = grpc.aio.secure_channel(
            self._lnd_grpc_url,
            self._combined_creds,
            options=opts,
        )
        self._lnd_stub = lnrpc.LightningStub(self._channel)
        self._router_stub = routerrpc.RouterStub(self._channel)
        self._wallet_unlocker = unlockerrpc.WalletUnlockerStub(self._channel)

        logger.debug("logger.Created LND gRPC stubs")

    def get_implementation_name(self) -> str:
        return "LND_GRPC"

    @logger.catch(exclude=(HTTPException,))
    async def _check_lnd_status(
        self,
        sleep_time: float = 2,
    ) -> AsyncGenerator[InitLnRepoUpdate, None]:
        logger.debug("_check_lnd_status() start")

        self._lnd_connect_error_debug_msg_sent = False

        # Create a temporary channel which will be destroyed at each iteration
        # Reason is that gRPC seems to only try and connect every 5 seconds to
        # the node if it is not running. To avoid the delay we create a new
        # channel each iteration.
        temp_channel = None
        temp_stub = None

        # We want to log the wallet locked error only once to avoid spamming the log
        wallet_locked_sent = False
        while True:
            try:
                if temp_channel is None:
                    if self._channel is not None:
                        temp_channel = self._channel
                        temp_stub = self._lnd_stub
                    else:
                        temp_channel = grpc.aio.secure_channel(
                            self._lnd_grpc_url, self._combined_creds
                        )
                        temp_stub = lnrpc.LightningStub(temp_channel)
                await temp_stub.GetInfo(ln.GetInfoRequest())

                if self._channel is None:
                    self._create_stubs()

                await self._init_queue.put(InitLnRepoUpdate(state=LnInitState.DONE))
                break
            except grpc.aio._call.AioRpcError as error:
                details = error.details()
                logger.debug(f"Waiting for LND daemon... Details {details}")

                if "failed to connect to all addresses" in details:
                    await self._init_queue.put(
                        InitLnRepoUpdate(
                            state=LnInitState.OFFLINE,
                            msg="Unable to connect to LND daemon, waiting...",
                        )
                    )

                    if not self._lnd_connect_error_debug_msg_sent:
                        logger.debug(self._lnd_connect_error_debug_msg)
                        self._lnd_connect_error_debug_msg_sent = True

                    await temp_channel.close()
                    temp_channel = None
                elif "waiting to start, RPC services not available" in details:
                    await self._init_queue.put(
                        InitLnRepoUpdate(
                            state=LnInitState.BOOTSTRAPPING,
                            msg=(
                                "Connected but waiting to start, RPC services "
                                "not available"
                            ),
                        )
                    )
                    await temp_channel.close()
                    temp_channel = None
                elif "wallet locked, unlock it to enable full RPC access" in details:
                    if not wallet_locked_sent:
                        logger.info(
                            (
                                "Wallet is locked. Unlock by calling "
                                "/lightning/unlock-wallet"
                            )
                        )

                    wallet_locked_sent = True
                    await self._init_queue.put(
                        InitLnRepoUpdate(
                            state=LnInitState.LOCKED,
                            msg="Wallet locked, unlock it to enable full RPC access",
                        )
                    )
                    if temp_channel != self._channel:
                        await temp_channel.close()
                        temp_channel = None
                elif (
                    "the RPC server is in the process of starting up, but not yet "
                    "ready to accept calls"
                ) in details:
                    # message from LND AFTER unlocking the wallet
                    await self._init_queue.put(
                        InitLnRepoUpdate(
                            state=LnInitState.BOOTSTRAPPING_AFTER_UNLOCK,
                            msg=(
                                "The RPC server is in the process of starting up, "
                                "but not yet ready to accept calls"
                            ),
                        )
                    )
                else:
                    logger.error(f"Unknown error: {details}")
                    raise

                logger.debug(f"_check_lnd_status() sleeping {sleep_time} seconds...")
                await asyncio.sleep(sleep_time)

        logger.debug("_check_lnd_status() done")

    @logger.catch(exclude=(HTTPException,))
    async def initialize(self) -> AsyncGenerator[InitLnRepoUpdate, None]:
        logger.trace("Unable to connect to LND daemon, waiting...")

        if self._initialized:
            logger.warning(
                (
                    "Connection already initialized. "
                    "This function must not be called twice."
                )
            )
            yield InitLnRepoUpdate(state=LnInitState.DONE)

        lnd_macaroon = config_get_hex_str(
            str(dconfig("BAPI_LND_MACAROON")), name="lnd_macaroon"
        )
        lnd_cert = bytes.fromhex(
            config_get_hex_str(str(dconfig("BAPI_LND_CERT")), name="lnd_cert")
        )

        def metadata_callback(context, callback):
            # for more info see grpc docs
            callback([("macaroon", lnd_macaroon)], None)

        lnd_grpc_ip = str(dconfig("BAPI_LND_GRPC_IP"))
        lnd_grpc_port = str(dconfig("BAPI_LND_GRPC_PORT"))
        self._lnd_grpc_url = lnd_grpc_ip + ":" + lnd_grpc_port

        auth_creds = grpc.metadata_call_credentials(metadata_callback)
        ssl_creds = grpc.ssl_channel_credentials(lnd_cert)
        self._combined_creds = grpc.composite_channel_credentials(ssl_creds, auth_creds)
        self._channel = None
        self._lnd_stub = None
        self._router_stub = None
        self._wallet_unlocker = None

        self._init_queue = asyncio.Queue()

        logger.info("Trying to connect to LND daemon ...")

        task = asyncio.create_task(self._check_lnd_status(sleep_time=2))

        while not self._initialized:
            res = await self._init_queue.get()  # type: InitLnRepoUpdate

            if (
                res.state == LnInitState.BOOTSTRAPPING_AFTER_UNLOCK
                and self._channel is None
            ):
                task.cancel()

                if self._channel is None:
                    # if res == _API_WALLET_UNLOCK_EVENT the endpoint function will have
                    # created the channel for us.
                    self._create_stubs()

                task = asyncio.create_task(self._check_lnd_status(sleep_time=0.5))
            elif res.state == LnInitState.DONE:
                self._initialized = True
                if not task.cancelled():
                    task.cancel()
            elif (
                res.state == LnInitState.OFFLINE
                or res.state == LnInitState.LOCKED
                or res.state == LnInitState.BOOTSTRAPPING
                or res.state == LnInitState.BOOTSTRAPPING_AFTER_UNLOCK
            ):
                pass  # do nothing here
            else:
                logger.warning(f"Unhandled initialization event: {res.model_dump()}")

            yield res

        logger.success("Initialization complete.")

    @logger.catch(exclude=(HTTPException,))
    async def get_wallet_balance(self) -> WalletBalance:
        logger.trace("get_wallet_balance() ")

        try:
            w_req = ln.WalletBalanceRequest()
            onchain = await self._lnd_stub.WalletBalance(w_req)

            c_req = ln.ChannelBalanceRequest()
            channel = await self._lnd_stub.ChannelBalance(c_req)

            return WalletBalance.from_lnd_grpc(onchain, channel)
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException,))
    async def list_all_tx(
        self, successful_only: bool, index_offset: int, max_tx: int, reversed: bool
    ) -> List[GenericTx]:
        logger.trace(
            (
                f"logger.list_all_tx(successful_only={successful_only}, "
                f"index_offset={index_offset}, max_tx={max_tx}, reversed={reversed})"
            )
        )

        # TODO: find a better caching strategy
        list_invoice_req = ln.ListInvoiceRequest(
            pending_only=successful_only,
            index_offset=0,
            num_max_invoices=0,
            reversed=reversed,
        )

        get_tx_req = ln.GetTransactionsRequest()

        list_payments_req = ln.ListPaymentsRequest(
            include_incomplete=not successful_only,
            index_offset=0,
            max_payments=0,
            reversed=reversed,
        )

        try:
            res = await asyncio.gather(
                *[
                    self._lnd_stub.ListInvoices(list_invoice_req),
                    self._lnd_stub.GetTransactions(get_tx_req),
                    self._lnd_stub.ListPayments(list_payments_req),
                ]
            )

            tx = []
            for i in res[0].invoices:
                tx.append(GenericTx.from_lnd_grpc_invoice(i))
            for t in res[1].transactions:
                tx.append(GenericTx.from_lnd_grpc_onchain_tx(t))
            for p in res[2].payments:
                comment = ""
                if p.payment_request in self._memo_cache:
                    comment = self._memo_cache[p.payment_request]
                else:
                    if p.payment_request is not None and p.payment_request != "":
                        pr = await self.decode_pay_request(p.payment_request)
                        if pr is None:
                            logger.error(
                                f"Unable to decode payment request {p.payment_request}"
                            )
                            continue

                        comment = pr.description
                        self._memo_cache[p.payment_request] = pr.description
                tx.append(GenericTx.from_lnd_grpc_payment(p, comment))

            def sortKey(e: GenericTx):
                return e.time_stamp

            tx.sort(key=sortKey)

            if reversed:
                tx.reverse()

            current_tx = len(tx)
            for i in range(current_tx):
                tx[i].index = i

            if max_tx == 0:
                max_tx = current_tx

            return tx[index_offset : index_offset + max_tx]
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException,))
    async def list_invoices(
        self,
        pending_only: bool,
        index_offset: int,
        num_max_invoices: int,
        reversed: bool,
    ):
        logger.trace("logger.list_invoices() ")

        try:
            req = ln.ListInvoiceRequest(
                pending_only=pending_only,
                index_offset=index_offset,
                num_max_invoices=num_max_invoices,
                reversed=reversed,
            )
            response = await self._lnd_stub.ListInvoices(req)
            return [Invoice.from_lnd_grpc(i) for i in response.invoices]
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException,))
    async def list_on_chain_tx(self) -> List[OnChainTransaction]:
        logger.trace("logger.list_on_chain_tx() ")

        try:
            req = ln.GetTransactionsRequest()
            response = await self._lnd_stub.GetTransactions(req)
            return [OnChainTransaction.from_lnd_grpc(t) for t in response.transactions]
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException,))
    async def list_payments(
        self,
        include_incomplete: bool,
        index_offset: int,
        max_payments: int,
        reversed: bool,
    ):
        logger.trace(
            (
                f"logger.list_payments(include_incomplete={include_incomplete}, "
                f"index_offset{index_offset}, max_payments={max_payments}, "
                f"reversed={reversed})"
            )
        )

        try:
            req = ln.ListPaymentsRequest(
                include_incomplete=include_incomplete,
                index_offset=index_offset,
                max_payments=max_payments,
                reversed=reversed,
            )
            response = await self._lnd_stub.ListPayments(req)
            return [Payment.from_lnd_grpc(p) for p in response.payments]
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException,))
    async def add_invoice(
        self,
        value_msat: int,
        memo: str = "",
        expiry: int = 3600,
        is_keysend: bool = False,
    ) -> Invoice:
        logger.trace(
            (
                f"logger.add_invoice(value_msat={value_msat}, memo={memo}, "
                f"expiry={expiry}, is_keysend={is_keysend})"
            )
        )

        try:
            i = ln.Invoice(
                memo=memo,
                value_msat=value_msat,
                expiry=expiry,
                is_keysend=is_keysend,
            )

            response = await self._lnd_stub.AddInvoice(i)

            # Can't use Invoice.from_lnd_grpc() here because
            # the response is not a standard invoice
            invoice = Invoice(
                memo=memo,
                expiry=expiry,
                r_hash=response.r_hash.hex(),
                payment_request=response.payment_request,
                add_index=str(response.add_index),
                payment_addr=response.payment_addr.hex(),
                state=InvoiceState.OPEN,
                is_keysend=is_keysend,
                value_msat=value_msat,
            )

            return invoice
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException,))
    async def decode_pay_request(self, pay_req: str) -> PaymentRequest:
        logger.trace(f"logger.decode_pay_request(pay_req={pay_req})")

        try:
            req = ln.PayReqString(pay_req=pay_req)
            res = await self._lnd_stub.DecodePayReq(req)
            return PaymentRequest.from_lnd_grpc(res)
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            details = error.details() or ""
            raise_for_pay_req_decode_error(details)
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=details
            )

    @logger.catch(exclude=(HTTPException,))
    async def get_fee_revenue(self) -> FeeRevenue:
        logger.trace("logger.get_fee_revenue()")

        req = ln.FeeReportRequest()
        res = await self._lnd_stub.FeeReport(req)
        return FeeRevenue.from_lnd_grpc(res)

    @logger.catch(exclude=(HTTPException,))
    async def new_address(self, input: NewAddressInput) -> str:
        logger.trace(f"logger.new_address(input={input})")

        t = 1 if input.type == OnchainAddressType.NP2WKH else 2
        try:
            req = ln.NewAddressRequest(type=t)
            response = await self._lnd_stub.NewAddress(req)
            return response.address
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException,))
    async def send_coins(self, input: SendCoinsInput) -> SendCoinsResponse:
        logger.trace(f"logger.send_coins(input={input})")

        try:
            r = ln.SendCoinsRequest(
                addr=input.address,
                amount=input.amount,
                target_conf=input.target_conf,
                sat_per_vbyte=input.sat_per_vbyte,
                min_confs=input.min_confs,
                label=input.label,
                send_all=input.send_all,
            )

            bi = await btc.get_blockchain_info()
            sendResponse = await self._lnd_stub.SendCoins(r)
            txResponse = await self._lnd_stub.GetTransactions(
                ln.GetTransactionsRequest(start_height=-1, end_height=bi.blocks)
            )

            tx = None
            for t in txResponse.transactions:
                if t.tx_hash == sendResponse.txid:
                    tx = t
                    break

            r = SendCoinsResponse.from_lnd_grpc(tx, input)
            await broadcast_msg(Event.LN_ONCHAIN_PAYMENT_STATUS, r.model_dump())
            return r
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            details = error.details()
            if details and details.find("invalid bech32 string") > -1:
                raise HTTPException(
                    status.HTTP_400_BAD_REQUEST,
                    detail=(
                        "Could not parse destination address, destination "
                        "should be a valid address."
                    ),
                )
            elif details and details.find("insufficient funds available") > -1:
                raise HTTPException(status.HTTP_412_PRECONDITION_FAILED, detail=details)
            else:
                raise HTTPException(
                    status.HTTP_500_INTERNAL_SERVER_ERROR, detail=details
                )

    @logger.catch(exclude=(HTTPException,))
    async def send_payment(
        self,
        pay_req: str,
        timeout_seconds: int,
        fee_limit_msat: int,
        amount_msat: Optional[int] = None,
    ) -> Payment:
        logger.trace(
            (
                f"logger.send_payment(pay_req={pay_req}, "
                f"timeout_seconds={timeout_seconds}, fee_limit_msat={fee_limit_msat}, "
                f"amount_msat={amount_msat})"
            )
        )

        try:
            r = router.SendPaymentRequest(
                payment_request=pay_req,
                timeout_seconds=timeout_seconds,
                fee_limit_msat=fee_limit_msat,
                amt_msat=amount_msat,
            )

            p = None
            async for response in self._router_stub.SendPaymentV2(r):
                p = Payment.from_lnd_grpc(response)
                await broadcast_msg(Event.LN_PAYMENT_STATUS, p.model_dump())
            return p
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            if (
                error.details() is not None
                and error.details().find("invalid bech32 string") > -1
            ):
                raise HTTPException(
                    status.HTTP_400_BAD_REQUEST, detail="Invalid payment request string"
                )
            elif (
                error.details() is not None
                and error.details().find("OPENSSL_internal:CERTIFICATE_VERIFY_FAILED.")
                > -1
            ):
                raise HTTPException(
                    status.HTTP_500_INTERNAL_SERVER_ERROR,
                    detail="Invalid LND credentials. SSL certificate verify failed.",
                )
            elif (
                error.details() is not None
                and error.details().find(
                    "amount must be specified when paying a zero amount invoice"
                )
                > -1
            ):
                raise HTTPException(
                    status.HTTP_400_BAD_REQUEST,
                    detail="amount must be specified when paying a zero amount invoice",
                )
            elif (
                error.details() is not None
                and error.details().find(
                    "amount must not be specified when paying a non-zero amount invoice"
                )
                > -1
            ):
                raise HTTPException(
                    status.HTTP_400_BAD_REQUEST,
                    detail=(
                        "amount must not be specified when paying a non-zero "
                        "amount invoice"
                    ),
                )
            elif (
                error.details() is not None
                and error.details().find("invoice is already paid") > -1
            ):
                raise HTTPException(
                    status.HTTP_409_CONFLICT, detail="invoice is already paid"
                )
            else:
                logger.error(error.details())
                raise HTTPException(
                    status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
                )

    @logger.catch(exclude=(HTTPException,))
    async def get_ln_info(self) -> LnInfo:
        logger.trace("logger.get_ln_info()")

        if not self._initialized:
            raise HTTPException(
                status.HTTP_503_SERVICE_UNAVAILABLE, detail="LND not fully initialized"
            )

        try:
            req = ln.GetInfoRequest()
            response = await self._lnd_stub.GetInfo(req)
            return LnInfo.from_lnd_grpc(self.get_implementation_name(), response)
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException,))
    async def _wait_wallet_fully_ready(self):
        logger.trace("logger._wait_wallet_fully_ready()")

        # This must only be called after unlocking the wallet.

        while True:
            try:
                info = await self._lnd_stub.GetInfo(ln.GetInfoRequest())

                if info is not None:
                    logger.debug(
                        (
                            "logger._wait_wallet_fully_ready() breaking out of "
                            "wait ready loop"
                        )
                    )
                    break
            except grpc.aio._call.AioRpcError as error:
                details = error.details()
                if (
                    "the RPC server is in the process of starting up, but not yet "
                    "ready to accept calls"
                ) in details:
                    # message from LND AFTER unlocking the wallet
                    await self._init_queue.put(
                        InitLnRepoUpdate(
                            state=LnInitState.BOOTSTRAPPING_AFTER_UNLOCK,
                            msg=(
                                "The RPC server is in the process of starting up, "
                                "but not yet ready to accept calls"
                            ),
                        )
                    )
                    await asyncio.sleep(0.1)
                else:
                    logger.error(f"logger.Unknown error: {details}")
                    raise

    @logger.catch(exclude=(HTTPException,))
    async def unlock_wallet(self, password: str) -> bool:
        logger.trace("logger.unlock_wallet(password=wedontlogpasswords)")

        try:
            if self._channel is None:
                self._create_stubs()

            req = unlocker.UnlockWalletRequest(wallet_password=bytes(password, "utf-8"))
            await self._wallet_unlocker.UnlockWallet(req)
            await self._wait_wallet_fully_ready()
            return True
        except grpc.aio._call.AioRpcError as error:
            if error.details().find("invalid passphrase") > -1:
                raise HTTPException(
                    status.HTTP_401_UNAUTHORIZED, detail=error.details()
                )
            elif error.details().find("wallet already unlocked") > -1:
                raise HTTPException(
                    status.HTTP_412_PRECONDITION_FAILED, detail=error.details()
                )
            else:
                raise HTTPException(
                    status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
                )

    @logger.catch(exclude=(HTTPException,))
    async def listen_invoices(self) -> AsyncGenerator[Invoice, None]:
        logger.trace("logger.listen_invoices()")

        request = ln.InvoiceSubscription()
        try:
            async for r in self._lnd_stub.SubscribeInvoices(request):
                yield Invoice.from_lnd_grpc(r)
        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException,))
    async def listen_forward_events(self) -> ForwardSuccessEvent:
        logger.trace("logger.listen_forward_events()")

        request = router.SubscribeHtlcEventsRequest()
        try:
            _fwd_cache = {}

            async for e in self._router_stub.SubscribeHtlcEvents(request):
                if e.event_type != 3:
                    continue

                evt = str(e)
                failed_event = "forward_fail_event" in evt or "link_fail_event" in evt
                if e.incoming_htlc_id not in _fwd_cache and not failed_event:
                    _fwd_cache[e.incoming_htlc_id] = e
                elif e.incoming_htlc_id in _fwd_cache and not failed_event:
                    if hasattr(e, "settle_event") and len(e.settle_event.preimage) > 0:
                        old_e = _fwd_cache[e.incoming_htlc_id]
                        del _fwd_cache[e.incoming_htlc_id]
                        amt_in_msat = old_e.forward_event.info.incoming_amt_msat
                        amt_out_msat = old_e.forward_event.info.outgoing_amt_msat
                        fee = amt_in_msat - amt_out_msat
                        yield ForwardSuccessEvent(
                            timestamp_ns=e.timestamp_ns,
                            chan_id_in=e.incoming_channel_id,
                            chan_id_out=e.outgoing_channel_id,
                            amt_in_msat=amt_in_msat,
                            amt_out_msat=amt_out_msat,
                            fee_msat=fee,
                        )
                elif failed_event and e.incoming_htlc_id in _fwd_cache:
                    del _fwd_cache[e.incoming_htlc_id]

        except grpc.aio._call.AioRpcError as error:
            _check_transient_ln_error(error)
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException,))
    async def channel_open(
        self, local_funding_amount: int, node_URI: str, target_confs: int
    ) -> str:
        logger.trace(
            (
                f"logger.channel_open(local_funding_amount={local_funding_amount}, "
                f"node_URI={node_URI}, target_confs={target_confs})"
            )
        )

        try:
            pubkey = node_URI.split("@")[0]
            host = node_URI.split("@")[1]

            # make sure to be connected to peer
            r = ln.ConnectPeerRequest(
                addr=ln.LightningAddress(pubkey=pubkey, host=host),
                perm=False,
                timeout=10,
            )
            try:
                await self._lnd_stub.ConnectPeer(r)
            except grpc.aio._call.AioRpcError as error:
                if (
                    error.details() is not None
                    and error.details().find("already connected to peer") > -1
                ):
                    logger.debug(f"already connected to peer {pubkey}")
                else:
                    raise error

            # open channel
            r = ln.OpenChannelRequest(
                node_pubkey=bytes.fromhex(pubkey),
                local_funding_amount=local_funding_amount,
                target_conf=target_confs,
            )
            async for response in self._lnd_stub.OpenChannel(r):
                return str(response.chan_pending.txid.hex())

        except grpc.aio._call.AioRpcError as error:
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException, NodeNotFoundError))
    async def peer_resolve_alias(self, node_pub: str) -> str:
        logger.trace(f"logger.peer_resolve_alias(node_pub={node_pub})")

        # get fresh list of peers and their aliases
        try:
            request = ln.NodeInfoRequest(pub_key=node_pub, include_channels=False)
            response = await self._lnd_stub.GetNodeInfo(request)

            return str(response.node.alias)
        except grpc.aio._call.AioRpcError as error:
            details = error.details()

            if "unable to find node" in details:
                raise NodeNotFoundError(node_pub)

            logger.error(details)
            raise HTTPException(status.HTTP_500_INTERNAL_SERVER_ERROR, detail=details)

    @logger.catch(exclude=(HTTPException,))
    async def channel_list(self) -> List[Channel]:
        logger.trace("logger.channel_list()")

        try:
            request = ln.ListChannelsRequest()
            response = await self._lnd_stub.ListChannels(request)

            channels = []
            for channel_grpc in response.channels:
                channel = Channel.from_lnd_grpc(channel_grpc)
                channel.peer_alias = await alias_or_empty(
                    self.peer_resolve_alias, channel.peer_publickey
                )
                channels.append(channel)

            request = ln.PendingChannelsRequest()
            response = await self._lnd_stub.PendingChannels(request)
            for channel_grpc in response.pending_open_channels:
                channel = Channel.from_lnd_grpc_pending(channel_grpc.channel)
                channel.peer_alias = await alias_or_empty(
                    self.peer_resolve_alias, channel.peer_publickey
                )
                channels.append(channel)

            return channels

        except grpc.aio._call.AioRpcError as error:
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )

    @logger.catch(exclude=(HTTPException,))
    async def channel_close(self, channel_id: int, force_close: bool) -> str:
        logger.trace(
            f"logger.channel_close(channel_id={channel_id}, force_close={force_close})"
        )

        if ":" not in channel_id:
            raise ValueError("channel_id must contain : for lnd")

        try:
            funding_txid = channel_id.split(":")[0]
            output_index = channel_id.split(":")[1]

            request = ln.CloseChannelRequest(
                channel_point=ln.ChannelPoint(
                    funding_txid_str=funding_txid, output_index=int(output_index)
                ),
                force=force_close,
                target_conf=6,
            )
            async for response in self._lnd_stub.CloseChannel(request):
                return str(response.close_pending.txid.hex())

        except grpc.aio._call.AioRpcError as error:
            raise HTTPException(
                status.HTTP_500_INTERNAL_SERVER_ERROR, detail=error.details()
            )