elvish/glob/parse.go

package glob

import (
	"bytes"
	"unicode/utf8"
)

func Parse(s string) Pattern {
	segments := []Segment{}
	add := func(seg Segment) {
		segments = append(segments, seg)
	}
	p := &parser{s, 0, 0}

rune:
	for {
		r := p.next()
		switch r {
		case EOF:
			break rune
		case '?':
			add(Segment{Question, ""})
		case '*':
			n := 1
			for p.next() == '*' {
				n++
			}
			p.backup()
			if n == 1 {
				add(Segment{Star, ""})
			} else {
				add(Segment{StarStar, ""})
			}
		case '/':
			for p.next() == '/' {
			}
			p.backup()
			add(Segment{Slash, ""})
		default:
			var literal bytes.Buffer
		literal:
			for {
				switch r {
				case '?', '*', '/', EOF:
					break literal
				case '\\':
					r = p.next()
					if r == EOF {
						break literal
					}
					literal.WriteRune(r)
				default:
					literal.WriteRune(r)
				}
				r = p.next()
			}
			p.backup()
			add(Segment{Literal, literal.String()})
		}
	}
	return Pattern{segments}
}

// XXX Contains duplicate code with parse/parser.go.

type parser struct {
	src     string
	pos     int
	overEOF int
}

const EOF rune = -1

func (ps *parser) next() rune {
	if ps.pos == len(ps.src) {
		ps.overEOF += 1
		return EOF
	}
	r, s := utf8.DecodeRuneInString(ps.src[ps.pos:])
	ps.pos += s
	return r
}

func (ps *parser) backup() {
	if ps.overEOF > 0 {
		ps.overEOF -= 1
		return
	}
	_, s := utf8.DecodeLastRuneInString(ps.src[:ps.pos])
	ps.pos -= s
}