github.com/mook-as/cf-cli@v7.0.0-beta.28.0.20200120190804-b91c115fae48+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.GetNewClientsAndConnectToCF(config, ui) 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 }