package scheme

import (
	certmanagerv1 "github.com/jetstack/cert-manager/pkg/apis/certmanager/v1"
	"github.com/pingcap/tidb-operator/pkg/apis/pingcap/v1alpha1"
	serverlessv1alpha1 "github.com/tidbcloud/aws-shared-provider/apis/serverless/v1alpha1"
	cloudv1 "github.com/tidbcloud/infra-api/api/v1"
	apiextensionsv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
	"k8s.io/apimachinery/pkg/runtime"
	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
	clientgoscheme "k8s.io/client-go/kubernetes/scheme"
)

// Scheme is used by client to visit kubernetes API.
var Scheme = runtime.NewScheme()

func init() {
	utilruntime.Must(clientgoscheme.AddToScheme(Scheme))
	utilruntime.Must(cloudv1.AddToScheme(Scheme))
	utilruntime.Must(apiextensionsv1.AddToScheme(Scheme))
	utilruntime.Must(v1alpha1.AddToScheme(Scheme))
	utilruntime.Must(certmanagerv1.AddToScheme(Scheme))
	utilruntime.Must(serverlessv1alpha1.AddToScheme(Scheme))
}

// Type is a helper function to type object for testing
// It is not needed for real logic.
func Type(obj runtime.Object) {
	gvks, _, err := Scheme.ObjectKinds(obj)
	if err != nil {
		// do nothing
		return
	}

	objKind := obj.GetObjectKind()
	old := objKind.GroupVersionKind()
	if old.Group != "" || old.Version != "" || old.Kind != "" {
		return
	}
	objKind.SetGroupVersionKind(gvks[0])
}