forked from TrueCloudLab/neoneo-go
16ce63e653
This makes rpc flags consistent across all commands, previously some commands used 'endpoint, e' and some 'rpc, r', some had ability to change timeout and some hadn't. Now 'rpc-endpoint, r' is used everywhere along with 'timeout, t'.
377 lines
9.4 KiB
Go
377 lines
9.4 KiB
Go
package wallet
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
|
|
"github.com/nspcc-dev/neo-go/cli/flags"
|
|
"github.com/nspcc-dev/neo-go/cli/options"
|
|
"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/smartcontract/context"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/wallet"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
func newNEP5Commands() []cli.Command {
|
|
balanceFlags := []cli.Flag{
|
|
walletPathFlag,
|
|
cli.StringFlag{
|
|
Name: "addr",
|
|
Usage: "Address to use",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "token",
|
|
Usage: "Token to use",
|
|
},
|
|
}
|
|
balanceFlags = append(balanceFlags, options.RPC...)
|
|
importFlags := []cli.Flag{
|
|
walletPathFlag,
|
|
cli.StringFlag{
|
|
Name: "token",
|
|
Usage: "Token contract hash in LE",
|
|
},
|
|
}
|
|
importFlags = append(importFlags, options.RPC...)
|
|
transferFlags := []cli.Flag{
|
|
walletPathFlag,
|
|
outFlag,
|
|
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",
|
|
},
|
|
}
|
|
transferFlags = append(transferFlags, options.RPC...)
|
|
return []cli.Command{
|
|
{
|
|
Name: "balance",
|
|
Usage: "get address balance",
|
|
UsageText: "balance --path <path> --rpc-endpoint <node> --timeout <time> --addr <addr> [--token <hash-or-name>]",
|
|
Action: getNEP5Balance,
|
|
Flags: balanceFlags,
|
|
},
|
|
{
|
|
Name: "import",
|
|
Usage: "import NEP5 token to a wallet",
|
|
UsageText: "import --path <path> --rpc-endpoint <node> --timeout <time> --token <hash>",
|
|
Action: importNEP5Token,
|
|
Flags: importFlags,
|
|
},
|
|
{
|
|
Name: "info",
|
|
Usage: "print imported NEP5 token info",
|
|
UsageText: "print --path <path> [--token <hash-or-name>]",
|
|
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 <path> <hash-or-name>",
|
|
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 <path> --rpc-endpoint <node> --timeout <time> --from <addr> --to <addr> --token <hash> --amount string",
|
|
Action: transferNEP5,
|
|
Flags: transferFlags,
|
|
},
|
|
}
|
|
}
|
|
|
|
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 := options.GetTimeoutContext(ctx)
|
|
defer cancel()
|
|
|
|
c, err := options.GetRPCClient(gctx, ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
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 := options.GetTimeoutContext(ctx)
|
|
defer cancel()
|
|
|
|
c, err := options.GetRPCClient(gctx, ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
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 := options.GetTimeoutContext(ctx)
|
|
defer cancel()
|
|
|
|
c, err := options.GetRPCClient(gctx, ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
tx, err := c.CreateNEP5TransferTx(acc, to, token.Hash, amount, gas)
|
|
if err != nil {
|
|
return cli.NewExitError(err, 1)
|
|
}
|
|
|
|
if outFile := ctx.String("out"); outFile != "" {
|
|
priv := acc.PrivateKey()
|
|
pub := priv.PublicKey()
|
|
sign := priv.Sign(tx.GetSignedPart())
|
|
scCtx := context.NewParameterContext("Neo.Core.ContractTransaction", tx)
|
|
if err := scCtx.AddSignature(acc.Contract, pub, sign); err != nil {
|
|
return cli.NewExitError(fmt.Errorf("can't add signature: %v", err), 1)
|
|
} else if data, err := json.Marshal(scCtx); err != nil {
|
|
return cli.NewExitError(fmt.Errorf("can't marshal tx to JSON: %v", err), 1)
|
|
} else if err := ioutil.WriteFile(outFile, data, 0644); err != nil {
|
|
return cli.NewExitError(fmt.Errorf("can't write tx to file: %v", err), 1)
|
|
}
|
|
} else {
|
|
_ = acc.SignTx(tx)
|
|
if err := c.SendRawTransaction(tx); err != nil {
|
|
return cli.NewExitError(err, 1)
|
|
}
|
|
}
|
|
|
|
fmt.Println(tx.Hash().StringLE())
|
|
return nil
|
|
}
|