github.com/franc20/ayesa_sap@v7.0.0-beta.28.0.20200124003224-302d4d52fa6c+incompatible/command/v7/disable_ssh_command.go (about)

     1  package v7
     2  
     3  import (
     4  	"code.cloudfoundry.org/cli/actor/sharedaction"
     5  	"code.cloudfoundry.org/cli/actor/v7action"
     6  	"code.cloudfoundry.org/cli/api/cloudcontroller/ccv3"
     7  	"code.cloudfoundry.org/cli/command"
     8  	"code.cloudfoundry.org/cli/command/flag"
     9  	"code.cloudfoundry.org/cli/command/v7/shared"
    10  	"code.cloudfoundry.org/clock"
    11  )
    12  
    13  //go:generate counterfeiter . DisableSSHActor
    14  
    15  type DisableSSHActor interface {
    16  	GetApplicationByNameAndSpace(appName string, spaceGUID string) (v7action.Application, v7action.Warnings, error)
    17  	GetAppFeature(appGUID string, featureName string) (ccv3.ApplicationFeature, v7action.Warnings, error)
    18  	UpdateAppFeature(app v7action.Application, enabled bool, featureName string) (v7action.Warnings, error)
    19  }
    20  
    21  type DisableSSHCommand struct {
    22  	RequiredArgs    flag.AppName `positional-args:"yes"`
    23  	usage           interface{}  `usage:"CF_NAME disable-ssh APP_NAME"`
    24  	relatedCommands interface{}  `related_commands:"disallow-space-ssh, space-ssh-allowed, ssh, ssh-enabled"`
    25  
    26  	UI          command.UI
    27  	Config      command.Config
    28  	SharedActor command.SharedActor
    29  	Actor       DisableSSHActor
    30  }
    31  
    32  func (cmd *DisableSSHCommand) Setup(config command.Config, ui command.UI) error {
    33  	cmd.UI = ui
    34  	cmd.Config = config
    35  	sharedActor := sharedaction.NewActor(config)
    36  	cmd.SharedActor = sharedActor
    37  
    38  	ccClient, _, err := shared.GetNewClientsAndConnectToCF(config, ui, "")
    39  	if err != nil {
    40  		return err
    41  	}
    42  	cmd.Actor = v7action.NewActor(ccClient, config, sharedActor, nil, clock.NewClock())
    43  	return nil
    44  }
    45  
    46  func (cmd *DisableSSHCommand) Execute(args []string) error {
    47  	err := cmd.SharedActor.CheckTarget(true, true)
    48  	if err != nil {
    49  		return err
    50  	}
    51  
    52  	username, err := cmd.Config.CurrentUserName()
    53  	if err != nil {
    54  		return err
    55  	}
    56  
    57  	cmd.UI.DisplayTextWithFlavor("Disabling ssh support for app {{.AppName}} as {{.CurrentUserName}}...", map[string]interface{}{
    58  		"AppName":         cmd.RequiredArgs.AppName,
    59  		"CurrentUserName": username,
    60  	})
    61  
    62  	app, getAppWarnings, err := cmd.Actor.GetApplicationByNameAndSpace(cmd.RequiredArgs.AppName, cmd.Config.TargetedSpace().GUID)
    63  	if err != nil {
    64  		return err
    65  	}
    66  	cmd.UI.DisplayWarnings(getAppWarnings)
    67  
    68  	appFeature, getAppFeatureWarnings, err := cmd.Actor.GetAppFeature(app.GUID, "ssh")
    69  	if err != nil {
    70  		return err
    71  	}
    72  
    73  	cmd.UI.DisplayWarnings(getAppFeatureWarnings)
    74  
    75  	if !appFeature.Enabled {
    76  		cmd.UI.DisplayTextWithFlavor("ssh support for app '{{.AppName}}' is already disabled.", map[string]interface{}{
    77  			"AppName": cmd.RequiredArgs.AppName,
    78  		})
    79  	}
    80  
    81  	updateSSHWarnings, err := cmd.Actor.UpdateAppFeature(app, false, "ssh")
    82  	if err != nil {
    83  		return err
    84  	}
    85  
    86  	cmd.UI.DisplayWarnings(updateSSHWarnings)
    87  	cmd.UI.DisplayOK()
    88  	return nil
    89  }