package auth import ( "net/http" "strings" "github.com/micro/go-micro/v2/auth" ) // CombinedAuthHandler wraps a server and authenticates requests func CombinedAuthHandler(h http.Handler) http.Handler { return authHandler{ handler: h, auth: auth.DefaultAuth, } } type authHandler struct { handler http.Handler auth auth.Auth } const ( // BearerScheme is the prefix in the auth header BearerScheme = "Bearer " ) func (h authHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { loginURL := h.auth.Options().LoginURL // Return if the user disabled auth on this endpoint excludes := h.auth.Options().Exclude if len(loginURL) > 0 { excludes = append(excludes, loginURL) } for _, e := range excludes { if e == req.URL.Path { h.handler.ServeHTTP(w, req) return } } var token string if header := req.Header.Get("Authorization"); len(header) > 0 { // Extract the auth token from the request if strings.HasPrefix(header, BearerScheme) { token = header[len(BearerScheme):] } } else { // Get the token out the cookies if not provided in headers if c, err := req.Cookie("micro-token"); err == nil && c != nil { token = strings.TrimPrefix(c.Value, auth.CookieName+"=") req.Header.Set("Authorization", BearerScheme+token) } } // If the token is valid, allow the request if _, err := h.auth.Verify(token); err == nil { h.handler.ServeHTTP(w, req) return } // If there is no auth login url set, 401 if loginURL == "" { w.WriteHeader(401) } // Redirect to the login path http.Redirect(w, req, loginURL, http.StatusTemporaryRedirect) }