forked from cloudfoundry/cli
/
set_quota.go
66 lines (55 loc) · 1.42 KB
/
set_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
package organization
import (
"cf/api"
"cf/configuration"
"cf/requirements"
"cf/terminal"
"errors"
"github.com/codegangsta/cli"
)
type SetQuota struct {
ui terminal.UI
config configuration.Reader
quotaRepo api.QuotaRepository
orgReq requirements.OrganizationRequirement
}
func NewSetQuota(ui terminal.UI, config configuration.Reader, quotaRepo api.QuotaRepository) (cmd *SetQuota) {
cmd = new(SetQuota)
cmd.ui = ui
cmd.config = config
cmd.quotaRepo = quotaRepo
return
}
func (cmd *SetQuota) GetRequirements(reqFactory requirements.Factory, c *cli.Context) (reqs []requirements.Requirement, err error) {
if len(c.Args()) != 2 {
err = errors.New("Incorrect Usage")
cmd.ui.FailWithUsage(c, "set-quota")
return
}
cmd.orgReq = reqFactory.NewOrganizationRequirement(c.Args()[0])
reqs = []requirements.Requirement{
reqFactory.NewLoginRequirement(),
cmd.orgReq,
}
return
}
func (cmd *SetQuota) Run(c *cli.Context) {
org := cmd.orgReq.GetOrganization()
quotaName := c.Args()[1]
quota, apiErr := cmd.quotaRepo.FindByName(quotaName)
if apiErr != nil {
cmd.ui.Failed(apiErr.Error())
return
}
cmd.ui.Say("Setting quota %s to org %s as %s...",
terminal.EntityNameColor(quota.Name),
terminal.EntityNameColor(org.Name),
terminal.EntityNameColor(cmd.config.Username()),
)
apiErr = cmd.quotaRepo.Update(org.Guid, quota.Guid)
if apiErr != nil {
cmd.ui.Failed(apiErr.Error())
return
}
cmd.ui.Ok()
}