package flag import ( clientFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/flag" "forge.cadoles.com/cadoles/bouncer/internal/store" "github.com/pkg/errors" "github.com/urfave/cli/v2" ) func WithProxyFlags(flags ...cli.Flag) []cli.Flag { baseFlags := clientFlag.ComposeFlags( ProxyName(), ) flags = append(flags, baseFlags...) return flags } const KeyProxyName = "proxy-name" func ProxyName() cli.Flag { return &cli.StringFlag{ Name: KeyProxyName, Usage: "use `PROXY_NAME` as targeted proxy", Value: "", Required: true, } } const KeyProxyTo = "proxy-to" func ProxyTo() cli.Flag { return &cli.StringFlag{ Name: KeyProxyTo, Usage: "Set `PROXY_TO` as proxy's destination url", Value: "", } } const KeyProxyFrom = "proxy-from" func ProxyFrom() cli.Flag { return &cli.StringSliceFlag{ Name: KeyProxyFrom, Usage: "Set `PROXY_FROM` as proxy's patterns to match incoming requests", Value: cli.NewStringSlice("*"), } } const KeyProxyWeight = "proxy-weight" func ProxyWeight() cli.Flag { return &cli.IntFlag{ Name: KeyProxyWeight, Usage: "Set `PROXY_WEIGHT` as proxy's weight", } } const KeyProxyEnabled = "proxy-enabled" func ProxyEnabled() cli.Flag { return &cli.BoolFlag{ Name: KeyProxyEnabled, Usage: "Enable or disable proxy", } } func AssertProxyName(ctx *cli.Context) (store.ProxyName, error) { rawProxyName := ctx.String(KeyProxyName) name, err := store.ValidateName(rawProxyName) if err != nil { return "", errors.WithStack(err) } return store.ProxyName(name), nil }