From c57795643b11f0f2a701174c5318088907a15ae7 Mon Sep 17 00:00:00 2001
From: Jesse de Wit <witdejesse@hotmail.com>
Date: Fri, 18 Nov 2022 16:41:30 +0100
Subject: [PATCH] add cln to main

---
 main.go   | 29 ++++++++++++++++++++++++-----
 server.go |  2 +-
 2 files changed, 25 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index 1a89553..4a2160e 100644
--- a/main.go
+++ b/main.go
@@ -24,8 +24,30 @@ func main() {
 		log.Fatalf("pgConnect() error: %v", err)
 	}
 
-	client = NewLndClient()
-	interceptor := NewLndHtlcInterceptor(client)
+	runCln := os.Getenv("RUN_CLN") == "true"
+	runLnd := os.Getenv("RUN_LND") == "true"
+
+	if runCln && runLnd {
+		log.Fatalf("One of RUN_CLN or RUN_LND must be true, not both.")
+	}
+
+	if !runCln && !runLnd {
+		log.Fatalf("Either RUN_CLN or RUN_LND must be true.")
+	}
+
+	var interceptor HtlcInterceptor
+	if runCln {
+		c := NewClnClient("lightningrpc", ".")
+		client = c
+		interceptor = NewClnHtlcInterceptor(c)
+	}
+
+	if runLnd {
+		c := NewLndClient()
+		client = c
+		interceptor = NewLndHtlcInterceptor(c)
+	}
+
 	s := NewGrpcServer()
 
 	info, err := client.GetInfo()
@@ -39,9 +61,6 @@ func main() {
 		nodePubkey = info.Pubkey
 	}
 
-	go forwardingHistorySynchronize(client)
-	go channelsSynchronize(client)
-
 	var wg sync.WaitGroup
 	wg.Add(2)
 
diff --git a/server.go b/server.go
index 9d2de8e..1d1ad8c 100644
--- a/server.go
+++ b/server.go
@@ -46,7 +46,7 @@ type server struct {
 }
 
 var (
-	client              *LndClient
+	client              LightningClient
 	openChannelReqGroup singleflight.Group
 	privateKey          *btcec.PrivateKey
 	publicKey           *btcec.PublicKey