github.com/cloudfoundry-attic/cli-with-i18n@v6.32.1-0.20171002233121-7401370d3b85+incompatible/command/v3/v3_set_env_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 . V3SetEnvActor
    17  
    18  type V3SetEnvActor interface {
    19  	CloudControllerAPIVersion() string
    20  	SetEnvironmentVariableByApplicationNameAndSpace(appName string, spaceGUID string, envPair v3action.EnvironmentVariablePair) (v3action.Warnings, error)
    21  }
    22  
    23  type V3SetEnvCommand struct {
    24  	RequiredArgs    flag.SetEnvironmentArgs `positional-args:"yes"`
    25  	usage           interface{}             `usage:"CF_NAME v3-set-env APP_NAME ENV_VAR_NAME ENV_VAR_VALUE"`
    26  	relatedCommands interface{}             `related_commands:"v3-apps, v3-env, v3-restart, set-running-environment-variable-group, set-staging-environment-variable-group, v3-stage, v3-unset-env"`
    27  
    28  	UI          command.UI
    29  	Config      command.Config
    30  	SharedActor command.SharedActor
    31  	Actor       V3SetEnvActor
    32  }
    33  
    34  func (cmd *V3SetEnvCommand) Setup(config command.Config, ui command.UI) error {
    35  	cmd.UI = ui
    36  	cmd.Config = config
    37  	cmd.SharedActor = sharedaction.NewActor(config)
    38  
    39  	ccClient, _, err := shared.NewClients(config, ui, true)
    40  	if err != nil {
    41  		if v3Err, ok := err.(ccerror.V3UnexpectedResponseError); ok && v3Err.ResponseCode == http.StatusNotFound {
    42  			return translatableerror.MinimumAPIVersionNotMetError{MinimumVersion: ccversion.MinVersionV3}
    43  		}
    44  
    45  		return err
    46  	}
    47  	cmd.Actor = v3action.NewActor(nil, ccClient, config)
    48  
    49  	return nil
    50  }
    51  
    52  func (cmd V3SetEnvCommand) Execute(args []string) error {
    53  	cmd.UI.DisplayText(command.ExperimentalWarning)
    54  	cmd.UI.DisplayNewline()
    55  
    56  	err := command.MinimumAPIVersionCheck(cmd.Actor.CloudControllerAPIVersion(), ccversion.MinVersionV3)
    57  	if err != nil {
    58  		return err
    59  	}
    60  
    61  	err = cmd.SharedActor.CheckTarget(cmd.Config, true, true)
    62  	if err != nil {
    63  		return shared.HandleError(err)
    64  	}
    65  
    66  	user, err := cmd.Config.CurrentUser()
    67  	if err != nil {
    68  		return shared.HandleError(err)
    69  	}
    70  
    71  	appName := cmd.RequiredArgs.AppName
    72  	cmd.UI.DisplayTextWithFlavor("Setting env variable {{.EnvVarName}} for app {{.AppName}} in org {{.OrgName}} / space {{.SpaceName}} as {{.Username}}...", map[string]interface{}{
    73  		"AppName":    appName,
    74  		"EnvVarName": cmd.RequiredArgs.EnvironmentVariableName,
    75  		"OrgName":    cmd.Config.TargetedOrganization().Name,
    76  		"SpaceName":  cmd.Config.TargetedSpace().Name,
    77  		"Username":   user.Name,
    78  	})
    79  
    80  	warnings, err := cmd.Actor.SetEnvironmentVariableByApplicationNameAndSpace(
    81  		appName,
    82  		cmd.Config.TargetedSpace().GUID,
    83  		v3action.EnvironmentVariablePair{
    84  			Key:   cmd.RequiredArgs.EnvironmentVariableName,
    85  			Value: string(cmd.RequiredArgs.EnvironmentVariableValue),
    86  		})
    87  	cmd.UI.DisplayWarnings(warnings)
    88  	if err != nil {
    89  		return shared.HandleError(err)
    90  	}
    91  
    92  	cmd.UI.DisplayOK()
    93  	cmd.UI.DisplayText("TIP: Use 'cf v3-stage {{.AppName}}' to ensure your env variable changes take effect.", map[string]interface{}{
    94  		"AppName": appName,
    95  	})
    96  
    97  	return nil
    98  }