Use ctx.isTokenNotFound

charithe · charithe · commit 8751a5781d4d · 2024-12-02T14:40:08.000Z
Signed-off-by: Charith Ellawala &lt;charith@cerbos.dev&gt;
diff --git a/parser/parser.go b/parser/parser.go
@@ -1063,12 +1063,10 @@ func (p *parser) parseDirective(ctx *context, g *TokenGroup) (*ast.DirectiveNode
 
 func (p *parser) parseComment(ctx *context) (ast.Node, error) {
 	cm := p.parseHeadComment(ctx)
-	tk := ctx.currentToken()
-	if tk == nil {
+	if ctx.isTokenNotFound() {
 		return cm, nil
 	}
-
-	node, err := p.parseToken(ctx, tk)
+	node, err := p.parseToken(ctx, ctx.currentToken())
 	if err != nil {
 		return nil, err
 	}

Original file line number	Diff line number	Diff line change
`@@ -1063,12 +1063,10 @@ func (p parser) parseDirective(ctx context, g TokenGroup) (ast.DirectiveNode`
`1063`	`1063`
`1064`	`1064`	`func (p parser) parseComment(ctx context) (ast.Node, error) {`
`1065`	`1065`	`cm := p.parseHeadComment(ctx)`
`1066`		`- tk := ctx.currentToken()`
`1067`		`- if tk == nil {`
	`1066`	`+ if ctx.isTokenNotFound() {`
`1068`	`1067`	`return cm, nil`
`1069`	`1068`	`}`
`1070`		`-`
`1071`		`- node, err := p.parseToken(ctx, tk)`
	`1069`	`+ node, err := p.parseToken(ctx, ctx.currentToken())`
`1072`	`1070`	`if err != nil {`
`1073`	`1071`	`return nil, err`
`1074`	`1072`	`}`