65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
package commonflags
|
|
|
|
import (
|
|
"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 = "WIF (NEP-2) string or 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>')"
|
|
|
|
Verbose = "verbose"
|
|
VerboseShorthand = "v"
|
|
VerboseUsage = "verbose output"
|
|
)
|
|
|
|
// Init adds common flags to the command:
|
|
// - GenerateKey
|
|
// - WalletPath
|
|
// - Account
|
|
// - RPC
|
|
func Init(cmd *cobra.Command) {
|
|
InitWithoutRPC(cmd)
|
|
|
|
ff := cmd.Flags()
|
|
ff.StringP(RPC, RPCShorthand, RPCDefault, RPCUsage)
|
|
}
|
|
|
|
// 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))
|
|
}
|