mirror of
https://github.com/henrygd/beszel.git
synced 2025-12-17 02:36:17 +01:00
43 lines
747 B
Go
43 lines
747 B
Go
package main
|
|
|
|
import (
|
|
"beszel"
|
|
"beszel/internal/agent"
|
|
"beszel/internal/update"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
// handle flags / subcommands
|
|
if len(os.Args) > 1 {
|
|
switch os.Args[1] {
|
|
case "-v":
|
|
fmt.Println(beszel.AppName+"-agent", beszel.Version)
|
|
case "update":
|
|
update.UpdateBeszelAgent()
|
|
}
|
|
os.Exit(0)
|
|
}
|
|
|
|
var pubKey []byte
|
|
if pubKeyEnv, exists := os.LookupEnv("KEY"); exists {
|
|
pubKey = []byte(pubKeyEnv)
|
|
} else {
|
|
log.Fatal("KEY environment variable is not set")
|
|
}
|
|
|
|
port := ":45876"
|
|
if p, exists := os.LookupEnv("PORT"); exists {
|
|
// allow passing an address in the form of "127.0.0.1:45876"
|
|
if !strings.Contains(p, ":") {
|
|
p = ":" + p
|
|
}
|
|
port = p
|
|
}
|
|
|
|
agent.NewAgent(pubKey, port).Run()
|
|
}
|