36 lines
971 B
Go
36 lines
971 B
Go
|
package auth
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Middleware returns HTTP middleware that validates Bearer tokens
|
||
|
|
func (s *JWTService) Middleware() func(http.Handler) http.Handler {
|
||
|
|
return func(next http.Handler) http.Handler {
|
||
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
authHeader := r.Header.Get("Authorization")
|
||
|
|
if authHeader == "" {
|
||
|
|
http.Error(w, `{"error":"missing authorization header"}`, http.StatusUnauthorized)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
parts := strings.SplitN(authHeader, " ", 2)
|
||
|
|
if len(parts) != 2 || strings.ToLower(parts[0]) != "bearer" {
|
||
|
|
http.Error(w, `{"error":"invalid authorization format"}`, http.StatusUnauthorized)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
tokenString := parts[1]
|
||
|
|
claims, err := s.ValidateToken(tokenString)
|
||
|
|
if err != nil {
|
||
|
|
http.Error(w, `{"error":"invalid token"}`, http.StatusUnauthorized)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
ctx := WithClaims(r.Context(), claims)
|
||
|
|
next.ServeHTTP(w, r.WithContext(ctx))
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|