Skip to content

Instantly share code, notes, and snippets.

@Kelwing

Kelwing/main.go

Created Dec 14, 2018
Embed
What would you like to do?
Tool for downloading a kubeconfig from DigitalOcean
package main
import (
"context"
"flag"
"io/ioutil"
"log"
"os"
"os/user"
"strings"
"github.com/digitalocean/godo"
"golang.org/x/oauth2"
)
type TokenSource struct {
AccessToken string
}
func (t *TokenSource) Token() (*oauth2.Token, error) {
token := &oauth2.Token{
AccessToken: t.AccessToken,
}
return token, nil
}
func main() {
pat := flag.String("token", "NONE", "Digital Ocean access token")
clusterID := flag.String("cluster", "NONE", "Digital Ocean cluster ID")
flag.Parse()
usr, err := user.Current()
tokenSource := &TokenSource{
AccessToken: *pat,
}
oauthClient := oauth2.NewClient(oauth2.NoContext, tokenSource)
client := godo.NewClient(oauthClient)
ctx := context.TODO()
config, _, err := client.Kubernetes.GetKubeConfig(ctx, *clusterID)
if err != nil {
log.Fatal(err)
}
kubeConfigFile := string(config.KubeconfigYAML)
var str strings.Builder
str.WriteString(usr.HomeDir)
str.WriteString("/.kube")
_ = os.Mkdir(str.String(), 0770)
str.WriteString("/config")
err = ioutil.WriteFile(str.String(), []byte(kubeConfigFile), 0640)
if err != nil {
log.Fatal(err)
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment