38 lines
696 B
Go
38 lines
696 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/rs/cors"
|
||
|
|
)
|
||
|
|
|
||
|
|
type mux struct {
|
||
|
|
*http.ServeMux
|
||
|
|
api *http.ServeMux
|
||
|
|
}
|
||
|
|
|
||
|
|
func httpMux(fns ...interface{ RegisterHTTP(*http.ServeMux) }) http.Handler {
|
||
|
|
mux := newMux()
|
||
|
|
for _, fn := range fns {
|
||
|
|
fn.RegisterHTTP(mux.ServeMux)
|
||
|
|
|
||
|
|
if fn, ok := fn.(interface{ RegisterAPIv1(*http.ServeMux) }); ok {
|
||
|
|
fn.RegisterAPIv1(mux.api)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return cors.AllowAll().Handler(mux)
|
||
|
|
}
|
||
|
|
func newMux() *mux {
|
||
|
|
mux := &mux{
|
||
|
|
api: http.NewServeMux(),
|
||
|
|
ServeMux: http.NewServeMux(),
|
||
|
|
}
|
||
|
|
mux.Handle("/api/v1/", http.StripPrefix("/api/v1/", mux.api))
|
||
|
|
|
||
|
|
return mux
|
||
|
|
}
|
||
|
|
func (m mux) HandleAPIv1(pattern string, handler http.Handler) {
|
||
|
|
m.api.Handle(pattern, handler)
|
||
|
|
}
|