github.com/randomtask1155/cli@v6.41.1-0.20181227003417-a98eed78cbde+incompatible/command/v6/v3_restart_command.go (about)

     1  package v6
     2  
     3  import (
     4  	"code.cloudfoundry.org/cli/actor/sharedaction"
     5  	"code.cloudfoundry.org/cli/actor/v3action"
     6  	"code.cloudfoundry.org/cli/api/cloudcontroller/ccversion"
     7  	"code.cloudfoundry.org/cli/command"
     8  	"code.cloudfoundry.org/cli/command/flag"
     9  	"code.cloudfoundry.org/cli/command/v6/shared"
    10  )
    11  
    12  //go:generate counterfeiter . V3RestartActor
    13  
    14  type V3RestartActor interface {
    15  	CloudControllerAPIVersion() string
    16  	GetApplicationByNameAndSpace(appName string, spaceGUID string) (v3action.Application, v3action.Warnings, error)
    17  	StartApplication(appGUID string) (v3action.Application, v3action.Warnings, error)
    18  	StopApplication(appGUID string) (v3action.Warnings, error)
    19  }
    20  
    21  type V3RestartCommand struct {
    22  	RequiredArgs        flag.AppName `positional-args:"yes"`
    23  	usage               interface{}  `usage:"CF_NAME v3-restart APP_NAME"`
    24  	envCFStartupTimeout interface{}  `environmentName:"CF_STARTUP_TIMEOUT" environmentDescription:"Max wait time for app instance startup, in minutes" environmentDefault:"5"`
    25  
    26  	UI          command.UI
    27  	Config      command.Config
    28  	SharedActor command.SharedActor
    29  	Actor       V3RestartActor
    30  }
    31  
    32  func (cmd *V3RestartCommand) Setup(config command.Config, ui command.UI) error {
    33  	cmd.UI = ui
    34  	cmd.Config = config
    35  	cmd.SharedActor = sharedaction.NewActor(config)
    36  
    37  	ccClient, _, err := shared.NewV3BasedClients(config, ui, true, "")
    38  	if err != nil {
    39  		return err
    40  	}
    41  	cmd.Actor = v3action.NewActor(ccClient, config, nil, nil)
    42  
    43  	return nil
    44  }
    45  
    46  func (cmd V3RestartCommand) Execute(args []string) error {
    47  	cmd.UI.DisplayWarning(command.ExperimentalWarning)
    48  
    49  	err := command.MinimumCCAPIVersionCheck(cmd.Actor.CloudControllerAPIVersion(), ccversion.MinVersionApplicationFlowV3)
    50  	if err != nil {
    51  		return err
    52  	}
    53  
    54  	err = cmd.SharedActor.CheckTarget(true, true)
    55  	if err != nil {
    56  		return err
    57  	}
    58  
    59  	user, err := cmd.Config.CurrentUser()
    60  	if err != nil {
    61  		return err
    62  	}
    63  
    64  	app, warnings, err := cmd.Actor.GetApplicationByNameAndSpace(cmd.RequiredArgs.AppName, cmd.Config.TargetedSpace().GUID)
    65  	cmd.UI.DisplayWarnings(warnings)
    66  	if err != nil {
    67  		return err
    68  	}
    69  
    70  	if app.Started() {
    71  		cmd.UI.DisplayTextWithFlavor("Stopping app {{.AppName}} in org {{.OrgName}} / space {{.SpaceName}} as {{.Username}}...", map[string]interface{}{
    72  			"AppName":   cmd.RequiredArgs.AppName,
    73  			"OrgName":   cmd.Config.TargetedOrganization().Name,
    74  			"SpaceName": cmd.Config.TargetedSpace().Name,
    75  			"Username":  user.Name,
    76  		})
    77  
    78  		warnings, err = cmd.Actor.StopApplication(app.GUID)
    79  		cmd.UI.DisplayWarnings(warnings)
    80  		if err != nil {
    81  			return err
    82  		}
    83  
    84  		cmd.UI.DisplayOK()
    85  	}
    86  
    87  	cmd.UI.DisplayTextWithFlavor("Starting app {{.AppName}} in org {{.OrgName}} / space {{.SpaceName}} as {{.Username}}...", map[string]interface{}{
    88  		"AppName":   cmd.RequiredArgs.AppName,
    89  		"OrgName":   cmd.Config.TargetedOrganization().Name,
    90  		"SpaceName": cmd.Config.TargetedSpace().Name,
    91  		"Username":  user.Name,
    92  	})
    93  
    94  	_, warnings, err = cmd.Actor.StartApplication(app.GUID)
    95  
    96  	cmd.UI.DisplayWarnings(warnings)
    97  	if err != nil {
    98  		return err
    99  	}
   100  
   101  	cmd.UI.DisplayOK()
   102  
   103  	return nil
   104  }