From bba752601b998d2f09ab6389626adac446b1441f Mon Sep 17 00:00:00 2001
From: ekzyis <ekzyis@ekzyis.com>
Date: Thu, 23 Nov 2023 00:13:51 +0100
Subject: [PATCH] Make network configurable

---
 main.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index defd744..f087f8d 100644
--- a/main.go
+++ b/main.go
@@ -24,6 +24,7 @@ func init() {
 		lndAddress     string
 		lndCert        string
 		lndMacaroonDir string
+		lndNetwork     string
 		db_            *db.DB
 		lnd_           *lnd.LNDClient
 		ctx            router.ServerContext
@@ -36,6 +37,7 @@ func init() {
 	flag.StringVar(&lndAddress, "LND_ADDRESS", "localhost:10001", "LND gRPC server address")
 	flag.StringVar(&lndCert, "LND_CERT", "", "Path to LND TLS certificate")
 	flag.StringVar(&lndMacaroonDir, "LND_MACAROON_DIR", "", "LND macaroon directory")
+	flag.StringVar(&lndNetwork, "LND_NETWORK", "regtest", "LND network")
 	env.Parse()
 	figlet()
 	log.Printf("Commit:      %s", env.CommitShortSha)
@@ -50,7 +52,7 @@ func init() {
 		LndAddress:  lndAddress,
 		TLSPath:     lndCert,
 		MacaroonDir: lndMacaroonDir,
-		Network:     lndclient.NetworkRegtest,
+		Network:     lndclient.Network(lndNetwork),
 	}); err != nil {
 		log.Printf("[warn] error connecting to LND: %v\n", err)
 		lnd_ = nil