-
-
Notifications
You must be signed in to change notification settings - Fork 1.5k
/
connect.go
65 lines (56 loc) · 1.61 KB
/
connect.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
package cmd
import (
"bufio"
"fmt"
"log"
"os"
"strings"
"github.com/spf13/cobra"
)
// connectCmd represents the connect command
var connectCmd = &cobra.Command{
Use: "connect",
Short: "Connect with a cluster",
Run: func(cmd *cobra.Command, args []string) {
secret := "*********"
fmt.Println("To keep the current value, press enter.")
if u, _ := cmd.Flags().GetString("url"); u != "" {
c.ClusterURL = u
} else if u := input("Cluster URL [" + c.ClusterURL + "]: "); u != "" {
c.ClusterURL = u
}
if u, _ := cmd.Flags().GetString("id"); u != "" {
c.ClientID = u
} else if u := input("Client ID [" + c.ClientID + "]: "); u != "" {
c.ClientID = u
}
if c.ClientSecret == "" {
secret = "empty"
}
if u, _ := cmd.Flags().GetString("secret"); u != "" {
fmt.Println("You should not provide secrets using command line flags. The secret might leak to bash history and similar systems.")
c.ClientSecret = u
} else if u := input("Client Secret [" + secret + "]: "); u != "" {
c.ClientSecret = u
}
if err := c.Persist(); err != nil {
log.Fatalf("Unable to save config file because %s.", err)
}
fmt.Println("Done.")
},
}
func input(message string) string {
reader := bufio.NewReader(os.Stdin)
fmt.Print(message)
s, err := reader.ReadString('\n')
if err != nil {
fatal(fmt.Sprintf("Could not read user input because %s.", err))
}
return strings.TrimSpace(s)
}
func init() {
RootCmd.AddCommand(connectCmd)
connectCmd.Flags().String("url", "", "The cluster URL")
connectCmd.Flags().String("id", "", "The client id")
connectCmd.Flags().String("secret", "", "The client secret")
}