From b958e9eefe903e7670d8de1f0c172abb09ace8a9 Mon Sep 17 00:00:00 2001
From: henrygd <hank@henrygd.me>
Date: Fri, 21 Feb 2025 00:56:22 -0500
Subject: [PATCH] fix: agent network not set to unix if addr provided by env
 var

---
 beszel/cmd/agent/agent.go      |  4 +++-
 beszel/cmd/agent/agent_test.go | 10 ++++++++++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/beszel/cmd/agent/agent.go b/beszel/cmd/agent/agent.go
index 631abc0..9622a67 100644
--- a/beszel/cmd/agent/agent.go
+++ b/beszel/cmd/agent/agent.go
@@ -117,6 +117,8 @@ func main() {
 
 	flag.Parse()
 
+	opts.addr = opts.getAddress()
+
 	var serverConfig agent.ServerOptions
 	var err error
 	serverConfig.Keys, err = opts.loadPublicKeys()
@@ -124,7 +126,7 @@ func main() {
 		log.Fatal("Failed to load public keys:", err)
 	}
 
-	serverConfig.Addr = opts.getAddress()
+	serverConfig.Addr = opts.addr
 	serverConfig.Network = opts.getNetwork()
 
 	agent := agent.NewAgent()
diff --git a/beszel/cmd/agent/agent_test.go b/beszel/cmd/agent/agent_test.go
index cc72fb7..31a48b2 100644
--- a/beszel/cmd/agent/agent_test.go
+++ b/beszel/cmd/agent/agent_test.go
@@ -54,6 +54,16 @@ func TestGetAddress(t *testing.T) {
 			},
 			expected: "7070",
 		},
+		{
+			name: "use unix socket from env var",
+			opts: cmdOptions{
+				addr: "",
+			},
+			envVars: map[string]string{
+				"ADDR": "/tmp/beszel.sock",
+			},
+			expected: "/tmp/beszel.sock",
+		},
 		{
 			name: "flag takes precedence over env vars",
 			opts: cmdOptions{