default header reset

README.md

@@ -27,7 +27,7 @@ func main() {
 	r.Use(middleware1, middleware2)
 
 	// let's add a route
-	r.Get("/hello", func(w http.ResponseWriter, r *http.Request) {
+	r.GET("/hello", func(w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte("i am route /hello"))
 	})
 	// r.Post(pattern string, h http.HandlerFunc)
@@ -37,7 +37,7 @@ func main() {
 	// you can inline middleware(s) to a route
 	r.
 		With(mwInline).
-		Get("/hello-2", func(w http.ResponseWriter, r *http.Request) {
+		GET("/hello-2", func(w http.ResponseWriter, r *http.Request) {
 			w.Write([]byte("i am route /hello-2 with my own middleware"))
 		})
 
@@ -64,13 +64,13 @@ func main() {
 	// create a group of few routes with their own middlewares
 	r.Group(func(grp *mux.Router) {
 		grp.Use(mwGroup)
-		grp.Get("/group", func(w http.ResponseWriter, r *http.Request) {
+		grp.GET("/group", func(w http.ResponseWriter, r *http.Request) {
 			w.Write([]byte("i am route /group"))
 		})
 	})
 
 	// catches all
-	r.Get("/", func(w http.ResponseWriter, r *http.Request) {
+	r.GET("/", func(w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte("hello there"))
 	})
 

example/main.go

@@ -25,7 +25,7 @@ func main() {
 	r.Use(middleware1, middleware2)
 
 	// let's add a route
-	r.Get("/hello", func(w http.ResponseWriter, r *http.Request) {
+	r.GET("/hello", func(w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte("i am route /hello"))
 	})
 	// r.Post(pattern string, h http.HandlerFunc)
@@ -35,7 +35,7 @@ func main() {
 	// you can inline middleware(s) to a route
 	r.
 		With(mwInline).
-		Get("/hello-2", func(w http.ResponseWriter, r *http.Request) {
+		GET("/hello-2", func(w http.ResponseWriter, r *http.Request) {
 			w.Write([]byte("i am route /hello-2 with my own middleware"))
 		})
 
@@ -62,13 +62,13 @@ func main() {
 	// create a group of few routes with their own middlewares
 	r.Group(func(grp *mux.Router) {
 		grp.Use(mwGroup)
-		grp.Get("/group", func(w http.ResponseWriter, r *http.Request) {
+		grp.GET("/group", func(w http.ResponseWriter, r *http.Request) {
 			w.Write([]byte("i am route /group"))
 		})
 	})
 
 	// catches all
-	r.Get("/", func(w http.ResponseWriter, r *http.Request) {
+	r.GET("/", func(w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte("hello there"))
 	})
 

router.go

@@ -14,9 +14,10 @@ type Router struct {
 }
 
 func NewRouter() *Router {
-	return &Router{
+	r := &Router{
 		mux: http.NewServeMux(),
 	}
+	return r
 }
 
 // Use will register middleware(s) with router stack
@@ -27,48 +28,46 @@ func (r *Router) Use(h ...func(http.Handler) http.Handler) {
 	r.middlewares = append(r.middlewares, h...)
 }
 
-// Get method route
+// GET method route
-func (r *Router) Get(pattern string, h http.HandlerFunc) {
+func (r *Router) GET(pattern string, h http.HandlerFunc) {
 	r.handlerFunc(http.MethodGet, pattern, h)
 }
 
-// Head method route
+// HEAD method route
-func (r *Router) Head(pattern string, h http.HandlerFunc) {
+func (r *Router) HEAD(pattern string, h http.HandlerFunc) {
 	r.handlerFunc(http.MethodHead, pattern, h)
 }
 
-// Post method route
+// POST method route
-func (r *Router) Post(pattern string, h http.HandlerFunc) {
+func (r *Router) POST(pattern string, h http.HandlerFunc) {
 	r.handlerFunc(http.MethodPost, pattern, h)
 }
 
-// Put method route
+// PUT method route
-func (r *Router) Put(pattern string, h http.HandlerFunc) {
+func (r *Router) PUT(pattern string, h http.HandlerFunc) {
 	r.handlerFunc(http.MethodPut, pattern, h)
 }
 
-// Patch method route
+// PATCH method route
-func (r *Router) Patch(pattern string, h http.HandlerFunc) {
+func (r *Router) PATCH(pattern string, h http.HandlerFunc) {
 	r.handlerFunc(http.MethodPatch, pattern, h)
 }
 
-// Delete method route
+// DELETE method route
-func (r *Router) Delete(pattern string, h http.HandlerFunc) {
+func (r *Router) DELETE(pattern string, h http.HandlerFunc) {
 	r.handlerFunc(http.MethodDelete, pattern, h)
 }
 
-// Connect method route
+// CONNECT method route
-func (r *Router) Connect(pattern string, h http.HandlerFunc) {
+func (r *Router) CONNECT(pattern string, h http.HandlerFunc) {
 	r.handlerFunc(http.MethodConnect, pattern, h)
-}
+} // OPTIONS method route
-
+func (r *Router) OPTIONS(pattern string, h http.HandlerFunc) {
-// Options method route
-func (r *Router) Options(pattern string, h http.HandlerFunc) {
 	r.handlerFunc(http.MethodOptions, pattern, h)
 }
 
-// Trace method route
+// TRACE method route
-func (r *Router) Trace(pattern string, h http.HandlerFunc) {
+func (r *Router) TRACE(pattern string, h http.HandlerFunc) {
 	r.handlerFunc(http.MethodTrace, pattern, h)
 }
 
@@ -144,23 +143,7 @@ func (r *Router) ServeHTTP(w http.ResponseWriter, req *http.Request) {
 	r.mux.ServeHTTP(w, req)
 }
 
-// Proxy are request and
+// TODO: proxy for aws lambda
-func (r *Router) Proxy(w http.ResponseWriter, req *http.Request) {
-	if r == nil {
-		panic("mux: method ServeHTTP called on nil")
-	}
-
-	h, pattern := r.mux.Handler(req)
-	if pattern == "" {
-		http.Error(w, "Not Found", http.StatusNotFound)
-		return
-	}
-
-	// ensure we run all the middlewares
-	h = stack(r.middlewares, h)
-	// serve
-	h.ServeHTTP(w, req)
-}
 
 // stack middlewares(http handler) in order they are passed (FIFO)
 func stack(middlewares []func(http.Handler) http.Handler, endpoint http.Handler) http.Handler {

router_serve.go

@@ -3,6 +3,7 @@ package mux
 import (
 	"context"
 	"errors"
+	"io"
 	"log/slog"
 	"net/http"
 	"os"
@@ -13,6 +14,21 @@ type ServeCB func(srv *http.Server) error
 
 // Serve with graceful shutdown
 func (r *Router) Serve(cb ServeCB) {
+	// catch all options
+	// lets get it thorugh all middlewares
+	r.OPTIONS("/", func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Content-Length", "0")
+		if r.ContentLength != 0 {
+			// Read up to 4KB of OPTIONS body (as mentioned in the
+			// spec as being reserved for future use), but anything
+			// over that is considered a waste of server resources
+			// (or an attack) and we abort and close the connection,
+			// courtesy of MaxBytesReader's EOF behavior.
+			mb := http.MaxBytesReader(w, r.Body, 4<<10)
+			io.Copy(io.Discard, mb)
+		}
+	})
+
 	srv := &http.Server{
 		Handler: r,
 	}