CONSOLE-4785: GraphQL Context Updates for multi-group impersonation

Update GraphQL context to support array-valued headers for multi-group
impersonation. Change headers type from map[string]string to map[string]interface{}
to support both single string values and []string arrays.

Author: Leo6Leo

pkg/graphql/resolver/utils.go

@@ -11,20 +11,31 @@ const HeadersKey string = "headers"
 
 func contextToHeaders(ctx context.Context, request *http.Request) {
 	if ctx.Value(HeadersKey) != nil {
-		headers, ok := ctx.Value(HeadersKey).(map[string]string)
+		headers, ok := ctx.Value(HeadersKey).(map[string]interface{})
 		if ok {
 			for key, value := range headers {
-				if value != "" {
-					request.Header.Add(key, value)
+				switch v := value.(type) {
+				case string:
+					if v != "" {
+						request.Header.Add(key, v)
+					}
+				case []string:
+					// Handle multiple values for the same header (e.g., Impersonate-Group)
+					for _, val := range v {
+						if val != "" {
+							request.Header.Add(key, val)
+						}
+					}
 				}
 			}
 		}
 	}
 }
 
 type initPayload struct {
-	ImpersonateUser  string `json:"Impersonate-User"`
-	ImpersonateGroup string `json:"Impersonate-Group"`
+	ImpersonateUser   string   `json:"Impersonate-User"`
+	ImpersonateGroup  string   `json:"Impersonate-Group"`
+	ImpersonateGroups []string `json:"Impersonate-Groups"`
 }
 
 func InitPayload(ctx context.Context, payload json.RawMessage) context.Context {
@@ -33,10 +44,19 @@ func InitPayload(ctx context.Context, payload json.RawMessage) context.Context {
 	if err != nil {
 		return ctx
 	}
-	headers, ok := ctx.Value(HeadersKey).(map[string]string)
+	headers, ok := ctx.Value(HeadersKey).(map[string]interface{})
 	if ok {
-		headers["Impersonate-User"] = initPayload.ImpersonateUser
-		headers["Impersonate-Group"] = initPayload.ImpersonateGroup
+		if initPayload.ImpersonateUser != "" {
+			headers["Impersonate-User"] = initPayload.ImpersonateUser
+		}
+		// Support both single group (backward compatibility) and multiple groups
+		if len(initPayload.ImpersonateGroups) > 0 {
+			groups := initPayload.ImpersonateGroups
+			groups = append(groups, "system:authenticated")
+			headers["Impersonate-Group"] = groups
+		} else if initPayload.ImpersonateGroup != "" {
+			headers["Impersonate-Group"] = []string{initPayload.ImpersonateGroup, "system:authenticated"}
+		}
 		ctx = context.WithValue(ctx, HeadersKey, headers)
 	}
 	return ctx

pkg/server/server.go

@@ -365,7 +365,7 @@ func (s *Server) HTTPHandler() (http.Handler, error) {
 	handler.InitPayload = resolver.InitPayload
 	graphQLHandler := handler.NewHandlerFunc(schema, gql.NewHttpHandler(schema))
 	handle("/api/graphql", authHandlerWithUser(func(user *auth.User, w http.ResponseWriter, r *http.Request) {
-		ctx := context.WithValue(context.Background(), resolver.HeadersKey, map[string]string{
+		ctx := context.WithValue(context.Background(), resolver.HeadersKey, map[string]interface{}{
 			"Authorization": fmt.Sprintf("Bearer %s", user.Token),
 		})
 		graphQLHandler(w, r.WithContext(ctx))