treerack/whitespace.go

package treerack

import (
	"fmt"
	"strconv"
	"strings"
)

const whitespaceName = ":ws"

func brokenRegistryError(err error) error {
	return fmt.Errorf("broken registry: %v", err)
}

func splitWhitespaceDefs(defs []definition) ([]definition, []definition) {
	var whitespaceDefs, nonWhitespaceDefs []definition
	for _, def := range defs {
		if def.commitType()&Whitespace != 0 {
			def.setCommitType(def.commitType() | Alias)
			whitespaceDefs = append(whitespaceDefs, def)
			continue
		}

		nonWhitespaceDefs = append(nonWhitespaceDefs, def)
	}

	return whitespaceDefs, nonWhitespaceDefs
}

func splitRoot(defs []definition) (definition, []definition) {
	var (
		root definition
		rest []definition
	)

	for _, def := range defs {
		if def.commitType()&Root != 0 {
			root = def
			continue
		}

		rest = append(rest, def)
	}

	return root, rest
}

func mergeWhitespaceDefs(ws []definition) definition {
	var names []string
	for _, def := range ws {
		names = append(names, def.nodeName())
	}

	return newChoice(whitespaceName, Alias, names)
}

func patchName(s ...string) string {
	return strings.Join(s, ":")
}

func applyWhitespaceToSeq(s *sequenceDefinition) []definition {
	var (
		defs  []definition
		items []SequenceItem
	)

	whitespace := SequenceItem{Name: whitespaceName, Min: 0, Max: -1}
	for i, item := range s.items {
		// TODO: there should not be max=0
		if item.Max >= 0 && item.Max <= 1 {
			if i > 0 {
				items = append(items, whitespace)
			}

			items = append(items, item)
			continue
		}

		singleItem := SequenceItem{Name: item.Name, Min: 1, Max: 1}

		restName := patchName(item.Name, s.nodeName(), "wsrest", strconv.Itoa(i))
		restDef := newSequence(restName, Alias, []SequenceItem{whitespace, singleItem})
		defs = append(defs, restDef)

		restItems := SequenceItem{Name: restName, Min: 0, Max: -1}
		if item.Min > 0 {
			restItems.Min = item.Min - 1
		}
		if item.Max > 0 {
			restItems.Min = item.Max - 1
		}

		if item.Min > 0 {
			if i > 0 {
				items = append(items, whitespace)
			}

			items = append(items, singleItem, restItems)
			continue
		}

		optItems := []SequenceItem{singleItem, restItems}
		if i > 0 {
			optItems = []SequenceItem{whitespace, singleItem, restItems}
		}

		optName := patchName(item.Name, s.nodeName(), "wsopt", strconv.Itoa(i))
		optDef := newSequence(optName, Alias, optItems)
		defs = append(defs, optDef)
		items = append(items, SequenceItem{Name: optName, Min: 0, Max: 1})
	}

	s = newSequence(s.nodeName(), s.commitType(), items)
	defs = append(defs, s)
	return defs
}

func applyWhitespaceToDefs(defs []definition) []definition {
	var defsWS []definition
	for _, def := range defs {
		if def.commitType()&NoWhitespace != 0 {
			defsWS = append(defsWS, def)
			continue
		}

		seq, ok := def.(*sequenceDefinition)
		if !ok {
			defsWS = append(defsWS, def)
			continue
		}

		defsWS = append(defsWS, applyWhitespaceToSeq(seq)...)
	}

	return defsWS
}

func applyWhitespaceToRoot(root definition) (definition, definition) {
	original, name := root, root.nodeName()
	wsName := patchName(name, "wsroot")

	original.setNodeName(wsName)
	original.setCommitType(original.commitType() &^ Root)
	original.setCommitType(original.commitType() | Alias)

	root = newSequence(name, Root, []SequenceItem{{
		Name: whitespaceName,
		Min:  0,
		Max:  -1,
	}, {
		Name: wsName,
		Min:  1,
		Max:  1,
	}, {
		Name: whitespaceName,
		Min:  0,
		Max:  -1,
	}})

	return original, root
}

func hasWhitespace(defs []definition) bool {
	for i := range defs {
		if defs[i].commitType()&Whitespace != 0 {
			return true
		}
	}

	return false
}

func applyWhitespace(defs []definition) ([]definition, definition) {
	whitespaceDefs, defs := splitWhitespaceDefs(defs)
	whitespace := mergeWhitespaceDefs(whitespaceDefs)

	defs = applyWhitespaceToDefs(defs)

	root, defs := splitRoot(defs)
	originalRoot, root := applyWhitespaceToRoot(root)

	defs = append(
		append(
			defs,
			whitespaceDefs...,
		),
		whitespace,
		originalRoot,
		root,
	)

	return defs, root
}