Import source code of lexer generator

From: https://github.com/nihei9/maleeni

1 files changed, 61 insertions, 0 deletions

diff --git a/driver/parser/token_stream.go b/driver/parser/token_stream.go
new file mode 100644
index 0000000..0bc9e32
--- /dev/null
+++ b/driver/parser/token_stream.go
@@ -0,0 +1,61 @@
+package parser
+
+import (
+	"io"
+
+	"github.com/nihei9/vartan/driver/lexer"
+	spec "github.com/nihei9/vartan/spec/grammar"
+)
+
+type vToken struct {
+	terminalID int
+	tok        *lexer.Token
+}
+
+func (t *vToken) TerminalID() int {
+	return t.terminalID
+}
+
+func (t *vToken) Lexeme() []byte {
+	return t.tok.Lexeme
+}
+
+func (t *vToken) EOF() bool {
+	return t.tok.EOF
+}
+
+func (t *vToken) Invalid() bool {
+	return t.tok.Invalid
+}
+
+func (t *vToken) Position() (int, int) {
+	return t.tok.Row, t.tok.Col
+}
+
+type tokenStream struct {
+	lex            *lexer.Lexer
+	kindToTerminal []int
+}
+
+func NewTokenStream(g *spec.CompiledGrammar, src io.Reader) (TokenStream, error) {
+	lex, err := lexer.NewLexer(lexer.NewLexSpec(g.Lexical), src)
+	if err != nil {
+		return nil, err
+	}
+
+	return &tokenStream{
+		lex:            lex,
+		kindToTerminal: g.Syntactic.KindToTerminal,
+	}, nil
+}
+
+func (l *tokenStream) Next() (VToken, error) {
+	tok, err := l.lex.Next()
+	if err != nil {
+		return nil, err
+	}
+	return &vToken{
+		terminalID: l.kindToTerminal[tok.KindID],
+		tok:        tok,
+	}, nil
+}