cmd take account with env var

This commit is contained in:
adman
2026-07-16 22:16:04 +02:00
parent 7f878f40a1
commit e324a95727
+9 -1
View File
@@ -3,6 +3,7 @@ package messenger
import ( import (
"flag" "flag"
"fmt" "fmt"
"os"
"strconv" "strconv"
"strings" "strings"
@@ -17,10 +18,17 @@ import (
"github.com/admin-else/strom/mc/proto_generated/v1_21_11" "github.com/admin-else/strom/mc/proto_generated/v1_21_11"
) )
func stror(a, b string) string {
if a != "" {
return a
}
return b
}
var ( var (
cmd = flag.NewFlagSet("status", flag.ContinueOnError) cmd = flag.NewFlagSet("status", flag.ContinueOnError)
connectToFlag = cmd.String("addr", "", "server address to connect to") connectToFlag = cmd.String("addr", "", "server address to connect to")
accFlag = cmd.String("acc", "strom_messenger", "the name to use if longer will be treated as ygg token") accFlag = cmd.String("acc", stror(os.Getenv("STROM_ACC"), "strom_messenger"), "the name to use if longer will be treated as ygg token")
) )
type Messenger struct { type Messenger struct {