package mux
import (
"net/http"
"strings"
)
// defCodeWriter is an http.ResponseWriter that writes the given status code by
// default instead of always defaulting to a 200.
type defCodeWriter struct {
http.ResponseWriter
code int
wrote bool
}
func (w *defCodeWriter) Write(p []byte) (int, error) {
if !w.wrote {
w.wrote = true
w.WriteHeader(w.code)
}
return w.ResponseWriter.Write(p)
}
func (w *defCodeWriter) WriteHeader(statusCode int) {
w.wrote = true
w.ResponseWriter.WriteHeader(statusCode)
}
func notFoundHandler(h http.Handler) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
h.ServeHTTP(&defCodeWriter{
ResponseWriter: w,
code: http.StatusNotFound,
}, r)
}
}
func defOptions(node node) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
var verbs []string
for v := range node.handlers {
verbs = append(verbs, v)
}
w.Header().Add("Allow", strings.Join(verbs, ","))
w.Write(nil)
})
}