git.iko.soy/typeable/oauth2-proxy
on f8ddc13 help sign in
typeable/oauth2-proxy/pkg/middleware/scope.go
33 lines 937 B history raw
1package middleware
3import (
4 "net/http"
6 "github.com/google/uuid"
7 "github.com/justinas/alice"
8 middlewareapi "github.com/oauth2-proxy/oauth2-proxy/v7/pkg/apis/middleware"
9)
11func NewScope(reverseProxy bool, idHeader string) alice.Constructor {
12 return func(next http.Handler) http.Handler {
13 return http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
14 scope := &middlewareapi.RequestScope{
15 ReverseProxy: reverseProxy,
16 RequestID: genRequestID(req, idHeader),
17 }
18 req = middlewareapi.AddRequestScope(req, scope)
19 next.ServeHTTP(rw, req)
20 })
21 }
24// genRequestID sets a request-wide ID for use in logging or error pages.
25// If a RequestID header is set, it uses that. Otherwise, it generates a random
26// UUID for the lifespan of the request.
27func genRequestID(req *http.Request, idHeader string) string {
28 rid := req.Header.Get(idHeader)
29 if rid != "" {
30 return rid
31 }
32 return uuid.New().String()