lexer.go

package uci

import (
	"fmt"
	"strings"
	"unicode/utf8"
)

const eof = -1

// stateFn represents the state of the scanner as a function that returns
// the next state.
//
// https://talks.golang.org/2011/lex.slide#19, #27 and following
type stateFn func(*lexer) stateFn

// lexer holds the state of the scanner.
//
// https://talks.golang.org/2011/lex.slide#22
type lexer struct {
	name  string    // used only in error reports
	input string    // the string being scanned
	start int       // start position of the current item
	pos   int       // current position in the input
	width int       // width of last rune read from input
	state stateFn   // current state (see *lexer.nextItem())
	items chan item // channel of scanned items
}

// lex starts the lexer
//
// https://talks.golang.org/2011/lex.slide#41
func lex(name, input string) *lexer {
	return &lexer{
		name:  name,
		input: input,
		state: lexKeyword,
		items: make(chan item, 2),
	}
}

// nextItem returns the next item from the input
//
// https://talks.golang.org/2011/lex.slide#41
func (l *lexer) nextItem() item {
	for l.state != nil {
		select {
		case it, ok := <-l.items:
			if ok {
				return it
			}
			return l.eof()

		default:
			s := l.state(l)
			l.state = s
			if s == nil {
				return l.stop()
			}
		}
	}
	return l.eof()
}

// stop closes the item channel, so that nextItem will (eventually)
// return an EOF token.
func (l *lexer) stop() item {
	it := l.eof()
	if l.items == nil {
		return it
	}
	if len(l.items) > 0 {
		it = <-l.items
	}
	close(l.items)
	l.items = nil
	return it
}

// eof directly returns an EOF token.
func (l *lexer) eof() item {
	return item{itemEOF, l.input[l.start:l.pos], l.pos}
}

// emit emits a token
//
// https://talks.golang.org/2011/lex.slide#25
func (l *lexer) emit(t itemType) {
	if l.pos > l.start {
		l.items <- item{t, l.input[l.start:l.pos], l.pos}
		l.start = l.pos
	}
}

// emitString emits a string token. it removes the surrounding quotes
func (l *lexer) emitString(t itemType) {
	if l.pos-1 > l.start+1 {
		l.items <- item{t, l.input[l.start+1 : l.pos-1], l.pos}
		l.start = l.pos
	}
}

// next returns the next rune in the input
//
// https://talks.golang.org/2011/lex.slide#31
func (l *lexer) next() (r rune) {
	if l.pos >= len(l.input) {
		l.width = 0
		return eof
	}
	r, l.width = utf8.DecodeRuneInString(l.input[l.pos:])
	l.pos += l.width
	return r
}

// ignore skips over the pending input before this point.
//
// https://talks.golang.org/2011/lex.slide#32
func (l *lexer) ignore() {
	l.start = l.pos
}

// backup steps back one rune. Can be called only once per call of next.
//
// https://talks.golang.org/2011/lex.slide#32
func (l *lexer) backup() {
	l.pos -= l.width
}

// peek returns but does not consume the next rune in the input.
func (l *lexer) peek() rune {
	r := l.next()
	l.backup()
	return r
}

// accept consumes the next rune if it's from the valid set.
//
// https://talks.golang.org/2011/lex.slide#34
func (l *lexer) accept(valid string) bool {
	if strings.ContainsRune(valid, l.next()) {
		return true
	}
	l.backup()
	return false
}

// acceptRun consumes a run of runes from the valid set.
//
// https://talks.golang.org/2011/lex.slide#34
func (l *lexer) acceptRun(valid string) {
	for strings.ContainsRune(valid, l.next()) {
	}
	l.backup()
}

// acceptComment consumes a line comment starting with # until eol or eof.
func (l *lexer) acceptComment() {
	if l.next() == '#' {
		for {
			r := l.next()
			if r == '\n' || r == eof {
				break
			}
		}
	}
	l.backup()
}

// acceptIdent consumes an UCI identifier [-_a-zA-Z0-9].
func (l *lexer) acceptIdent() {
	for {
		r := l.next()
		if !(r == '-' || r == '_' || 'a' <= r && r <= 'z' || 'A' <= r && r <= 'Z' || '0' <= r && r <= '9') {
			l.backup()
			break
		}
	}
}

// consumeWhitespace consumes (and ignores) space and tab characters.
func (l *lexer) consumeWhitespace() {
	for isSpace(l.peek()) {
		l.next()
	}
	l.ignore()
}

func isSpace(r rune) bool {
	return r == ' ' || r == '\t'
}

// errorf returns an error token and terminates the scan by passing back
// a nil pointer that will be the next state, terminating l.run.
//
// https://talks.golang.org/2011/lex.slide#37
func (l *lexer) errorf(format string, args ...interface{}) stateFn {
	l.items <- item{itemError, fmt.Sprintf(format, args...), l.pos}
	return nil
}

// rest returns the not-yet ingested part of l.input
func (l *lexer) rest() string {
	return l.input[l.pos:]
}

func lexKeyword(l *lexer) stateFn {
	l.acceptRun(" \t\n")
	l.ignore()
	switch curr := l.rest(); {
	case strings.HasPrefix(curr, "#"):
		return lexComment
	case strings.HasPrefix(curr, string(kwPackage)):
		return lexPackage
	case strings.HasPrefix(curr, string(kwConfig)):
		return lexConfig
	case strings.HasPrefix(curr, string(kwOption)):
		return lexOption
	case strings.HasPrefix(curr, string(kwList)):
		return lexList
	}
	if l.next() == eof {
		l.emit(itemEOF)
	} else {
		l.errorf("expected keyword (package, config, option, list) or eof")
	}
	return nil
}

func lexComment(l *lexer) stateFn {
	l.acceptComment()
	l.ignore()
	return lexKeyword
}

func lexPackage(l *lexer) stateFn {
	l.pos += len(kwPackage)
	l.emit(itemPackage)
	return lexPackageName
}

func lexPackageName(l *lexer) stateFn {
	for {
		switch r := l.next(); {
		case r == eof || r == '\n':
			return l.errorf("incomplete package name")
		case isSpace(r):
			l.ignore()
		case r == '\'' || r == '"':
			l.backup()
			return lexQuoted
		}
	}
}

func lexConfig(l *lexer) stateFn {
	l.pos += len(kwConfig)
	l.emit(itemConfig)
	l.consumeWhitespace()
	return lexConfigType
}

func lexConfigType(l *lexer) stateFn {
	l.acceptIdent()
	l.emit(itemIdent)
	l.consumeWhitespace()
	return lexOptionalName
}

func lexOptionalName(l *lexer) stateFn {
	switch r := l.next(); {
	case r == '\n':
		l.ignore()
	case r == '"' || r == '\'':
		l.backup()
		return lexQuoted
	default:
		l.acceptIdent()
		l.emit(itemString)
	}
	return lexKeyword
}

func lexOption(l *lexer) stateFn {
	l.pos += len(kwOption)
	l.emit(itemOption)
	l.consumeWhitespace()
	return lexOptionName
}

func lexList(l *lexer) stateFn {
	l.pos += len(kwList)
	l.emit(itemList)
	l.consumeWhitespace()
	return lexOptionName
}

func lexOptionName(l *lexer) stateFn {
	l.acceptIdent()
	l.emit(itemIdent)
	l.consumeWhitespace()
	return lexValue
}

func lexValue(l *lexer) stateFn {
	if r := l.peek(); r == '"' || r == '\'' {
		return lexQuoted
	}
	return lexUnquoted
}

// lexQuote scans a quoted string.
func lexQuoted(l *lexer) stateFn {
	q := l.next()
	if q != '"' && q != '\'' {
		return l.errorf("expected quotation")
	}

Loop:
	for {
		switch l.next() {
		case '\\':
			if r := l.next(); r != eof {
				break // switch
			}
			fallthrough
		case eof, '\n':
			return l.errorf("unterminated quoted string")
		case q:
			break Loop
		}
	}
	l.emitString(itemString)
	l.consumeWhitespace()
	return lexKeyword
}

func lexUnquoted(l *lexer) stateFn {
Loop:
	for {
		switch l.next() {
		case '\\':
			if r := l.next(); r != eof {
				break
			}
			fallthrough
		case eof:
			return l.errorf("unterminated unquoted string")
		case ' ', '\n':
			break Loop
		}
	}
	l.backup()
	l.emit(itemString)
	l.consumeWhitespace()
	l.accept("\n")
	l.ignore()
	return lexKeyword
}