mirror of https://github.com/status-im/go-waku.git
76 lines
1.4 KiB
Go
76 lines
1.4 KiB
Go
package cliutils
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
)
|
|
|
|
type AddressValue struct {
|
|
Value *common.Address
|
|
}
|
|
|
|
func (v *AddressValue) Set(value string) error {
|
|
if !common.IsHexAddress(value) {
|
|
return errors.New("invalid ethereum address")
|
|
}
|
|
|
|
*v.Value = common.HexToAddress(value)
|
|
return nil
|
|
}
|
|
|
|
func (v *AddressValue) String() string {
|
|
if v.Value == nil {
|
|
return ""
|
|
}
|
|
return (*v.Value).Hex()
|
|
}
|
|
|
|
type PrivateKeyValue struct {
|
|
Value **ecdsa.PrivateKey
|
|
}
|
|
|
|
func (v *PrivateKeyValue) Set(value string) error {
|
|
prvKey, err := crypto.ToECDSA(common.FromHex(value))
|
|
if err != nil {
|
|
return errors.New("invalid private key")
|
|
}
|
|
|
|
*v.Value = prvKey
|
|
|
|
return nil
|
|
}
|
|
|
|
func (v *PrivateKeyValue) String() string {
|
|
if v.Value == nil || *v.Value == nil {
|
|
return ""
|
|
}
|
|
return "0x" + common.Bytes2Hex(crypto.FromECDSA(*v.Value))
|
|
}
|
|
|
|
type ChoiceValue struct {
|
|
Choices []string // the choices that this value can take
|
|
Value *string // the actual value
|
|
}
|
|
|
|
func (v *ChoiceValue) Set(value string) error {
|
|
for _, choice := range v.Choices {
|
|
if strings.Compare(choice, value) == 0 {
|
|
*v.Value = value
|
|
return nil
|
|
}
|
|
}
|
|
return fmt.Errorf("%s is not a valid option. need %+v", value, v.Choices)
|
|
}
|
|
|
|
func (v *ChoiceValue) String() string {
|
|
if v.Value == nil {
|
|
return ""
|
|
}
|
|
return *v.Value
|
|
}
|