package k8s

import (
	"context"

	"gitee.com/neuro-netw0rk/server-golib/k8s/informer"

	"gitee.com/neuro-netw0rk/server-golib/log"

	"k8s.io/client-go/kubernetes"
	"k8s.io/client-go/rest"
	"k8s.io/client-go/tools/clientcmd"
	"k8s.io/metrics/pkg/client/clientset/versioned"
)

type Conf struct {
	KubeConfigPath  string `yaml:"kubeConfigPath"`
	Pvc             string `yaml:"pvc"`
	Namespace       string `yaml:"namespace"`
	ResourceMonitor string `yaml:"resourceMonitor"`
}

type ClientSets struct {
	NeuroK8sClient ICoreV1Store
	K8sClient      kubernetes.Interface
	MetricsClient  versioned.Interface
	InformerClient informer.Storer
	stop           chan struct{}
}

func NewClientSets(conf Conf, stop chan struct{}) (*ClientSets, error) {
	ctx := context.Background()
	var (
		config *rest.Config
		err    error
	)

	config, err = clientcmd.BuildConfigFromFlags("", conf.KubeConfigPath)
	if err != nil {
		log.C(ctx).Warnf("Can't create config %v", err)
		return nil, err
	}

	clientSet, err := kubernetes.NewForConfig(config)
	if err != nil {
		log.C(ctx).Warnf("Can't create clientSet %v", err)
		return nil, err
	}

	metricSet, err := versioned.NewForConfig(config)
	if err != nil {
		log.C(ctx).Warnf("Can't create metricSet %v", err)
		return nil, err
	}

	informerStore, err := informer.NewInformerStore(stop, clientSet)
	if err != nil {
		log.C(ctx).Warnf("create informer store error: %v", err)
		return nil, err
	}

	k8sClient := NewCoreV1Store(clientSet, metricSet, informerStore, conf.KubeConfigPath)

	return &ClientSets{
		NeuroK8sClient: k8sClient,
		K8sClient:      clientSet,
		MetricsClient:  metricSet,
		InformerClient: informerStore,
		stop:           stop,
	}, nil
}

func (c *ClientSets) Stop() {
	<-c.stop
}