github.com/Bytom/bytom@v1.1.2-0.20210127130405-ae40204c0b09/cmd/bytomcli/commands/account.go

package commands

import (
	"os"
	"strings"

	"github.com/spf13/cobra"
	jww "github.com/spf13/jwalterweatherman"

	"github.com/bytom/bytom/crypto/ed25519/chainkd"
	"github.com/bytom/bytom/util"
)

func init() {
	createAccountCmd.PersistentFlags().IntVarP(&accountQuorum, "quorom", "q", 1, "quorum must be greater than 0 and less than or equal to the number of signers")
	createAccountCmd.PersistentFlags().StringVarP(&accountToken, "access", "a", "", "access token")

	updateAccountAliasCmd.PersistentFlags().StringVar(&accountID, "id", "", "account ID")
	updateAccountAliasCmd.PersistentFlags().StringVar(&accountAlias, "alias", "", "account alias")

	listAccountsCmd.PersistentFlags().StringVar(&accountID, "id", "", "account ID")
	listAccountsCmd.PersistentFlags().StringVar(&accountAlias, "alias", "", "account alias")

	deleteAccountCmd.PersistentFlags().StringVar(&accountID, "id", "", "account ID")
	deleteAccountCmd.PersistentFlags().StringVar(&accountAlias, "alias", "", "account alias")

	listAddressesCmd.PersistentFlags().StringVar(&accountID, "id", "", "account ID")
	listAddressesCmd.PersistentFlags().StringVar(&accountAlias, "alias", "", "account alias")

	listBalancesCmd.PersistentFlags().StringVar(&accountID, "id", "", "account ID")
	listBalancesCmd.PersistentFlags().StringVar(&accountAlias, "alias", "", "account alias")

	listUnspentOutputsCmd.PersistentFlags().StringVar(&accountID, "account_id", "", "account ID")
	listUnspentOutputsCmd.PersistentFlags().StringVar(&accountAlias, "account_alias", "", "account alias")
	listUnspentOutputsCmd.PersistentFlags().StringVar(&outputID, "id", "", "ID of unspent output")
	listUnspentOutputsCmd.PersistentFlags().BoolVar(&unconfirmed, "unconfirmed", false, "list unconfirmed unspent outputs")
	listUnspentOutputsCmd.PersistentFlags().BoolVar(&smartContract, "contract", false, "list smart contract unspent outputs")
	listUnspentOutputsCmd.PersistentFlags().IntVar(&from, "from", 0, "the starting position of a page")
	listUnspentOutputsCmd.PersistentFlags().IntVar(&count, "count", 0, "the longest count per page")
}

var (
	accountID     = ""
	accountAlias  = ""
	accountQuorum = 1
	accountToken  = ""
	outputID      = ""
	smartContract = false
	from          = 0
	count         = 0
)

var createAccountCmd = &cobra.Command{
	Use:   "create-account <alias> <xpub(s)>",
	Short: "Create an account",
	Args:  cobra.MinimumNArgs(2),
	Run: func(cmd *cobra.Command, args []string) {
		ins := accountIns{}

		for _, x := range args[1:] {
			xpub := chainkd.XPub{}
			if err := xpub.UnmarshalText([]byte(x)); err != nil {
				jww.ERROR.Println(err)
				os.Exit(util.ErrLocalExe)
			}
			ins.RootXPubs = append(ins.RootXPubs, xpub)
		}

		ins.Quorum = accountQuorum
		ins.Alias = args[0]
		ins.AccessToken = accountToken

		data, exitCode := util.ClientCall("/create-account", &ins)
		if exitCode != util.Success {
			os.Exit(exitCode)
		}

		printJSON(data)
	},
}

var updateAccountAliasCmd = &cobra.Command{
	Use:   "update-account-alias <newAlias>",
	Short: "update account alias",
	Args:  cobra.ExactArgs(1),
	Run: func(cmd *cobra.Command, args []string) {
		var ins = struct {
			AccountID    string `json:"account_id"`
			AccountAlias string `json:"account_alias"`
			NewAlias     string `json:"new_alias"`
		}{AccountID: accountID, AccountAlias: accountAlias, NewAlias: args[0]}

		_, exitCode := util.ClientCall("/update-account-alias", &ins)
		if exitCode != util.Success {
			os.Exit(exitCode)
		}

		jww.FEEDBACK.Println("Successfully update account alias")
	},
}

var listAccountsCmd = &cobra.Command{
	Use:   "list-accounts",
	Short: "List the existing accounts",
	Args:  cobra.NoArgs,
	Run: func(cmd *cobra.Command, args []string) {
		filter := struct {
			ID    string `json:"id"`
			Alias string `json:"alias"`
		}{ID: accountID, Alias: accountAlias}

		data, exitCode := util.ClientCall("/list-accounts", &filter)
		if exitCode != util.Success {
			os.Exit(exitCode)
		}

		printJSONList(data)
	},
}

var deleteAccountCmd = &cobra.Command{
	Use:   "delete-account <[accountAlias]|[accountID]>",
	Short: "Delete the existing account",
	Args:  cobra.NoArgs,
	Run: func(cmd *cobra.Command, args []string) {
		var ins = struct {
			AccountID    string `json:"account_id"`
			AccountAlias string `json:"account_alias"`
		}{AccountID: accountID, AccountAlias: accountAlias}

		if _, exitCode := util.ClientCall("/delete-account", &ins); exitCode != util.Success {
			os.Exit(exitCode)
		}

		jww.FEEDBACK.Println("Successfully delete account")
	},
}

var createAccountReceiverCmd = &cobra.Command{
	Use:   "create-account-receiver <accountAlias> [accountID]",
	Short: "Create an account receiver",
	Args:  cobra.RangeArgs(1, 2),
	Run: func(cmd *cobra.Command, args []string) {
		var ins = struct {
			AccountID    string `json:"account_id"`
			AccountAlias string `json:"account_alias"`
		}{AccountAlias: args[0]}

		if len(args) == 2 {
			ins.AccountID = args[1]
		}

		data, exitCode := util.ClientCall("/create-account-receiver", &ins)
		if exitCode != util.Success {
			os.Exit(exitCode)
		}

		printJSON(data)
	},
}

var listAddressesCmd = &cobra.Command{
	Use:   "list-addresses",
	Short: "List the account addresses",
	Args:  cobra.NoArgs,
	Run: func(cmd *cobra.Command, args []string) {
		var ins = struct {
			AccountID    string `json:"account_id"`
			AccountAlias string `json:"account_alias"`
		}{AccountID: accountID, AccountAlias: accountAlias}

		data, exitCode := util.ClientCall("/list-addresses", &ins)
		if exitCode != util.Success {
			os.Exit(exitCode)
		}

		printJSONList(data)
	},
}

var validateAddressCmd = &cobra.Command{
	Use:   "validate-address <address>",
	Short: "validate the account addresses",
	Args:  cobra.ExactArgs(1),
	Run: func(cmd *cobra.Command, args []string) {
		var ins = struct {
			Address string `json:"address"`
		}{Address: args[0]}

		data, exitCode := util.ClientCall("/validate-address", &ins)
		if exitCode != util.Success {
			os.Exit(exitCode)
		}

		printJSON(data)
	},
}

var listPubKeysCmd = &cobra.Command{
	Use:   "list-pubkeys <accountInfo> [publicKey]",
	Short: "list the account pubkeys",
	Args:  cobra.RangeArgs(1, 2),
	Run: func(cmd *cobra.Command, args []string) {
		var ins = struct {
			AccountID    string `json:"account_id"`
			AccountAlias string `json:"account_alias"`
			PublicKey    string `json:"public_key"`
		}{}

		if len(args[0]) == 13 && strings.HasPrefix(args[0], "0") {
			ins.AccountID = args[0]
		} else {
			ins.AccountAlias = args[0]
		}

		if len(args) == 2 {
			ins.PublicKey = args[1]
		}

		data, exitCode := util.ClientCall("/list-pubkeys", &ins)
		if exitCode != util.Success {
			os.Exit(exitCode)
		}

		printJSON(data)
	},
}

var listBalancesCmd = &cobra.Command{
	Use:   "list-balances",
	Short: "List the accounts balances",
	Args:  cobra.NoArgs,
	Run: func(cmd *cobra.Command, args []string) {
		var filter = struct {
			AccountID    string `json:"account_id"`
			AccountAlias string `json:"account_alias"`
		}{AccountID: accountID, AccountAlias: accountAlias}

		data, exitCode := util.ClientCall("/list-balances", &filter)
		if exitCode != util.Success {
			os.Exit(exitCode)
		}

		printJSONList(data)
	},
}

var listUnspentOutputsCmd = &cobra.Command{
	Use:   "list-unspent-outputs",
	Short: "List the accounts unspent outputs",
	Args:  cobra.NoArgs,
	Run: func(cmd *cobra.Command, args []string) {
		filter := struct {
			AccountID     string `json:"account_id"`
			AccountAlias  string `json:"account_alias"`
			ID            string `json:"id"`
			Unconfirmed   bool   `json:"unconfirmed"`
			SmartContract bool   `json:"smart_contract"`
			From          uint   `json:"from"`
			Count         uint   `json:"count"`
		}{
			AccountID:     accountID,
			AccountAlias:  accountAlias,
			ID:            outputID,
			Unconfirmed:   unconfirmed,
			SmartContract: smartContract,
			From:          uint(from),
			Count:         uint(count),
		}

		data, exitCode := util.ClientCall("/list-unspent-outputs", &filter)
		if exitCode != util.Success {
			os.Exit(exitCode)
		}

		printJSONList(data)
	},
}