forked from TrueCloudLab/neoneo-go
cli: add nep11 remove
command
This commit is contained in:
parent
5fdb8e2a01
commit
3742e38399
3 changed files with 31 additions and 1 deletions
|
@ -1,6 +1,7 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"io"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -61,4 +62,14 @@ func TestNEP11Import(t *testing.T) {
|
||||||
checkNNSInfo(t)
|
checkNNSInfo(t)
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
|
t.Run("Remove", func(t *testing.T) {
|
||||||
|
e.In.WriteString("y\r")
|
||||||
|
e.Run(t, "neo-go", "wallet", "nep11", "remove",
|
||||||
|
"--wallet", walletPath, "--token", nnsContractHash.StringLE())
|
||||||
|
e.Run(t, "neo-go", "wallet", "nep11", "info",
|
||||||
|
"--wallet", walletPath)
|
||||||
|
_, err := e.Out.ReadString('\n')
|
||||||
|
require.Equal(t, err, io.EOF)
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
|
@ -24,6 +24,17 @@ func newNEP11Commands() []cli.Command {
|
||||||
tokenFlag,
|
tokenFlag,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
Name: "remove",
|
||||||
|
Usage: "remove NEP11 token from the wallet",
|
||||||
|
UsageText: "remove --wallet <path> --token <hash-or-name>",
|
||||||
|
Action: removeNEP11Token,
|
||||||
|
Flags: []cli.Flag{
|
||||||
|
walletPathFlag,
|
||||||
|
tokenFlag,
|
||||||
|
forceFlag,
|
||||||
|
},
|
||||||
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -34,3 +45,7 @@ func importNEP11Token(ctx *cli.Context) error {
|
||||||
func printNEP11Info(ctx *cli.Context) error {
|
func printNEP11Info(ctx *cli.Context) error {
|
||||||
return printNEPInfo(ctx, manifest.NEP11StandardName)
|
return printNEPInfo(ctx, manifest.NEP11StandardName)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func removeNEP11Token(ctx *cli.Context) error {
|
||||||
|
return removeNEPToken(ctx, manifest.NEP11StandardName)
|
||||||
|
}
|
||||||
|
|
|
@ -345,13 +345,17 @@ func printNEPInfo(ctx *cli.Context, standard string) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
func removeNEP17Token(ctx *cli.Context) error {
|
func removeNEP17Token(ctx *cli.Context) error {
|
||||||
|
return removeNEPToken(ctx, manifest.NEP17StandardName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func removeNEPToken(ctx *cli.Context, standard string) error {
|
||||||
wall, err := openWallet(ctx.String("wallet"))
|
wall, err := openWallet(ctx.String("wallet"))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return cli.NewExitError(err, 1)
|
return cli.NewExitError(err, 1)
|
||||||
}
|
}
|
||||||
defer wall.Close()
|
defer wall.Close()
|
||||||
|
|
||||||
token, err := getMatchingToken(ctx, wall, ctx.String("token"), manifest.NEP17StandardName)
|
token, err := getMatchingToken(ctx, wall, ctx.String("token"), standard)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return cli.NewExitError(err, 1)
|
return cli.NewExitError(err, 1)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue