From f8dccbec808722216db2f97402dd79b76306ae57 Mon Sep 17 00:00:00 2001
From: Owen Schwartz <owen@txv.io>
Date: Wed, 29 Jan 2025 22:15:28 -0500
Subject: [PATCH] Fix save config

---
 main.go             | 5 -----
 websocket/client.go | 4 ++++
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/main.go b/main.go
index bff80ac..c475448 100644
--- a/main.go
+++ b/main.go
@@ -289,11 +289,6 @@ func main() {
 	loggerLevel := parseLogLevel(logLevel)
 	logger.GetLogger().SetLevel(parseLogLevel(logLevel))
 
-	// Validate required fields
-	if endpoint == "" || id == "" || secret == "" {
-		logger.Fatal("endpoint, id, and secret are required either via CLI flags or environment variables")
-	}
-
 	// parse the mtu string into an int
 	mtuInt, err = strconv.Atoi(mtu)
 	if err != nil {
diff --git a/websocket/client.go b/websocket/client.go
index 879a109..8a7d3f9 100644
--- a/websocket/client.go
+++ b/websocket/client.go
@@ -305,6 +305,10 @@ func (c *Client) establishConnection() error {
 	go c.readPump()
 
 	if c.onConnect != nil {
+		err := c.saveConfig()
+		if err != nil {
+			logger.Error("Failed to save config: %v", err)
+		}
 		if err := c.onConnect(); err != nil {
 			logger.Error("OnConnect callback failed: %v", err)
 		}