Expose paren Position on DefStmt (#504)

google · Sep 11, 2023 · 95963e0 · 95963e0
1 parent c9e9adf
commit 95963e0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/syntax/parse.go b/syntax/parse.go
@@ -177,15 +177,17 @@ func (p *parser) parseStmt(stmts []Stmt) []Stmt {
 func (p *parser) parseDefStmt() Stmt {
 	defpos := p.nextToken() // consume DEF
 	id := p.parseIdent()
-	p.consume(LPAREN)
+	lparen := p.consume(LPAREN)
 	params := p.parseParams()
-	p.consume(RPAREN)
+	rparen := p.consume(RPAREN)
 	p.consume(COLON)
 	body := p.parseSuite()
 	return &DefStmt{
 		Def:    defpos,
 		Name:   id,
+		Lparen: lparen,
 		Params: params,
+		Rparen: rparen,
 		Body:   body,
 	}
 }

diff --git a/syntax/syntax.go b/syntax/syntax.go
@@ -123,7 +123,9 @@ type DefStmt struct {
 	commentsRef
 	Def    Position
 	Name   *Ident
+	Lparen Position
 	Params []Expr // param = ident | ident=expr | * | *ident | **ident
+	Rparen Position
 	Body   []Stmt
 
 	Function interface{} // a *resolve.Function, set by resolver