updated BuildEndNode to accomodate quotes

Signed-off-by: quobix <dave@quobix.com>
daveshanley · Apr 3, 2024 · 3a990a7 · 3a990a7
1 parent d3cd35a
commit 3a990a7
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/utils/end_node.go b/utils/end_node.go
@@ -11,5 +11,10 @@ func BuildEndNode(node *yaml.Node) *yaml.Node {
 	if node == nil {
 		return nil
 	}
-	return &yaml.Node{Line: node.Line, Column: node.Column + len(node.Value), Kind: node.Kind, Value: node.Value}
+	modifier := 0
+	// quotes need to be accounted for
+	if node.Style == yaml.DoubleQuotedStyle || node.Style == yaml.SingleQuotedStyle {
+		modifier = 2
+	}
+	return &yaml.Node{Line: node.Line, Column: node.Column + len(node.Value) + modifier, Kind: node.Kind, Value: node.Value}
 }