Team LBAC: Move middleware to enterprise (#76969)

* Team LBAC: Move middleware to enterprise * Remove ds proxy part * Move utils to enterprise
2023-10-24 14:06:18 +03:00
parent 575981201c
commit cad3c43bb1
5 changed files with 2 additions and 303 deletions
@@ -1,131 +0,0 @@
-package clientmiddleware
-
-import (
-	"context"
-
-	"github.com/grafana/grafana-plugin-sdk-go/backend"
-	"github.com/grafana/grafana/pkg/components/simplejson"
-	"github.com/grafana/grafana/pkg/infra/appcontext"
-	"github.com/grafana/grafana/pkg/plugins"
-	"github.com/grafana/grafana/pkg/services/contexthandler"
-	"github.com/grafana/grafana/pkg/services/datasources"
-	"github.com/grafana/grafana/pkg/util/proxyutil"
-)
-
-// NewTeamHTTPHeaderMiddleware creates a new plugins.ClientMiddleware that will
-// set headers based on teams user is member of.
-func NewTeamHTTPHeadersMiddleware() plugins.ClientMiddleware {
-	return plugins.ClientMiddlewareFunc(func(next plugins.Client) plugins.Client {
-		return &TeamHTTPHeadersMiddleware{
-			next: next,
-		}
-	})
-}
-
-type TeamHTTPHeadersMiddleware struct {
-	next plugins.Client
-}
-
-func (m *TeamHTTPHeadersMiddleware) QueryData(ctx context.Context, req *backend.QueryDataRequest) (*backend.QueryDataResponse, error) {
-	if req == nil {
-		return m.next.QueryData(ctx, req)
-	}
-
-	err := m.setHeaders(ctx, req.PluginContext, req)
-	if err != nil {
-		return nil, err
-	}
-
-	return m.next.QueryData(ctx, req)
-}
-
-func (m *TeamHTTPHeadersMiddleware) CallResource(ctx context.Context, req *backend.CallResourceRequest, sender backend.CallResourceResponseSender) error {
-	if req == nil {
-		return m.next.CallResource(ctx, req, sender)
-	}
-
-	err := m.setHeaders(ctx, req.PluginContext, req)
-	if err != nil {
-		return err
-	}
-
-	return m.next.CallResource(ctx, req, sender)
-}
-
-func (m *TeamHTTPHeadersMiddleware) CheckHealth(ctx context.Context, req *backend.CheckHealthRequest) (*backend.CheckHealthResult, error) {
-	if req == nil {
-		return m.next.CheckHealth(ctx, req)
-	}
-
-	// NOTE: might not be needed to set headers. we want to for now set these headers
-	err := m.setHeaders(ctx, req.PluginContext, req)
-	if err != nil {
-		return nil, err
-	}
-
-	return m.next.CheckHealth(ctx, req)
-}
-
-func (m *TeamHTTPHeadersMiddleware) CollectMetrics(ctx context.Context, req *backend.CollectMetricsRequest) (*backend.CollectMetricsResult, error) {
-	return m.next.CollectMetrics(ctx, req)
-}
-
-func (m *TeamHTTPHeadersMiddleware) SubscribeStream(ctx context.Context, req *backend.SubscribeStreamRequest) (*backend.SubscribeStreamResponse, error) {
-	return m.next.SubscribeStream(ctx, req)
-}
-
-func (m *TeamHTTPHeadersMiddleware) PublishStream(ctx context.Context, req *backend.PublishStreamRequest) (*backend.PublishStreamResponse, error) {
-	return m.next.PublishStream(ctx, req)
-}
-
-func (m *TeamHTTPHeadersMiddleware) RunStream(ctx context.Context, req *backend.RunStreamRequest, sender *backend.StreamSender) error {
-	return m.next.RunStream(ctx, req, sender)
-}
-
-func (m *TeamHTTPHeadersMiddleware) setHeaders(ctx context.Context, pCtx backend.PluginContext, req interface{}) error {
-	reqCtx := contexthandler.FromContext(ctx)
-	// if request not for a datasource or no HTTP request context skip middleware
-	if req == nil || pCtx.DataSourceInstanceSettings == nil || reqCtx == nil || reqCtx.Req == nil {
-		return nil
-	}
-
-	settings := pCtx.DataSourceInstanceSettings
-	jsonDataBytes, err := simplejson.NewJson(settings.JSONData)
-	if err != nil {
-		return err
-	}
-
-	ds := &datasources.DataSource{
-		ID:       settings.ID,
-		OrgID:    pCtx.OrgID,
-		JsonData: jsonDataBytes,
-		Updated:  settings.Updated,
-	}
-
-	signedInUser, err := appcontext.User(ctx)
-	if err != nil {
-		return nil // no user
-	}
-
-	teamHTTPHeaders, err := proxyutil.GetTeamHTTPHeaders(ds, signedInUser.GetTeams())
-	if err != nil {
-		return err
-	}
-
-	switch t := req.(type) {
-	case *backend.QueryDataRequest:
-		for key, value := range teamHTTPHeaders {
-			t.SetHTTPHeader(key, value)
-		}
-	case *backend.CheckHealthRequest:
-		for key, value := range teamHTTPHeaders {
-			t.SetHTTPHeader(key, value)
-		}
-	case *backend.CallResourceRequest:
-		for key, value := range teamHTTPHeaders {
-			t.SetHTTPHeader(key, value)
-		}
-	}
-
-	return nil
-}