// Copyright 2020-2021 Clastix Labs
// SPDX-License-Identifier: Apache-2.0

package utils

import (
	"path/filepath"
	"strconv"

	"k8s.io/client-go/kubernetes"
	"k8s.io/client-go/rest"
	"k8s.io/client-go/tools/clientcmd"
	"k8s.io/client-go/util/homedir"
)

func GetK8sVersion() (major, minor int, ver string, err error) {
	cfg, err := rest.InClusterConfig()
	if err != nil {
		kubeconfig := filepath.Join(homedir.HomeDir(), ".kube", "config")
		cfg, err = clientcmd.BuildConfigFromFlags("", kubeconfig)
	}
	if err != nil {
		return 0, 0, "", err
	}
	client, err := kubernetes.NewForConfig(cfg)
	if err != nil {
		return 0, 0, "", err
	}

	version, err := client.Discovery().ServerVersion()
	if err != nil {
		return 0, 0, "", err
	}
	major, _ = strconv.Atoi(version.Major)
	minor, _ = strconv.Atoi(version.Minor)
	ver = version.String()
	return
}