feat: pro config option

BoltzExchange · Feb 3, 2025 · 70d0bcd · 70d0bcd
1 parent fac5e57
commit 70d0bcd
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 5 deletions.
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -85,6 +85,7 @@ type Config struct {
 
 	Proxy string `long:"proxy" description:"Proxy URL to use for all Boltz API requests"`
 
+	Pro        bool `long:"pro" description:"Use boltz pro API"`
 	ReferralId string
 
 	MaxZeroConfAmount        *uint64 `long:"max-zeroconf-amount" description:"Maximum amount of sats to accept 0-conf"`
@@ -247,6 +248,10 @@ func LoadConfig(dataDir string) (*Config, error) {
 	cfg.RPC.AdminMacaroonPath = utils.ExpandDefaultPath(macaroonDir, cfg.RPC.AdminMacaroonPath, "admin.macaroon")
 	cfg.RPC.ReadonlyMacaroonPath = utils.ExpandDefaultPath(macaroonDir, cfg.RPC.ReadonlyMacaroonPath, "readonly.macaroon")
 
+	if cfg.Pro {
+		cfg.ReferralId = "pro"
+	}
+
 	createDirIfNotExists(cfg.DataDir)
 	createDirIfNotExists(macaroonDir)
 

diff --git a/internal/rpcserver/server.go b/internal/rpcserver/server.go
@@ -342,7 +342,7 @@ func initBoltz(cfg *config.Config, network *boltz.Network) (*boltz.Api, error) {
 		logger.Info("Using configured Boltz endpoint: " + boltzUrl)
 	}
 
-	boltzApi := &boltz.Api{URL: boltzUrl}
+	boltzApi := &boltz.Api{URL: boltzUrl, Referral: cfg.ReferralId}
 	if cfg.Proxy != "" {
 		proxy, err := url.Parse(cfg.Proxy)
 		if err != nil {

diff --git a/pkg/boltz/api.go b/pkg/boltz/api.go
@@ -14,8 +14,9 @@ import (
 )
 
 type Api struct {
-	URL    string
-	Client http.Client
+	URL      string
+	Client   http.Client
+	Referral string
 
 	DisablePartialSignatures bool
 }
@@ -713,7 +714,13 @@ func (boltz *Api) FetchBolt12Invoice(offer string, amountSat uint64) (string, er
 }
 
 func (boltz *Api) sendGetRequest(endpoint string, response interface{}) error {
-	res, err := boltz.Client.Get(boltz.URL + endpoint)
+	request, err := http.NewRequest("GET", boltz.URL+endpoint, nil)
+	if err != nil {
+		return err
+	}
+	request.Header.Set("Referral", boltz.Referral)
+
+	res, err := boltz.Client.Do(request)
 
 	if err != nil {
 		return err
@@ -728,12 +735,18 @@ func (boltz *Api) sendGetRequestV2(endpoint string, response interface{}) error
 
 func (boltz *Api) sendPostRequest(endpoint string, requestBody interface{}, response interface{}) error {
 	rawBody, err := json.Marshal(requestBody)
+	if err != nil {
+		return err
+	}
 
+	request, err := http.NewRequest("POST", boltz.URL+endpoint, bytes.NewBuffer(rawBody))
 	if err != nil {
 		return err
 	}
+	request.Header.Set("Referral", boltz.Referral)
+	request.Header.Set("Content-Type", "application/json")
 
-	res, err := boltz.Client.Post(boltz.URL+"/v2"+endpoint, "application/json", bytes.NewBuffer(rawBody))
+	res, err := boltz.Client.Do(request)
 
 	if err != nil {
 		return err