-
Notifications
You must be signed in to change notification settings - Fork 927
/
enable_service_access_command.go
118 lines (97 loc) · 4.66 KB
/
enable_service_access_command.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
package v6
import (
"code.cloudfoundry.org/cli/actor/sharedaction"
"code.cloudfoundry.org/cli/actor/v2action"
"code.cloudfoundry.org/cli/command"
"code.cloudfoundry.org/cli/command/flag"
"code.cloudfoundry.org/cli/command/translatableerror"
"code.cloudfoundry.org/cli/command/v6/shared"
)
//go:generate counterfeiter . EnableServiceAccessActor
type EnableServiceAccessActor interface {
EnablePlanForOrg(serviceName, servicePlanName, orgName, serviceBrokerName string) (v2action.Warnings, error)
EnableServiceForOrg(serviceName, orgName, serviceBrokerName string) (v2action.Warnings, error)
EnablePlanForAllOrgs(serviceName, servicePlanName, serviceBrokerName string) (v2action.Warnings, error)
EnableServiceForAllOrgs(serviceName, serviceBrokerName string) (v2action.Warnings, error)
}
type EnableServiceAccessCommand struct {
RequiredArgs flag.Service `positional-args:"yes"`
ServiceBroker string `short:"b" description:"Enable access to a service from a particular service broker. Required when service name is ambiguous"`
Organization string `short:"o" description:"Enable access for a specified organization"`
ServicePlan string `short:"p" description:"Enable access to a specified service plan"`
usage interface{} `usage:"CF_NAME enable-service-access SERVICE [-b BROKER] [-p PLAN] [-o ORG]"`
relatedCommands interface{} `related_commands:"marketplace, service-access, service-brokers"`
UI command.UI
SharedActor command.SharedActor
Actor EnableServiceAccessActor
Config command.Config
}
func (cmd *EnableServiceAccessCommand) Setup(config command.Config, ui command.UI) error {
cmd.UI = ui
cmd.Config = config
cmd.SharedActor = sharedaction.NewActor(config)
ccClient, uaaClient, err := shared.GetNewClientsAndConnectToCF(config, ui)
if err != nil {
return err
}
cmd.Actor = v2action.NewActor(ccClient, uaaClient, config)
return nil
}
func (cmd EnableServiceAccessCommand) Execute(args []string) error {
if len(args) > 0 {
return translatableerror.TooManyArgumentsError{
ExtraArgument: args[0],
}
}
err := cmd.SharedActor.CheckTarget(false, false)
if err != nil {
return err
}
user, err := cmd.Config.CurrentUser()
if err != nil {
return err
}
serviceBrokerName := cmd.ServiceBroker
serviceName := cmd.RequiredArgs.ServiceOffering
servicePlanName := cmd.ServicePlan
orgName := cmd.Organization
var warnings v2action.Warnings
cmd.UI.DisplayTextWithFlavor(messages[enableServiceAccessOptions{servicePlanName != "", orgName != "", serviceBrokerName != ""}],
map[string]interface{}{
"ServicePlan": servicePlanName,
"Service": serviceName,
"ServiceBroker": serviceBrokerName,
"Organization": orgName,
"User": user.Name,
})
if servicePlanName != "" && orgName != "" {
warnings, err = cmd.Actor.EnablePlanForOrg(serviceName, servicePlanName, orgName, serviceBrokerName)
} else if orgName != "" {
warnings, err = cmd.Actor.EnableServiceForOrg(serviceName, orgName, serviceBrokerName)
} else if servicePlanName != "" {
warnings, err = cmd.Actor.EnablePlanForAllOrgs(serviceName, servicePlanName, serviceBrokerName)
} else {
warnings, err = cmd.Actor.EnableServiceForAllOrgs(serviceName, serviceBrokerName)
}
cmd.UI.DisplayWarnings(warnings)
if err != nil {
return err
}
cmd.UI.DisplayOK()
return nil
}
type enableServiceAccessOptions struct {
Plan bool
Org bool
Broker bool
}
var messages = map[enableServiceAccessOptions]string{
{Plan: true, Org: true, Broker: false}: "Enabling access to plan {{.ServicePlan}} of service {{.Service}} for org {{.Organization}} as {{.User}}...",
{Plan: false, Org: true, Broker: false}: "Enabling access to all plans of service {{.Service}} for the org {{.Organization}} as {{.User}}...",
{Plan: true, Org: false, Broker: false}: "Enabling access of plan {{.ServicePlan}} for service {{.Service}} as {{.User}}...",
{Plan: false, Org: false, Broker: false}: "Enabling access to all plans of service {{.Service}} for all orgs as {{.User}}...",
{Plan: true, Org: true, Broker: true}: "Enabling access to plan {{.ServicePlan}} of service {{.Service}} from broker {{.ServiceBroker}} for org {{.Organization}} as {{.User}}...",
{Plan: false, Org: true, Broker: true}: "Enabling access to all plans of service {{.Service}} from broker {{.ServiceBroker}} for the org {{.Organization}} as {{.User}}...",
{Plan: true, Org: false, Broker: true}: "Enabling access to plan {{.ServicePlan}} for service {{.Service}} from broker {{.ServiceBroker}} for all orgs as {{.User}}...",
{Plan: false, Org: false, Broker: true}: "Enabling access to all plans of service {{.Service}} from broker {{.ServiceBroker}} for all orgs as {{.User}}...",
}