repository cleanup

config/config.go

@@ -0,0 +1,143 @@
+package config
+
+import (
+	"log"
+	"net"
+
+	"github.com/spf13/viper"
+)
+
+// Config holds the global server-wide config.
+type Config struct {
+	HostIP  string `mapstructure:"host_ip"`
+	BinPath string `mapstructure:"bin_path"`
+	DevMode bool
+
+	DevModeOptions DevModeOptions
+	Discord        Discord
+	Database       Database
+	Launcher       Launcher
+	Sign           Sign
+	Entrance       Entrance
+}
+
+// DevModeOptions holds various debug/temporary options for use while developing Erupe.
+type DevModeOptions struct {
+	ServerName          string // To get specific instance server about (Current Players/Event Week)
+	CleanDB             bool   // Automatically wipes the DB on server reset.
+	MaxLauncherHR       bool   // Sets the HR returned in the launcher to HR9 so that you can join non-beginner worlds.
+	FixedStageID        bool   // Causes all move_stage to use the ID sl1Ns200p0a0u0 to get you into all stages
+	LogInboundMessages  bool   // Log all messages sent to the server
+	LogOutboundMessages bool   // Log all messages sent to the clients
+	MaxHexdumpLength    int    // Maximum number of bytes printed when logs are enabled
+	DivaEvent           int    // Diva Defense event status
+	FestaEvent          int    // Hunter's Festa event status
+	TournamentEvent     int    // VS Tournament event status
+	MezFesEvent         bool   // MezFes status
+	SaveDumps           SaveDumpOptions
+}
+
+type SaveDumpOptions struct {
+	Enabled   bool
+	OutputDir string
+}
+
+// Discord holds the discord integration config.
+type Discord struct {
+	Enabled           bool
+	BotToken          string
+	ServerID          string
+	RealtimeChannelID string
+	DevRoles          []string
+	DevMode           bool
+}
+
+// Database holds the postgres database config.
+type Database struct {
+	Host     string
+	Port     int
+	User     string
+	Password string
+	Database string
+}
+
+// Launcher holds the launcher server config.
+type Launcher struct {
+	Port                     int
+	UseOriginalLauncherFiles bool
+}
+
+// Sign holds the sign server config.
+type Sign struct {
+	Port int
+}
+
+// Entrance holds the entrance server config.
+type Entrance struct {
+	Port    uint16
+	Entries []EntranceServerInfo
+}
+
+// EntranceServerInfo represents an entry in the serverlist.
+type EntranceServerInfo struct {
+	IP          string
+	Type        uint8  // Server type. 0=?, 1=open, 2=cities, 3=newbie, 4=bar
+	Season      uint8  // Server activity. 0 = green, 1 = orange, 2 = blue
+	Recommended uint8  // Something to do with server recommendation on 0, 3, and 5.
+	Name        string // Server name, 66 byte null terminated Shift-JIS(JP) or Big5(TW).
+	Description string // Server description
+	// 4096(PC, PS3/PS4)?, 8258(PC, PS3/PS4)?, 8192 == nothing?
+	// THIS ONLY EXISTS IF Binary8Header.type == "SV2", NOT "SVR"!
+	AllowedClientFlags uint32
+
+	Channels []EntranceChannelInfo
+}
+
+// EntranceChannelInfo represents an entry in a server's channel list.
+type EntranceChannelInfo struct {
+	Port           uint16
+	MaxPlayers     uint16
+	CurrentPlayers uint16
+}
+
+// getOutboundIP4 gets the preferred outbound ip4 of this machine
+// From https://stackoverflow.com/a/37382208
+func getOutboundIP4() net.IP {
+	conn, err := net.Dial("udp4", "8.8.8.8:80")
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer conn.Close()
+
+	localAddr := conn.LocalAddr().(*net.UDPAddr)
+
+	return localAddr.IP.To4()
+}
+
+// LoadConfig loads the given config toml file.
+func LoadConfig() (*Config, error) {
+	viper.SetConfigName("config")
+	viper.AddConfigPath(".")
+
+	viper.SetDefault("DevModeOptions.SaveDumps", SaveDumpOptions{
+		Enabled:   false,
+		OutputDir: "savedata",
+	})
+
+	err := viper.ReadInConfig()
+	if err != nil {
+		return nil, err
+	}
+
+	c := &Config{}
+	err = viper.Unmarshal(c)
+	if err != nil {
+		return nil, err
+	}
+
+	if c.HostIP == "" {
+		c.HostIP = getOutboundIP4().To4().String()
+	}
+
+	return c, nil
+}