github.com/arunkumar7540/cli@v6.45.0+incompatible/command/v7/env_command.go (about)

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