package scheme

import (
	cciamv1beta1 "github.com/GoogleCloudPlatform/k8s-config-connector/pkg/clients/generated/apis/iam/v1beta1"
	cck8sv1alpha1 "github.com/GoogleCloudPlatform/k8s-config-connector/pkg/clients/generated/apis/k8s/v1alpha1"
	ec2v1alpha1 "github.com/crossplane/provider-aws/apis/ec2/v1alpha1"
	ec2v1beta1 "github.com/crossplane/provider-aws/apis/ec2/v1beta1"
	awskmsv1alpha1 "github.com/crossplane/provider-aws/apis/kms/v1alpha1"
	lambdav1alpha1 "github.com/crossplane/provider-aws/apis/lambda/v1alpha1"
	awsdnsv1alpha1 "github.com/crossplane/provider-aws/apis/route53/v1alpha1"
	s3v1beta1 "github.com/crossplane/provider-aws/apis/s3/v1beta1"
	certmanagerv1 "github.com/jetstack/cert-manager/pkg/apis/certmanager/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"

	"github.com/pingcap/tidb-operator/pkg/apis/pingcap/v1alpha1"
	cloudv1 "github.com/tidbcloud/infra-api/api/v1"
)

// 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(awskmsv1alpha1.AddToScheme(Scheme))
	utilruntime.Must(awsdnsv1alpha1.SchemeBuilder.AddToScheme(Scheme))
	utilruntime.Must(ec2v1alpha1.SchemeBuilder.AddToScheme(Scheme))
	utilruntime.Must(ec2v1beta1.SchemeBuilder.AddToScheme(Scheme))
	utilruntime.Must(lambdav1alpha1.SchemeBuilder.AddToScheme(Scheme))
	utilruntime.Must(certmanagerv1.AddToScheme(Scheme))
	utilruntime.Must(s3v1beta1.SchemeBuilder.AddToScheme(Scheme))
	utilruntime.Must(cck8sv1alpha1.SchemeBuilder.AddToScheme(Scheme))
	utilruntime.Must(cciamv1beta1.SchemeBuilder.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])
}