grafana/pkg/registry/apis/datasource/standalone_services.go

package datasource

import (
	"context"
	"path/filepath"

	"github.com/grafana/grafana/pkg/api/routing"
	"github.com/grafana/grafana/pkg/bus"
	"github.com/grafana/grafana/pkg/infra/kvstore"
	"github.com/grafana/grafana/pkg/infra/localcache"
	"github.com/grafana/grafana/pkg/infra/tracing"
	"github.com/grafana/grafana/pkg/infra/usagestats/service"
	"github.com/grafana/grafana/pkg/plugins"
	pCfg "github.com/grafana/grafana/pkg/plugins/config"
	"github.com/grafana/grafana/pkg/plugins/manager/loader"
	"github.com/grafana/grafana/pkg/plugins/manager/pipeline/bootstrap"
	"github.com/grafana/grafana/pkg/plugins/manager/pipeline/discovery"
	"github.com/grafana/grafana/pkg/plugins/manager/pipeline/initialization"
	"github.com/grafana/grafana/pkg/plugins/manager/pipeline/termination"
	"github.com/grafana/grafana/pkg/plugins/manager/pipeline/validation"
	"github.com/grafana/grafana/pkg/plugins/manager/registry"
	"github.com/grafana/grafana/pkg/plugins/manager/signature"
	"github.com/grafana/grafana/pkg/plugins/manager/sources"
	"github.com/grafana/grafana/pkg/services/accesscontrol/acimpl"
	"github.com/grafana/grafana/pkg/services/accesscontrol/ossaccesscontrol"
	"github.com/grafana/grafana/pkg/services/datasources/guardian"
	datasourceService "github.com/grafana/grafana/pkg/services/datasources/service"
	"github.com/grafana/grafana/pkg/services/encryption/provider"
	encryptionService "github.com/grafana/grafana/pkg/services/encryption/service"
	"github.com/grafana/grafana/pkg/services/featuremgmt"
	"github.com/grafana/grafana/pkg/services/kmsproviders/osskmsproviders"
	"github.com/grafana/grafana/pkg/services/org/orgimpl"
	"github.com/grafana/grafana/pkg/services/pluginsintegration/config"
	"github.com/grafana/grafana/pkg/services/pluginsintegration/pluginstore"
	"github.com/grafana/grafana/pkg/services/quota/quotaimpl"
	"github.com/grafana/grafana/pkg/services/secrets/database"
	kvstoreService "github.com/grafana/grafana/pkg/services/secrets/kvstore"
	"github.com/grafana/grafana/pkg/services/secrets/manager"
	"github.com/grafana/grafana/pkg/services/sqlstore"
	"github.com/grafana/grafana/pkg/services/sqlstore/migrations"
	"github.com/grafana/grafana/pkg/services/supportbundles/bundleregistry"
	"github.com/grafana/grafana/pkg/services/team/teamimpl"
	"github.com/grafana/grafana/pkg/services/user/userimpl"
	"github.com/grafana/grafana/pkg/setting"
)

func apiBuilderServices(cfg *setting.Cfg, features featuremgmt.FeatureToggles, pluginID string) (
	*acimpl.AccessControl,
	*pluginstore.Service,
	*datasourceService.Service,
	*datasourceService.CacheServiceImpl,
	error,
) {
	accessControl := acimpl.ProvideAccessControl(cfg)
	cacheService := localcache.ProvideService()
	tracingService, err := tracing.ProvideService(cfg)
	if err != nil {
		return nil, nil, nil, nil, err
	}
	routeRegisterImpl := routing.ProvideRegister()
	featureManager, err := featuremgmt.ProvideManagerService(cfg)
	if err != nil {
		return nil, nil, nil, nil, err
	}

	inProcBus := bus.ProvideBus(tracingService)
	ossMigrations := migrations.ProvideOSSMigrations(features)
	sqlStore, err := sqlstore.ProvideService(cfg, features, ossMigrations, inProcBus, tracingService)
	if err != nil {
		return nil, nil, nil, nil, err
	}

	kvStore := kvstore.ProvideService(sqlStore)
	featureToggles := featuremgmt.ProvideToggles(featureManager)
	bundleRegistry := bundleregistry.ProvideService()

	quota := quotaimpl.ProvideService(sqlStore, cfg)
	orgService, err := orgimpl.ProvideService(sqlStore, cfg, quota)
	if err != nil {
		return nil, nil, nil, nil, err
	}
	teamService := teamimpl.ProvideService(sqlStore, cfg)
	userService, err := userimpl.ProvideService(sqlStore, orgService, cfg, teamService, cacheService, quota, bundleRegistry)
	if err != nil {
		return nil, nil, nil, nil, err
	}

	acimplService, err := acimpl.ProvideService(cfg, sqlStore, routeRegisterImpl, cacheService, accessControl, userService, featureToggles)
	if err != nil {
		return nil, nil, nil, nil, err
	}
	usageStats, err := service.ProvideService(cfg, kvStore, routeRegisterImpl, tracingService, accessControl, acimplService, bundleRegistry)
	if err != nil {
		return nil, nil, nil, nil, err
	}
	secretsStoreImpl := database.ProvideSecretsStore(sqlStore)
	providerProvider := provider.ProvideEncryptionProvider()
	serviceService, err := encryptionService.ProvideEncryptionService(providerProvider, usageStats, cfg)
	if err != nil {
		return nil, nil, nil, nil, err
	}
	kmsProviders := osskmsproviders.ProvideService(serviceService, cfg, featureToggles)
	secretsService, err := manager.ProvideSecretsService(secretsStoreImpl, kmsProviders, serviceService, cfg, featureToggles, usageStats)
	if err != nil {
		return nil, nil, nil, nil, err
	}
	ossImpl := setting.ProvideProvider(cfg)
	pluginCfg, err := config.ProvideConfig(ossImpl, cfg, featureToggles)
	if err != nil {
		return nil, nil, nil, nil, err
	}
	pluginRegistry := registry.ProvideService()
	quotaService := quotaimpl.ProvideService(sqlStore, cfg)
	pluginLoader, err := createLoader(pluginCfg, pluginRegistry)
	if err != nil {
		return nil, nil, nil, nil, err
	}
	pluginStore, err := pluginstore.ProvideService(pluginRegistry, newPluginSource(cfg, pluginID), pluginLoader)
	if err != nil {
		return nil, nil, nil, nil, err
	}
	secretsKVStore, err := kvstoreService.ProvideService(sqlStore, secretsService, pluginStore, kvStore, featureToggles, cfg)
	if err != nil {
		return nil, nil, nil, nil, err
	}
	dsPermissionsService := ossaccesscontrol.ProvideDatasourcePermissionsService()
	dsService, err := datasourceService.ProvideService(sqlStore, secretsService, secretsKVStore, cfg, featureToggles, accessControl, dsPermissionsService, quotaService, pluginStore)
	if err != nil {
		return nil, nil, nil, nil, err
	}

	ossProvider := guardian.ProvideGuardian()
	cacheServiceImpl := datasourceService.ProvideCacheService(cacheService, sqlStore, ossProvider)

	return accessControl, pluginStore, dsService, cacheServiceImpl, nil
}

var _ sources.Registry = (*pluginSource)(nil)

type pluginSource struct {
	cfg      *setting.Cfg
	pluginID string
}

func newPluginSource(cfg *setting.Cfg, pluginID string) *pluginSource {
	return &pluginSource{
		cfg:      cfg,
		pluginID: pluginID,
	}
}

func (t *pluginSource) List(_ context.Context) []plugins.PluginSource {
	p := filepath.Join(t.cfg.StaticRootPath, "app/plugins/datasource", t.pluginID)
	return []plugins.PluginSource{sources.NewLocalSource(plugins.ClassCore, []string{p})}
}

func createLoader(cfg *pCfg.Cfg, pr registry.Service) (loader.Service, error) {
	d := discovery.New(cfg, discovery.Opts{
		FindFilterFuncs: []discovery.FindFilterFunc{
			func(ctx context.Context, _ plugins.Class, b []*plugins.FoundBundle) ([]*plugins.FoundBundle, error) {
				return discovery.NewDuplicatePluginFilterStep(pr).Filter(ctx, b)
			},
		},
	})
	b := bootstrap.New(cfg, bootstrap.Opts{
		DecorateFuncs: []bootstrap.DecorateFunc{}, // no decoration required
	})
	v := validation.New(cfg, validation.Opts{
		ValidateFuncs: []validation.ValidateFunc{
			validation.SignatureValidationStep(signature.NewValidator(signature.NewUnsignedAuthorizer(cfg))),
		},
	})
	i := initialization.New(cfg, initialization.Opts{
		InitializeFuncs: []initialization.InitializeFunc{
			initialization.PluginRegistrationStep(pr),
		},
	})
	t, err := termination.New(cfg, termination.Opts{
		TerminateFuncs: []termination.TerminateFunc{
			termination.DeregisterStep(pr),
		},
	})
	if err != nil {
		return nil, err
	}

	return loader.New(d, b, v, i, t), nil
}