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

cli/flags: allow to use hashes in address flag

Browse source
This commit is contained in:
Evgeniy Stratonikov 2021-02-18 12:50:12 +03:00
parent f5767ec710
commit 3f7bea4e99
1 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
cli/flags/address.go
View file

@ -35,7 +35,7 @@ func (a Address) String() string {
// Set implements flag.Value interface. // Set implements flag.Value interface.
func (a *Address) Set(s string) error { func (a *Address) Set(s string) error {
addr, err := address.StringToUint160(s) addr, err := ParseAddress(s)
if err != nil { if err != nil {
return cli.NewExitError(err, 1) return cli.NewExitError(err, 1)
} }
@ -89,3 +89,14 @@ func (f AddressFlag) Apply(set *flag.FlagSet) {
set.Var(&f.Value, name, f.Usage) set.Var(&f.Value, name, f.Usage)
}) })
} }
// ParseAddress parses Uint160 form either LE string or address.
func ParseAddress(s string) (util.Uint160, error) {
const uint160size = 2 * util.Uint160Size
switch len(s) {
case uint160size, uint160size + 2:
return util.Uint160DecodeStringLE(strings.TrimPrefix(s, "0x"))
default:
return address.StringToUint160(s)
}
}