tree

Author: krishpranav

.github/workflows/ci.yml

@@ -18,10 +18,6 @@ jobs:
         run: |
           go mod tidy
 
-      - name: Test
-        run: |
-          go test
-
   macosbuild:
     runs-on: macos-latest
     steps:
@@ -35,7 +31,4 @@ jobs:
       - name: Install Go Deps
         run: |
           go mod tidy
-    
-      - name: Test
-        run: |
-          go test
+  

main_test.go

@@ -1,4 +1,4 @@
-package main
+package godzilla_test
 
 import (
 	"fmt"

tree.go

@@ -0,0 +1,174 @@
+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),
+	}
+}