github.com/loafoe/cli@v7.1.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/command/flag"
     9  	log "github.com/sirupsen/logrus"
    10  )
    11  
    12  type EnvCommand struct {
    13  	BaseCommand
    14  
    15  	RequiredArgs    flag.EnvironmentArgs `positional-args:"yes"`
    16  	usage           interface{}          `usage:"CF_NAME env APP_NAME"`
    17  	relatedCommands interface{}          `related_commands:"app, apps, set-env, unset-env, running-environment-variable-group, staging-environment-variable-group"`
    18  }
    19  
    20  func (cmd EnvCommand) Execute(_ []string) error {
    21  	err := cmd.SharedActor.CheckTarget(true, true)
    22  	if err != nil {
    23  		return err
    24  	}
    25  
    26  	user, err := cmd.Config.CurrentUser()
    27  	if err != nil {
    28  		return err
    29  	}
    30  
    31  	appName := cmd.RequiredArgs.AppName
    32  	cmd.UI.DisplayTextWithFlavor("Getting env variables for app {{.AppName}} in org {{.OrgName}} / space {{.SpaceName}} as {{.Username}}...", map[string]interface{}{
    33  		"AppName":   appName,
    34  		"OrgName":   cmd.Config.TargetedOrganization().Name,
    35  		"SpaceName": cmd.Config.TargetedSpace().Name,
    36  		"Username":  user.Name,
    37  	})
    38  
    39  	envGroups, warnings, err := cmd.Actor.GetEnvironmentVariablesByApplicationNameAndSpace(
    40  		appName,
    41  		cmd.Config.TargetedSpace().GUID,
    42  	)
    43  	cmd.UI.DisplayWarnings(warnings)
    44  	if err != nil {
    45  		return err
    46  	}
    47  
    48  	if len(envGroups.System) > 0 || len(envGroups.Application) > 0 {
    49  		cmd.UI.DisplayHeader("System-Provided:")
    50  		err = cmd.displaySystem(envGroups.System)
    51  		if err != nil {
    52  			log.Errorln("error formatting system provided:", err)
    53  		}
    54  		if len(envGroups.Application) > 0 {
    55  			cmd.UI.DisplayNewline()
    56  			err = cmd.displaySystem(envGroups.Application)
    57  			if err != nil {
    58  				log.Errorln("error formatting application:", err)
    59  			}
    60  		}
    61  	} else {
    62  		cmd.UI.DisplayText("No system-provided env variables have been set")
    63  	}
    64  	cmd.UI.DisplayNewline()
    65  
    66  	if len(envGroups.EnvironmentVariables) > 0 {
    67  		cmd.UI.DisplayHeader("User-Provided:")
    68  		cmd.displayEnvGroup(envGroups.EnvironmentVariables)
    69  	} else {
    70  		cmd.UI.DisplayText("No user-provided env variables have been set")
    71  	}
    72  	cmd.UI.DisplayNewline()
    73  
    74  	if len(envGroups.Running) > 0 {
    75  		cmd.UI.DisplayHeader("Running Environment Variable Groups:")
    76  		cmd.displayEnvGroup(envGroups.Running)
    77  	} else {
    78  		cmd.UI.DisplayText("No running env variables have been set")
    79  	}
    80  	cmd.UI.DisplayNewline()
    81  
    82  	if len(envGroups.Staging) > 0 {
    83  		cmd.UI.DisplayHeader("Staging Environment Variable Groups:")
    84  		cmd.displayEnvGroup(envGroups.Staging)
    85  	} else {
    86  		cmd.UI.DisplayText("No staging env variables have been set")
    87  	}
    88  	cmd.UI.DisplayNewline()
    89  
    90  	return nil
    91  }
    92  
    93  func (cmd EnvCommand) displayEnvGroup(group map[string]interface{}) {
    94  	keys := sortKeys(group)
    95  
    96  	for _, key := range keys {
    97  		cmd.UI.DisplayText(fmt.Sprintf("%s: %v", key, group[key]))
    98  	}
    99  }
   100  
   101  func sortKeys(group map[string]interface{}) []string {
   102  	keys := make([]string, 0, len(group))
   103  	for key := range group {
   104  		keys = append(keys, key)
   105  	}
   106  	sort.Strings(keys)
   107  	return keys
   108  }
   109  
   110  func (cmd EnvCommand) displaySystem(group map[string]interface{}) error {
   111  	for key, val := range group {
   112  		jsonVal, err := json.MarshalIndent(val, "", " ")
   113  		if err != nil {
   114  			return err
   115  		}
   116  		cmd.UI.DisplayText(fmt.Sprintf("%s: %s", key, jsonVal))
   117  	}
   118  	return nil
   119  }