diff --git a/firebase.go b/firebase.go index 2c462ba..baa77df 100644 --- a/firebase.go +++ b/firebase.go @@ -1,4 +1,4 @@ -package firebase // import "github.com/captaincodeman/go-firebase" +package firebase import ( "fmt" diff --git a/server.go b/server.go index 4629948..32cc7bb 100644 --- a/server.go +++ b/server.go @@ -17,9 +17,17 @@ type ( generateURI string verifyURI string allowedOrigins []string + serveMux *http.ServeMux } ) +// ServerServeMux Uses the existing ServeMux +func ServerServeMux(m *http.ServeMux) func(*Server) { + return func(s *Server) { + s.serveMux = m + } +} + // ServerGenerateURI Sets URI for the token generation func ServerGenerateURI(uri string) func(*Server) { return func(s *Server) { @@ -66,6 +74,9 @@ func (a *Auth) Server(claimsFn CreateClaimsFunc, options ...func(*Server)) http. // endpoints to issue and verify tokens m := http.NewServeMux() + if s.serveMux != nil { + m = s.serveMux + } m.HandleFunc(s.generateURI, s.generateHandler) m.HandleFunc(s.verifyURI, s.verifyHandler) @@ -137,3 +148,4 @@ func (s *Server) verifyHandler(w http.ResponseWriter, r *http.Request) { enc := json.NewEncoder(w) enc.Encode(token.Claims()) } +