-
Notifications
You must be signed in to change notification settings - Fork 927
/
unset_space_quota.go
86 lines (71 loc) · 2.43 KB
/
unset_space_quota.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
package spacequota
import (
"fmt"
"code.cloudfoundry.org/cli/cf/api/spacequotas"
"code.cloudfoundry.org/cli/cf/api/spaces"
"code.cloudfoundry.org/cli/cf/commandregistry"
"code.cloudfoundry.org/cli/cf/configuration/coreconfig"
"code.cloudfoundry.org/cli/cf/flags"
. "code.cloudfoundry.org/cli/cf/i18n"
"code.cloudfoundry.org/cli/cf/requirements"
"code.cloudfoundry.org/cli/cf/terminal"
)
type UnsetSpaceQuota struct {
ui terminal.UI
config coreconfig.Reader
quotaRepo spacequotas.SpaceQuotaRepository
spaceRepo spaces.SpaceRepository
}
func init() {
commandregistry.Register(&UnsetSpaceQuota{})
}
func (cmd *UnsetSpaceQuota) MetaData() commandregistry.CommandMetadata {
return commandregistry.CommandMetadata{
Name: "unset-space-quota",
Description: T("Unassign a quota from a space"),
Usage: []string{
T("CF_NAME unset-space-quota SPACE QUOTA\n\n"),
},
}
}
func (cmd *UnsetSpaceQuota) Requirements(requirementsFactory requirements.Factory, fc flags.FlagContext) ([]requirements.Requirement, error) {
if len(fc.Args()) != 2 {
cmd.ui.Failed(T("Incorrect Usage. Requires SPACE and QUOTA as arguments\n\n") + commandregistry.Commands.CommandUsage("unset-space-quota"))
return nil, fmt.Errorf("Incorrect usage: %d arguments of %d required", len(fc.Args()), 2)
}
reqs := []requirements.Requirement{
requirementsFactory.NewLoginRequirement(),
requirementsFactory.NewTargetedOrgRequirement(),
}
return reqs, nil
}
func (cmd *UnsetSpaceQuota) SetDependency(deps commandregistry.Dependency, pluginCall bool) commandregistry.Command {
cmd.ui = deps.UI
cmd.config = deps.Config
cmd.spaceRepo = deps.RepoLocator.GetSpaceRepository()
cmd.quotaRepo = deps.RepoLocator.GetSpaceQuotaRepository()
return cmd
}
func (cmd *UnsetSpaceQuota) Execute(c flags.FlagContext) error {
spaceName := c.Args()[0]
quotaName := c.Args()[1]
space, err := cmd.spaceRepo.FindByName(spaceName)
if err != nil {
return err
}
quota, err := cmd.quotaRepo.FindByName(quotaName)
if err != nil {
return err
}
cmd.ui.Say(T("Unassigning space quota {{.QuotaName}} from space {{.SpaceName}} as {{.Username}}...",
map[string]interface{}{
"QuotaName": terminal.EntityNameColor(quota.Name),
"SpaceName": terminal.EntityNameColor(space.Name),
"Username": terminal.EntityNameColor(cmd.config.Username())}))
err = cmd.quotaRepo.UnassignQuotaFromSpace(space.GUID, quota.GUID)
if err != nil {
return err
}
cmd.ui.Ok()
return nil
}