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

     1  package v3
     2  
     3  import (
     4  	"net/http"
     5  
     6  	"code.cloudfoundry.org/cli/actor/actionerror"
     7  	"code.cloudfoundry.org/cli/actor/sharedaction"
     8  	"code.cloudfoundry.org/cli/actor/v3action"
     9  	"code.cloudfoundry.org/cli/api/cloudcontroller/ccerror"
    10  	"code.cloudfoundry.org/cli/api/cloudcontroller/ccversion"
    11  	"code.cloudfoundry.org/cli/command"
    12  	"code.cloudfoundry.org/cli/command/flag"
    13  	"code.cloudfoundry.org/cli/command/translatableerror"
    14  	"code.cloudfoundry.org/cli/command/v3/shared"
    15  )
    16  
    17  //go:generate counterfeiter . V3UnsetEnvActor
    18  
    19  type V3UnsetEnvActor interface {
    20  	CloudControllerAPIVersion() string
    21  	UnsetEnvironmentVariableByApplicationNameAndSpace(appName string, spaceGUID string, EnvironmentVariableName string) (v3action.Warnings, error)
    22  }
    23  
    24  type V3UnsetEnvCommand struct {
    25  	RequiredArgs    flag.UnsetEnvironmentArgs `positional-args:"yes"`
    26  	usage           interface{}               `usage:"CF_NAME v3-unset-env APP_NAME ENV_VAR_NAME"`
    27  	relatedCommands interface{}               `related_commands:"v3-apps, v3-env, v3-restart, v3-set-env, v3-stage"`
    28  
    29  	UI          command.UI
    30  	Config      command.Config
    31  	SharedActor command.SharedActor
    32  	Actor       V3UnsetEnvActor
    33  }
    34  
    35  func (cmd *V3UnsetEnvCommand) Setup(config command.Config, ui command.UI) error {
    36  	cmd.UI = ui
    37  	cmd.Config = config
    38  	cmd.SharedActor = sharedaction.NewActor(config)
    39  
    40  	ccClient, _, err := shared.NewClients(config, ui, true)
    41  	if err != nil {
    42  		if v3Err, ok := err.(ccerror.V3UnexpectedResponseError); ok && v3Err.ResponseCode == http.StatusNotFound {
    43  			return translatableerror.MinimumAPIVersionNotMetError{MinimumVersion: ccversion.MinVersionV3}
    44  		}
    45  
    46  		return err
    47  	}
    48  	cmd.Actor = v3action.NewActor(ccClient, config, nil, nil)
    49  
    50  	return nil
    51  }
    52  
    53  func (cmd V3UnsetEnvCommand) Execute(args []string) error {
    54  	cmd.UI.DisplayWarning(command.ExperimentalWarning)
    55  
    56  	err := command.MinimumAPIVersionCheck(cmd.Actor.CloudControllerAPIVersion(), ccversion.MinVersionV3)
    57  	if err != nil {
    58  		return err
    59  	}
    60  
    61  	err = cmd.SharedActor.CheckTarget(true, true)
    62  	if err != nil {
    63  		return err
    64  	}
    65  
    66  	user, err := cmd.Config.CurrentUser()
    67  	if err != nil {
    68  		return err
    69  	}
    70  
    71  	appName := cmd.RequiredArgs.AppName
    72  	cmd.UI.DisplayTextWithFlavor("Removing env variable {{.EnvVarName}} from 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.UnsetEnvironmentVariableByApplicationNameAndSpace(
    81  		appName,
    82  		cmd.Config.TargetedSpace().GUID,
    83  		cmd.RequiredArgs.EnvironmentVariableName,
    84  	)
    85  	cmd.UI.DisplayWarnings(warnings)
    86  	if err != nil {
    87  		switch errVal := err.(type) {
    88  		case actionerror.EnvironmentVariableNotSetError:
    89  			cmd.UI.DisplayText(errVal.Error())
    90  		default:
    91  			return err
    92  		}
    93  	}
    94  
    95  	cmd.UI.DisplayOK()
    96  	if err == nil {
    97  		cmd.UI.DisplayText("TIP: Use 'cf v3-stage {{.AppName}}' to ensure your env variable changes take effect.", map[string]interface{}{
    98  			"AppName": appName,
    99  		})
   100  	}
   101  
   102  	return nil
   103  }