forked from cloudfoundry-community/cloudfoundry-cli
-
Notifications
You must be signed in to change notification settings - Fork 0
/
create_security_group.go
104 lines (87 loc) · 3.11 KB
/
create_security_group.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
package securitygroup
import (
"strings"
. "github.com/cloudfoundry/cli/cf/i18n"
"github.com/cloudfoundry/cli/cf/api/security_groups"
"github.com/cloudfoundry/cli/cf/command_metadata"
"github.com/cloudfoundry/cli/cf/configuration/core_config"
"github.com/cloudfoundry/cli/cf/errors"
"github.com/cloudfoundry/cli/cf/requirements"
"github.com/cloudfoundry/cli/cf/terminal"
"github.com/cloudfoundry/cli/json"
"github.com/codegangsta/cli"
)
type CreateSecurityGroup struct {
ui terminal.UI
securityGroupRepo security_groups.SecurityGroupRepo
configRepo core_config.Reader
}
func NewCreateSecurityGroup(ui terminal.UI, configRepo core_config.Reader, securityGroupRepo security_groups.SecurityGroupRepo) CreateSecurityGroup {
return CreateSecurityGroup{
ui: ui,
configRepo: configRepo,
securityGroupRepo: securityGroupRepo,
}
}
func (cmd CreateSecurityGroup) Metadata() command_metadata.CommandMetadata {
primaryUsage := T("CF_NAME create-security-group SECURITY_GROUP PATH_TO_JSON_RULES_FILE")
secondaryUsage := T(` The provided path can be an absolute or relative path to a file. The file should have
a single array with JSON objects inside describing the rules. The JSON Base Object is
omitted and only the square brackets and associated child object are required in the file.
Valid json file example:
[
{
"protocol": "tcp",
"destination": "10.244.1.18",
"ports": "3306"
}
]`)
return command_metadata.CommandMetadata{
Name: "create-security-group",
Description: T("Create a security group"),
Usage: strings.Join([]string{primaryUsage, secondaryUsage}, "\n\n"),
}
}
func (cmd CreateSecurityGroup) GetRequirements(requirementsFactory requirements.Factory, context *cli.Context) ([]requirements.Requirement, error) {
if len(context.Args()) != 2 {
cmd.ui.FailWithUsage(context)
}
requirements := []requirements.Requirement{requirementsFactory.NewLoginRequirement()}
return requirements, nil
}
func (cmd CreateSecurityGroup) Run(context *cli.Context) {
name := context.Args()[0]
pathToJSONFile := context.Args()[1]
rules, err := json.ParseJsonArray(pathToJSONFile)
if err != nil {
cmd.ui.Failed(T(`Incorrect json format: file: {{.JSONFile}}
Valid json file example:
[
{
"protocol": "tcp",
"destination": "10.244.1.18",
"ports": "3306"
}
]`, map[string]interface{}{"JSONFile": pathToJSONFile}))
}
cmd.ui.Say(T("Creating security group {{.security_group}} as {{.username}}",
map[string]interface{}{
"security_group": terminal.EntityNameColor(name),
"username": terminal.EntityNameColor(cmd.configRepo.Username()),
}))
err = cmd.securityGroupRepo.Create(name, rules)
httpErr, ok := err.(errors.HttpError)
if ok && httpErr.ErrorCode() == errors.SECURITY_GROUP_EXISTS {
cmd.ui.Ok()
cmd.ui.Warn(T("Security group {{.security_group}} {{.error_message}}",
map[string]interface{}{
"security_group": terminal.EntityNameColor(name),
"error_message": terminal.WarningColor(T("already exists")),
}))
return
}
if err != nil {
cmd.ui.Failed(err.Error())
}
cmd.ui.Ok()
}