short: handle trailing slash

changkun · changkun · commit 370874cc4af7 · 2020-10-30T23:18:48.000+01:00
diff --git a/short.go b/short.go
@@ -127,7 +127,7 @@ func (s *server) sHandler(w http.ResponseWriter, r *http.Request) {
 			http.Redirect(w, r, "/404.html", http.StatusTemporaryRedirect)
 		}
 	}()
-	alias := strings.TrimPrefix(r.URL.Path, conf.S.Prefix)
+	alias := strings.TrimSuffix(strings.TrimPrefix(r.URL.Path, conf.S.Prefix), "/")
 	if alias == "" {
 		err = s.stats(ctx, w)
 		return

Original file line number	Diff line number	Diff line change
`@@ -127,7 +127,7 @@ func (s server) sHandler(w http.ResponseWriter, r http.Request) {`
`127`	`127`	`http.Redirect(w, r, "/404.html", http.StatusTemporaryRedirect)`
`128`	`128`	`}`
`129`	`129`	`}()`
`130`		`- alias := strings.TrimPrefix(r.URL.Path, conf.S.Prefix)`
	`130`	`+ alias := strings.TrimSuffix(strings.TrimPrefix(r.URL.Path, conf.S.Prefix), "/")`
`131`	`131`	`if alias == "" {`
`132`	`132`	`err = s.stats(ctx, w)`
`133`	`133`	`return`