wip

Author: mikefarah

pkg/yqlib/operator_sort.go

@@ -24,30 +24,55 @@ func sortByOperator(d *dataTreeNavigator, context Context, expressionNode *Expre
 	for el := context.MatchingNodes.Front(); el != nil; el = el.Next() {
 		candidate := el.Value.(*CandidateNode)
 
-		if candidate.Kind != SequenceNode {
-			return context, fmt.Errorf("node at path [%v] is not an array (it's a %v)", candidate.GetNicePath(), candidate.Tag)
-		}
+		var sortableArray sortableNodeArray
+
+		if candidate.Kind == MappingNode {
+
+			sortableArray = make(sortableNodeArray, len(candidate.Content)/2)
+			log.Warningf("Sorting map: %v", NodeToString(candidate))
+			for i := 1; i < len(candidate.Content); i = i + 2 {
 
-		sortableArray := make(sortableNodeArray, len(candidate.Content))
+				originalNode := candidate.Content[i]
+				compareContext, err := d.GetMatchingNodes(context.SingleReadonlyChildContext(originalNode), expressionNode.RHS)
+				if err != nil {
+					return Context{}, err
+				}
 
-		for i, originalNode := range candidate.Content {
+				sortableArray[i/2] = sortableNode{Node: originalNode, CompareContext: compareContext, dateTimeLayout: context.GetDateTimeLayout()}
 
-			compareContext, err := d.GetMatchingNodes(context.SingleReadonlyChildContext(originalNode), expressionNode.RHS)
-			if err != nil {
-				return Context{}, err
 			}
 
-			sortableArray[i] = sortableNode{Node: originalNode, CompareContext: compareContext, dateTimeLayout: context.GetDateTimeLayout()}
+		} else if candidate.Kind == SequenceNode {
+			sortableArray = make(sortableNodeArray, len(candidate.Content))
+
+			for i, originalNode := range candidate.Content {
+
+				compareContext, err := d.GetMatchingNodes(context.SingleReadonlyChildContext(originalNode), expressionNode.RHS)
+				if err != nil {
+					return Context{}, err
+				}
+
+				sortableArray[i] = sortableNode{Node: originalNode, CompareContext: compareContext, dateTimeLayout: context.GetDateTimeLayout()}
 
+			}
+		} else {
+			return context, fmt.Errorf("node at path [%v] is not an array or map (it's a %v)", candidate.GetNicePath(), candidate.Tag)
 		}
 
 		sort.Stable(sortableArray)
 
-		sortedList := candidate.CreateReplacementWithComments(SequenceNode, "!!seq", candidate.Style)
-
-		for _, sortedNode := range sortableArray {
-			sortedList.AddChild(sortedNode.Node)
+		sortedList := candidate.CopyWithoutContent()
+		if candidate.Kind == MappingNode {
+			for _, sortedNode := range sortableArray {
+				sortedList.AddKeyValueChild(sortedNode.Node.Key, sortedNode.Node)
+			}
+		} else if candidate.Kind == SequenceNode {
+			for _, sortedNode := range sortableArray {
+				sortedList.AddChild(sortedNode.Node)
+			}
 		}
+
+		// convert array of value nodes back to map
 		results.PushBack(sortedList)
 	}
 	return context.ChildContext(results), nil