bloat/config/config.go

package config

import (
	"bufio"
	"errors"
	"io"
	"os"
	"strings"
	"web/model"
)

type config struct {
	ListenAddress        string
	ClientName           string
	ClientScope          string
	ClientWebsite        string
	StaticDirectory      string
	TemplatesGlobPattern string
	DatabasePath         string
	CustomCSS            string
	PostFormats          []model.PostFormat
	Logfile              string
}

func (c *config) IsValid() bool {
	if len(c.ListenAddress) < 1 ||
		len(c.ClientName) < 1 ||
		len(c.ClientScope) < 1 ||
		len(c.ClientWebsite) < 1 ||
		len(c.StaticDirectory) < 1 ||
		len(c.TemplatesGlobPattern) < 1 ||
		len(c.DatabasePath) < 1 {
		return false
	}
	return true
}

func getDefaultConfig() *config {
	return &config{
		ListenAddress:        ":8080",
		ClientName:           "web",
		ClientScope:          "read write follow",
		ClientWebsite:        "http://localhost:8080",
		StaticDirectory:      "static",
		TemplatesGlobPattern: "templates/*",
		DatabasePath:         "database.db",
		CustomCSS:            "",
		PostFormats: []model.PostFormat{
			model.PostFormat{"Plain Text", "text/plain"},
			model.PostFormat{"HTML", "text/html"},
			model.PostFormat{"Markdown", "text/markdown"},
			model.PostFormat{"BBCode", "text/bbcode"},
		},
		Logfile: "",
	}
}

func Parse(r io.Reader) (c *config, err error) {
	c = getDefaultConfig()
	scanner := bufio.NewScanner(r)
	for scanner.Scan() {
		line := strings.TrimSpace(scanner.Text())

		if len(line) < 1 {
			continue
		}

		index := strings.IndexRune(line, '#')
		if index == 0 {
			continue
		}

		index = strings.IndexRune(line, '=')
		if index < 1 {
			return nil, errors.New("invalid config key")
		}

		key := strings.TrimSpace(line[:index])
		val := strings.TrimSpace(line[index+1 : len(line)])

		switch key {
		case "listen_address":
			c.ListenAddress = val
		case "client_name":
			c.ClientName = val
		case "client_scope":
			c.ClientScope = val
		case "client_website":
			c.ClientWebsite = val
		case "static_directory":
			c.StaticDirectory = val
		case "templates_glob_pattern":
			c.TemplatesGlobPattern = val
		case "database_path":
			c.DatabasePath = val
		case "custom_css":
			c.CustomCSS = val
		case "post_formats":
			vals := strings.Split(val, ",")
			var formats []model.PostFormat
			for _, v := range vals {
				pair := strings.Split(v, ":")
				if len(pair) != 2 {
					return nil, errors.New("invalid config key " + key)
				}
				n := strings.TrimSpace(pair[0])
				t := strings.TrimSpace(pair[1])
				if len(n) < 1 || len(t) < 1 {
					return nil, errors.New("invalid config key " + key)
				}
				formats = append(formats, model.PostFormat{
					Name: n,
					Type: t,
				})
			}
			c.PostFormats = formats
		case "logfile":
			c.Logfile = val
		default:
			return nil, errors.New("invliad config key " + key)
		}
	}

	return
}

func ParseFile(file string) (c *config, err error) {
	f, err := os.Open(file)
	if err != nil {
		return
	}
	defer f.Close()

	info, err := f.Stat()
	if err != nil {
		return
	}

	if info.IsDir() {
		return nil, errors.New("invalid config file")
	}

	return Parse(f)
}