99 lines
2.3 KiB
Go
99 lines
2.3 KiB
Go
|
package flag
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"forge.cadoles.com/cadoles/bouncer/internal/format"
|
||
|
|
"forge.cadoles.com/cadoles/bouncer/internal/format/table"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"github.com/urfave/cli/v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
func ComposeFlags(flags ...cli.Flag) []cli.Flag {
|
||
|
|
baseFlags := []cli.Flag{
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "server",
|
||
|
|
Aliases: []string{"s"},
|
||
|
|
Usage: "use `SERVER` as server url",
|
||
|
|
Value: "http://127.0.0.1:8081",
|
||
|
|
},
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "format",
|
||
|
|
Aliases: []string{"f"},
|
||
|
|
Usage: fmt.Sprintf("use `FORMAT` as output format (available: %s)", format.Available()),
|
||
|
|
Value: string(table.Format),
|
||
|
|
},
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "output-mode",
|
||
|
|
Aliases: []string{"m"},
|
||
|
|
Usage: fmt.Sprintf("use `MODE` as output mode (available: %s)", []format.OutputMode{format.OutputModeCompact, format.OutputModeWide}),
|
||
|
|
Value: string(format.OutputModeCompact),
|
||
|
|
},
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "token",
|
||
|
|
Aliases: []string{"t"},
|
||
|
|
EnvVars: []string{`BOUNCER_TOKEN`},
|
||
|
|
Usage: "use `TOKEN` as authentication token",
|
||
|
|
},
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "token-file",
|
||
|
|
EnvVars: []string{`BOUNCER_TOKEN_FILE`},
|
||
|
|
Usage: "use `TOKEN_FILE` as file containing the authentication token",
|
||
|
|
Value: ".bouncer-token",
|
||
|
|
TakesFile: true,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
flags = append(flags, baseFlags...)
|
||
|
|
|
||
|
|
return flags
|
||
|
|
}
|
||
|
|
|
||
|
|
type BaseFlags struct {
|
||
|
|
ServerURL string
|
||
|
|
Format format.Format
|
||
|
|
OutputMode format.OutputMode
|
||
|
|
Token string
|
||
|
|
TokenFile string
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetBaseFlags(ctx *cli.Context) *BaseFlags {
|
||
|
|
serverURL := ctx.String("server")
|
||
|
|
rawFormat := ctx.String("format")
|
||
|
|
rawOutputMode := ctx.String("output-mode")
|
||
|
|
tokenFile := ctx.String("token-file")
|
||
|
|
token := ctx.String("token")
|
||
|
|
|
||
|
|
return &BaseFlags{
|
||
|
|
ServerURL: serverURL,
|
||
|
|
Format: format.Format(rawFormat),
|
||
|
|
OutputMode: format.OutputMode(rawOutputMode),
|
||
|
|
Token: token,
|
||
|
|
TokenFile: tokenFile,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetToken(flags *BaseFlags) (string, error) {
|
||
|
|
if flags.Token != "" {
|
||
|
|
return flags.Token, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
if flags.TokenFile == "" {
|
||
|
|
return "", nil
|
||
|
|
}
|
||
|
|
|
||
|
|
rawToken, err := ioutil.ReadFile(flags.TokenFile)
|
||
|
|
if err != nil && !errors.Is(err, os.ErrNotExist) {
|
||
|
|
return "", errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if rawToken == nil {
|
||
|
|
return "", nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return strings.TrimSpace(string(rawToken)), nil
|
||
|
|
}
|