package proxy import ( "net/url" "os" "forge.cadoles.com/cadoles/bouncer/internal/client" "forge.cadoles.com/cadoles/bouncer/internal/command/client/apierr" clientFlag "forge.cadoles.com/cadoles/bouncer/internal/command/client/flag" proxyFlag "forge.cadoles.com/cadoles/bouncer/internal/command/client/proxy/flag" "forge.cadoles.com/cadoles/bouncer/internal/format" "github.com/pkg/errors" "github.com/urfave/cli/v2" ) func UpdateCommand() *cli.Command { return &cli.Command{ Name: "update", Usage: "Update proxy", Flags: proxyFlag.WithProxyFlags( &cli.StringFlag{ Name: "to", Usage: "Set `TO` as proxy's destination url", }, &cli.StringSliceFlag{ Name: "from", Usage: "Set `FROM` as proxy's patterns to match incoming requests", }, &cli.BoolFlag{ Name: "enabled", Usage: "Enable or disable proxy", }, &cli.IntFlag{ Name: "weight", Usage: "Set `WEIGHT` as proxy's weight", }, ), Action: func(ctx *cli.Context) error { baseFlags := clientFlag.GetBaseFlags(ctx) token, err := clientFlag.GetToken(baseFlags) if err != nil { return errors.WithStack(apierr.Wrap(err)) } proxyName, err := proxyFlag.AssertProxyName(ctx) if err != nil { return errors.WithStack(err) } opts := &client.UpdateProxyOptions{} if ctx.IsSet("to") { to := ctx.String("to") if _, err := url.Parse(to); err != nil { return errors.Wrap(err, "'to' parameter should be a valid url") } opts.To = &to } from := ctx.StringSlice("from") if from != nil { opts.From = from } if ctx.IsSet("weight") { weight := ctx.Int("weight") opts.Weight = &weight } if ctx.IsSet("enabled") { enabled := ctx.Bool("enabled") opts.Enabled = &enabled } client := client.New(baseFlags.ServerURL, client.WithToken(token)) proxy, err := client.UpdateProxy(ctx.Context, proxyName, opts) if err != nil { return errors.WithStack(apierr.Wrap(err)) } hints := proxyHints(baseFlags.OutputMode) if err := format.Write(baseFlags.Format, os.Stdout, hints, proxy); err != nil { return errors.WithStack(err) } return nil }, } }