go/oauth2-proxy
1
0
Fork 0
mirror of https://github.com/oauth2-proxy/oauth2-proxy.git synced 2025-12-15 23:41:26 +02:00
Code Issues Releases Activity

Track the ReverseProxy option in the request Scope

Browse Source 
This allows for proper handling of reverse proxy based headers throughout
the lifecycle of a request.
This commit is contained in:
Nick Meves
2020-12-23 17:42:02 -08:00
parent 8e02fac2cc
commit b625de9490
4 changed files with 29 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
pkg/middleware/scope.go
View File

@@ -6,26 +6,26 @@ import (
"github.com/justinas/alice"
middlewareapi "github.com/oauth2-proxy/oauth2-proxy/v7/pkg/apis/middleware"
"github.com/oauth2-proxy/oauth2-proxy/v7/pkg/apis/options"
)
type scopeKey string
// requestScopeKey uses a typed string to reduce likelihood of clasing
// requestScopeKey uses a typed string to reduce likelihood of clashing
// with other context keys
const requestScopeKey scopeKey = "request-scope"
func NewScope() alice.Constructor {
return addScope
}
// addScope injects a new request scope into the request context.
func addScope(next http.Handler) http.Handler {
return http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
scope := &middlewareapi.RequestScope{}
contextWithScope := context.WithValue(req.Context(), requestScopeKey, scope)
requestWithScope := req.WithContext(contextWithScope)
next.ServeHTTP(rw, requestWithScope)
})
func NewScope(opts *options.Options) alice.Constructor {
return func(next http.Handler) http.Handler {
return http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
scope := &middlewareapi.RequestScope{
ReverseProxy: opts.ReverseProxy,
}
contextWithScope := context.WithValue(req.Context(), requestScopeKey, scope)
requestWithScope := req.WithContext(contextWithScope)
next.ServeHTTP(rw, requestWithScope)
})
}
}
// GetRequestScope returns the current request scope from the given request