-
Notifications
You must be signed in to change notification settings - Fork 927
/
disable_ssh_command.go
58 lines (46 loc) · 1.48 KB
/
disable_ssh_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
package v7
import (
"code.cloudfoundry.org/cli/command/flag"
)
type DisableSSHCommand struct {
BaseCommand
RequiredArgs flag.AppName `positional-args:"yes"`
usage interface{} `usage:"CF_NAME disable-ssh APP_NAME"`
relatedCommands interface{} `related_commands:"disallow-space-ssh, space-ssh-allowed, ssh, ssh-enabled"`
}
func (cmd *DisableSSHCommand) Execute(args []string) error {
err := cmd.SharedActor.CheckTarget(true, true)
if err != nil {
return err
}
username, err := cmd.Config.CurrentUserName()
if err != nil {
return err
}
cmd.UI.DisplayTextWithFlavor("Disabling ssh support for app {{.AppName}} as {{.CurrentUserName}}...", map[string]interface{}{
"AppName": cmd.RequiredArgs.AppName,
"CurrentUserName": username,
})
app, getAppWarnings, err := cmd.Actor.GetApplicationByNameAndSpace(cmd.RequiredArgs.AppName, cmd.Config.TargetedSpace().GUID)
if err != nil {
return err
}
cmd.UI.DisplayWarnings(getAppWarnings)
appFeature, getAppFeatureWarnings, err := cmd.Actor.GetAppFeature(app.GUID, "ssh")
if err != nil {
return err
}
cmd.UI.DisplayWarnings(getAppFeatureWarnings)
if !appFeature.Enabled {
cmd.UI.DisplayTextWithFlavor("ssh support for app '{{.AppName}}' is already disabled.", map[string]interface{}{
"AppName": cmd.RequiredArgs.AppName,
})
}
updateSSHWarnings, err := cmd.Actor.UpdateAppFeature(app, false, "ssh")
if err != nil {
return err
}
cmd.UI.DisplayWarnings(updateSSHWarnings)
cmd.UI.DisplayOK()
return nil
}