github.com/mook-as/cf-cli@v7.0.0-beta.28.0.20200120190804-b91c115fae48+incompatible/command/v7/enable_feature_flag_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/flag"
     8  	"code.cloudfoundry.org/cli/command/v7/shared"
     9  	"code.cloudfoundry.org/clock"
    10  )
    11  
    12  //go:generate counterfeiter . EnableFeatureFlagActor
    13  
    14  type EnableFeatureFlagActor interface {
    15  	EnableFeatureFlag(flagName string) (v7action.Warnings, error)
    16  }
    17  
    18  type EnableFeatureFlagCommand struct {
    19  	RequiredArgs    flag.Feature `positional-args:"yes"`
    20  	usage           interface{}  `usage:"CF_NAME enable-feature-flag FEATURE_FLAG_NAME"`
    21  	relatedCommands interface{}  `related_commands:"disable-feature-flag, feature-flag, feature-flags"`
    22  
    23  	UI          command.UI
    24  	Config      command.Config
    25  	SharedActor command.SharedActor
    26  	Actor       EnableFeatureFlagActor
    27  }
    28  
    29  func (cmd *EnableFeatureFlagCommand) Setup(config command.Config, ui command.UI) error {
    30  	cmd.UI = ui
    31  	cmd.Config = config
    32  	sharedActor := sharedaction.NewActor(config)
    33  	cmd.SharedActor = sharedActor
    34  
    35  	ccClient, uaaClient, err := shared.GetNewClientsAndConnectToCF(config, ui, "")
    36  	if err != nil {
    37  		return err
    38  	}
    39  	cmd.Actor = v7action.NewActor(ccClient, config, sharedActor, uaaClient, clock.NewClock())
    40  
    41  	return nil
    42  }
    43  
    44  func (cmd EnableFeatureFlagCommand) Execute(args []string) error {
    45  	err := cmd.SharedActor.CheckTarget(false, false)
    46  	if err != nil {
    47  		return err
    48  	}
    49  
    50  	user, err := cmd.Config.CurrentUser()
    51  	if err != nil {
    52  		return err
    53  	}
    54  
    55  	cmd.UI.DisplayTextWithFlavor("Enabling feature flag {{.FeatureFlag}} as {{.Username}}...", map[string]interface{}{
    56  		"FeatureFlag": cmd.RequiredArgs.Feature,
    57  		"Username":    user.Name,
    58  	})
    59  
    60  	warnings, err := cmd.Actor.EnableFeatureFlag(cmd.RequiredArgs.Feature)
    61  	cmd.UI.DisplayWarnings(warnings)
    62  	if err != nil {
    63  		return err
    64  	}
    65  
    66  	cmd.UI.DisplayOK()
    67  	return nil
    68  }