122 lines
2.3 KiB
Go
122 lines
2.3 KiB
Go
package keyer
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"fmt"
|
|
"os"
|
|
"text/tabwriter"
|
|
|
|
"github.com/mr-tron/base58"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/hash"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
)
|
|
|
|
type (
|
|
Dashboard struct {
|
|
privKey *keys.PrivateKey
|
|
pubKey *keys.PublicKey
|
|
scriptHash3 util.Uint160
|
|
|
|
multisigKeys keys.PublicKeys
|
|
}
|
|
)
|
|
|
|
func (d Dashboard) PrettyPrint(uncompressed, useHex bool) {
|
|
var (
|
|
data []byte
|
|
|
|
privKey, pubKey, wif, wallet3, sh3, shBE3, multiSigAddr string
|
|
)
|
|
|
|
w := tabwriter.NewWriter(os.Stdout, 0, 0, 1, ' ', 0)
|
|
|
|
if d.privKey != nil {
|
|
privKey = d.privKey.String()
|
|
|
|
wif = d.privKey.WIF()
|
|
if useHex {
|
|
wif = base58ToHex(wif)
|
|
}
|
|
}
|
|
|
|
if d.pubKey != nil {
|
|
if uncompressed {
|
|
data = d.pubKey.UncompressedBytes()
|
|
} else {
|
|
data = d.pubKey.Bytes()
|
|
}
|
|
|
|
pubKey = hex.EncodeToString(data)
|
|
}
|
|
|
|
if !d.scriptHash3.Equals(util.Uint160{}) {
|
|
sh3 = d.scriptHash3.StringLE()
|
|
shBE3 = d.scriptHash3.StringBE()
|
|
wallet3 = address.Uint160ToString(d.scriptHash3)
|
|
|
|
if useHex {
|
|
wallet3 = base58ToHex(wallet3)
|
|
}
|
|
}
|
|
|
|
if len(d.multisigKeys) != 0 {
|
|
u160, err := scriptHashFromMultikey(d.multisigKeys)
|
|
if err != nil {
|
|
panic("can't create multisig redeem script")
|
|
}
|
|
|
|
multiSigAddr = address.Uint160ToString(u160)
|
|
}
|
|
|
|
if privKey != "" {
|
|
fmt.Fprintf(w, "PrivateKey\t%s\n", privKey)
|
|
}
|
|
|
|
if pubKey != "" {
|
|
fmt.Fprintf(w, "PublicKey\t%s\n", pubKey)
|
|
}
|
|
|
|
if wif != "" {
|
|
fmt.Fprintf(w, "WIF\t%s\n", wif)
|
|
}
|
|
|
|
if wallet3 != "" {
|
|
fmt.Fprintf(w, "Wallet3.0\t%s\n", wallet3)
|
|
}
|
|
|
|
if sh3 != "" {
|
|
fmt.Fprintf(w, "ScriptHash3.0\t%s\n", sh3)
|
|
}
|
|
|
|
if shBE3 != "" {
|
|
fmt.Fprintf(w, "ScriptHash3.0BE\t%s\n", shBE3)
|
|
}
|
|
|
|
if multiSigAddr != "" {
|
|
fmt.Fprintf(w, "MultiSigAddress\t%s\n", multiSigAddr)
|
|
}
|
|
|
|
w.Flush()
|
|
}
|
|
|
|
func base58ToHex(data string) string {
|
|
val, err := base58.Decode(data)
|
|
if err != nil {
|
|
panic("produced incorrect base58 value")
|
|
}
|
|
|
|
return hex.EncodeToString(val)
|
|
}
|
|
|
|
func scriptHashFromMultikey(k keys.PublicKeys) (util.Uint160, error) {
|
|
script, err := smartcontract.CreateDefaultMultiSigRedeemScript(k)
|
|
if err != nil {
|
|
return util.Uint160{}, fmt.Errorf("can't create multisig redeem script: %w", err)
|
|
}
|
|
|
|
return hash.Hash160(script), nil
|
|
}
|