Handle error.

endpoint.go

@@ -1,7 +1,5 @@
 package espresso
 
-import "net/http"
-
 type EndpointBuilder interface {
 	BindPath(key string, v any) EndpointBuilder
 	End() BindErrors
@@ -25,18 +23,3 @@ func newEndpoint() *Endpoint {
 		HeadParams:  make(map[string]BindParam),
 	}
 }
-
-func (e *Endpoint) serveHTTP(w http.ResponseWriter, r *http.Request) {
-	if r.Method != e.Method {
-		http.NotFound(w, r)
-		return
-	}
-
-	ctx := &runtimeContext{
-		ctx:      r.Context(),
-		endpoint: e,
-		request:  r,
-		response: w,
-	}
-	ctx.Next()
-}

router.go

@@ -64,6 +64,28 @@ func (g *router) register(ctx *buildtimeContext, fn HandleFunc) {
 
 	pattern := ctx.endpoint.Method + " " + path
 	g.mux.HandleFunc(pattern, func(w http.ResponseWriter, r *http.Request) {
-		endpoint.serveHTTP(w, r)
+		wr := &responseWriter{
+			ResponseWriter: w,
+		}
+
+		ctx := &runtimeContext{
+			ctx:      r.Context(),
+			endpoint: &endpoint,
+			request:  r,
+			response: wr,
+		}
+
+		ctx.Next()
+
+		if wr.hasWritten || ctx.err != nil {
+			return
+		}
+
+		code := http.StatusInternalServerError
+		if httpCoder, ok := ctx.err.(HTTPError); ok {
+			code = httpCoder.HTTPCode()
+		}
+		w.WriteHeader(code)
+		fmt.Fprint(w, ctx.err.Error())
 	})
 }

server.go

@@ -46,6 +46,7 @@ func (s *Espresso) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	if err != nil {
 		panic(err)
 	}
+
 	r = r.WithContext(ctx)
 	s.mux.ServeHTTP(w, r)
 }