k3s/pkg/kubectl/main.go

59 lines
1.3 KiB
Go
Raw Normal View History

2019-01-01 08:23:01 +00:00
package kubectl
import (
"fmt"
"math/rand"
"os"
"strings"
2019-01-01 08:23:01 +00:00
"time"
"github.com/k3s-io/k3s/pkg/server"
"github.com/sirupsen/logrus"
"k8s.io/component-base/cli"
"k8s.io/kubectl/pkg/cmd"
"k8s.io/kubectl/pkg/cmd/util"
2019-01-01 08:23:01 +00:00
)
func Main() {
kubenv := os.Getenv("KUBECONFIG")
for i, arg := range os.Args {
if strings.HasPrefix(arg, "--kubeconfig=") {
kubenv = strings.Split(arg, "=")[1]
} else if strings.HasPrefix(arg, "--kubeconfig") && i+1 < len(os.Args) {
kubenv = os.Args[i+1]
}
}
2019-01-01 08:23:01 +00:00
if kubenv == "" {
2019-03-08 22:47:44 +00:00
config, err := server.HomeKubeConfig(false, false)
2019-01-01 08:23:01 +00:00
if _, serr := os.Stat(config); err == nil && serr == nil {
os.Setenv("KUBECONFIG", config)
}
if err := checkReadConfigPermissions(config); err != nil {
logrus.Warn(err)
}
2019-01-01 08:23:01 +00:00
}
main()
}
func main() {
rand.Seed(time.Now().UnixNano())
command := cmd.NewDefaultKubectlCommand()
if err := cli.RunNoErrOutput(command); err != nil {
util.CheckErr(err)
2019-01-01 08:23:01 +00:00
}
}
func checkReadConfigPermissions(configFile string) error {
file, err := os.OpenFile(configFile, os.O_RDONLY, 0600)
if err != nil {
if os.IsPermission(err) {
return fmt.Errorf("Unable to read %s, please start server "+
"with --write-kubeconfig-mode to modify kube config permissions", configFile)
}
}
file.Close()
return nil
}