From 1c4533f1f2951a123d5251bbf49e7bce7b785005 Mon Sep 17 00:00:00 2001
From: Ben Morgenstren <ben.morgenstern@gmail.com>
Date: Tue, 14 Jan 2025 22:13:36 -0500
Subject: [PATCH] adding reading agent ssh key from file

---
 beszel/cmd/agent/agent.go | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/beszel/cmd/agent/agent.go b/beszel/cmd/agent/agent.go
index 6dfa653..6a78395 100644
--- a/beszel/cmd/agent/agent.go
+++ b/beszel/cmd/agent/agent.go
@@ -7,6 +7,7 @@ import (
 	"log"
 	"os"
 	"strings"
+	"io/ioutil"
 )
 
 func main() {
@@ -25,7 +26,16 @@ func main() {
 	if pubKeyEnv, exists := os.LookupEnv("KEY"); exists {
 		pubKey = []byte(pubKeyEnv)
 	} else {
-		log.Fatal("KEY environment variable is not set")
+		keyFile := os.Getenv("KEY_FILE")
+		if keyFile != "" {
+			if keyData, err := ioutil.ReadFile(keyFile); err == nil {
+				pubKey = keyData
+			} else {
+				log.Fatalf("Failed to read key from file '%s': %v", keyFile, err)
+			}
+		} else {
+			log.Fatal("KEY environment variable is not set, and KEY_FILE environment variable is not set")
+		}
 	}
 
 	addr := ":45876"