Added String requirement to Nodes

RealA10N · Aug 2, 2024 · f3d974f · f3d974f
1 parent dd21fe9
commit f3d974f
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/parse/node.go b/parse/node.go
@@ -12,7 +12,11 @@ type UnmanagedTokenView = view.UnmanagedView[lex.Token, uint32]
 type TokenViewContext = view.ViewContext[lex.Token]
 
 type Node interface {
+	// Return a reference to the node substring in the source code
 	View() source.UnmanagedSourceView
+
+	// Regenerate ("format") the code to a unique, single representation.
+	String(ctx source.SourceContext) string
 }
 
 type NodeParser[T any] interface {

diff --git a/parse/type.go b/parse/type.go
@@ -13,6 +13,10 @@ func (n TypeNode) View() source.UnmanagedSourceView {
 	return n.view
 }
 
+func (n TypeNode) String(ctx source.SourceContext) string {
+	return string(n.view.Raw(ctx))
+}
+
 type TypeParser struct{}
 
 func (TypeParser) Parse(v *TokenView) (node TypeNode, err ParsingError) {

diff --git a/parse/type_test.go b/parse/type_test.go
@@ -10,6 +10,15 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestTypeNodeStringer(t *testing.T) {
+	typView, ctx := source.NewSourceView("$i32").Detach()
+	typTok := lex.Token{Type: lex.TypToken, View: typView}
+	tkns := view.NewView[lex.Token, uint32]([]lex.Token{typTok})
+	node, err := parse.TypeParser{}.Parse(&tkns)
+	assert.Nil(t, err)
+	assert.Equal(t, "$i32", node.String(ctx))
+}
+
 func TestTypeParserSimpleCase(t *testing.T) {
 	typView, ctx := source.NewSourceView("$i32").Detach()
 	typTkn := lex.Token{Type: lex.TypToken, View: typView}