123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128 |
- package config
- import (
- "bufio"
- "errors"
- "io"
- "os"
- "strings"
- "bloat/model"
- )
- type config struct {
- ListenAddress string
- ClientName string
- ClientScope string
- ClientWebsite string
- SingleInstance string
- StaticDirectory string
- TemplatesPath 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.TemplatesPath) < 1 ||
- len(c.DatabasePath) < 1 {
- return false
- }
- return true
- }
- func Parse(r io.Reader) (c *config, err error) {
- c = new(config)
- 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 "single_instance":
- c.SingleInstance = val
- case "static_directory":
- c.StaticDirectory = val
- case "templates_path":
- c.TemplatesPath = 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 "log_file":
- c.LogFile = val
- default:
- return nil, errors.New("invalid 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)
- }
|