-
Notifications
You must be signed in to change notification settings - Fork 13
/
setup.go
148 lines (126 loc) · 3.17 KB
/
setup.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
package utils
import (
"fmt"
"os"
"path/filepath"
"strconv"
"github.com/AlecAivazis/survey/v2"
"golang.org/x/crypto/ssh/terminal"
"gopkg.in/yaml.v2"
"github.com/qingstor/qsctl/v2/constants"
)
// InputConfig is the struct for setup config input
type InputConfig struct {
AccessKeyID string `yaml:"access_key_id"`
SecretAccessKey string `yaml:"secret_access_key"`
Host string `yaml:"host"`
Port string `yaml:"port"`
Protocol string `yaml:"protocol"`
}
// NewInputConfig setup InputConfig and return the struct
func NewInputConfig() InputConfig {
return InputConfig{
AccessKeyID: "",
SecretAccessKey: "",
Host: constants.DefaultHost,
Port: constants.DefaultPort,
Protocol: constants.DefaultProtocol,
}
}
var keyPrompt = []*survey.Question{
{
Name: "AccessKeyID",
Prompt: &survey.Input{Message: "AccessKeyID:"},
Validate: survey.Required,
},
{
Name: "SecretAccessKey",
Prompt: &survey.Password{Message: "SecretAccessKey:"},
Validate: survey.Required,
},
}
var isPublicCloud = true
var publicCloudPrompt = &survey.Confirm{
Message: "Apply qsctl for QingStor public cloud?",
}
var privatePrompt = []*survey.Question{
{
Name: "Host",
Prompt: &survey.Input{Message: "Host:"},
Validate: survey.Required,
},
{
Name: "Port",
Prompt: &survey.Input{Message: "Port:"},
Validate: func(ans interface{}) error {
if v, ok := ans.(string); ok {
if _, err := strconv.Atoi(v); err != nil {
return fmt.Errorf("cannot parse port from your input <%v>: [%w]", ans, err)
}
return nil
}
return fmt.Errorf("cannot transfer port from non-string input, please check your input")
},
},
{
Name: "Protocol",
Prompt: &survey.Select{
Message: "Protocol:",
Options: []string{"http", "https"},
},
Validate: survey.Required,
},
}
var confirm = false
var confirmPrompt = &survey.Confirm{
Message: "Confirm your config?",
}
// SetupConfigInteractive setup input config interactively
func SetupConfigInteractive() (fileName string, err error) {
in := NewInputConfig()
if err = survey.Ask(keyPrompt, &in); err != nil {
return "", err
}
if err = survey.AskOne(publicCloudPrompt, &isPublicCloud); err != nil {
return "", err
}
if !isPublicCloud {
if err = survey.Ask(privatePrompt, &in); err != nil {
return "", err
}
}
if err = survey.AskOne(confirmPrompt, &confirm); err != nil {
return "", err
}
if !confirm {
return "", fmt.Errorf("config not confirmed")
}
b, err := yaml.Marshal(in)
if err != nil {
return "", err
}
homeDir, err := os.UserHomeDir()
if err != nil {
return "", err
}
fileName = filepath.Join(homeDir, ".qingstor/config.yaml")
if err = os.MkdirAll(filepath.Dir(fileName), 0755); err != nil {
return "", err
}
f, err := os.Create(fileName)
if err != nil {
return "", err
}
defer func() {
_ = f.Close()
}()
if _, err = f.Write(b); err != nil {
return "", err
}
return fileName, nil
}
// IsInteractiveEnable checks whether qsctl run interactively by
// checking stdin and stdout is terminal or not
func IsInteractiveEnable() bool {
return terminal.IsTerminal(int(os.Stdout.Fd())) && terminal.IsTerminal(int(os.Stdin.Fd()))
}