github.com/nimakaviani/cli@v6.37.1-0.20180619223813-e734901a73fa+incompatible/command/v3/v3_restart_command.go (about)

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