Fix panic when parsing unattached comment (#565)

* Fix panic when parsing unattached comment --------- Signed-off-by: Charith Ellawala <[email protected]>
goccy · Dec 2, 2024 · 5d22eed · 5d22eed
1 parent 2ab584e
commit 5d22eed
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/parser/parser.go b/parser/parser.go
@@ -1063,6 +1063,9 @@ func (p *parser) parseDirective(ctx *context, g *TokenGroup) (*ast.DirectiveNode
 
 func (p *parser) parseComment(ctx *context) (ast.Node, error) {
 	cm := p.parseHeadComment(ctx)
+	if ctx.isTokenNotFound() {
+		return cm, nil
+	}
 	node, err := p.parseToken(ctx, ctx.currentToken())
 	if err != nil {
 		return nil, err

diff --git a/parser/parser_test.go b/parser/parser_test.go
@@ -1471,6 +1471,14 @@ foo: | # comment
 			yaml: `
 foo: > # comment
   x: 42
+`,
+		},
+		{
+			name: "unattached comment",
+			yaml: `
+# This comment is in its own document
+---
+a: b
 `,
 		},
 	}
-Original file line number
+Diff line change
@@ Expand Up / @@ -1471,6 +1471,14 @@ foo: | # comment @@
     			yaml: `
     foo: > # comment
       x: 42
+    `,
+    		},
+    		{
+    			name: "unattached comment",
+    			yaml: `
+    # This comment is in its own document
+    ---
+    a: b
     `,
     		},
     	}
@@ Expand Down @@