mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2024-11-19 18:11:30 +01:00
93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
from os import getenv
|
|
import lnd_grpc # https://github.com/willcl-ark/lnd_grpc/blob/master/lnd_grpc/lightning.py
|
|
from .base import InvoiceResponse, PaymentResponse, PaymentStatus, Wallet
|
|
|
|
|
|
class LndWallet(Wallet):
|
|
"""https://api.lightning.community/rest/index.html#lnd-rest-api-reference"""
|
|
|
|
def __init__(self):
|
|
|
|
endpoint = getenv("LND_GRPC_ENDPOINT")
|
|
self.endpoint = endpoint[:-1] if endpoint.endswith("/") else endpoint
|
|
self.port = getenv("LND_GRPC_PORT")
|
|
self.auth_admin = getenv("LND_ADMIN_MACAROON")
|
|
self.auth_invoice = getenv("LND_INVOICE_MACAROON")
|
|
self.auth_read = getenv("LND_READ_MACAROON")
|
|
self.auth_cert = getenv("LND_CERT")
|
|
|
|
def create_invoice(self, amount: int, mem: str = "") -> InvoiceResponse:
|
|
|
|
lnd_rpc = lnd_grpc.Client(
|
|
lnd_dir = None,
|
|
macaroon_path = self.auth_invoice,
|
|
tls_cert_path = self.auth_cert,
|
|
network = 'mainnet',
|
|
grpc_host = self.endpoint,
|
|
grpc_port = self.port
|
|
)
|
|
|
|
lndResponse = lnd_rpc.add_invoice(
|
|
memo = mem,
|
|
value = amount,
|
|
expiry = 600,
|
|
private = True
|
|
)
|
|
|
|
ok, checking_id, payment_request, error_message = True, lndResponse.r_hash, lndResponse.payment_request, None
|
|
|
|
return InvoiceResponse(ok, checking_id, payment_request, error_message)
|
|
|
|
def pay_invoice(self, bolt11: str) -> PaymentResponse:
|
|
|
|
lnd_rpc = lnd_grpc.Client(
|
|
lnd_dir = None,
|
|
macaroon_path = self.auth_admin,
|
|
tls_cert_path = self.auth_cert,
|
|
network = 'mainnet',
|
|
grpc_host = self.endpoint,
|
|
grpc_port = self.port
|
|
)
|
|
|
|
payinvoice = lnd_rpc.pay_invoice( # https://github.com/willcl-ark/lnd_grpc/blob/cf938c51c201f078e8bbe9e19ffc2d038f3abf7f/lnd_grpc/lightning.py#L439
|
|
payment_request = _payreq,
|
|
)
|
|
|
|
ok, checking_id, fee_msat, error_message = True, None, 0, None
|
|
|
|
if payinvoice.payment_error:
|
|
ok, error_message = False, payinvoice.payment_error
|
|
else:
|
|
checking_id = payinvoice.payment_hash
|
|
|
|
return PaymentResponse(ok, checking_id, fee_msat, error_message)
|
|
|
|
def get_invoice_status(self, checking_id: str) -> PaymentStatus:
|
|
|
|
lnd_rpc = lnd_grpc.Client(
|
|
lnd_dir = None,
|
|
macaroon_path = self.auth_invoice,
|
|
tls_cert_path = self.auth_cert,
|
|
network = 'mainnet',
|
|
grpc_host = self.endpoint,
|
|
grpc_port = self.port
|
|
)
|
|
|
|
for _response in lnd_rpc.subscribe_single_invoice(_hash):
|
|
# myQueue.put(_response) # ???
|
|
|
|
if _response.state == 1:
|
|
|
|
return PaymentStatus("settled")
|
|
|
|
invoiceThread = threading.Thread(
|
|
target=detectPayment,
|
|
args=[lndResponse.r_hash, ],
|
|
daemon=True
|
|
)
|
|
invoiceThread.start()
|
|
|
|
def get_payment_status(self, checking_id: str) -> PaymentStatus:
|
|
|
|
return PaymentStatus(True)
|