-
Notifications
You must be signed in to change notification settings - Fork 503
/
cmd_helm.go
151 lines (135 loc) · 4.11 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
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/rpc/v2/daemon"
"github.com/telepresenceio/telepresence/v2/pkg/client/cache"
"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)
},
}
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,
}
ha.request, ha.kubeFlags = initConnectRequest(cmd)
return cmd
}
func (ha *helmArgs) run(cmd *cobra.Command, args []string) error {
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
}
request := &connector.HelmRequest{
Type: ha.cmdType,
ValuePaths: ha.values,
ConnectRequest: ha.request,
}
addKubeconfigEnv(request.ConnectRequest)
// always disconnect to ensure that there are no running intercepts etc.
_ = cliutil.Disconnect(cmd.Context(), false, false)
doQuit := false
err := cliutil.WithNetwork(cmd.Context(), func(ctx context.Context, daemonClient daemon.DaemonClient) error {
return cliutil.WithConnector(ctx, func(ctx context.Context, connectorClient connector.ConnectorClient) error {
status, _ := connectorClient.Status(ctx, &empty.Empty{})
resp, err := connectorClient.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)
return nil
})
})
if err == nil && doQuit {
err = cliutil.Disconnect(cmd.Context(), true, 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
}