🧑‍💻 Created ShiftPath urlutil

internal/urlutil/shift_path.go

@@ -0,0 +1,22 @@
+package urlutil
+
+import (
+	"path"
+	"strings"
+)
+
+// ShiftPath splits off the first component of p, which will be cleaned of
+// relative components before processing. head will never contain a slash and
+// tail will always be a rooted path without trailing slash.
+//
+// See: https://blog.merovius.de/posts/2017-06-18-how-not-to-use-an-http-router/
+func ShiftPath(p string) (head, tail string) {
+	p = path.Clean("/" + p)
+
+	i := strings.Index(p[1:], "/") + 1
+	if i <= 0 {
+		return p[1:], "/"
+	}
+
+	return p[1:i], p[i:]
+}

internal/urlutil/shift_path_test.go

@@ -0,0 +1,37 @@
+package urlutil_test
+
+import (
+	"testing"
+
+	"source.toby3d.me/toby3d/pub/internal/urlutil"
+)
+
+func TestShiftPath(t *testing.T) {
+	t.Parallel()
+
+	for name, tc := range map[string]struct {
+		input, expHead, expTail string
+	}{
+		"root":    {input: "/", expHead: "", expTail: "/"},
+		"file":    {input: "/foo", expHead: "foo", expTail: "/"},
+		"dir":     {input: "/foo/", expHead: "foo", expTail: "/"},
+		"dirfile": {input: "/foo/bar", expHead: "foo", expTail: "/bar"},
+		"subdir":  {input: "/foo/bar/", expHead: "foo", expTail: "/bar"},
+	} {
+		name, tc := name, tc
+
+		t.Run(name, func(t *testing.T) {
+			t.Parallel()
+
+			head, tail := urlutil.ShiftPath(tc.input)
+
+			if head != tc.expHead {
+				t.Errorf("ShiftPath(%s) = '%s', want '%s'", tc.input, head, tc.expHead)
+			}
+
+			if tail != tc.expTail {
+				t.Errorf("ShiftPath(%s) = '%s', want '%s'", tc.input, tail, tc.expTail)
+			}
+		})
+	}
+}