-
Notifications
You must be signed in to change notification settings - Fork 509
/
cmd_helm.go
158 lines (141 loc) · 4.05 KB
/
cmd_helm.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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
package cli
import (
"context"
"fmt"
"path/filepath"
"github.com/spf13/cobra"
"github.com/spf13/pflag"
empty "google.golang.org/protobuf/types/known/emptypb"
"github.com/telepresenceio/telepresence/rpc/v2/connector"
"github.com/telepresenceio/telepresence/v2/pkg/client/cache"
"github.com/telepresenceio/telepresence/v2/pkg/client/cli/ann"
"github.com/telepresenceio/telepresence/v2/pkg/client/cli/cliutil"
"github.com/telepresenceio/telepresence/v2/pkg/client/errcat"
)
func helmCommand() *cobra.Command {
cmd := &cobra.Command{
Use: "helm",
}
cmd.AddCommand(helmInstallCommand(), helmUninstallCommand())
return cmd
}
type helmArgs struct {
cmdType connector.HelmRequest_Type
values []string
request *connector.ConnectRequest
kubeFlags *pflag.FlagSet
}
func helmInstallCommand() *cobra.Command {
var upgrade bool
ha := &helmArgs{
cmdType: connector.HelmRequest_INSTALL,
}
cmd := &cobra.Command{
Use: "install",
Args: cobra.NoArgs,
Short: "Install telepresence traffic manager",
RunE: func(cmd *cobra.Command, args []string) error {
if upgrade {
ha.cmdType = connector.HelmRequest_UPGRADE
}
return ha.run(cmd, args)
},
Annotations: map[string]string{
ann.UserDaemon: ann.Required,
},
}
flags := cmd.Flags()
flags.BoolVarP(&upgrade, "upgrade", "u", false, "replace the traffic manager if it already exists")
flags.StringSliceVarP(&ha.values, "values", "f", []string{}, "specify values in a YAML file or a URL (can specify multiple)")
ha.request, ha.kubeFlags = initConnectRequest(cmd)
return cmd
}
func helmUninstallCommand() *cobra.Command {
ha := &helmArgs{
cmdType: connector.HelmRequest_UNINSTALL,
}
cmd := &cobra.Command{
Use: "uninstall",
Args: cobra.NoArgs,
Short: "Uninstall telepresence traffic manager",
RunE: ha.run,
Annotations: map[string]string{
ann.UserDaemon: ann.Required,
},
}
ha.request, ha.kubeFlags = initConnectRequest(cmd)
return cmd
}
func (ha *helmArgs) run(cmd *cobra.Command, _ []string) error {
if err := cliutil.InitCommand(cmd); err != nil {
return err
}
ha.request.KubeFlags = kubeFlagMap(ha.kubeFlags)
for i, path := range ha.values {
absPath, err := filepath.Abs(path)
if err != nil {
return fmt.Errorf("--values path %q not valid: %w", path, err)
}
ha.values[i] = absPath
}
// always disconnect to ensure that there are no running intercepts etc.
ctx := cmd.Context()
_ = cliutil.Disconnect(ctx, false)
doQuit := false
userD := cliutil.GetUserDaemon(ctx)
status, _ := userD.Status(ctx, &empty.Empty{})
request := &connector.HelmRequest{
Type: ha.cmdType,
ValuePaths: ha.values,
ConnectRequest: ha.request,
}
cliutil.AddKubeconfigEnv(request.ConnectRequest)
resp, err := userD.Helm(ctx, request)
if err != nil {
return err
}
if err = errcat.FromResult(resp); err != nil {
return err
}
var msg string
switch ha.cmdType {
case connector.HelmRequest_INSTALL:
msg = "installed"
case connector.HelmRequest_UPGRADE:
msg = "upgraded"
case connector.HelmRequest_UNINSTALL:
if status != nil {
if err = removeClusterFromUserCache(ctx, status); err != nil {
return err
}
}
doQuit = true
msg = "uninstalled"
}
fmt.Fprintf(cmd.OutOrStdout(), "\nTraffic Manager %s successfully\n", msg)
if err == nil && doQuit {
err = cliutil.Disconnect(cmd.Context(), true)
}
return err
}
func removeClusterFromUserCache(ctx context.Context, connInfo *connector.ConnectInfo) (err error) {
// Login token is affined to the traffic-manager that just got removed. The user-info
// in turn, is info obtained using that token so both are removed here as a
// consequence of removing the manager.
if err := cliutil.EnsureLoggedOut(ctx); err != nil {
return err
}
// Delete the ingress info for the cluster if it exists.
ingresses, err := cache.LoadIngressesFromUserCache(ctx)
if err != nil {
return err
}
key := connInfo.ClusterServer + "/" + connInfo.ClusterContext
if _, ok := ingresses[key]; ok {
delete(ingresses, key)
if err = cache.SaveIngressesToUserCache(ctx, ingresses); err != nil {
return err
}
}
return nil
}