nyx/config/config.go

package config

import (
	"flag"
	"gopkg.in/yaml.v2"
	"io/ioutil"
	"os"
)

var configFileName = "./config.yml"

func init() {
	flag.StringVar(&configFileName, "config", "./config.yml", "Config File Location")
}

type Config struct {
	Site              SiteConfig    `yaml:"site"`             // Site/HTML Configuration
	DB                DBConfig      `yaml:"db"`               // Database Configuration
	HostnameWhiteList []string      `yaml:"hosts"`            // List of valid hostnames, ignored if empty
	ListenOn          string        `yaml:"listen_on"`        // Address & Port to use
	MasterSecret      string        `yaml:"secret"`           // Master Secret for keychain
	DisableSecurity   bool          `yaml:"disable_security"` // Disables various flags to ensure non-HTTPS requests work
	Captcha           CaptchaConfig `yaml:"captcha"`
}

const (
	CaptchaRecaptcha = "recaptcha"
	CaptchaDisabled  = "disabled"
)

type CaptchaConfig struct {
	Mode     string            `yaml:"mode"` // Captcha Mode
	Settings map[string]string `yaml:"settings,inline"`
}

type SiteConfig struct {
	Title        string `yaml:"title"`       // Site Title
	Description  string `yaml:"description"` // Site Description
	PrimaryColor string `yaml:"color"`       // Primary Color for Size
}

type DBConfig struct {
	File     string `yaml:"file"`
	ReadOnly bool   `yaml:"read_only"`
}

func Load() (*Config, error) {
	var config = &Config{
		Site: SiteConfig{
			Title:        "NyxChan",
			PrimaryColor: "#78909c",
			Description:  "NyxChan Default Configuration",
		},
		DB: DBConfig{
			File:     ":memory:",
			ReadOnly: false,
		},
		HostnameWhiteList: []string{},
		ListenOn:          ":8080",
		MasterSecret:      "changeme",
		DisableSecurity:   false,
		Captcha: CaptchaConfig{
			Mode: CaptchaDisabled,
		},
	}
	if _, err := os.Stat(configFileName); os.IsNotExist(err) {
		return config, nil
	}
	dat, err := ioutil.ReadFile(configFileName)
	if err != nil {
		return nil, err
	}
	err = yaml.Unmarshal(dat, config)
	if err != nil {
		return nil, err
	}
	return config, nil
}

func (c Config) IsHostNameValid(hostname string) bool {
	if c.HostnameWhiteList == nil {
		return true
	}
	if len(c.HostnameWhiteList) == 0 {
		return true
	}
	for _, v := range c.HostnameWhiteList {
		if v == hostname {
			return true
		}
	}
	return false
}