Initial commit

core/config.go

@@ -0,0 +1,81 @@
+package core
+
+import (
+	"bytes"
+	"os"
+
+	"github.com/BurntSushi/toml"
+)
+
+type ForwardRule struct {
+	Protocol      string `toml:"protocol"`
+	TailscalePort int    `toml:"tailscale_port"`
+	LocalAddr     string `toml:"local_addr"`
+}
+
+type ConnectRule struct {
+	Protocol  string `toml:"protocol"`
+	LocalPort int    `toml:"local_port"`
+	LocalAddr string `toml:"local_addr"`
+	DstAddr   string `toml:"dst_addr"`
+	LanEnable *bool  `toml:"lan_enable"`
+	LanMotd   string `toml:"lan_motd"`
+}
+
+type Core struct {
+	AuthKey      string `toml:"auth_key"`
+	ControlURL   string `toml:"control_url"`
+	Hostname     string `toml:"hostname"`
+	Ephemeral    bool   `toml:"ephemeral"`
+	AcceptRoutes bool   `toml:"accept_routes"`
+}
+
+func (r ConnectRule) LANEnabled() bool {
+	if r.LanEnable != nil {
+		return *r.LanEnable
+	}
+	return r.Protocol == "minecraft"
+}
+
+func (r ConnectRule) LANMotdOr(def string) string {
+	if r.LanMotd != "" {
+		return r.LanMotd
+	}
+	return def
+}
+
+type Config struct {
+	Core    Core                     `toml:"core"`
+	Forward map[string][]ForwardRule `toml:"forward"`
+	Connect map[string][]ConnectRule `toml:"connect"`
+}
+
+func LoadConfig(path string) (*Config, error) {
+	hostname, err := os.Hostname()
+	if err != nil {
+		hostname = "unknown"
+	}
+	cfg := &Config{
+		Core: Core{
+			Hostname:     hostname,
+			Ephemeral:    true,
+			AcceptRoutes: true,
+		},
+		Forward: make(map[string][]ForwardRule),
+		Connect: make(map[string][]ConnectRule),
+	}
+	if _, err = os.Stat(path); err != nil {
+		// write a basic config
+		buf := new(bytes.Buffer)
+		err = toml.NewEncoder(buf).Encode(cfg)
+		if err != nil {
+			return nil, err
+		}
+		err = os.WriteFile(path, buf.Bytes(), 0644)
+	}
+	_, err = toml.DecodeFile(path, cfg)
+	if err != nil {
+		return nil, err
+	}
+	return cfg, nil
+}