yaml.go

package usage

import (
	"bytes"
	"strings"

	yamlv3 "gopkg.in/yaml.v3"
)

const yamlCommentMark = "00__"

// markNodeAsComment marks a node as a comment which we then post process later to add the #
// We could use the yamlv3 FootComment/LineComment but this gets complicated with indentation
// especially when we have edge cases like resources that are fully commented out
func markNodeAsComment(node *yamlv3.Node) {
	node.Value = yamlCommentMark + node.Value
}

// isNodeMarkedAsCommented returns true if the node is marked as commented
func isNodeMarkedAsCommented(node *yamlv3.Node) bool {
	return strings.HasPrefix(node.Value, yamlCommentMark)
}

// replaceCommentMarks replaces the comment marks in the YAML document with #
func replaceCommentMarks(b []byte) []byte {
	return bytes.ReplaceAll(b, []byte(yamlCommentMark), []byte("# "))
}