yq/pkg/yqlib/operator_sort_keys.go

package yqlib

import (
	"sort"

	yaml "gopkg.in/yaml.v3"
)

func sortKeysOperator(d *dataTreeNavigator, context Context, expressionNode *ExpressionNode) (Context, error) {

	for el := context.MatchingNodes.Front(); el != nil; el = el.Next() {
		candidate := el.Value.(*CandidateNode)
		rhs, err := d.GetMatchingNodes(context.SingleChildContext(candidate), expressionNode.Rhs)
		if err != nil {
			return Context{}, err
		}

		for childEl := rhs.MatchingNodes.Front(); childEl != nil; childEl = childEl.Next() {
			node := unwrapDoc(childEl.Value.(*CandidateNode).Node)
			if node.Kind == yaml.MappingNode {
				sortKeys(node)
			}
			if err != nil {
				return Context{}, err
			}
		}

	}
	return context, nil
}

func sortKeys(node *yaml.Node) {
	keys := make([]string, len(node.Content)/2)
	keyBucket := map[string]*yaml.Node{}
	valueBucket := map[string]*yaml.Node{}
	var contents = node.Content
	for index := 0; index < len(contents); index = index + 2 {
		key := contents[index]
		value := contents[index+1]
		keys[index/2] = key.Value
		keyBucket[key.Value] = key
		valueBucket[key.Value] = value
	}
	sort.Strings(keys)
	sortedContent := make([]*yaml.Node, len(node.Content))
	for index := 0; index < len(keys); index = index + 1 {
		keyString := keys[index]
		sortedContent[index*2] = keyBucket[keyString]
		sortedContent[1+(index*2)] = valueBucket[keyString]
	}
	node.Content = sortedContent
}