accountcreateadd.go

// Copyright (c) 2022 IoTeX Foundation
// This source code is provided 'as is' and no warranties are given as to title or non-infringement, merchantability
// or fitness for purpose and, to the extent permitted by law, all liability for your use of the code is disclaimed.
// This source code is governed by Apache License 2.0 that can be found in the LICENSE file.

package account

import (
	"fmt"

	"github.com/pkg/errors"
	"github.com/spf13/cobra"

	"github.com/iotexproject/iotex-core/v2/ioctl"
	"github.com/iotexproject/iotex-core/v2/ioctl/config"
	"github.com/iotexproject/iotex-core/v2/ioctl/validator"
)

// Multi-language support
var (
	_createAddCmdShorts = map[config.Language]string{
		config.English: "Create new account for ioctl",
		config.Chinese: "为ioctl创建新账户",
	}
	_createAddCmdUses = map[config.Language]string{
		config.English: "createadd ALIAS",
		config.Chinese: "createadd 别名",
	}
	_invalidAlias = map[config.Language]string{
		config.English: "invalid alias",
		config.Chinese: "无效别名",
	}
	_aliasHasAlreadyUsed = map[config.Language]string{
		config.English: "** Alias \"%s\" has already used for %s\n" +
			"Overwriting the account will keep the previous keystore file stay, " +
			"but bind the alias to the new one.\nWould you like to continue?\n",
		config.Chinese: "** 这个别名 \"%s\" 已被 %s 使用!\n" +
			"复写帐户后先前的 keystore 文件将会留存!\n" +
			"但底下的别名将绑定为新的。您是否要继续？",
	}
	_outputMessage = map[config.Language]string{
		config.English: "New account \"%s\" is created.\n" +
			"Please Keep your password, or you will lose your private key.",
		config.Chinese: "新帐户 \"%s\" 已建立。\n" +
			"请保护好您的密码，否则您会失去您的私钥。",
	}
)

// NewAccountCreateAdd represents the account createadd command
func NewAccountCreateAdd(client ioctl.Client) *cobra.Command {
	use, _ := client.SelectTranslation(_createAddCmdUses)
	short, _ := client.SelectTranslation(_createAddCmdShorts)
	_invalidAlias, _ := client.SelectTranslation(_invalidAlias)
	_aliasHasAlreadyUsed, _ := client.SelectTranslation(_aliasHasAlreadyUsed)
	infoQuit, _ := client.SelectTranslation(_infoQuit)
	failToWriteToConfigFile, _ := client.SelectTranslation(_failToWriteToConfigFile)
	failToGenerateNewPrivateKey, _ := client.SelectTranslation(_failToGenerateNewPrivateKey)
	failToGenerateNewPrivateKeySm2, _ := client.SelectTranslation(_failToGenerateNewPrivateKeySm2)
	_outputMessage, _ := client.SelectTranslation(_outputMessage)

	return &cobra.Command{
		Use:   use,
		Short: short,
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			cmd.SilenceUsage = true

			if err := validator.ValidateAlias(args[0]); err != nil {
				return errors.Wrap(err, _invalidAlias)
			}

			if addr, ok := client.Config().Aliases[args[0]]; ok {
				confirmed, err := client.AskToConfirm(fmt.Sprintf(_aliasHasAlreadyUsed, args[0], addr))
				if err != nil {
					return errors.Wrap(err, "failed to ask confirm")
				}
				if !confirmed {
					cmd.Println(infoQuit)
					return nil
				}
			}

			var addr string
			var err error
			if client.IsCryptoSm2() {
				addr, err = newAccountSm2(client, cmd, args[0])
				if err != nil {
					return errors.Wrap(err, failToGenerateNewPrivateKey)
				}
			} else {
				addr, err = newAccount(client, cmd, args[0])
				if err != nil {
					return errors.Wrap(err, failToGenerateNewPrivateKeySm2)
				}
			}
			if err := client.SetAliasAndSave(args[0], addr); err != nil {
				return errors.Wrapf(err, failToWriteToConfigFile)
			}
			cmd.Println(fmt.Sprintf(_outputMessage, args[0]))
			return nil
		},
	}
}