Middleware: Add team metadata to HTTP handlers (#71010)

Signed-off-by: bergquist <carl.bergquist@gmail.com>

pkg/middleware/requestmeta/request_metadata.go

@@ -0,0 +1,75 @@
+package requestmeta
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/grafana/grafana/pkg/web"
+)
+
+const (
+	TeamAlerting = "alerting"
+	TeamAuth     = "auth"
+	TeamCore     = "core"
+)
+
+type rMDContextKey struct{}
+
+type RequestMetaData struct {
+	Team string
+}
+
+var requestMetaDataContextKey = rMDContextKey{}
+
+// SetupRequestMetadata injects defaul request metadata values
+// on the request context.
+func SetupRequestMetadata() web.Middleware {
+	return func(next http.Handler) http.Handler {
+		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			rmd := defaultRequestMetadata()
+
+			ctx := context.WithValue(r.Context(), requestMetaDataContextKey, rmd)
+			*r = *r.WithContext(ctx)
+
+			next.ServeHTTP(w, r)
+		})
+	}
+}
+
+// GetRequestMetaData returns the request metadata for the context.
+// if request metadata is missing it will return the default values.
+func GetRequestMetaData(ctx context.Context) *RequestMetaData {
+	val := ctx.Value(requestMetaDataContextKey)
+
+	value, ok := val.(*RequestMetaData)
+	if ok {
+		return value
+	}
+
+	return defaultRequestMetadata()
+}
+
+// SetRequestMetaData returns an `web.Handler` that overrides the request metadata
+// with the provided param.
+func SetRequestMetaData(rmd RequestMetaData) web.Handler {
+	return func(w http.ResponseWriter, r *http.Request) {
+		v := GetRequestMetaData(r.Context())
+		if rmd.Team != "" {
+			v.Team = rmd.Team
+		}
+	}
+}
+
+// SetOwner returns an `web.Handler` that sets the team name for an request.
+func SetOwner(team string) web.Handler {
+	return func(w http.ResponseWriter, r *http.Request) {
+		v := GetRequestMetaData(r.Context())
+		v.Team = team
+	}
+}
+
+func defaultRequestMetadata() *RequestMetaData {
+	return &RequestMetaData{
+		Team: TeamCore,
+	}
+}