TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

cli: use AddressFlag for nep17 import

Browse source
This commit is contained in:
Anna Shaleva 2021-04-16 12:29:46 +03:00
parent 92c6708fa0
commit ced90e0510
2 changed files with 13 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
cli/wallet/nep17.go
View file

@ -40,9 +40,9 @@ func newNEP17Commands() []cli.Command {
balanceFlags = append(balanceFlags, options.RPC...)
importFlags := []cli.Flag{
walletPathFlag,
cli.StringFlag{
flags.AddressFlag{
Name: "token",
Usage: "Token contract hash in LE",
Usage: "Token contract address or hash in LE",
},
}
importFlags = append(importFlags, options.RPC...)
@ -257,10 +257,11 @@ func importNEP17Token(ctx *cli.Context) error {
}
defer wall.Close()
tokenHash, err := flags.ParseAddress(ctx.String("token"))
if err != nil {
return cli.NewExitError(fmt.Errorf("invalid token contract hash: %w", err), 1)
tokenHashFlag := ctx.Generic("token").(*flags.Address)
if !tokenHashFlag.IsSet {
return cli.NewExitError("token contract hash was not set", 1)
}
tokenHash := tokenHashFlag.Uint160()
for _, t := range wall.Extra.Tokens {
if t.Hash.Equals(tokenHash) {