fankes/beszel
2
0
Fork 0
mirror of https://github.com/fankes/beszel.git synced 2025-10-19 01:39:34 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

adding reading agent ssh key from file

Browse Source
This commit is contained in:
Ben Morgenstren
2025-01-14 22:13:36 -05:00
parent 5d88599c9a
commit 1c4533f1f2
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
beszel/cmd/agent/agent.go
View File

@@ -7,6 +7,7 @@ import (
"log" "log"
"os" "os"
"strings" "strings"
"io/ioutil"
) )
func main() { func main() {
@@ -25,7 +26,16 @@ func main() {
if pubKeyEnv, exists := os.LookupEnv("KEY"); exists { if pubKeyEnv, exists := os.LookupEnv("KEY"); exists {
pubKey = []byte(pubKeyEnv) pubKey = []byte(pubKeyEnv)
} else { } 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" addr := ":45876"