grafana/apps/plugins/pkg/app/storage_wrapper.go

package app

import (
	"fmt"

	"github.com/emicklei/go-restful/v3"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/apimachinery/pkg/runtime/schema"
	"k8s.io/apiserver/pkg/registry/rest"
	genericserver "k8s.io/apiserver/pkg/server"

	appsdkapiserver "github.com/grafana/grafana-app-sdk/k8s/apiserver"

	grafanarest "github.com/grafana/grafana/pkg/apiserver/rest"
)

var _ appsdkapiserver.GenericAPIServer = (*customStorageWrapper)(nil)

type customStorageWrapper struct {
	wrapped appsdkapiserver.GenericAPIServer
	replace map[schema.GroupVersionResource]rest.Storage
}

func (c *customStorageWrapper) InstallAPIGroup(
	apiGroupInfo *genericserver.APIGroupInfo,
) error {
	if apiGroupInfo == nil || apiGroupInfo.VersionedResourcesStorageMap == nil {
		return fmt.Errorf("apiGroupInfo cannot be nil")
	}

	// Override NegotiatedSerializer to exclude protobuf support.
	// Our generated types do not implement protobuf encoding, so we exclude protobuf
	// to prevent namespace deletion failures and marshalling errors.
	// See: https://github.com/kubernetes/kubernetes/issues/86666
	if apiGroupInfo.NegotiatedSerializer != nil {
		apiGroupInfo.NegotiatedSerializer = &noProtobufWrapper{apiGroupInfo.NegotiatedSerializer}
	}

	for gvr, storage := range c.replace {
		if _, ok := apiGroupInfo.VersionedResourcesStorageMap[gvr.Version]; !ok {
			apiGroupInfo.VersionedResourcesStorageMap[gvr.Version] = map[string]rest.Storage{}
		}
		apiGroupInfo.VersionedResourcesStorageMap[gvr.Version][gvr.Resource] = storage
	}
	return c.wrapped.InstallAPIGroup(apiGroupInfo)
}

type noProtobufWrapper struct {
	runtime.NegotiatedSerializer
}

func (w *noProtobufWrapper) SupportedMediaTypes() []runtime.SerializerInfo {
	base := w.NegotiatedSerializer.SupportedMediaTypes()
	var supported []runtime.SerializerInfo
	for _, info := range base {
		if grafanarest.NoProtobuf(info) {
			supported = append(supported, info)
		}
	}
	return supported
}

// RegisteredWebServices implements apiserver.GenericAPIServer.
func (c *customStorageWrapper) RegisteredWebServices() []*restful.WebService {
	return []*restful.WebService{}
}