37 lines
782 B
Go
37 lines
782 B
Go
|
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"
|
||
|
|
)
|
||
|
|
|
||
|
|
const FlagProxyName = "proxy-name"
|
||
|
|
|
||
|
|
func WithProxyFlags(flags ...cli.Flag) []cli.Flag {
|
||
|
|
baseFlags := clientFlag.ComposeFlags(
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: FlagProxyName,
|
||
|
|
Usage: "use `PROXY_NAME` as targeted proxy",
|
||
|
|
Value: "",
|
||
|
|
Required: true,
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
flags = append(flags, baseFlags...)
|
||
|
|
|
||
|
|
return flags
|
||
|
|
}
|
||
|
|
|
||
|
|
func AssertProxyName(ctx *cli.Context) (store.ProxyName, error) {
|
||
|
|
rawProxyName := ctx.String(FlagProxyName)
|
||
|
|
|
||
|
|
name, err := store.ValidateName(rawProxyName)
|
||
|
|
if err != nil {
|
||
|
|
return "", errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return store.ProxyName(name), nil
|
||
|
|
}
|