feat: add cmd flags

2026-04-23 21:50:53 +02:00
parent 24be1cedeb
commit 5e2043d54e
8 changed files with 120 additions and 156 deletions
@@ -0,0 +1,101 @@
+package bootstrap
+
+import (
+	"flag"
+	"os"
+	"reflect"
+	"strconv"
+)
+
+type Environment struct {
+	Version  string `env:"VERSION" default:"0.0.1" flag:"version"`
+	Codename string `env:"CODENAME" default:"Magical Anomaly" flag:"codename"`
+	LogLevel string `env:"LOG_LEVEL" default:"debug" flag:"log-level"`
+
+	DataDirectory         string `env:"DATA_DIR" default:"./data" flag:"data-dir" usage:"option to specify where the state data gets stored"`
+	ContentDirectory      string `env:"CONTENT_DIR" default:"./content" flag:"content-dir" usage:"option to specify where the content gets stored"`
+	Hostname              string `env:"HOSTNAME" default:"0.0.0.0" flag:"hostname" usage:"option specify the address/hostname to bind the api server to"`
+	Port                  int    `env:"PORT" default:"8080" flag:"port" usage:"option to specify the port to bind the api server to"`
+	AuthenticationEnabled bool   `env:"AUTH_ENABLED" default:"true" flag:"auth" usage:"option to disable authentication"`
+
+	WatchdogEnabled  bool   `env:"WATCHDOG_ENABLED" default:"true" flag:"watchdog" usage:"option to disable watchdog"`
+	WatchdogInterval int    `env:"WATCHDOG_INTERVAL" default:"60" flag:"watchdog-interval" usage:"option to specify the interval in second(s) on which watchdog runs"`
+	WatchdogSyncMode string `env:"WATCHDOG_SYNC_MODE" default:"strict" flag:"watchdog-mode" usage:"option to specify the mode watchdog will run with: strict|sync|dry"`
+}
+
+func loadFromEnv(env *Environment) {
+	v := reflect.ValueOf(env).Elem()
+	t := v.Type()
+
+	for i := 0; i < t.NumField(); i++ {
+		field := t.Field(i)
+		valueField := v.Field(i)
+
+		// It is of great importance we set these fields above
+		key := field.Tag.Get("env")
+		fallback := field.Tag.Get("default")
+
+		// If the variable is not found or is empty
+		raw, ok := os.LookupEnv(key)
+		if !ok || raw == "" {
+			raw = fallback
+		}
+
+		switch valueField.Kind() {
+		case reflect.String:
+			valueField.SetString(raw)
+
+		case reflect.Int:
+			if parsed, err := strconv.Atoi(raw); err == nil {
+				valueField.SetInt(int64(parsed))
+			}
+
+		case reflect.Bool:
+			if parsed, err := strconv.ParseBool(raw); err == nil {
+				valueField.SetBool(parsed)
+			}
+		}
+	}
+}
+
+func bindFlags(env *Environment) {
+	v := reflect.ValueOf(env).Elem()
+	t := v.Type()
+
+	for i := 0; i < t.NumField(); i++ {
+		field := t.Field(i)
+		valueField := v.Field(i)
+
+		flagName := field.Tag.Get("flag")
+		flagUsage := field.Tag.Get("usage")
+		if flagName == "" {
+			continue
+		}
+
+		switch valueField.Kind() {
+		case reflect.String:
+			flag.StringVar(valueField.Addr().Interface().(*string), flagName,
+				valueField.String(), flagUsage,
+			)
+		case reflect.Int:
+			flag.IntVar(valueField.Addr().Interface().(*int), flagName,
+				int(valueField.Int()), flagUsage,
+			)
+		case reflect.Bool:
+			flag.BoolVar(
+				valueField.Addr().Interface().(*bool), flagName,
+				valueField.Bool(), flagUsage,
+			)
+		}
+	}
+}
+
+func LoadConfig() Environment {
+	var env Environment
+
+	loadFromEnv(&env)
+	bindFlags(&env)
+	flag.Parse()
+
+	return env
+}