tree.go

package godzilla

import (
	"strings"
)

type nodeType uint8

const (
	static nodeType = iota
	root
	param
	catchAll
)

type node struct {
	path     string
	param    *node
	children map[string]*node
	nType    nodeType
	handlers handlersChain
}

func (n *node) addRoute(path string, handlers handlersChain) {
	currentNode := n
	originalPath := path
	path = path[1:]

	paramNames := make(map[string]bool)

	for {
		pathLen := len(path)
		if pathLen == 0 {
			if currentNode.handlers != nil {
				panic("handlers are already registered for path '" + originalPath + "'")
			}

			routeHandlers := make(handlersChain, len(handlers))
			copy(routeHandlers, handlers)

			currentNode.handlers = routeHandlers
			break
		}

		segmentDelimiter := strings.Index(path, "/")
		if segmentDelimiter == -1 {
			segmentDelimiter = pathLen
		}

		pathSegment := path[:segmentDelimiter]
		if pathSegment[0] == ':' || pathSegment[0] == '*' {

			if len(currentNode.children) > 0 {
				panic("parameter " + pathSegment +
					" conflicts with existing static children in path '" +
					originalPath + "'")
			}

			if currentNode.param != nil {
				if currentNode.param.path[0] == '*' {
					panic("parameter " + pathSegment +
						" conflicts with catch all (*) route in path '" +
						originalPath + "'")
				} else if currentNode.param.path != pathSegment {
					panic("parameter " + pathSegment + " in new path '" +
						originalPath + "' conflicts with existing wildcard '" +
						currentNode.param.path)
				}
			}

			if currentNode.param == nil {
				var nType nodeType
				if pathSegment[0] == '*' {
					nType = catchAll
					if pathLen > 1 {
						panic("catch all (*) routes are only allowed " +
							"at the end of the path in path '" +
							originalPath + "'")
					}
				} else {
					nType = param
					if _, ok := paramNames[pathSegment]; ok {
						panic("parameter " + pathSegment +
							" must be unique in path '" + originalPath + "'")
					} else {
						paramNames[pathSegment] = true
					}
				}

				currentNode.param = &node{
					path:     pathSegment,
					nType:    nType,
					children: make(map[string]*node),
				}
			}
			currentNode = currentNode.param
		} else {

			if currentNode.param != nil {
				panic(pathSegment + "' conflicts with existing parameter " +
					currentNode.param.path + " in path '" + originalPath + "'")
			}
			if child, ok := currentNode.children[pathSegment]; ok {
				currentNode = child

			} else {
				child = &node{
					path:     pathSegment,
					nType:    static,
					children: make(map[string]*node),
				}
				currentNode.children[pathSegment] = child
				currentNode = child
			}
		}

		if pathLen > segmentDelimiter {
			segmentDelimiter++
		}
		path = path[segmentDelimiter:]
	}
}

func (n *node) matchRoute(path string, ctx *context) handlersChain {
	pathLen := len(path)
	if pathLen > 0 && path[0] != '/' {
		return nil
	}

	currentNode := n

	if pathLen > 0 {
		path = path[1:]
	}

	for {
		pathLen = len(path)

		if pathLen == 0 || currentNode.nType == catchAll {
			return currentNode.handlers
		}
		segmentDelimiter := strings.Index(path, "/")
		if segmentDelimiter == -1 {
			segmentDelimiter = pathLen
		}
		pathSegment := path[:segmentDelimiter]

		if pathLen > segmentDelimiter {
			segmentDelimiter++
		}
		path = path[segmentDelimiter:]

		if currentNode.param != nil {
			currentNode = currentNode.param
			ctx.paramValues[currentNode.path[1:]] = pathSegment
			continue
		}

		if child, ok := currentNode.children[pathSegment]; ok {
			currentNode = child
			continue
		}

		return nil
	}
}

func createRootNode() *node {
	return &node{
		nType:    root,
		path:     "/",
		children: make(map[string]*node),
	}
}