config: use shlex
This simplifies parsing and allows quoting words. Closes: https://todo.sr.ht/~emersion/soju/43
This commit is contained in:
parent
f2a0775208
commit
886e2d3cf2
112
config/config.go
112
config/config.go
@ -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
|
|
||||||
}
|
|
||||||
|
Loading…
Reference in New Issue
Block a user