github.com/nimakaviani/cli@v6.37.1-0.20180619223813-e734901a73fa+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(ccClient, config, nil, nil)
    48  
    49  	return nil
    50  }
    51  
    52  func (cmd V3SetEnvCommand) Execute(args []string) error {
    53  	cmd.UI.DisplayWarning(command.ExperimentalWarning)
    54  
    55  	err := command.MinimumAPIVersionCheck(cmd.Actor.CloudControllerAPIVersion(), ccversion.MinVersionV3)
    56  	if err != nil {
    57  		return err
    58  	}
    59  
    60  	err = cmd.SharedActor.CheckTarget(true, true)
    61  	if err != nil {
    62  		return err
    63  	}
    64  
    65  	user, err := cmd.Config.CurrentUser()
    66  	if err != nil {
    67  		return err
    68  	}
    69  
    70  	appName := cmd.RequiredArgs.AppName
    71  	cmd.UI.DisplayTextWithFlavor("Setting env variable {{.EnvVarName}} for app {{.AppName}} in org {{.OrgName}} / space {{.SpaceName}} as {{.Username}}...", map[string]interface{}{
    72  		"AppName":    appName,
    73  		"EnvVarName": cmd.RequiredArgs.EnvironmentVariableName,
    74  		"OrgName":    cmd.Config.TargetedOrganization().Name,
    75  		"SpaceName":  cmd.Config.TargetedSpace().Name,
    76  		"Username":   user.Name,
    77  	})
    78  
    79  	warnings, err := cmd.Actor.SetEnvironmentVariableByApplicationNameAndSpace(
    80  		appName,
    81  		cmd.Config.TargetedSpace().GUID,
    82  		v3action.EnvironmentVariablePair{
    83  			Key:   cmd.RequiredArgs.EnvironmentVariableName,
    84  			Value: string(cmd.RequiredArgs.EnvironmentVariableValue),
    85  		})
    86  	cmd.UI.DisplayWarnings(warnings)
    87  	if err != nil {
    88  		return err
    89  	}
    90  
    91  	cmd.UI.DisplayOK()
    92  	cmd.UI.DisplayText("TIP: Use 'cf v3-stage {{.AppName}}' to ensure your env variable changes take effect.", map[string]interface{}{
    93  		"AppName": appName,
    94  	})
    95  
    96  	return nil
    97  }