From a50c14f83bacf1d9ff3ce64ff33d3824457174bd Mon Sep 17 00:00:00 2001
From: Matt Curtis <matt@obscu.ro>
Date: Wed, 18 Dec 2024 10:30:04 +0000
Subject: [PATCH] edb-connect: fix edb default host and allow caller to set it

---
 tools/edbconnect/main/main.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/tools/edbconnect/main/main.go b/tools/edbconnect/main/main.go
index 42b2c265b8..eaa5caa3ec 100644
--- a/tools/edbconnect/main/main.go
+++ b/tools/edbconnect/main/main.go
@@ -12,6 +12,12 @@ import (
 )
 
 func main() {
+	// optionally set edbHost from cli flag, default to obscuronode-edgelessdb-0
+	edbHost := "obscuronode-edgelessdb-0"
+	if len(os.Args) > 1 {
+		edbHost = os.Args[1]
+	}
+
 	fmt.Println("Retrieving Edgeless DB credentials...")
 	creds, found, err := edgelessdb.LoadCredentialsFromFile()
 	if err != nil {
@@ -29,9 +35,9 @@ func main() {
 	}
 	fmt.Println("TLS config created. Connecting to Edgeless DB...")
 	testlog.SetupSysOut()
-	db, err := edgelessdb.ConnectToEdgelessDB("obscuronode-edgelessdb", cfg, testlog.Logger())
+	db, err := edgelessdb.ConnectToEdgelessDB(edbHost, cfg, testlog.Logger())
 	if err != nil {
-		fmt.Println("Error connecting to Edgeless DB:", err)
+		fmt.Printf("Error connecting to Edgeless DB at %s: %v\n", edbHost, err)
 		panic(err)
 	}
 	fmt.Println("Connected to Edgeless DB.")