configurable email header names

tgerakitis · Apr 4, 2020 · 07c26c8 · 07c26c8
1 parent ffa3b03
commit 07c26c8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/internal/config.go b/internal/config.go
@@ -40,6 +40,7 @@ type Config struct {
 	CookieDomains           []CookieDomain       `long:"cookie-domain" env:"COOKIE_DOMAIN" description:"Domain to set auth cookie on, can be set multiple times"`
 	InsecureCookie          bool                 `long:"insecure-cookie" env:"INSECURE_COOKIE" description:"Use insecure cookies"`
 	CookieName              string               `long:"cookie-name" env:"COOKIE_NAME" default:"_forward_auth" description:"ID Cookie Name"`
+	EmailHeaderNames        CommaSeparatedList   `long:"email-header-names" env:"EMAIL_HEADER_NAMES" default:"X-Forwarded-User" description:"Headers which will contain the users email address"`
 	UserCookieName          string               `long:"user-cookie-name" env:"USER_COOKIE_NAME" default:"_forward_auth_name" description:"User Cookie Name"`
 	CSRFCookieName          string               `long:"csrf-cookie-name" env:"CSRF_COOKIE_NAME" default:"_forward_auth_csrf" description:"CSRF Cookie Name"`
 	GroupsSessionName       string               `long:"groups-session-name" env:"GROUPS_SESSION_NAME" default:"_forward_auth_claims" description:"Groups Session Name"`

diff --git a/internal/server.go b/internal/server.go
@@ -178,7 +178,9 @@ func (s *Server) AuthHandler(rule string) http.HandlerFunc {
 
 		// Valid request
 		logger.Debugf("Allow request from %s", email)
-		w.Header().Set("X-Forwarded-User", email)
+		for _, headerName := range config.EmailHeaderNames {
+			w.Header().Set(headerName, email)
+		}
 
 		if config.EnableImpersonation {
 			// Set impersonation headers