Fix a bug ast.Merge removes comments from SequenceNode (#570)

* fix: fix a bug that merge of sequences loses comments * fix: append null if ValueHeadComments is empty
goccy · Dec 5, 2024 · a91fb5b · a91fb5b
1 parent 5d22eed
commit a91fb5b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ast/ast.go b/ast/ast.go
@@ -1475,6 +1475,11 @@ func (n *SequenceNode) Merge(target *SequenceNode) {
 	column := n.Start.Position.Column - target.Start.Position.Column
 	target.AddColumn(column)
 	n.Values = append(n.Values, target.Values...)
+	if len(target.ValueHeadComments) == 0 {
+		n.ValueHeadComments = append(n.ValueHeadComments, make([]*CommentGroupNode, len(target.Values))...)
+		return
+	}
+	n.ValueHeadComments = append(n.ValueHeadComments, target.ValueHeadComments...)
 }
 
 // SetIsFlowStyle set value to IsFlowStyle field recursively.