github.com/mook-as/cf-cli@v7.0.0-beta.28.0.20200120190804-b91c115fae48+incompatible/command/v7/feature_flags_command.go (about)

     1  package v7
     2  
     3  import (
     4  	"code.cloudfoundry.org/cli/actor/sharedaction"
     5  	"code.cloudfoundry.org/cli/actor/v7action"
     6  	"code.cloudfoundry.org/cli/command"
     7  	"code.cloudfoundry.org/cli/command/v7/shared"
     8  	"code.cloudfoundry.org/cli/util/ui"
     9  	"code.cloudfoundry.org/clock"
    10  )
    11  
    12  //go:generate counterfeiter . FeatureFlagsActor
    13  
    14  type FeatureFlagsActor interface {
    15  	GetFeatureFlags() ([]v7action.FeatureFlag, v7action.Warnings, error)
    16  }
    17  
    18  type FeatureFlagsCommand struct {
    19  	usage           interface{} `usage:"CF_NAME feature-flags"`
    20  	relatedCommands interface{} `related_commands:"disable-feature-flag, enable-feature-flag"`
    21  
    22  	UI          command.UI
    23  	Config      command.Config
    24  	SharedActor command.SharedActor
    25  	Actor       FeatureFlagsActor
    26  }
    27  
    28  func (cmd *FeatureFlagsCommand) Setup(config command.Config, ui command.UI) error {
    29  	cmd.UI = ui
    30  	cmd.Config = config
    31  	cmd.SharedActor = sharedaction.NewActor(config)
    32  
    33  	ccClient, _, err := shared.GetNewClientsAndConnectToCF(config, ui, "")
    34  	if err != nil {
    35  		return err
    36  	}
    37  	cmd.Actor = v7action.NewActor(ccClient, config, nil, nil, clock.NewClock())
    38  
    39  	return nil
    40  }
    41  
    42  func (cmd FeatureFlagsCommand) Execute(args []string) error {
    43  	err := cmd.SharedActor.CheckTarget(false, false)
    44  	if err != nil {
    45  		return err
    46  	}
    47  
    48  	user, err := cmd.Config.CurrentUser()
    49  	if err != nil {
    50  		return err
    51  	}
    52  
    53  	cmd.UI.DisplayTextWithFlavor("Getting feature flags as {{.Username}}...", map[string]interface{}{
    54  		"Username": user.Name,
    55  	})
    56  	cmd.UI.DisplayNewline()
    57  
    58  	flags, warnings, err := cmd.Actor.GetFeatureFlags()
    59  	cmd.UI.DisplayWarnings(warnings)
    60  	if err != nil {
    61  		return err
    62  	}
    63  
    64  	cmd.displayTable(flags)
    65  
    66  	return nil
    67  }
    68  
    69  func (cmd FeatureFlagsCommand) displayTable(featureFlags []v7action.FeatureFlag) {
    70  	if len(featureFlags) > 0 {
    71  		var keyValueTable = [][]string{
    72  			{"name", "state"},
    73  		}
    74  		for _, flag := range featureFlags {
    75  			state := "disabled"
    76  			if flag.Enabled {
    77  				state = "enabled"
    78  			}
    79  			keyValueTable = append(keyValueTable, []string{flag.Name, state})
    80  		}
    81  
    82  		cmd.UI.DisplayTableWithHeader("", keyValueTable, ui.DefaultTableSpacePadding)
    83  	}
    84  }