NetworkTool/server/internal/config/config.go

package config

import (
	"flag"
	"fmt"
	"net"
)

const ServerVersion = "2026.05.13.1700"

type Config struct {
	HTTPHost         string
	HTTPPort         int
	UDPHost          string
	UDPPort          int
	MaintenanceIP    string
	MaintenanceCIDR  string
	AdminPassword    string
	ServerVersion    string
	DeviceIDFallback string
}

func Load(args []string) Config {
	cfg := Config{
		HTTPHost:         "0.0.0.0",
		HTTPPort:         48888,
		UDPHost:          "0.0.0.0",
		UDPPort:          50000,
		MaintenanceIP:    "",
		MaintenanceCIDR:  "",
		AdminPassword:    "Dt123$",
		ServerVersion:    ServerVersion,
		DeviceIDFallback: "networktool-device",
	}

	fs := flag.NewFlagSet("networktool-server", flag.ContinueOnError)
	fs.StringVar(&cfg.MaintenanceIP, "ip", cfg.MaintenanceIP, "maintenance IPv4 address, must be in 169.254.0.0/16 when set")
	fs.IntVar(&cfg.HTTPPort, "port", cfg.HTTPPort, "HTTP listen port")
	fs.StringVar(&cfg.AdminPassword, "password", cfg.AdminPassword, "admin password")
	_ = fs.Parse(args)

	if cfg.MaintenanceIP != "" {
		parsed := net.ParseIP(cfg.MaintenanceIP)
		ipv4 := parsed.To4()
		if parsed == nil || ipv4 == nil || ipv4[0] != 169 || ipv4[1] != 254 {
			panic(fmt.Sprintf("invalid maintenance ip: %s", cfg.MaintenanceIP))
		}
		cfg.MaintenanceCIDR = fmt.Sprintf("%s/16", cfg.MaintenanceIP)
	}

	return cfg
}