rest.go

package lightning

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"net/url"
	"strconv"
	"strings"

	utils "github.com/bolt-observer/go_common/utils"
	"github.com/lightningnetwork/lnd/lnrpc"
)

// GetDoFunc = signature for Do function.
type GetDoFunc func(req *http.Request) (*http.Response, error)

// HTTPAPI struct.
type HTTPAPI struct {
	DoFunc GetDoFunc
	client *http.Client
}

// Do - invokes HTTP request.
func (h *HTTPAPI) Do(req *http.Request) (*http.Response, error) {
	if h.DoFunc != nil {
		return h.DoFunc(req)
	} else if h.client != nil {
		return h.client.Do(req)
	}

	return nil, fmt.Errorf("no way to fulfill request")
}

// NewHTTPAPI returns a new HTTPAPI.
func NewHTTPAPI() *HTTPAPI {
	return &HTTPAPI{client: nil, DoFunc: nil}
}

// SetTransport - sets HTTP transport.
func (h *HTTPAPI) SetTransport(transport *http.Transport) {
	h.client = &http.Client{Transport: transport}
}

// HTTPGetInfo - invokes GetInfo method.
func (h *HTTPAPI) HTTPGetInfo(ctx context.Context, req *http.Request) (*GetInfoResponseOverride, error) {
	var info GetInfoResponseOverride

	req = req.WithContext(ctx)

	u, err := url.Parse(fmt.Sprintf("%s/v1/getinfo", req.URL))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u

	err = h.doGetRequest(req, &info)
	if err != nil {
		return nil, err
	}

	return &info, nil
}

// HTTPGetChannels - invokes GetChannels method.
func (h *HTTPAPI) HTTPGetChannels(ctx context.Context, req *http.Request) (*Channels, error) {
	var channels Channels

	req = req.WithContext(ctx)

	u, err := url.Parse(fmt.Sprintf("%s/v1/channels", req.URL))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u
	req.Method = http.MethodGet

	err = h.doGetRequest(req, &channels)
	if err != nil {
		return nil, err
	}

	return &channels, nil
}

// HTTPGetGraph - invokes GetGraph method.
func (h *HTTPAPI) HTTPGetGraph(ctx context.Context, req *http.Request, unannounced bool) (*Graph, error) {
	var graph Graph

	req = req.WithContext(ctx)

	u, err := url.Parse(fmt.Sprintf("%s/v1/graph?include_unannounced=%s", req.URL, strconv.FormatBool(unannounced)))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u
	req.Method = http.MethodGet

	err = h.doGetRequest(req, &graph)
	if err != nil {
		return nil, err
	}

	return &graph, nil
}

// HTTPGetNodeInfo - invokes GetNodeInfo method.
func (h *HTTPAPI) HTTPGetNodeInfo(ctx context.Context, req *http.Request, pubKey string, channels bool) (*GetNodeInfoOverride, error) {
	var nodeinfo GetNodeInfoOverride

	req = req.WithContext(ctx)

	u, err := url.Parse(fmt.Sprintf("%s/v1/graph/node/%s?include_channels=%s", req.URL, pubKey, strconv.FormatBool(channels)))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u
	req.Method = http.MethodGet

	err = h.doGetRequest(req, &nodeinfo)
	if err != nil {
		return nil, err
	}

	return &nodeinfo, nil
}

// HTTPGetChanInfo - invokes GetChanInfo method.
func (h *HTTPAPI) HTTPGetChanInfo(ctx context.Context, req *http.Request, chanID uint64) (*GraphEdgeOverride, error) {
	var chaninfo GraphEdgeOverride

	req = req.WithContext(ctx)

	u, err := url.Parse(fmt.Sprintf("%s/v1/graph/edge/%d", req.URL, chanID))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u
	req.Method = http.MethodGet

	err = h.doGetRequest(req, &chaninfo)
	if err != nil {
		return nil, err
	}

	return &chaninfo, nil
}

// HTTPForwardEvents - invokes ForwardEvents method.
func (h *HTTPAPI) HTTPForwardEvents(ctx context.Context, req *http.Request, input *ForwardingHistoryRequestOverride) (*ForwardingHistoryResponseOverride, error) {
	var data ForwardingHistoryResponseOverride

	req = req.WithContext(ctx)
	req.Method = http.MethodPost

	u, err := url.Parse(fmt.Sprintf("%s/v1/switch", req.URL))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	s, err := json.Marshal(input)
	if err != nil {
		return nil, err
	}
	b := bytes.NewBuffer(s)

	req.URL = u
	req.Body = io.NopCloser(b)

	resp, err := h.Do(req)
	if err != nil {
		return nil, fmt.Errorf("http request failed %s", err)
	}

	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("http got error %d", resp.StatusCode)
	}

	decoder := json.NewDecoder(resp.Body)

	err = decoder.Decode(&data)
	if err != nil {
		return nil, fmt.Errorf("got error %v", err)
	}

	return &data, nil
}

// HTTPSubscribeHtlcEvents - invokes SubscribeHtlcEvents method.
func (h *HTTPAPI) HTTPSubscribeHtlcEvents(ctx context.Context, req *http.Request) (<-chan *HtlcEventOverride, error) {
	req = req.WithContext(ctx)

	u, err := url.Parse(fmt.Sprintf("%s/v2/router/htlcevents", req.URL))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u
	req.Method = http.MethodGet

	resp, err := h.Do(req)
	if err != nil {
		return nil, fmt.Errorf("http request failed %s", err)
	}

	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("http got error %d", resp.StatusCode)
	}

	outchan := make(chan *HtlcEventOverride)

	go func() {
		var data HtlcEventOverride
		defer resp.Body.Close()
		decoder := json.NewDecoder(resp.Body)
		for {
			if ctx.Err() != nil {
				return
			}

			err := decoder.Decode(&data)
			if err != nil {
				return
			}

			outchan <- &data
		}
	}()

	return outchan, nil
}

// HTTPListInvoices - invokes ListInvoices method.
func (h *HTTPAPI) HTTPListInvoices(ctx context.Context, req *http.Request, input *ListInvoiceRequestOverride) (*ListInvoiceResponseOverride, error) {
	var data ListInvoiceResponseOverride

	req = req.WithContext(ctx)

	u, err := url.Parse(fmt.Sprintf("%s/v1/invoices?pending_only=%v&index_offset=%s&num_max_invoices=%s&reversed=%v", req.URL, input.PendingOnly, input.IndexOffset,
		input.NumMaxInvoices, input.Reversed))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u

	err = h.doGetRequest(req, &data)
	if err != nil {
		return nil, err
	}

	return &data, nil
}

// HTTPListPayments - invokes ListPayments method.
func (h *HTTPAPI) HTTPListPayments(ctx context.Context, req *http.Request, input *ListPaymentsRequestOverride) (*ListPaymentsResponseOverride, error) {
	var data ListPaymentsResponseOverride

	req = req.WithContext(ctx)

	u, err := url.Parse(fmt.Sprintf("%s/v1/payments?include_incomplete=%v&index_offset=%s&max_payments=%s&reversed=%v", req.URL, input.IncludeIncomplete, input.IndexOffset,
		input.MaxPayments, input.Reversed))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u

	err = h.doGetRequest(req, &data)
	if err != nil {
		return nil, err
	}

	return &data, nil
}

// HTTPPeers - invokes peers method.
func (h *HTTPAPI) HTTPPeers(ctx context.Context, req *http.Request, input *ConnectPeerRequestOverride) error {
	req = req.WithContext(ctx)
	req.Method = http.MethodPost

	u, err := url.Parse(fmt.Sprintf("%s/v1/peers", req.URL))
	if err != nil {
		return fmt.Errorf("invalid url %s", err)
	}

	s, err := json.Marshal(input)
	if err != nil {
		return err
	}
	b := bytes.NewBuffer(s)

	req.URL = u
	req.Body = io.NopCloser(b)

	resp, err := h.Do(req)
	if err != nil {
		return fmt.Errorf("http request failed %s", err)
	}

	defer resp.Body.Close()

	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return err
	}

	alreadyConnected := strings.Contains(string(body), "already connected to peer")
	if resp.StatusCode != http.StatusOK && !alreadyConnected {
		return fmt.Errorf("http got error %d %s", resp.StatusCode, body)
	}

	return nil
}

// GetHTTPRequest - generic method for doing HTTP requests.
func (h *HTTPAPI) GetHTTPRequest(getData GetDataCall) (*http.Request, *http.Transport, error) {
	if getData == nil {
		return nil, nil, fmt.Errorf("getData is nil")
	}

	data, err := getData()
	if err != nil {
		return nil, nil, err
	}

	myurl := data.Endpoint
	if !strings.HasPrefix(data.Endpoint, "http") {
		myurl = fmt.Sprintf("https://%s", data.Endpoint)
	}

	req, err := http.NewRequest(http.MethodGet, myurl, nil)
	if err != nil {
		return nil, nil, fmt.Errorf("new request %s", err.Error())
	}
	req.Header.Set("Grpc-Metadata-macaroon", data.MacaroonHex)

	certBytes, err := utils.SafeBase64Decode(data.CertificateBase64)
	if err != nil {
		return nil, nil, fmt.Errorf("base64 decode error %s", err)
	}

	verification := PublicCAorCert
	if data.CertVerificationType != nil {
		verification = CertificateVerification(*data.CertVerificationType)
	}

	tls, err := getTLSConfig(certBytes, data.Endpoint, verification)
	if err != nil {
		return nil, nil, fmt.Errorf("getTlsConfig failed %v", err)
	}

	trans := &http.Transport{
		TLSClientConfig: tls,
	}

	return req, trans, nil
}

func (h *HTTPAPI) doGetRequest(req *http.Request, data any) error {
	if req != nil {
		req.Method = http.MethodGet
	}
	resp, err := h.Do(req)
	if err != nil {
		return fmt.Errorf("http request failed %s", err)
	}

	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		return fmt.Errorf("http got error %d", resp.StatusCode)
	}

	decoder := json.NewDecoder(resp.Body)

	err = decoder.Decode(&data)
	if err != nil {
		return fmt.Errorf("got error %v", err)
	}

	return nil
}

// HTTPNewAddress - invokes NewAddress method.
func (h *HTTPAPI) HTTPNewAddress(ctx context.Context, req *http.Request, input *lnrpc.NewAddressRequest) (*lnrpc.NewAddressResponse, error) {
	var resp lnrpc.NewAddressResponse

	req = req.WithContext(ctx)

	t := uint32(input.Type)

	u, err := url.Parse(fmt.Sprintf("%s/v1/newaddress?type=%d", req.URL, t))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u

	err = h.doGetRequest(req, &resp)
	if err != nil {
		return nil, err
	}

	return &resp, nil
}

// HTTPBalance - invokes Balance method.
func (h *HTTPAPI) HTTPBalance(ctx context.Context, req *http.Request) (*WalletBalanceResponseOverride, error) {
	var resp WalletBalanceResponseOverride

	req = req.WithContext(ctx)

	u, err := url.Parse(fmt.Sprintf("%s/v1/balance/blockchain", req.URL))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u

	err = h.doGetRequest(req, &resp)
	if err != nil {
		return nil, err
	}

	return &resp, nil
}

func doPostRequest[U, V any](ctx context.Context, h *HTTPAPI, req *http.Request, uri string, input *U) (*V, error) {
	var reply V

	req = req.WithContext(ctx)
	req.Method = http.MethodPost

	u, err := url.Parse(uri)
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	s, err := json.Marshal(input)
	if err != nil {
		return nil, err
	}
	b := bytes.NewBuffer(s)

	req.URL = u
	req.Body = io.NopCloser(b)

	resp, err := h.Do(req)
	if err != nil {
		return nil, fmt.Errorf("http request failed %s", err)
	}

	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("http got error %d", resp.StatusCode)
	}

	decoder := json.NewDecoder(resp.Body)

	err = decoder.Decode(&reply)
	if err != nil {
		return nil, fmt.Errorf("got error %v", err)
	}

	return &reply, nil
}

// HTTPSendCoins - invokes SendCoins method.
func (h *HTTPAPI) HTTPSendCoins(ctx context.Context, req *http.Request, input *SendCoinsRequestOverride) (*lnrpc.SendCoinsResponse, error) {
	return doPostRequest[SendCoinsRequestOverride, lnrpc.SendCoinsResponse](ctx, h, req, fmt.Sprintf("%s/v1/transactions", req.URL), input)
}

// HTTPPayInvoice - invokes PayInvoice method.
func (h *HTTPAPI) HTTPPayInvoice(ctx context.Context, req *http.Request, input *SendPaymentRequestOverride) (*PaymentOverride, error) {
	return doPostRequest[SendPaymentRequestOverride, PaymentOverride](ctx, h, req, fmt.Sprintf("%s/v2/router/send", req.URL), input)
}

// HTTPTrackPayment - invokes TrackPayment method.
func (h *HTTPAPI) HTTPTrackPayment(ctx context.Context, req *http.Request, input *TrackPaymentRequestOverride) (*PaymentOverride, error) {
	var reply PaymentOverride

	req = req.WithContext(ctx)
	req.Method = http.MethodPost

	u, err := url.Parse(fmt.Sprintf("%s/v2/router/track/%s?no_inflight_updates=%v", req.URL, input.PaymentHash, strconv.FormatBool(input.NoInflightUpdates)))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u

	resp, err := h.Do(req)
	if err != nil {
		return nil, fmt.Errorf("http request failed %s", err)
	}

	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("http got error %d", resp.StatusCode)
	}

	decoder := json.NewDecoder(resp.Body)

	err = decoder.Decode(&reply)
	if err != nil {
		return nil, fmt.Errorf("got error %v", err)
	}

	return &reply, nil
}

// HTTPAddInvoice - invokes AddInvoice method.
func (h *HTTPAPI) HTTPAddInvoice(ctx context.Context, req *http.Request, input *InvoiceOverride) (*AddInvoiceResponseOverride, error) {
	return doPostRequest[InvoiceOverride, AddInvoiceResponseOverride](ctx, h, req, fmt.Sprintf("%s/v1/invoices", req.URL), input)
}

// HTTPLookupInvoice - invokes LookupInvoice method.
func (h *HTTPAPI) HTTPLookupInvoice(ctx context.Context, req *http.Request, paymentHash string) (*InvoiceOverride, error) {
	var reply InvoiceOverride

	req = req.WithContext(ctx)

	u, err := url.Parse(fmt.Sprintf("%s/v1/invoice/%s", req.URL, paymentHash))
	if err != nil {
		return nil, fmt.Errorf("invalid url %s", err)
	}

	req.URL = u

	err = h.doGetRequest(req, &reply)
	if err != nil {
		return nil, err
	}

	return &reply, nil
}