Plugins: Support MT app registration (#113348)

apps/plugins/pkg/app/app.go

@@ -5,17 +5,22 @@ import (
 	"fmt"
 
 	"github.com/grafana/grafana-app-sdk/app"
+	"github.com/grafana/grafana-app-sdk/k8s"
+	appsdkapiserver "github.com/grafana/grafana-app-sdk/k8s/apiserver"
 	"github.com/grafana/grafana-app-sdk/operator"
 	"github.com/grafana/grafana-app-sdk/simple"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	"k8s.io/apiserver/pkg/registry/generic"
+	"k8s.io/apiserver/pkg/registry/rest"
+	restclient "k8s.io/client-go/rest"
 	"k8s.io/klog/v2"
 
+	pluginsappapis "github.com/grafana/grafana/apps/plugins/pkg/apis"
 	pluginsv0alpha1 "github.com/grafana/grafana/apps/plugins/pkg/apis/plugins/v0alpha1"
 	"github.com/grafana/grafana/apps/plugins/pkg/app/meta"
 )
 
 func New(cfg app.Config) (app.App, error) {
-	cfg.KubeConfig.APIPath = "apis"
-
 	specificConfig, ok := cfg.SpecificConfig.(*PluginAppConfig)
 	if !ok {
 		return nil, fmt.Errorf("invalid config type")
@@ -60,3 +65,77 @@ func New(cfg app.Config) (app.App, error) {
 type PluginAppConfig struct {
 	MetaProviderManager *meta.ProviderManager
 }
+
+func ProvideAppInstaller(
+	metaProviderManager *meta.ProviderManager,
+) (appsdkapiserver.AppInstaller, error) {
+	specificConfig := &PluginAppConfig{
+		MetaProviderManager: metaProviderManager,
+	}
+	provider := simple.NewAppProvider(pluginsappapis.LocalManifest(), specificConfig, New)
+	appConfig := app.Config{
+		KubeConfig:     restclient.Config{}, // this will be overridden by the installer's InitializeApp method
+		ManifestData:   *pluginsappapis.LocalManifest().ManifestData,
+		SpecificConfig: specificConfig,
+	}
+	defaultInstaller, err := appsdkapiserver.NewDefaultAppInstaller(provider, appConfig, pluginsappapis.NewGoTypeAssociator())
+	if err != nil {
+		return nil, err
+	}
+
+	appInstaller := &pluginAppInstaller{
+		AppInstaller: defaultInstaller,
+		metaManager:  metaProviderManager,
+		ready:        make(chan struct{}),
+	}
+	return appInstaller, nil
+}
+
+type pluginAppInstaller struct {
+	appsdkapiserver.AppInstaller
+	metaManager *meta.ProviderManager
+
+	// restConfig is set during InitializeApp and used by the client factory
+	restConfig *restclient.Config
+	ready      chan struct{}
+}
+
+func (p *pluginAppInstaller) InitializeApp(restConfig restclient.Config) error {
+	if p.restConfig == nil {
+		p.restConfig = &restConfig
+		close(p.ready)
+	}
+	return p.AppInstaller.InitializeApp(restConfig)
+}
+
+func (p *pluginAppInstaller) InstallAPIs(
+	server appsdkapiserver.GenericAPIServer,
+	restOptsGetter generic.RESTOptionsGetter,
+) error {
+	// Create a client factory function that will be called lazily when the client is needed.
+	// This uses the rest config from the app, which is set during InitializeApp.
+	clientFactory := func(ctx context.Context) (*pluginsv0alpha1.PluginClient, error) {
+		<-p.ready
+		if p.restConfig == nil {
+			return nil, fmt.Errorf("rest config not yet initialized, app must be initialized before client can be created")
+		}
+
+		clientGenerator := k8s.NewClientRegistry(*p.restConfig, k8s.DefaultClientConfig())
+		client, err := pluginsv0alpha1.NewPluginClientFromGenerator(clientGenerator)
+		if err != nil {
+			return nil, fmt.Errorf("failed to create plugin client: %w", err)
+		}
+
+		return client, nil
+	}
+
+	pluginMetaGVR := pluginsv0alpha1.PluginMetaKind().GroupVersionResource()
+	replacedStorage := map[schema.GroupVersionResource]rest.Storage{
+		pluginMetaGVR: NewPluginMetaStorage(p.metaManager, clientFactory),
+	}
+	wrappedServer := &customStorageWrapper{
+		wrapped: server,
+		replace: replacedStorage,
+	}
+	return p.AppInstaller.InstallAPIs(wrappedServer, restOptsGetter)
+}