github.com/jenspinney/cli@v6.42.1-0.20190207184520-7450c600020e+incompatible/command/v6/restage_command.go (about)

     1  package v6
     2  
     3  import (
     4  	"github.com/cloudfoundry/noaa/consumer"
     5  
     6  	"code.cloudfoundry.org/cli/actor/sharedaction"
     7  	"code.cloudfoundry.org/cli/actor/v2action"
     8  	"code.cloudfoundry.org/cli/actor/v2v3action"
     9  	"code.cloudfoundry.org/cli/actor/v3action"
    10  	"code.cloudfoundry.org/cli/command"
    11  	"code.cloudfoundry.org/cli/command/flag"
    12  	"code.cloudfoundry.org/cli/command/v6/shared"
    13  	sharedV3 "code.cloudfoundry.org/cli/command/v6/shared"
    14  	log "github.com/sirupsen/logrus"
    15  )
    16  
    17  //go:generate counterfeiter . RestageActor
    18  
    19  type RestageActor interface {
    20  	GetApplicationByNameAndSpace(name string, spaceGUID string) (v2action.Application, v2action.Warnings, error)
    21  	GetApplicationSummaryByNameAndSpace(name string, spaceGUID string) (v2action.ApplicationSummary, v2action.Warnings, error)
    22  	RestageApplication(app v2action.Application, client v2action.NOAAClient) (<-chan *v2action.LogMessage, <-chan error, <-chan v2action.ApplicationStateChange, <-chan string, <-chan error)
    23  }
    24  
    25  type RestageCommand struct {
    26  	RequiredArgs        flag.AppName `positional-args:"yes"`
    27  	usage               interface{}  `usage:"CF_NAME restage APP_NAME"`
    28  	relatedCommands     interface{}  `related_commands:"restart"`
    29  	envCFStagingTimeout interface{}  `environmentName:"CF_STAGING_TIMEOUT" environmentDescription:"Max wait time for buildpack staging, in minutes" environmentDefault:"15"`
    30  	envCFStartupTimeout interface{}  `environmentName:"CF_STARTUP_TIMEOUT" environmentDescription:"Max wait time for app instance startup, in minutes" environmentDefault:"5"`
    31  
    32  	UI                      command.UI
    33  	Config                  command.Config
    34  	SharedActor             command.SharedActor
    35  	Actor                   RestageActor
    36  	ApplicationSummaryActor shared.ApplicationSummaryActor
    37  	NOAAClient              *consumer.Consumer
    38  }
    39  
    40  func (cmd *RestageCommand) Setup(config command.Config, ui command.UI) error {
    41  	cmd.UI = ui
    42  	cmd.Config = config
    43  	cmd.SharedActor = sharedaction.NewActor(config)
    44  	sharedActor := sharedaction.NewActor(config)
    45  
    46  	ccClient, uaaClient, err := shared.NewClients(config, ui, true)
    47  	if err != nil {
    48  		return err
    49  	}
    50  
    51  	ccClientV3, _, err := sharedV3.NewV3BasedClients(config, ui, true, "")
    52  	if err != nil {
    53  		return err
    54  	}
    55  
    56  	v2Actor := v2action.NewActor(ccClient, uaaClient, config)
    57  	v3Actor := v3action.NewActor(ccClientV3, config, sharedActor, nil)
    58  
    59  	cmd.Actor = v2action.NewActor(ccClient, uaaClient, config)
    60  	cmd.ApplicationSummaryActor = v2v3action.NewActor(v2Actor, v3Actor)
    61  
    62  	cmd.NOAAClient = shared.NewNOAAClient(ccClient.DopplerEndpoint(), config, uaaClient, ui)
    63  
    64  	return nil
    65  }
    66  
    67  func (cmd RestageCommand) Execute(args []string) error {
    68  	err := cmd.SharedActor.CheckTarget(true, true)
    69  	if err != nil {
    70  		return err
    71  	}
    72  
    73  	user, err := cmd.Config.CurrentUser()
    74  	if err != nil {
    75  		return err
    76  	}
    77  
    78  	cmd.UI.DisplayWarning("This action will cause app downtime.")
    79  	cmd.UI.DisplayTextWithFlavor("Restaging app {{.AppName}} in org {{.OrgName}} / space {{.SpaceName}} as {{.CurrentUser}}...",
    80  		map[string]interface{}{
    81  			"AppName":     cmd.RequiredArgs.AppName,
    82  			"OrgName":     cmd.Config.TargetedOrganization().Name,
    83  			"SpaceName":   cmd.Config.TargetedSpace().Name,
    84  			"CurrentUser": user.Name,
    85  		})
    86  
    87  	app, warnings, err := cmd.Actor.GetApplicationByNameAndSpace(cmd.RequiredArgs.AppName, cmd.Config.TargetedSpace().GUID)
    88  	cmd.UI.DisplayWarnings(warnings)
    89  	if err != nil {
    90  		return err
    91  	}
    92  
    93  	messages, logErrs, appState, apiWarnings, errs := cmd.Actor.RestageApplication(app, cmd.NOAAClient)
    94  	err = shared.PollStart(cmd.UI, cmd.Config, messages, logErrs, appState, apiWarnings, errs)
    95  	if err != nil {
    96  		return err
    97  	}
    98  
    99  	cmd.UI.DisplayNewline()
   100  	log.WithField("v3_api_version", cmd.ApplicationSummaryActor.CloudControllerV3APIVersion()).Debug("using v3 for app display")
   101  	appSummary, v3Warnings, err := cmd.ApplicationSummaryActor.GetApplicationSummaryByNameAndSpace(cmd.RequiredArgs.AppName, cmd.Config.TargetedSpace().GUID, true)
   102  	cmd.UI.DisplayWarnings(v3Warnings)
   103  	if err != nil {
   104  		return err
   105  	}
   106  	shared.NewAppSummaryDisplayer2(cmd.UI).AppDisplay(appSummary, true)
   107  
   108  	return nil
   109  }