config.go

package main

import (
	"os"

	"gopkg.in/yaml.v2"
)

type Tablist struct {
	Header []string `yaml:"header"`
	Footer []string `yaml:"footer"`
}

type Messages struct {
	NotInWhitelist          string `yaml:"not_in_whitelist"`
	Banned                  string `yaml:"banned"`
	ServerFull              string `yaml:"server_full"`
	AlreadyPlaying          string `yaml:"already_playing"`
	PlayerJoin              string `yaml:"player_join"`
	PlayerLeave             string `yaml:"player_leave"`
	UnknownCommand          string `yaml:"unknown_command"`
	ProtocolNew             string `yaml:"protocol_new"`
	ProtocolOld             string `yaml:"protocol_old"`
	InsufficientPermissions string `yaml:"insufficient_permissions"`
	ReloadComplete          string `yaml:"reload_complete"`
	ServerClosed            string `yaml:"server_closed"`
	OnlineMode              string `yaml:"online_mode"`
}

type Icon struct {
	Path   string `yaml:"path"`
	Enable bool   `yaml:"enable"`
}

type Chat struct {
	Format string `yaml:"format"`
	Colors bool   `yaml:"colors"`
	Enable bool   `yaml:"enable"`
}

type Whitelist struct {
	Enforce bool `yaml:"enforce"`
	Enable  bool `yaml:"enable"`
}

type Config struct {
	ServerName         string    `yaml:"server_name"`
	ServerIP           string    `yaml:"server_ip"`
	ServerPort         int       `yaml:"server_port"`
	ViewDistance       int       `yaml:"view_distance"`
	SimulationDistance int       `yaml:"simulation_distance"`
	MOTD               string    `yaml:"motd"`
	Icon               Icon      `yaml:"icon"`
	Whitelist          Whitelist `yaml:"whitelist"`
	Gamemode           string    `yaml:"gamemode"`
	Hardcore           bool      `yaml:"hardcore"`
	MaxPlayers         int       `yaml:"max_players"`
	Online             bool      `yaml:"online_mode"`
	Tablist            Tablist   `yaml:"tablist"`
	Chat               Chat      `yaml:"chat"`
	Messages           Messages  `yaml:"messages"`
}

func LoadConfig() *Config {
	config := &Config{}

	file, err := os.Open("config.yml")
	if err != nil {
		file.Close()
		config = &Config{
			ServerName: "Dynamite",
			ServerIP:   "0.0.0.0",
			ServerPort: 25565,
			MOTD:       "A Dynamite Minecraft Server",
			Whitelist: Whitelist{
				Enforce: false,
				Enable:  false,
			},
			Gamemode:           "survival",
			Hardcore:           false,
			MaxPlayers:         200,
			Online:             true,
			ViewDistance:       10,
			SimulationDistance: 10,
			Messages: Messages{
				NotInWhitelist:          "You are not whitelisted.",
				Banned:                  "You are banned from this server.",
				ServerFull:              "The server is full.",
				AlreadyPlaying:          "You are already playing on this server with a different client.",
				PlayerJoin:              "§e%player% has joined the game",
				PlayerLeave:             "§e%player% has left the game",
				UnknownCommand:          "§cUnknown command. Please use '/help' for a list of commands.",
				ProtocolNew:             "Your protocol is too new!",
				ProtocolOld:             "Your protocol is too old!",
				InsufficientPermissions: "§cYou aren't permitted to use this command.",
				ReloadComplete:          "§aReload complete.",
				ServerClosed:            "Server closed.",
				OnlineMode:              "The server is in online mode.",
			},
			Icon: Icon{
				Path:   "server-icon.png",
				Enable: false,
			},
			Tablist: Tablist{
				Header: []string{},
				Footer: []string{},
			},
			Chat: Chat{
				Colors: false,
				Format: "<%player_prefix%%player%> %message%",
				Enable: true,
			},
		}
		file, _ := os.Create("config.yml")
		e := yaml.NewEncoder(file)
		e.Encode(&config)
		return config
	}
	defer file.Close()

	d := yaml.NewDecoder(file)

	if err := d.Decode(&config); err != nil {
		return nil
	}

	return config
}