qitmeer58checkflag.go

// Copyright 2017-2018 The qitmeer developers
// Use of this source code is governed by an ISC
// license that can be found in the LICENSE file.
package qx

import (
	"encoding/hex"
	"github.com/Qitmeer/qitmeer/params"
)

type QitmeerBase58checkVersionFlag struct {
	Ver  []byte
	flag string
}

func (n *QitmeerBase58checkVersionFlag) Set(s string) error {
	n.Ver = []byte{}
	switch s {
	case "mainnet":
		n.Ver = append(n.Ver, params.MainNetParams.PubKeyHashAddrID[0:]...)
	case "privnet":
		n.Ver = append(n.Ver, params.PrivNetParams.PubKeyHashAddrID[0:]...)
	case "testnet":
		n.Ver = append(n.Ver, params.TestNetParams.PubKeyHashAddrID[0:]...)
	case "mixnet":
		n.Ver = append(n.Ver, params.MixNetParams.PubKeyHashAddrID[0:]...)
	default:
		v, err := hex.DecodeString(s)
		if err != nil {
			return err
		}
		n.Ver = append(n.Ver, v...)
	}
	n.flag = s
	return nil
}

func (n *QitmeerBase58checkVersionFlag) String() string {
	return n.flag
}