mirror of
https://github.com/turbot/steampipe.git
synced 2026-02-14 19:00:10 -05:00
64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package cmdconfig
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/fatih/color"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var requiredColor = color.New(color.Bold).SprintfFunc()
|
|
|
|
type flagOpt func(c *cobra.Command, name string, key string)
|
|
|
|
// FlagOptions :: shortcut for common flag options
|
|
var FlagOptions = struct {
|
|
Required func() flagOpt
|
|
Hidden func() flagOpt
|
|
Deprecated func(string) flagOpt
|
|
NoOptDefVal func(string) flagOpt
|
|
WithShortHand func(string) flagOpt
|
|
}{
|
|
Required: requiredOpt,
|
|
Hidden: hiddenOpt,
|
|
Deprecated: deprecatedOpt,
|
|
NoOptDefVal: noOptDefValOpt,
|
|
WithShortHand: withShortHand,
|
|
}
|
|
|
|
// Helper function to mark a flag as required
|
|
func requiredOpt() flagOpt {
|
|
return func(c *cobra.Command, name, key string) {
|
|
c.MarkFlagRequired(key)
|
|
key = fmt.Sprintf("required.%s", key)
|
|
viper.GetViper().Set(key, true)
|
|
u := c.Flag(name).Usage
|
|
c.Flag(name).Usage = fmt.Sprintf("%s %s", u, requiredColor("(required)"))
|
|
}
|
|
}
|
|
|
|
func hiddenOpt() flagOpt {
|
|
return func(c *cobra.Command, name, key string) {
|
|
c.Flag(name).Hidden = true
|
|
}
|
|
}
|
|
|
|
func deprecatedOpt(replacement string) flagOpt {
|
|
return func(c *cobra.Command, name, key string) {
|
|
c.Flag(name).Deprecated = fmt.Sprintf("please use %s", replacement)
|
|
}
|
|
}
|
|
|
|
func noOptDefValOpt(noOptDefVal string) flagOpt {
|
|
return func(c *cobra.Command, name, key string) {
|
|
c.Flag(name).NoOptDefVal = noOptDefVal
|
|
}
|
|
}
|
|
|
|
func withShortHand(shorthand string) flagOpt {
|
|
return func(c *cobra.Command, name, key string) {
|
|
c.Flag(name).Shorthand = shorthand
|
|
}
|
|
}
|