aboutsummaryrefslogtreecommitdiffstats
path: root/cmd/puppeth/puppeth.go
blob: 3af1751eaee7c79019a4a5fba1efa9dcd021a13a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
// Copyright 2017 The go-ethereum Authors
// This file is part of go-ethereum.
//
// go-ethereum 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.
//
// go-ethereum 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 go-ethereum. If not, see <http://www.gnu.org/licenses/>.

// puppeth is a command to assemble and maintain private networks.
package main

import (
    "encoding/json"
    "io/ioutil"
    "math/rand"
    "os"
    "strings"
    "time"

    "github.com/ethereum/go-ethereum/cmd/utils"
    "github.com/ethereum/go-ethereum/core"
    "github.com/ethereum/go-ethereum/log"
    "gopkg.in/urfave/cli.v1"
)

// main is just a boring entry point to set up the CLI app.
func main() {
    app := cli.NewApp()
    app.Name = "puppeth"
    app.Usage = "assemble and maintain private Ethereum networks"
    app.Flags = []cli.Flag{
        cli.StringFlag{
            Name:  "network",
            Usage: "name of the network to administer (no spaces or hyphens, please)",
        },
        cli.IntFlag{
            Name:  "loglevel",
            Value: 3,
            Usage: "log level to emit to the screen",
        },
    }
    app.Commands = []cli.Command{
        cli.Command{
            Action:    utils.MigrateFlags(convert),
            Name:      "convert",
            Usage:     "Convert from geth genesis into chainspecs for other nodes.",
            ArgsUsage: "<geth-genesis.json>",
        },
    }
    app.Action = func(c *cli.Context) error {
        // Set up the logger to print everything and the random generator
        log.Root().SetHandler(log.LvlFilterHandler(log.Lvl(c.Int("loglevel")), log.StreamHandler(os.Stdout, log.TerminalFormat(true))))
        rand.Seed(time.Now().UnixNano())

        network := c.String("network")
        if strings.Contains(network, " ") || strings.Contains(network, "-") || strings.ToLower(network) != network {
            log.Crit("No spaces, hyphens or capital letters allowed in network name")
        }
        // Start the wizard and relinquish control
        makeWizard(c.String("network")).run()
        return nil
    }
    app.Run(os.Args)
}

func convert(ctx *cli.Context) error {
    // Ensure we have a source genesis
    log.Root().SetHandler(log.LvlFilterHandler(log.LvlInfo, log.StreamHandler(os.Stdout, log.TerminalFormat(true))))
    if len(ctx.Args()) != 1 {
        utils.Fatalf("No geth genesis provided")
    }
    blob, err := ioutil.ReadFile(ctx.Args().First())
    if err != nil {
        utils.Fatalf("Could not read file: %v", err)
    }

    var genesis core.Genesis
    if err := json.Unmarshal(blob, &genesis); err != nil {
        utils.Fatalf("Failed parsing genesis: %v", err)
    }
    basename := strings.TrimRight(ctx.Args().First(), ".json")
    convertGenesis(&genesis, basename, basename, []string{})
    return nil
}