+++ /dev/null
-package commands
-
-import (
- "fmt"
- "os"
- "regexp"
-
- "github.com/spf13/cobra"
-
- "github.com/vapor/util"
-)
-
-// bytomcli usage template
-var usageTemplate = `Usage:{{if .Runnable}}
- {{.UseLine}}{{end}}{{if .HasAvailableSubCommands}}
- {{.CommandPath}} [command]{{end}}{{if gt (len .Aliases) 0}}
-
-Aliases:
- {{.NameAndAliases}}{{end}}{{if .HasExample}}
-
-Examples:
-{{.Example}}{{end}}{{if .HasAvailableSubCommands}}
-
-Available Commands:
- {{range .Commands}}{{if (and .IsAvailableCommand (.Name | WalletDisable))}}
- {{rpad .Name .NamePadding }} {{.Short}}{{end}}{{end}}
-
- available with wallet enable:
- {{range .Commands}}{{if (and .IsAvailableCommand (.Name | WalletEnable))}}
- {{rpad .Name .NamePadding }} {{.Short}}{{end}}{{end}}{{end}}{{if .HasAvailableLocalFlags}}
-
-Flags:
-{{.LocalFlags.FlagUsages | trimTrailingWhitespaces}}{{end}}{{if .HasAvailableInheritedFlags}}
-
-Global Flags:
-{{.InheritedFlags.FlagUsages | trimTrailingWhitespaces}}{{end}}{{if .HasHelpSubCommands}}
-
-Additional help topics:{{range .Commands}}{{if .IsAdditionalHelpTopicCommand}}
- {{rpad .CommandPath .CommandPathPadding}} {{.Short}}{{end}}{{end}}{{end}}{{if .HasAvailableSubCommands}}
-
-Use "{{.CommandPath}} [command] --help" for more information about a command.{{end}}
-`
-
-// commandError is an error used to signal different error situations in command handling.
-type commandError struct {
- s string
- userError bool
-}
-
-func (c commandError) Error() string {
- return c.s
-}
-
-func (c commandError) isUserError() bool {
- return c.userError
-}
-
-func newUserError(a ...interface{}) commandError {
- return commandError{s: fmt.Sprintln(a...), userError: true}
-}
-
-func newSystemError(a ...interface{}) commandError {
- return commandError{s: fmt.Sprintln(a...), userError: false}
-}
-
-func newSystemErrorF(format string, a ...interface{}) commandError {
- return commandError{s: fmt.Sprintf(format, a...), userError: false}
-}
-
-// Catch some of the obvious user errors from Cobra.
-// We don't want to show the usage message for every error.
-// The below may be to generic. Time will show.
-var userErrorRegexp = regexp.MustCompile("argument|flag|shorthand")
-
-func isUserError(err error) bool {
- if cErr, ok := err.(commandError); ok && cErr.isUserError() {
- return true
- }
-
- return userErrorRegexp.MatchString(err.Error())
-}
-
-// BytomcliCmd is Bytomcli's root command.
-// Every other command attached to BytomcliCmd is a child command to it.
-var BytomcliCmd = &cobra.Command{
- Use: "bytomcli",
- Short: "Bytomcli is a commond line client for bytom core (a.k.a. bytomd)",
- Run: func(cmd *cobra.Command, args []string) {
- if len(args) < 1 {
- cmd.SetUsageTemplate(usageTemplate)
- cmd.Usage()
- }
- },
-}
-
-// Execute adds all child commands to the root command BytomcliCmd and sets flags appropriately.
-func Execute() {
-
- AddCommands()
- AddTemplateFunc()
-
- if _, err := BytomcliCmd.ExecuteC(); err != nil {
- os.Exit(util.ErrLocalExe)
- }
-}
-
-// AddCommands adds child commands to the root command BytomcliCmd.
-func AddCommands() {
- BytomcliCmd.AddCommand(createAccessTokenCmd)
- BytomcliCmd.AddCommand(listAccessTokenCmd)
- BytomcliCmd.AddCommand(deleteAccessTokenCmd)
- BytomcliCmd.AddCommand(checkAccessTokenCmd)
-
- BytomcliCmd.AddCommand(createAccountCmd)
- BytomcliCmd.AddCommand(deleteAccountCmd)
- BytomcliCmd.AddCommand(listAccountsCmd)
- BytomcliCmd.AddCommand(createAccountReceiverCmd)
- BytomcliCmd.AddCommand(listAddressesCmd)
- BytomcliCmd.AddCommand(validateAddressCmd)
- BytomcliCmd.AddCommand(listPubKeysCmd)
-
- BytomcliCmd.AddCommand(createAssetCmd)
- BytomcliCmd.AddCommand(getAssetCmd)
- BytomcliCmd.AddCommand(listAssetsCmd)
- BytomcliCmd.AddCommand(updateAssetAliasCmd)
-
- BytomcliCmd.AddCommand(getTransactionCmd)
- BytomcliCmd.AddCommand(listTransactionsCmd)
-
- BytomcliCmd.AddCommand(getUnconfirmedTransactionCmd)
- BytomcliCmd.AddCommand(listUnconfirmedTransactionsCmd)
- BytomcliCmd.AddCommand(decodeRawTransactionCmd)
-
- BytomcliCmd.AddCommand(listUnspentOutputsCmd)
- BytomcliCmd.AddCommand(listBalancesCmd)
-
- BytomcliCmd.AddCommand(rescanWalletCmd)
- BytomcliCmd.AddCommand(walletInfoCmd)
-
- BytomcliCmd.AddCommand(buildTransactionCmd)
- BytomcliCmd.AddCommand(signTransactionCmd)
- BytomcliCmd.AddCommand(submitTransactionCmd)
- BytomcliCmd.AddCommand(estimateTransactionGasCmd)
-
- BytomcliCmd.AddCommand(getBlockCountCmd)
- BytomcliCmd.AddCommand(getBlockHashCmd)
- BytomcliCmd.AddCommand(getBlockCmd)
- BytomcliCmd.AddCommand(getBlockHeaderCmd)
- BytomcliCmd.AddCommand(getDifficultyCmd)
- BytomcliCmd.AddCommand(getHashRateCmd)
-
- BytomcliCmd.AddCommand(createKeyCmd)
- BytomcliCmd.AddCommand(deleteKeyCmd)
- BytomcliCmd.AddCommand(listKeysCmd)
- BytomcliCmd.AddCommand(resetKeyPwdCmd)
- BytomcliCmd.AddCommand(checkKeyPwdCmd)
-
- BytomcliCmd.AddCommand(signMsgCmd)
- BytomcliCmd.AddCommand(verifyMsgCmd)
- BytomcliCmd.AddCommand(decodeProgCmd)
-
- BytomcliCmd.AddCommand(createTransactionFeedCmd)
- BytomcliCmd.AddCommand(listTransactionFeedsCmd)
- BytomcliCmd.AddCommand(deleteTransactionFeedCmd)
- BytomcliCmd.AddCommand(getTransactionFeedCmd)
- BytomcliCmd.AddCommand(updateTransactionFeedCmd)
-
- BytomcliCmd.AddCommand(isMiningCmd)
- BytomcliCmd.AddCommand(setMiningCmd)
-
- BytomcliCmd.AddCommand(netInfoCmd)
- BytomcliCmd.AddCommand(gasRateCmd)
-
- BytomcliCmd.AddCommand(versionCmd)
-}
-
-// AddTemplateFunc adds usage template to the root command BytomcliCmd.
-func AddTemplateFunc() {
- walletEnableCmd := []string{
- createAccountCmd.Name(),
- listAccountsCmd.Name(),
- deleteAccountCmd.Name(),
- createAccountReceiverCmd.Name(),
- listAddressesCmd.Name(),
- validateAddressCmd.Name(),
- listPubKeysCmd.Name(),
-
- createAssetCmd.Name(),
- getAssetCmd.Name(),
- listAssetsCmd.Name(),
- updateAssetAliasCmd.Name(),
-
- createKeyCmd.Name(),
- deleteKeyCmd.Name(),
- listKeysCmd.Name(),
- resetKeyPwdCmd.Name(),
- checkKeyPwdCmd.Name(),
- signMsgCmd.Name(),
-
- buildTransactionCmd.Name(),
- signTransactionCmd.Name(),
-
- getTransactionCmd.Name(),
- listTransactionsCmd.Name(),
- listUnspentOutputsCmd.Name(),
- listBalancesCmd.Name(),
-
- rescanWalletCmd.Name(),
- walletInfoCmd.Name(),
- }
-
- cobra.AddTemplateFunc("WalletEnable", func(cmdName string) bool {
- for _, name := range walletEnableCmd {
- if name == cmdName {
- return true
- }
- }
- return false
- })
-
- cobra.AddTemplateFunc("WalletDisable", func(cmdName string) bool {
- for _, name := range walletEnableCmd {
- if name == cmdName {
- return false
- }
- }
- return true
- })
-}
OSDN Git Service
