forked from TrueCloudLab/frostfs-node
90 lines
2.3 KiB
Go
90 lines
2.3 KiB
Go
package commonflags
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// Common CLI flag keys, shorthands, default
|
|
// values and their usage descriptions.
|
|
const (
|
|
GenerateKey = "generate-key"
|
|
generateKeyShorthand = "g"
|
|
generateKeyDefault = false
|
|
generateKeyUsage = "Generate new private key"
|
|
|
|
WalletPath = "wallet"
|
|
WalletPathShorthand = "w"
|
|
WalletPathDefault = ""
|
|
WalletPathUsage = "Path to the wallet or binary key"
|
|
|
|
Account = "address"
|
|
AccountShorthand = ""
|
|
AccountDefault = ""
|
|
AccountUsage = "Address of wallet account"
|
|
|
|
RPC = "rpc-endpoint"
|
|
RPCShorthand = "r"
|
|
RPCDefault = ""
|
|
RPCUsage = "Remote node address (as 'multiaddr' or '<host>:<port>')"
|
|
|
|
Timeout = "timeout"
|
|
TimeoutShorthand = "t"
|
|
TimeoutDefault = 15 * time.Second
|
|
TimeoutUsage = "Timeout for an operation"
|
|
|
|
Verbose = "verbose"
|
|
VerboseShorthand = "v"
|
|
VerboseUsage = "Verbose output"
|
|
|
|
ForceFlag = "force"
|
|
ForceFlagShorthand = "f"
|
|
|
|
CIDFlag = "cid"
|
|
CIDFlagUsage = "Container ID."
|
|
|
|
OIDFlag = "oid"
|
|
OIDFlagUsage = "Object ID."
|
|
|
|
TracingFlag = "trace"
|
|
TracingFlagUsage = "Generate trace ID and print it."
|
|
)
|
|
|
|
// Init adds common flags to the command:
|
|
// - GenerateKey,
|
|
// - WalletPath,
|
|
// - Account,
|
|
// - RPC,
|
|
// - Tracing,
|
|
// - Timeout.
|
|
func Init(cmd *cobra.Command) {
|
|
InitWithoutRPC(cmd)
|
|
|
|
ff := cmd.Flags()
|
|
ff.StringP(RPC, RPCShorthand, RPCDefault, RPCUsage)
|
|
ff.Bool(TracingFlag, false, TracingFlagUsage)
|
|
ff.DurationP(Timeout, TimeoutShorthand, TimeoutDefault, TimeoutUsage)
|
|
}
|
|
|
|
// InitWithoutRPC is similar to Init but doesn't create the RPC flag.
|
|
func InitWithoutRPC(cmd *cobra.Command) {
|
|
ff := cmd.Flags()
|
|
|
|
ff.BoolP(GenerateKey, generateKeyShorthand, generateKeyDefault, generateKeyUsage)
|
|
ff.StringP(WalletPath, WalletPathShorthand, WalletPathDefault, WalletPathUsage)
|
|
ff.StringP(Account, AccountShorthand, AccountDefault, AccountUsage)
|
|
}
|
|
|
|
// Bind binds common command flags to the viper.
|
|
func Bind(cmd *cobra.Command) {
|
|
ff := cmd.Flags()
|
|
|
|
_ = viper.BindPFlag(GenerateKey, ff.Lookup(GenerateKey))
|
|
_ = viper.BindPFlag(WalletPath, ff.Lookup(WalletPath))
|
|
_ = viper.BindPFlag(Account, ff.Lookup(Account))
|
|
_ = viper.BindPFlag(RPC, ff.Lookup(RPC))
|
|
_ = viper.BindPFlag(Timeout, ff.Lookup(Timeout))
|
|
}
|