type Wallet struct {
	//私鑰
	Private *ecdsa.PrivateKey
	//約定，這里的PubKey不存儲(chǔ)原始的公鑰，而是存儲(chǔ)X和Y拼接的字符串，在校驗(yàn)端重新拆分（參考r,s傳遞）
	PubKey []byte
}

func NewWallet() *Wallet {
	//創(chuàng)建曲線
	curve := elliptic.P256()
	//生成私鑰
	privateKey, err := ecdsa.GenerateKey(curve, rand.Reader)
	if err != nil {
		log.Panic(err)
	}

	//生成公鑰
	pubKeyOrig := privateKey.PublicKey

	//拼接X(jué), Y
	pubKey := append(pubKeyOrig.X.Bytes(), pubKeyOrig.Y.Bytes()...)

	return &Wallet{Private: privateKey, PubKey: pubKey}
}

//定一個(gè) Wallets結(jié)構(gòu)，它保存所有的wallet以及它的地址
type Wallets struct {
	//map[地址]錢(qián)包
	WalletsMap map[string]*Wallet
}

func NewWallets() *Wallets {
  var ws Wallets
  ws.WalletsMap = make(map[string]*Wallet)
  //加載本地錢(qián)包，把新建的錢(qián)包保存到本地
  ws.loadFile()
  return &ws
}

func (ws *Wallets) CreateWallet() string {
	//創(chuàng)建一個(gè)錢(qián)包
	wallet := NewWallet()
	address := wallet.NewAddress()

	//添加到錢(qián)包集
	ws.WalletsMap[address] = wallet

	//保存包本地
	ws.saveToFile()
	//返回創(chuàng)建錢(qián)包的地址
	return address
}

func (ws *Wallets) saveToFile() {

	var buffer bytes.Buffer

	gob.Register(elliptic.P256())

	encoder := gob.NewEncoder(&buffer)
	err := encoder.Encode(ws)
	//一定要注意校驗(yàn)?。。?
	if err != nil {
		log.Panic(err)
	}

	ioutil.WriteFile(walletFile, buffer.Bytes(), 0600)
}

func (ws *Wallets) loadFile() {
	//在讀取之前，要先確認(rèn)文件是否在，如果不存在，直接退出
	_, err := os.Stat(walletFile)
	if os.IsNotExist(err) {
		return
	}

	//讀取內(nèi)容
	content, err := ioutil.ReadFile(walletFile)
	if err != nil {
		log.Panic(err)
	}

	//解碼
	gob.Register(elliptic.P256())

	decoder := gob.NewDecoder(bytes.NewReader(content))

	var wsLocal Wallets

	err = decoder.Decode(&wsLocal)
	if err != nil {
		log.Panic(err)
	}

	ws.WalletsMap = wsLocal.WalletsMap
}

func (ws *Wallets) ListAllAddresses() []string {
	var addresses []string
	//遍歷錢(qián)包，將所有的key取出來(lái)返回
	for address := range ws.WalletsMap {
		addresses = append(addresses, address)
	}

	return addresses
}

//生成地址
func (w *Wallet) NewAddress() string {
	//錢(qián)包公鑰
	pubKey := w.PubKey

	//計(jì)算公鑰哈希和ripe160
	rip160HashValue := HashPubKey(pubKey)
	//主網(wǎng)版本號(hào)為0x00
	version := byte(00)
	//拼接version
	payload := append([]byte{version}, rip160HashValue...)

	//校驗(yàn)碼checksum
	checkCode := CheckSum(payload)

	//拼接版本、哈希值、校驗(yàn)碼、25字節(jié)數(shù)據(jù)
	payload = append(payload, checkCode...)

	//base58編碼
	address := base58.Encode(payload)

	return address
}