config: use shlex

This simplifies parsing and allows quoting words.

Closes: https://todo.sr.ht/~emersion/soju/43
This commit is contained in:
Simon Ser 2020-04-01 14:57:27 +02:00
parent f2a0775208
commit 886e2d3cf2
No known key found for this signature in database
GPG Key ID: 0FDE7BE0E88F5E48

View File

@ -5,8 +5,8 @@ import (
"fmt" "fmt"
"io" "io"
"os" "os"
"strings"
"unicode" "github.com/google/shlex"
) )
type TLS struct { type TLS struct {
@ -46,10 +46,22 @@ func Load(path string) (*Server, error) {
} }
func Parse(r io.Reader) (*Server, error) { func Parse(r io.Reader) (*Server, error) {
p := parser{br: bufio.NewReader(r)} scanner := bufio.NewScanner(r)
directives, err := p.file()
if err != nil { var directives []directive
return nil, fmt.Errorf("failed to parse config file: %v", err) for scanner.Scan() {
words, err := shlex.Split(scanner.Text())
if err != nil {
return nil, fmt.Errorf("failed to parse config file: %v", err)
} else if len(words) == 0 {
continue
}
name, params := words[0], words[1:]
directives = append(directives, directive{name, params})
}
if err := scanner.Err(); err != nil {
return nil, fmt.Errorf("failed to read config file: %v", err)
} }
srv := Defaults() srv := Defaults()
@ -99,91 +111,3 @@ func (d *directive) parseParams(out ...*string) error {
} }
return nil return nil
} }
type parser struct {
br *bufio.Reader
}
func (p *parser) skipSpace() error {
for {
r, _, err := p.br.ReadRune()
if err == io.EOF {
break
} else if err != nil {
return err
}
if !unicode.IsSpace(r) || r == '\n' {
p.br.UnreadRune()
break
}
}
return nil
}
func (p *parser) atom() (string, error) {
var sb strings.Builder
for {
r, _, err := p.br.ReadRune()
if err == io.EOF && sb.Len() > 0 {
break
} else if err != nil {
return "", err
}
if unicode.IsSpace(r) {
p.br.UnreadRune()
if err := p.skipSpace(); err != nil {
return "", err
}
break
}
sb.WriteRune(r)
}
return sb.String(), nil
}
func (p *parser) directive() (*directive, error) {
name, err := p.atom()
if err == io.EOF {
return nil, io.EOF
} else if err != nil {
return nil, fmt.Errorf("failed to read directive name: %v", err)
}
var params []string
for {
r, _, err := p.br.ReadRune()
if err == io.EOF {
break
} else if err != nil {
return nil, err
}
if r == '\n' {
break
}
p.br.UnreadRune()
param, err := p.atom()
if err == io.EOF {
break
} else if err != nil {
return nil, fmt.Errorf("failed to read directive param: %v", err)
}
params = append(params, param)
}
return &directive{name, params}, nil
}
func (p *parser) file() ([]directive, error) {
var l []directive
for {
d, err := p.directive()
if err == io.EOF {
break
} else if err != nil {
return nil, err
}
l = append(l, *d)
}
return l, nil
}