path: root/test/keygen.go

package main

import (
    "encoding/json"
    "fmt"
    "io/ioutil"
    "math/big"
    "os"
    "strconv"

    "github.com/dexon-foundation/dexon/common"
    "github.com/dexon-foundation/dexon/core"
    "github.com/dexon-foundation/dexon/crypto"
)

const genesisFile = "genesis.json"

var preFundAmount *big.Int = new(big.Int).Mul(big.NewInt(1e18), big.NewInt(1e8))

var preFundAddresss = []string{
    "0x2a9D669e4791845EED01D4c0ffF3B927cC94A884",
    "0x1245A8672FA881Cf858eF01D34c42B55D9b263fF",
    "0xe0F859340353854693F537ea337106a33E9FeAB0",
}

func main() {
    data, err := ioutil.ReadFile(genesisFile)
    if err != nil {
        panic(err)
    }

    genesis := new(core.Genesis)
    if err := json.Unmarshal(data, &genesis); err != nil {
        panic(err)
    }

    // Clear previous allocation.
    genesis.Alloc = make(map[common.Address]core.GenesisAccount)

    count, err := strconv.Atoi(os.Args[1])
    if err != nil {
        panic(err)
    }
    for _, addr := range preFundAddresss {
        address := common.HexToAddress(addr)
        genesis.Alloc[address] = core.GenesisAccount{
            Balance: preFundAmount,
            Staked:  big.NewInt(0),
        }
        fmt.Printf("Created account %s\n", address.String())
    }
    for i := 0; i < count; i++ {
        privKey, err := crypto.GenerateKey()
        if err != nil {
            panic(err)
        }
        crypto.SaveECDSA(fmt.Sprintf("keystore/test%d.key", i), privKey)

        address := crypto.PubkeyToAddress(privKey.PublicKey)
        genesis.Alloc[address] = core.GenesisAccount{
            Balance:   new(big.Int).Mul(big.NewInt(1e18), big.NewInt(1e6)),
            Staked:    new(big.Int).Mul(big.NewInt(1e18), big.NewInt(5e5)),
            PublicKey: crypto.FromECDSAPub(&privKey.PublicKey),
            NodeInfo: core.NodeInfo{
                Name:     fmt.Sprintf("DEXON Test Node %d", i),
                Email:    fmt.Sprintf("dexon%d@dexon.org", i),
                Location: "Taipei, Taiwan",
                Url:      "https://dexon.org",
            },
        }
        fmt.Printf("Created account %s\n", address.String())
    }

    data, err = json.MarshalIndent(genesis, "", "  ")
    if err != nil {
        panic(err)
    }

    if err := ioutil.WriteFile(genesisFile, data, 0644); err != nil {
        panic(err)
    }
    fmt.Println("Done.")
}