59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
package mux
|
|
|
|
import (
|
|
"net/http"
|
|
)
|
|
|
|
type mux struct {
|
|
*http.ServeMux
|
|
api *http.ServeMux
|
|
wellknown *http.ServeMux
|
|
handler http.Handler
|
|
}
|
|
|
|
func (mux *mux) Add(fns ...interface{ RegisterHTTP(*http.ServeMux) }) {
|
|
for _, fn := range fns {
|
|
// log.Printf("HTTP: %T", fn)
|
|
fn.RegisterHTTP(mux.ServeMux)
|
|
|
|
if fn, ok := fn.(interface{ RegisterAPIv1(*http.ServeMux) }); ok {
|
|
// log.Printf("APIv1: %T", fn)
|
|
fn.RegisterAPIv1(mux.api)
|
|
}
|
|
|
|
if fn, ok := fn.(interface{ RegisterWellKnown(*http.ServeMux) }); ok {
|
|
// log.Printf("WellKnown: %T", fn)
|
|
fn.RegisterWellKnown(mux.wellknown)
|
|
}
|
|
|
|
if fn, ok := fn.(interface{ RegisterMiddleware(http.Handler) http.Handler }); ok {
|
|
hdlr := mux.handler
|
|
// log.Printf("WellKnown: %T", fn)
|
|
mux.handler = fn.RegisterMiddleware(hdlr)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (mux *mux) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
mux.handler.ServeHTTP(w, r)
|
|
}
|
|
|
|
|
|
func New() *mux {
|
|
mux := &mux{
|
|
api: http.NewServeMux(),
|
|
wellknown: http.NewServeMux(),
|
|
ServeMux: http.NewServeMux(),
|
|
}
|
|
mux.Handle("/v1/", http.StripPrefix("/v1", mux.api))
|
|
mux.Handle("/api/v1/", http.StripPrefix("/api/v1", mux.api))
|
|
mux.Handle("/.well-known/", http.StripPrefix("/.well-known", mux.wellknown))
|
|
mux.handler = mux.ServeMux
|
|
|
|
return mux
|
|
}
|
|
|
|
type RegisterHTTP func(*http.ServeMux)
|
|
|
|
func (fn RegisterHTTP) RegisterHTTP(mux *http.ServeMux) { fn(mux) }
|