K8s: APIGroupBuilder App SDK Support (#95638)

pkg/services/apiserver/builder/runner/runner.go

@@ -0,0 +1,123 @@
+package runner
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/grafana/grafana-app-sdk/app"
+	"github.com/grafana/grafana-app-sdk/resource"
+	"github.com/grafana/grafana/pkg/services/apiserver/builder"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	"k8s.io/client-go/rest"
+)
+
+type RunnerConfig struct {
+	RestConfigGetter func(context.Context) *rest.Config
+	APIRegistrar     builder.APIRegistrar
+}
+
+func NewAPIGroupRunner(cfg RunnerConfig, providers ...app.Provider) (*APIGroupRunner, error) {
+	groups := []appBuilderGroup{}
+	for _, provider := range providers {
+		created, err := newAppBuilderGroup(cfg, provider)
+		if err != nil {
+			return nil, err
+		}
+		groups = append(groups, created)
+	}
+	return &APIGroupRunner{
+		config:      cfg,
+		groups:      groups,
+		initialized: make(chan struct{}),
+	}, nil
+}
+
+type APIGroupRunner struct {
+	config      RunnerConfig
+	groups      []appBuilderGroup
+	initialized chan struct{}
+}
+
+func (r *APIGroupRunner) Run(ctx context.Context) error {
+	<-r.initialized
+	runner := app.NewMultiRunner()
+	for _, g := range r.groups {
+		runner.AddRunnable(g.app.Runner())
+	}
+	return runner.Run(ctx)
+}
+
+func (r *APIGroupRunner) Init(ctx context.Context) error {
+	defer close(r.initialized)
+	restConfig := r.config.RestConfigGetter(ctx)
+	if restConfig == nil {
+		return fmt.Errorf("rest config is nil")
+	}
+	for i, g := range r.groups {
+		customCfg := g.builders[0].config.CustomConfig
+		appConfig := app.Config{
+			KubeConfig:     *restConfig,
+			ManifestData:   *g.provider.Manifest().ManifestData,
+			SpecificConfig: customCfg,
+		}
+		app, err := g.provider.NewApp(appConfig)
+		if err != nil {
+			return err
+		}
+		r.groups[i].setApp(app)
+	}
+	return nil
+}
+
+func (r *APIGroupRunner) GetBuilders() []AppBuilder {
+	builders := []AppBuilder{}
+	for _, g := range r.groups {
+		builders = append(builders, g.builders...)
+	}
+	return builders
+}
+
+type appBuilderGroup struct {
+	builders []AppBuilder
+	provider app.Provider
+	app      app.App
+}
+
+func newAppBuilderGroup(cfg RunnerConfig, provider app.Provider) (appBuilderGroup, error) {
+	manifest := provider.Manifest()
+	if manifest.Location.Type != app.ManifestLocationEmbedded {
+		return appBuilderGroup{}, fmt.Errorf("app: %s has unsupported manifest location type: %s", manifest.ManifestData.AppName, manifest.Location.Type)
+	}
+
+	group := appBuilderGroup{
+		provider: provider,
+		builders: []AppBuilder{},
+	}
+
+	appBuilderConfig, ok := provider.SpecificConfig().(*AppBuilderConfig)
+	if !ok {
+		return group, fmt.Errorf("provider's SpecificConfig is not of type *AppBuilderConfig, got %T", provider.SpecificConfig())
+	}
+
+	for gv, kinds := range appBuilderConfig.ManagedKinds {
+		confCopy := *appBuilderConfig
+		confCopy.ManagedKinds = map[schema.GroupVersion]resource.Kind{
+			gv: kinds,
+		}
+		confCopy.groupVersion = gv
+		b, err := NewAppBuilder(confCopy)
+		if err != nil {
+			return group, err
+		}
+		group.builders = append(group.builders, b)
+		cfg.APIRegistrar.RegisterAPI(&b)
+	}
+	return group, nil
+}
+
+func (g *appBuilderGroup) setApp(app app.App) {
+	g.app = app
+	for i, _ := range g.builders {
+		g.builders[i].setApp(app)
+	}
+}