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

     1  package v3
     2  
     3  import (
     4  	"encoding/json"
     5  	"fmt"
     6  	"net/http"
     7  
     8  	"code.cloudfoundry.org/cli/actor/sharedaction"
     9  	"code.cloudfoundry.org/cli/actor/v3action"
    10  	"code.cloudfoundry.org/cli/api/cloudcontroller/ccerror"
    11  	"code.cloudfoundry.org/cli/api/cloudcontroller/ccversion"
    12  	"code.cloudfoundry.org/cli/command"
    13  	"code.cloudfoundry.org/cli/command/flag"
    14  	"code.cloudfoundry.org/cli/command/translatableerror"
    15  	"code.cloudfoundry.org/cli/command/v3/shared"
    16  )
    17  
    18  //go:generate counterfeiter . V3EnvActor
    19  
    20  type V3EnvActor interface {
    21  	CloudControllerAPIVersion() string
    22  	GetEnvironmentVariablesByApplicationNameAndSpace(appName string, spaceGUID string) (v3action.EnvironmentVariableGroups, v3action.Warnings, error)
    23  }
    24  
    25  type V3EnvCommand struct {
    26  	RequiredArgs    flag.EnvironmentArgs `positional-args:"yes"`
    27  	usage           interface{}          `usage:"CF_NAME v3-env APP_NAME"`
    28  	relatedCommands interface{}          `related_commands:"v3-app, v3-apps, v3-set-env, v3-unset-env, running-environment-variable-group, staging-environment-variable-group"`
    29  
    30  	UI          command.UI
    31  	Config      command.Config
    32  	SharedActor command.SharedActor
    33  	Actor       V3EnvActor
    34  }
    35  
    36  func (cmd *V3EnvCommand) Setup(config command.Config, ui command.UI) error {
    37  	cmd.UI = ui
    38  	cmd.Config = config
    39  	cmd.SharedActor = sharedaction.NewActor(config)
    40  
    41  	ccClient, _, err := shared.NewClients(config, ui, true)
    42  	if err != nil {
    43  		if v3Err, ok := err.(ccerror.V3UnexpectedResponseError); ok && v3Err.ResponseCode == http.StatusNotFound {
    44  			return translatableerror.MinimumAPIVersionNotMetError{MinimumVersion: ccversion.MinVersionV3}
    45  		}
    46  
    47  		return err
    48  	}
    49  	cmd.Actor = v3action.NewActor(ccClient, config, nil, nil)
    50  
    51  	return nil
    52  }
    53  
    54  func (cmd V3EnvCommand) Execute(args []string) error {
    55  	cmd.UI.DisplayWarning(command.ExperimentalWarning)
    56  
    57  	err := command.MinimumAPIVersionCheck(cmd.Actor.CloudControllerAPIVersion(), ccversion.MinVersionV3)
    58  	if err != nil {
    59  		return err
    60  	}
    61  
    62  	err = cmd.SharedActor.CheckTarget(true, true)
    63  	if err != nil {
    64  		return err
    65  	}
    66  
    67  	user, err := cmd.Config.CurrentUser()
    68  	if err != nil {
    69  		return err
    70  	}
    71  
    72  	appName := cmd.RequiredArgs.AppName
    73  	cmd.UI.DisplayTextWithFlavor("Getting env variables for app {{.AppName}} in org {{.OrgName}} / space {{.SpaceName}} as {{.Username}}...", map[string]interface{}{
    74  		"AppName":   appName,
    75  		"OrgName":   cmd.Config.TargetedOrganization().Name,
    76  		"SpaceName": cmd.Config.TargetedSpace().Name,
    77  		"Username":  user.Name,
    78  	})
    79  	cmd.UI.DisplayNewline()
    80  
    81  	envGroups, warnings, err := cmd.Actor.GetEnvironmentVariablesByApplicationNameAndSpace(
    82  		appName,
    83  		cmd.Config.TargetedSpace().GUID,
    84  	)
    85  	cmd.UI.DisplayWarnings(warnings)
    86  	if err != nil {
    87  		return err
    88  	}
    89  
    90  	if len(envGroups.System) > 0 || len(envGroups.Application) > 0 {
    91  		cmd.UI.DisplayHeader("System-Provided:")
    92  		cmd.displayEnvGroup(envGroups.System)
    93  		if len(envGroups.Application) > 0 {
    94  			cmd.UI.DisplayNewline()
    95  			cmd.displayEnvGroup(envGroups.Application)
    96  		}
    97  	} else {
    98  		cmd.UI.DisplayText("No system-provided env variables have been set")
    99  	}
   100  	cmd.UI.DisplayNewline()
   101  
   102  	if len(envGroups.EnvironmentVariables) > 0 {
   103  		cmd.UI.DisplayHeader("User-Provided:")
   104  		cmd.displayEnvGroup(envGroups.EnvironmentVariables)
   105  	} else {
   106  		cmd.UI.DisplayText("No user-provided env variables have been set")
   107  	}
   108  	cmd.UI.DisplayNewline()
   109  
   110  	if len(envGroups.Running) > 0 {
   111  		cmd.UI.DisplayHeader("Running Environment Variable Groups:")
   112  		cmd.displayEnvGroup(envGroups.Running)
   113  	} else {
   114  		cmd.UI.DisplayText("No running env variables have been set")
   115  	}
   116  	cmd.UI.DisplayNewline()
   117  
   118  	if len(envGroups.Staging) > 0 {
   119  		cmd.UI.DisplayHeader("Staging Environment Variable Groups:")
   120  		cmd.displayEnvGroup(envGroups.Staging)
   121  	} else {
   122  		cmd.UI.DisplayText("No staging env variables have been set")
   123  	}
   124  	cmd.UI.DisplayNewline()
   125  
   126  	return nil
   127  }
   128  
   129  func (cmd V3EnvCommand) displayEnvGroup(group map[string]interface{}) error {
   130  	for key, val := range group {
   131  		valJSON, err := json.MarshalIndent(val, "", " ")
   132  		if err != nil {
   133  			return err
   134  		}
   135  		cmd.UI.DisplayText(fmt.Sprintf("%s: %s", key, string(valJSON)))
   136  	}
   137  
   138  	return nil
   139  }