71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
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)
|
|
}
|