package wallet import ( "errors" "fmt" "github.com/nspcc-dev/neo-go/cli/flags" "github.com/nspcc-dev/neo-go/pkg/encoding/address" "github.com/nspcc-dev/neo-go/pkg/rpc/client" "github.com/nspcc-dev/neo-go/pkg/util" "github.com/nspcc-dev/neo-go/pkg/wallet" "github.com/urfave/cli" ) func newNEP5Commands() []cli.Command { return []cli.Command{ { Name: "balance", Usage: "get address balance", UsageText: "balance --path --rpc --addr [--token ]", Action: getNEP5Balance, Flags: []cli.Flag{ walletPathFlag, rpcFlag, timeoutFlag, cli.StringFlag{ Name: "addr", Usage: "Address to use", }, cli.StringFlag{ Name: "token", Usage: "Token to use", }, }, }, { Name: "import", Usage: "import NEP5 token to a wallet", UsageText: "import --path --rpc --token ", Action: importNEP5Token, Flags: []cli.Flag{ walletPathFlag, rpcFlag, cli.StringFlag{ Name: "token", Usage: "Token contract hash in LE", }, }, }, { Name: "info", Usage: "print imported NEP5 token info", UsageText: "print --path [--token ]", Action: printNEP5Info, Flags: []cli.Flag{ walletPathFlag, cli.StringFlag{ Name: "token", Usage: "Token name or hash", }, }, }, { Name: "remove", Usage: "remove NEP5 token from the wallet", UsageText: "remove --path ", Action: removeNEP5Token, Flags: []cli.Flag{ walletPathFlag, cli.StringFlag{ Name: "token", Usage: "Token name or hash", }, forceFlag, }, }, { Name: "transfer", Usage: "transfer NEP5 tokens", UsageText: "transfer --path --rpc --from --to --token --amount string", Action: transferNEP5, Flags: []cli.Flag{ walletPathFlag, rpcFlag, timeoutFlag, fromAddrFlag, toAddrFlag, cli.StringFlag{ Name: "token", Usage: "Token to use", }, cli.StringFlag{ Name: "amount", Usage: "Amount of asset to send", }, flags.Fixed8Flag{ Name: "gas", Usage: "Amount of GAS to attach to a tx", }, }, }, } } func getNEP5Balance(ctx *cli.Context) error { wall, err := openWallet(ctx.String("path")) if err != nil { return cli.NewExitError(err, 1) } defer wall.Close() addr := ctx.String("addr") addrHash, err := address.StringToUint160(addr) if err != nil { return cli.NewExitError(fmt.Errorf("invalid address: %v", err), 1) } acc := wall.GetAccount(addrHash) if acc == nil { return cli.NewExitError(fmt.Errorf("can't find account for the address: %s", addr), 1) } gctx, cancel := getGoContext(ctx) defer cancel() c, err := client.New(gctx, ctx.String("rpc"), client.Options{}) if err != nil { return cli.NewExitError(err, 1) } var token *wallet.Token name := ctx.String("token") if name != "" { token, err = getMatchingToken(wall, name) if err != nil { token, err = getMatchingTokenRPC(c, addrHash, name) if err != nil { return cli.NewExitError(err, 1) } } } balances, err := c.GetNEP5Balances(addrHash) if err != nil { return cli.NewExitError(err, 1) } for i := range balances.Balances { asset := balances.Balances[i].Asset if name != "" && !token.Hash.Equals(asset) { continue } fmt.Printf("TokenHash: %s\n", asset) fmt.Printf("\tAmount : %s\n", balances.Balances[i].Amount) fmt.Printf("\tUpdated: %d\n", balances.Balances[i].LastUpdated) } return nil } func getMatchingToken(w *wallet.Wallet, name string) (*wallet.Token, error) { return getMatchingTokenAux(func(i int) *wallet.Token { return w.Extra.Tokens[i] }, len(w.Extra.Tokens), name) } func getMatchingTokenRPC(c *client.Client, addr util.Uint160, name string) (*wallet.Token, error) { bs, err := c.GetNEP5Balances(addr) if err != nil { return nil, err } get := func(i int) *wallet.Token { t, _ := c.NEP5TokenInfo(bs.Balances[i].Asset) return t } return getMatchingTokenAux(get, len(bs.Balances), name) } func getMatchingTokenAux(get func(i int) *wallet.Token, n int, name string) (*wallet.Token, error) { var token *wallet.Token var count int for i := 0; i < n; i++ { t := get(i) if t != nil && (t.Name == name || t.Symbol == name || t.Address() == name || t.Hash.StringLE() == name) { if count == 1 { printTokenInfo(token) printTokenInfo(t) return nil, errors.New("multiple matching tokens found") } count++ token = t } } if count == 0 { return nil, errors.New("token was not found") } return token, nil } func importNEP5Token(ctx *cli.Context) error { wall, err := openWallet(ctx.String("path")) if err != nil { return cli.NewExitError(err, 1) } defer wall.Close() tokenHash, err := util.Uint160DecodeStringLE(ctx.String("token")) if err != nil { return cli.NewExitError(fmt.Errorf("invalid token contract hash: %v", err), 1) } for _, t := range wall.Extra.Tokens { if t.Hash.Equals(tokenHash) { printTokenInfo(t) return cli.NewExitError("token already exists", 1) } } gctx, cancel := getGoContext(ctx) defer cancel() c, err := client.New(gctx, ctx.String("rpc"), client.Options{}) if err != nil { return cli.NewExitError(err, 1) } tok, err := c.NEP5TokenInfo(tokenHash) if err != nil { return cli.NewExitError(fmt.Errorf("can't receive token info: %v", err), 1) } wall.AddToken(tok) if err := wall.Save(); err != nil { return cli.NewExitError(err, 1) } printTokenInfo(tok) return nil } func printTokenInfo(tok *wallet.Token) { fmt.Printf("Name:\t%s\n", tok.Name) fmt.Printf("Symbol:\t%s\n", tok.Symbol) fmt.Printf("Hash:\t%s\n", tok.Hash.StringLE()) fmt.Printf("Decimals: %d\n", tok.Decimals) fmt.Printf("Address: %s\n", tok.Address()) } func printNEP5Info(ctx *cli.Context) error { wall, err := openWallet(ctx.String("path")) if err != nil { return cli.NewExitError(err, 1) } defer wall.Close() if name := ctx.String("token"); name != "" { token, err := getMatchingToken(wall, name) if err != nil { return cli.NewExitError(err, 1) } printTokenInfo(token) return nil } for i, t := range wall.Extra.Tokens { if i > 0 { fmt.Println() } printTokenInfo(t) } return nil } func removeNEP5Token(ctx *cli.Context) error { wall, err := openWallet(ctx.String("path")) if err != nil { return cli.NewExitError(err, 1) } defer wall.Close() name := ctx.Args().First() if name == "" { return cli.NewExitError("token must be specified", 1) } token, err := getMatchingToken(wall, name) if err != nil { return cli.NewExitError(err, 1) } if !ctx.Bool("force") { if ok := askForConsent(); !ok { return nil } } if err := wall.RemoveToken(token.Hash); err != nil { return cli.NewExitError(fmt.Errorf("can't remove token: %v", err), 1) } else if err := wall.Save(); err != nil { return cli.NewExitError(fmt.Errorf("error while saving wallet: %v", err), 1) } return nil } func transferNEP5(ctx *cli.Context) error { wall, err := openWallet(ctx.String("path")) if err != nil { return cli.NewExitError(err, 1) } defer wall.Close() fromFlag := ctx.Generic("from").(*flags.Address) from := fromFlag.Uint160() acc := wall.GetAccount(from) if acc == nil { return cli.NewExitError(fmt.Errorf("can't find account for the address: %s", fromFlag), 1) } gctx, cancel := getGoContext(ctx) defer cancel() c, err := client.New(gctx, ctx.String("rpc"), client.Options{}) if err != nil { return cli.NewExitError(err, 1) } toFlag := ctx.Generic("to").(*flags.Address) to := toFlag.Uint160() token, err := getMatchingToken(wall, ctx.String("token")) if err != nil { fmt.Println("Can't find matching token in the wallet. Querying RPC-node for balances.") token, err = getMatchingTokenRPC(c, from, ctx.String("token")) if err != nil { return cli.NewExitError(err, 1) } } amount, err := util.FixedNFromString(ctx.String("amount"), int(token.Decimals)) if err != nil { return cli.NewExitError(fmt.Errorf("invalid amount: %v", err), 1) } gas := flags.Fixed8FromContext(ctx, "gas") if pass, err := readPassword("Password > "); err != nil { return cli.NewExitError(err, 1) } else if err := acc.Decrypt(pass); err != nil { return cli.NewExitError(err, 1) } hash, err := c.TransferNEP5(acc, to, token, amount, gas) if err != nil { return cli.NewExitError(err, 1) } fmt.Println(hash.StringLE()) return nil }