Filter out empty path components (#1)

thnxdev · Apr 21, 2023 · c8cbe20 · c8cbe20
1 parent e2a0ba3
commit c8cbe20
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/cmd/happy/main.go b/cmd/happy/main.go
@@ -609,7 +609,12 @@ func (t *tree) String() string {
 
 func (t *tree) Write(w *codewriter.Writer, earlyExit string, visitor func(w *codewriter.Writer, endpoint endpoint)) {
 	w.Import("strings")
-	w.L(`parts := strings.Split(r.URL.Path, "/")`)
+	w.L(`parts := []string{}`)
+	w.L(`for i, p := range strings.Split(r.URL.Path, "/") {`)
+	w.L(`	if i == 0 || p != "" {`)
+	w.L(`		parts = append(parts, p)`)
+	w.L(`	}`)
+	w.L(`}`)
 	w.L(`var params []string`)
 	w.L(`_ = params`)
 	w.L(`switch parts[0] {`)

diff --git a/testdata/main_api.go b/testdata/main_api.go