Merge pull request #201 from Bytom/v0.1

[bytom/vapor.git] / cmd / bytomd / commands / init.go

diff --git a/cmd/bytomd/commands/init.go b/cmd/bytomd/commands/init.go
new file mode 100644 (file)
index 0000000..4003047
--- /dev/null
+++ b/cmd/bytomd/commands/init.go
@@ -0,0 +1,65 @@
+package commands
+
+import (
+       "encoding/hex"
+       "io/ioutil"
+       "os"
+       "path"
+
+       log "github.com/sirupsen/logrus"
+       "github.com/spf13/cobra"
+
+       cfg "github.com/vapor/config"
+       "github.com/vapor/crypto/ed25519/chainkd"
+)
+
+var initFilesCmd = &cobra.Command{
+       Use:   "init",
+       Short: "Initialize blockchain",
+       Run:   initFiles,
+}
+
+func init() {
+       initFilesCmd.Flags().String("chain_id", config.ChainID, "Select [vapor] or [solonet]")
+
+       RootCmd.AddCommand(initFilesCmd)
+}
+
+func initFiles(cmd *cobra.Command, args []string) {
+       configFilePath := path.Join(config.RootDir, "config.toml")
+       if _, err := os.Stat(configFilePath); !os.IsNotExist(err) {
+               log.WithFields(log.Fields{"module": logModule, "config": configFilePath}).Panic("Already exists config file.")
+       }
+
+       switch config.ChainID {
+       case "vapor":
+               cfg.EnsureRoot(config.RootDir, config.ChainID)
+       default:
+               cfg.EnsureRoot(config.RootDir, "solonet")
+       }
+
+       //generate the federation config file
+       fedFilePath := config.FederationFile()
+       if _, err := os.Stat(fedFilePath); os.IsNotExist(err) {
+               if err := cfg.ExportFederationFile(fedFilePath, config); err != nil {
+                       log.WithFields(log.Fields{"module": logModule, "config": fedFilePath, "error": err}).Panic("fail on export federation file")
+               }
+       }
+
+       //generate the node private key
+       keyFilePath := path.Join(config.RootDir, config.PrivateKeyFile)
+       if _, err := os.Stat(keyFilePath); os.IsNotExist(err) {
+               xprv, err := chainkd.NewXPrv(nil)
+               if err != nil {
+                       log.WithFields(log.Fields{"module": logModule, "err": err}).Panic("fail on generate private key")
+               }
+
+               if err := ioutil.WriteFile(keyFilePath, []byte(hex.EncodeToString(xprv[:])), 0600); err != nil {
+                       log.WithFields(log.Fields{"module": logModule, "err": err}).Panic("fail on save private key")
+               }
+
+               log.WithFields(log.Fields{"pubkey": xprv.XPub()}).Info("success generate private")
+       }
+
+       log.WithFields(log.Fields{"module": logModule, "config": configFilePath}).Info("Initialized bytom")
+}