github.com/colincross/blueprint@v0.0.0-20150626231830-9c067caf2eb5/parser/sort.go

// Copyright 2014 Google Inc. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package parser

import (
	"sort"
	"text/scanner"
)

func SortLists(file *File) {
	for _, def := range file.Defs {
		if assignment, ok := def.(*Assignment); ok {
			sortListsInValue(assignment.Value, file)
		} else if module, ok := def.(*Module); ok {
			for _, prop := range module.Properties {
				sortListsInValue(prop.Value, file)
			}
		}
	}
	sort.Sort(commentsByOffset(file.Comments))
}

func SortList(file *File, value Value) {
	for i := 0; i < len(value.ListValue); i++ {
		// Find a set of values on contiguous lines
		line := value.ListValue[i].Pos.Line
		var j int
		for j = i + 1; j < len(value.ListValue); j++ {
			if value.ListValue[j].Pos.Line > line+1 {
				break
			}
			line = value.ListValue[j].Pos.Line
		}

		nextPos := value.EndPos
		if j < len(value.ListValue) {
			nextPos = value.ListValue[j].Pos
		}
		sortSubList(value.ListValue[i:j], nextPos, file)
		i = j - 1
	}
}

func ListIsSorted(value Value) bool {
	for i := 0; i < len(value.ListValue); i++ {
		// Find a set of values on contiguous lines
		line := value.ListValue[i].Pos.Line
		var j int
		for j = i + 1; j < len(value.ListValue); j++ {
			if value.ListValue[j].Pos.Line > line+1 {
				break
			}
			line = value.ListValue[j].Pos.Line
		}

		if !subListIsSorted(value.ListValue[i:j]) {
			return false
		}
		i = j - 1
	}

	return true
}

func sortListsInValue(value Value, file *File) {
	if value.Variable != "" {
		return
	}

	if value.Expression != nil {
		sortListsInValue(value.Expression.Args[0], file)
		sortListsInValue(value.Expression.Args[1], file)
		return
	}

	if value.Type == Map {
		for _, p := range value.MapValue {
			sortListsInValue(p.Value, file)
		}
		return
	} else if value.Type != List {
		return
	}

	SortList(file, value)
}

func sortSubList(values []Value, nextPos scanner.Position, file *File) {
	l := make(elemList, len(values))
	for i, v := range values {
		if v.Type != String {
			panic("list contains non-string element")
		}
		n := nextPos
		if i < len(values)-1 {
			n = values[i+1].Pos
		}
		l[i] = elem{v.StringValue, i, v.Pos, n}
	}

	sort.Sort(l)

	copyValues := append([]Value{}, values...)
	copyComments := append([]Comment{}, file.Comments...)

	curPos := values[0].Pos
	for i, e := range l {
		values[i] = copyValues[e.i]
		values[i].Pos = curPos
		for j, c := range copyComments {
			if c.Pos.Offset > e.pos.Offset && c.Pos.Offset < e.nextPos.Offset {
				file.Comments[j].Pos.Line = curPos.Line
				file.Comments[j].Pos.Offset += values[i].Pos.Offset - e.pos.Offset
			}
		}

		curPos.Offset += e.nextPos.Offset - e.pos.Offset
		curPos.Line++
	}
}

func subListIsSorted(values []Value) bool {
	prev := ""
	for _, v := range values {
		if v.Type != String {
			panic("list contains non-string element")
		}
		if prev > v.StringValue {
			return false
		}
		prev = v.StringValue
	}

	return true
}

type elem struct {
	s       string
	i       int
	pos     scanner.Position
	nextPos scanner.Position
}

type elemList []elem

func (l elemList) Len() int {
	return len(l)
}

func (l elemList) Swap(i, j int) {
	l[i], l[j] = l[j], l[i]
}

func (l elemList) Less(i, j int) bool {
	return l[i].s < l[j].s
}

type commentsByOffset []Comment

func (l commentsByOffset) Len() int {
	return len(l)
}

func (l commentsByOffset) Less(i, j int) bool {
	return l[i].Pos.Offset < l[j].Pos.Offset
}

func (l commentsByOffset) Swap(i, j int) {
	l[i], l[j] = l[j], l[i]
}