Print token names to help reading parser output

Author: msp301

query/parser.go

@@ -36,7 +36,7 @@ func (p *Parser) Parse() *graph.Graph {
 	for p.currentToken.Type != END {
 		ast.AddVertex(graph.Vertex{
 			Id:    uint64(p.position + 1),
-			Label: fmt.Sprint(p.currentToken.Type),
+			Label: p.currentToken.String(),
 		})
 		p.readToken()
 	}

query/tokens.go

@@ -1,5 +1,7 @@
 package query
 
+import "fmt"
+
 const (
 	TERM = iota
 	AND
@@ -18,3 +20,28 @@ type Token struct {
 	Type  TokenType
 	Value string
 }
+
+func (t *Token) String() string {
+	switch t.Type {
+	case TERM:
+		return "TERM"
+	case AND:
+		return "AND"
+	case OR:
+		return "OR"
+	case NOT:
+		return "NOT"
+	case LEFT_BRACKET:
+		return "("
+	case RIGHT_BRACKET:
+		return ")"
+	case SINGLE_QUOTE:
+		return "'"
+	case DOUBLE_QUOTE:
+		return "\""
+	case END:
+		return "END"
+	default:
+		panic(fmt.Sprintf("Unknown token: %d", t.Type))
+	}
+}