github.com/fastly/cli@v1.7.2-0.20240304164155-9d0f1d77c3bf/pkg/commands/logging/datadog/update.go (about)

     1  package datadog
     2  
     3  import (
     4  	"io"
     5  
     6  	"github.com/fastly/go-fastly/v9/fastly"
     7  
     8  	"github.com/fastly/cli/pkg/argparser"
     9  	"github.com/fastly/cli/pkg/commands/logging/common"
    10  	"github.com/fastly/cli/pkg/errors"
    11  	"github.com/fastly/cli/pkg/global"
    12  	"github.com/fastly/cli/pkg/manifest"
    13  	"github.com/fastly/cli/pkg/text"
    14  )
    15  
    16  // UpdateCommand calls the Fastly API to update a Datadog logging endpoint.
    17  type UpdateCommand struct {
    18  	argparser.Base
    19  	Manifest manifest.Data
    20  
    21  	// Required.
    22  	EndpointName   string // Can't shadow argparser.Base method Name().
    23  	ServiceName    argparser.OptionalServiceNameID
    24  	ServiceVersion argparser.OptionalServiceVersion
    25  
    26  	// Optional.
    27  	AutoClone         argparser.OptionalAutoClone
    28  	NewName           argparser.OptionalString
    29  	Token             argparser.OptionalString
    30  	Region            argparser.OptionalString
    31  	Format            argparser.OptionalString
    32  	FormatVersion     argparser.OptionalInt
    33  	ResponseCondition argparser.OptionalString
    34  	Placement         argparser.OptionalString
    35  }
    36  
    37  // NewUpdateCommand returns a usable command registered under the parent.
    38  func NewUpdateCommand(parent argparser.Registerer, g *global.Data) *UpdateCommand {
    39  	c := UpdateCommand{
    40  		Base: argparser.Base{
    41  			Globals: g,
    42  		},
    43  	}
    44  	c.CmdClause = parent.Command("update", "Update a Datadog logging endpoint on a Fastly service version")
    45  
    46  	// Required.
    47  	c.CmdClause.Flag("name", "The name of the Datadog logging object").Short('n').Required().StringVar(&c.EndpointName)
    48  	c.RegisterFlag(argparser.StringFlagOpts{
    49  		Name:        argparser.FlagVersionName,
    50  		Description: argparser.FlagVersionDesc,
    51  		Dst:         &c.ServiceVersion.Value,
    52  		Required:    true,
    53  	})
    54  
    55  	// Optional.
    56  	c.RegisterAutoCloneFlag(argparser.AutoCloneFlagOpts{
    57  		Action: c.AutoClone.Set,
    58  		Dst:    &c.AutoClone.Value,
    59  	})
    60  	c.CmdClause.Flag("auth-token", "The API key from your Datadog account").Action(c.Token.Set).StringVar(&c.Token.Value)
    61  	common.Format(c.CmdClause, &c.Format)
    62  	common.FormatVersion(c.CmdClause, &c.FormatVersion)
    63  	c.CmdClause.Flag("new-name", "New name of the Datadog logging object").Action(c.NewName.Set).StringVar(&c.NewName.Value)
    64  	common.Placement(c.CmdClause, &c.Placement)
    65  	c.CmdClause.Flag("region", "The region that log data will be sent to. One of US, US3, US5, or EU. Defaults to US if undefined").Action(c.Region.Set).StringVar(&c.Region.Value)
    66  	common.ResponseCondition(c.CmdClause, &c.ResponseCondition)
    67  	c.RegisterFlag(argparser.StringFlagOpts{
    68  		Name:        argparser.FlagServiceIDName,
    69  		Description: argparser.FlagServiceIDDesc,
    70  		Dst:         &g.Manifest.Flag.ServiceID,
    71  		Short:       's',
    72  	})
    73  	c.RegisterFlag(argparser.StringFlagOpts{
    74  		Action:      c.ServiceName.Set,
    75  		Name:        argparser.FlagServiceName,
    76  		Description: argparser.FlagServiceDesc,
    77  		Dst:         &c.ServiceName.Value,
    78  	})
    79  	return &c
    80  }
    81  
    82  // ConstructInput transforms values parsed from CLI flags into an object to be used by the API client library.
    83  func (c *UpdateCommand) ConstructInput(serviceID string, serviceVersion int) (*fastly.UpdateDatadogInput, error) {
    84  	input := fastly.UpdateDatadogInput{
    85  		ServiceID:      serviceID,
    86  		ServiceVersion: serviceVersion,
    87  		Name:           c.EndpointName,
    88  	}
    89  
    90  	if c.NewName.WasSet {
    91  		input.NewName = &c.NewName.Value
    92  	}
    93  
    94  	if c.Token.WasSet {
    95  		input.Token = &c.Token.Value
    96  	}
    97  
    98  	if c.Region.WasSet {
    99  		input.Region = &c.Region.Value
   100  	}
   101  
   102  	if c.Format.WasSet {
   103  		input.Format = &c.Format.Value
   104  	}
   105  
   106  	if c.FormatVersion.WasSet {
   107  		input.FormatVersion = &c.FormatVersion.Value
   108  	}
   109  
   110  	if c.ResponseCondition.WasSet {
   111  		input.ResponseCondition = &c.ResponseCondition.Value
   112  	}
   113  
   114  	if c.Placement.WasSet {
   115  		input.Placement = &c.Placement.Value
   116  	}
   117  
   118  	return &input, nil
   119  }
   120  
   121  // Exec invokes the application logic for the command.
   122  func (c *UpdateCommand) Exec(_ io.Reader, out io.Writer) error {
   123  	serviceID, serviceVersion, err := argparser.ServiceDetails(argparser.ServiceDetailsOpts{
   124  		AutoCloneFlag:      c.AutoClone,
   125  		APIClient:          c.Globals.APIClient,
   126  		Manifest:           *c.Globals.Manifest,
   127  		Out:                out,
   128  		ServiceNameFlag:    c.ServiceName,
   129  		ServiceVersionFlag: c.ServiceVersion,
   130  		VerboseMode:        c.Globals.Flags.Verbose,
   131  	})
   132  	if err != nil {
   133  		c.Globals.ErrLog.AddWithContext(err, map[string]any{
   134  			"Service ID":      serviceID,
   135  			"Service Version": errors.ServiceVersion(serviceVersion),
   136  		})
   137  		return err
   138  	}
   139  
   140  	input, err := c.ConstructInput(serviceID, fastly.ToValue(serviceVersion.Number))
   141  	if err != nil {
   142  		c.Globals.ErrLog.Add(err)
   143  		return err
   144  	}
   145  
   146  	datadog, err := c.Globals.APIClient.UpdateDatadog(input)
   147  	if err != nil {
   148  		c.Globals.ErrLog.Add(err)
   149  		return err
   150  	}
   151  
   152  	text.Success(out,
   153  		"Updated Datadog logging endpoint %s (service %s version %d)",
   154  		fastly.ToValue(datadog.Name),
   155  		fastly.ToValue(datadog.ServiceID),
   156  		fastly.ToValue(datadog.ServiceVersion),
   157  	)
   158  	return nil
   159  }