-
Notifications
You must be signed in to change notification settings - Fork 4
/
kubeconfig.go
126 lines (106 loc) · 2.72 KB
/
kubeconfig.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package google
import (
"bytes"
"encoding/base64"
"fmt"
"html/template"
"io/ioutil"
"os"
"path/filepath"
log "github.com/Sirupsen/logrus"
container "google.golang.org/api/container/v1"
)
const kubeconfigTemplate = `apiVersion: v1
clusters:
- cluster:
certificate-authority: {{.CA}}
server: {{.Server}}
name: {{.Cluster}}
contexts:
- context:
cluster: {{.Cluster}}
user: {{.User}}
name: {{.Context}}
current-context: {{.Context}}
kind: Config
preferences: {}
users:
- name: {{.User}}
user:
auth-provider:
name: gcp
`
type configOptions struct {
CA string
Server string
Cluster string
User string
Context string
TokenFile string
}
func (g *GCPCloud) K8sConfigPath() (string, error) {
cname, ok := os.LookupEnv("DATACOL_CLUSTER")
if !ok {
return "", fmt.Errorf("DATACOL_CLUSTER env var not found")
}
return cacheKubeConfig(g.DeploymentName, g.Project, g.DefaultZone, cname)
}
func cacheKubeConfig(sName, project, zone, cname string) (string, error) {
dir, err := os.Getwd()
if err != nil {
return "", err
}
filename := filepath.Join(dir, "kubeconfig")
if _, err = os.Stat(filename); err != nil {
if os.IsNotExist(err) {
log.Debugf("creating kubeconfig in %s ...", dir)
svc, err := container.New(httpClient(sName))
if err != nil {
return filename, fmt.Errorf("container client %s", err)
}
ret, err := svc.Projects.Zones.Clusters.Get(project, zone, cname).Do()
if err != nil {
return filename, err
}
return filename, generateClusterConfig(sName, dir, ret)
} else {
return filename, err
}
}
return filename, nil
}
func generateClusterConfig(rackName, baseDir string, c *container.Cluster) error {
tmpl, err := template.New("kubeconfig").Parse(kubeconfigTemplate)
if err != nil {
return fmt.Errorf("error reading config template: %v", err)
}
if err := os.MkdirAll(baseDir, 0700); err != nil {
return err
}
kubeconfigFile := filepath.Join(baseDir, "kubeconfig")
certsDir := baseDir
// Base64 encoded ca
caDecodedPath := filepath.Join(certsDir, "ca.pem")
caDecoded, err := base64.StdEncoding.DecodeString(c.MasterAuth.ClusterCaCertificate)
if err != nil {
return fmt.Errorf("error decoding ca file for kubeconfig: %v", err)
}
if err := ioutil.WriteFile(caDecodedPath, caDecoded, 0700); err != nil {
return err
}
copts := &configOptions{
CA: caDecodedPath,
Server: "https://" + c.Endpoint,
User: rackName,
Context: rackName,
Cluster: rackName,
}
var kubeconfig bytes.Buffer
if err = tmpl.Execute(&kubeconfig, copts); err != nil {
return err
}
if err = ioutil.WriteFile(kubeconfigFile, kubeconfig.Bytes(), 0644); err != nil {
return fmt.Errorf("error writing kubeconfig file: %v", err)
}
return nil
}