42 lines
1.0 KiB
Go
42 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/antlr4-go/antlr/v4"
|
|
"github.com/wyst-lang/wyst/parser"
|
|
)
|
|
|
|
func Parse(code string) (antlr.ParseTree, *parser.WystParser) {
|
|
chars := antlr.NewInputStream(code)
|
|
lexer := parser.NewWystLexer(chars)
|
|
stream := antlr.NewCommonTokenStream(lexer, 0)
|
|
|
|
p := parser.NewWystParser(stream)
|
|
// p.AddErrorListener(antlr.NewDiagnosticErrorListener(true).WithContext(p))
|
|
|
|
tree := p.Expr()
|
|
return tree, p
|
|
}
|
|
|
|
func iterateTree(node antlr.Tree, wparser *parser.WystParser, depth int) {
|
|
if ruleContext, ok := node.(antlr.RuleContext); ok {
|
|
ruleIndex := ruleContext.GetRuleIndex()
|
|
ruleName := wparser.RuleNames[ruleIndex]
|
|
fmt.Printf("%sRule: %s\n", indent(depth), ruleName)
|
|
}
|
|
|
|
if parseTree, ok := node.(antlr.ParseTree); ok {
|
|
fmt.Printf("%sNode: %s\n", indent(depth), parseTree.GetText())
|
|
}
|
|
|
|
for i := 0; i < node.GetChildCount(); i++ {
|
|
child := node.GetChild(i)
|
|
iterateTree(child, wparser, depth+1)
|
|
}
|
|
}
|
|
|
|
func indent(depth int) string {
|
|
return fmt.Sprintf("%s", string(make([]byte, depth*2)))
|
|
}
|