-
Notifications
You must be signed in to change notification settings - Fork 0
/
formatter.go
89 lines (78 loc) · 2.13 KB
/
formatter.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
package clusterregistrationtokens
import (
"fmt"
"net/url"
"github.com/rancher/norman/types"
"github.com/rancher/rancher/pkg/settings"
"github.com/rancher/rancher/pkg/systemtemplate"
)
const (
commandFormat = "kubectl apply -f %s"
insecureCommandFormat = "curl --insecure -sfL %s | kubectl apply -f -"
nodeCommandFormat = "sudo docker run -d --privileged --restart=unless-stopped --net=host -v /etc/kubernetes:/etc/kubernetes -v /var/run:/var/run %s --server %s --token %s%s"
)
func Formatter(request *types.APIContext, resource *types.RawResource) {
ca := systemtemplate.CAChecksum()
if ca != "" {
ca = " --ca-checksum " + ca
}
token, _ := resource.Values["token"].(string)
if token != "" {
url := getURL(request, token)
resource.Values["insecureCommand"] = fmt.Sprintf(insecureCommandFormat, url)
resource.Values["command"] = fmt.Sprintf(commandFormat, url)
resource.Values["nodeCommand"] = fmt.Sprintf(nodeCommandFormat,
settings.AgentImage.Get(),
getRootURL(request),
token,
ca)
resource.Values["token"] = token
resource.Values["manifestUrl"] = url
}
}
func NodeCommand(token string) string {
ca := systemtemplate.CAChecksum()
if ca != "" {
ca = " --ca-checksum " + ca
}
return fmt.Sprintf(nodeCommandFormat,
settings.AgentImage.Get(),
getRootURL(nil),
token,
ca)
}
func getRootURL(request *types.APIContext) string {
serverURL := settings.ServerURL.Get()
if serverURL == "" {
if request != nil {
serverURL = request.URLBuilder.RelativeToRoot("")
}
} else {
u, err := url.Parse(serverURL)
if err != nil {
if request != nil {
serverURL = request.URLBuilder.RelativeToRoot("")
}
} else {
u.Path = ""
serverURL = u.String()
}
}
return serverURL
}
func getURL(request *types.APIContext, token string) string {
path := "/v3/import/" + token + ".yaml"
serverURL := settings.ServerURL.Get()
if serverURL == "" {
serverURL = request.URLBuilder.RelativeToRoot(path)
} else {
u, err := url.Parse(serverURL)
if err != nil {
serverURL = request.URLBuilder.RelativeToRoot(path)
} else {
u.Path = path
serverURL = u.String()
}
}
return serverURL
}