feat: add HeaderSessionType constant and set it in Session middleware (#54)

Author: klaidliadon

common.go

@@ -19,7 +19,8 @@ import (
 )
 
 const (
-	HeaderAccessKey = "X-Access-Key"
+	HeaderAccessKey   = "X-Access-Key"
+	HeaderSessionType = "Session-Type"
 )
 
 type AccessKeyFunc func(*http.Request) string

middleware.go

@@ -146,9 +146,9 @@ func Session(cfg Options) func(next http.Handler) http.Handler {
 			if _, ok := GetSessionType(ctx); ok {
 				// Track this as a SpecialKey session for now.
 				// TODO: Remove once node-gateway SpecialKey support is gone.
+				w.Header().Set(HeaderSessionType, "SpecialKey")
 				httplog.SetAttrs(ctx, slog.String("sessionType", "SpecialKey"))
 				requestsCounter.Inc(sessionLabels{SessionType: "SpecialKey", RateLimited: "false"})
-
 				next.ServeHTTP(w, r)
 				return
 			}
@@ -185,7 +185,7 @@ func Session(cfg Options) func(next http.Handler) http.Handler {
 				httplog.SetAttrs(ctx, slog.String("account", accountClaim))
 
 				if cfg.UserStore != nil {
-					user, isAdmin, err := cfg.UserStore.GetUser(ctx, accountClaim)
+					user, isAdmin, err := cfg.UserStore.GetUser(context.WithValue(ctx, proto.HTTPRequestCtxKey, r), accountClaim)
 					if err != nil {
 						cfg.ErrHandler(r, w, err)
 						return
@@ -260,6 +260,7 @@ func Session(cfg Options) func(next http.Handler) http.Handler {
 			}
 
 			ctx = WithSessionType(ctx, sessionType)
+			w.Header().Set(HeaderSessionType, sessionType.String())
 			httplog.SetAttrs(ctx, slog.String("sessionType", sessionType.String()))
 
 			ww, ok := w.(middleware.WrapResponseWriter)