github.com/franc20/ayesa_sap@v7.0.0-beta.28.0.20200124003224-302d4d52fa6c+incompatible/command/v7/org_quotas_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/clock"
     9  )
    10  
    11  //go:generate counterfeiter . OrgQuotasActor
    12  
    13  type OrgQuotasActor interface {
    14  	GetOrganizationQuotas() ([]v7action.OrganizationQuota, v7action.Warnings, error)
    15  }
    16  
    17  type OrgQuotasCommand struct {
    18  	usage           interface{} `usage:"CF_NAME org-quotas"`
    19  	relatedCommands interface{} `related_commands:"org-quota"`
    20  
    21  	UI          command.UI
    22  	Config      command.Config
    23  	SharedActor command.SharedActor
    24  	Actor       OrgQuotasActor
    25  }
    26  
    27  func (cmd *OrgQuotasCommand) Setup(config command.Config, ui command.UI) error {
    28  	cmd.UI = ui
    29  	cmd.Config = config
    30  	sharedActor := sharedaction.NewActor(config)
    31  	cmd.SharedActor = sharedActor
    32  
    33  	ccClient, uaaClient, err := shared.GetNewClientsAndConnectToCF(config, ui, "")
    34  	if err != nil {
    35  		return err
    36  	}
    37  	cmd.Actor = v7action.NewActor(ccClient, config, sharedActor, uaaClient, clock.NewClock())
    38  
    39  	return nil
    40  }
    41  
    42  func (cmd OrgQuotasCommand) 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 org quotas as {{.Username}}...", map[string]interface{}{
    54  		"Username": user.Name,
    55  	})
    56  	cmd.UI.DisplayNewline()
    57  
    58  	quotas, warnings, err := cmd.Actor.GetOrganizationQuotas()
    59  	cmd.UI.DisplayWarnings(warnings)
    60  	if err != nil {
    61  		return err
    62  	}
    63  
    64  	quotaDisplayer := shared.NewQuotaDisplayer(cmd.UI)
    65  	quotaDisplayer.DisplayQuotasTable(quotas)
    66  
    67  	return nil
    68  }