mirror of https://github.com/fluxcd/flux2.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
1.3 KiB
Go
69 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
"k8s.io/client-go/kubernetes"
|
|
_ "k8s.io/client-go/plugin/pkg/client/auth"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
)
|
|
|
|
var VERSION = "0.0.1"
|
|
|
|
var rootCmd = &cobra.Command{
|
|
Use: "tk",
|
|
Short: "Kubernetes CD assembler",
|
|
Version: VERSION,
|
|
}
|
|
|
|
var (
|
|
kubeconfig string
|
|
)
|
|
|
|
func init() {
|
|
if home := homeDir(); home != "" {
|
|
rootCmd.PersistentFlags().StringVarP(&kubeconfig, "kubeconfig", "", filepath.Join(home, ".kube", "config"),
|
|
"path to the kubeconfig file")
|
|
} else {
|
|
checkCmd.PersistentFlags().StringVarP(&kubeconfig, "kubeconfig", "", "",
|
|
"absolute path to the kubeconfig file")
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
log.SetFlags(0)
|
|
|
|
rootCmd.SetArgs(os.Args[1:])
|
|
if err := rootCmd.Execute(); err != nil {
|
|
e := err.Error()
|
|
fmt.Println(strings.ToUpper(e[:1]) + e[1:])
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func homeDir() string {
|
|
if h := os.Getenv("HOME"); h != "" {
|
|
return h
|
|
}
|
|
return os.Getenv("USERPROFILE") // windows
|
|
}
|
|
|
|
func NewKubernetesClient() (*kubernetes.Clientset, error) {
|
|
config, err := clientcmd.BuildConfigFromFlags("", kubeconfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
client, err := kubernetes.NewForConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return client, nil
|
|
}
|