account.go

/*
 * Copyright (C) 2019 Zilliqa
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
package core

import (
	"bufio"
	"github.com/Zilliqa/gozilliqa-sdk/bech32"
	"github.com/Zilliqa/gozilliqa-sdk/keytools"
	"github.com/Zilliqa/gozilliqa-sdk/util"
	"os"
	"strings"
)

type Accounts []Account

func Split(accounts Accounts, lim int) [][]Account {
	var chunk []Account
	chunks := make([][]Account, 0, len(accounts)/lim+1)
	for len(accounts) >= lim {
		chunk, accounts = accounts[:lim], accounts[lim:]
		chunks = append(chunks, chunk)
	}

	if len(accounts) > 0 {
		chunks = append(chunks, accounts[:len(accounts)])
	}

	return chunks
}

func LoadFrom(path string) (Accounts, error) {
	var accounts Accounts
	file, err := os.Open(path)
	if err != nil {
		return nil, err
	}
	defer file.Close()

	scanner := bufio.NewScanner(file)
	for scanner.Scan() {
		line := scanner.Text()
		privates := strings.Split(line, " ")
		accs, err := fromPrivateKeys(privates)
		if err != nil {
			return nil, err
		}

		accounts = append(accounts, accs...)
	}

	if err := scanner.Err(); err != nil {
		return nil, err
	}

	return accounts, nil
}

func GeneratePrivateKeys(number int64) ([]keytools.PrivateKey, error) {
	var i int64
	var keys []keytools.PrivateKey
	for i = 0; i < number; i++ {
		key, err := keytools.GeneratePrivateKey()
		if err != nil {
			return nil, err
		}
		keys = append(keys, key)
	}
	return keys, nil
}

func fromPrivateKeys(privates []string) ([]Account, error) {
	var accounts Accounts
	for _, v := range privates {
		private := util.DecodeHex(v)
		publicKey := keytools.GetPublicKeyFromPrivateKey(private, true)
		public := util.EncodeHex(publicKey)
		address := keytools.GetAddressFromPublic(publicKey)
		bech32, err := bech32.ToBech32Address(address)
		if err != nil {
			return nil, err
		}
		account := Account{
			PrivateKey:    v,
			PublicKey:     public,
			Address:       address,
			Bech32Address: bech32,
		}
		accounts = append(accounts, account)
	}
	return accounts, nil
}