add cmd, update config.go

cmd/cmd.go

@@ -0,0 +1,82 @@
+package cmd
+
+import (
+	"crypto/ecdsa"
+	"encoding/hex"
+	"fmt"
+	"io/ioutil"
+
+	"github.com/arnaucube/slowlorisdb/config"
+	"github.com/arnaucube/slowlorisdb/core"
+	"github.com/arnaucube/slowlorisdb/db"
+	"github.com/arnaucube/slowlorisdb/node"
+	log "github.com/sirupsen/logrus"
+	"github.com/urfave/cli"
+)
+
+var Commands = []cli.Command{
+	{
+		Name:    "create",
+		Aliases: []string{},
+		Usage:   "create the node",
+		Action:  cmdCreate,
+	},
+	{
+		Name:    "start",
+		Aliases: []string{},
+		Usage:   "start the node",
+		Action:  cmdStart,
+	},
+}
+
+func cmdCreate(c *cli.Context) error {
+	log.Info("creating new keys of the node")
+	privK, err := core.NewKey()
+	if err != nil {
+		return err
+	}
+	fmt.Println(privK)
+	return nil
+}
+
+func cmdStart(c *cli.Context) error {
+	conf, err := config.MustRead(c)
+	if err != nil {
+		return err
+	}
+
+	dir, err := ioutil.TempDir("", conf.DbPath)
+	if err != nil {
+		return err
+	}
+	db, err := db.New(dir)
+	if err != nil {
+		return err
+	}
+
+	// parse AuthNodes from the config file
+	var authNodes []*ecdsa.PublicKey
+	for _, authNode := range conf.AuthNodes {
+		packedPubK, err := hex.DecodeString(authNode)
+		if err != nil {
+			return err
+		}
+		pubK := core.UnpackPubK(packedPubK)
+		authNodes = append(authNodes, pubK)
+	}
+
+	bc := core.NewPoABlockchain(db, authNodes)
+
+	// TODO parse privK from path in the config file
+	privK, err := core.NewKey()
+	node, err := node.NewNode(privK, bc, true)
+	if err != nil {
+		return err
+	}
+	err = node.Start()
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

config/config.go

@@ -8,13 +8,15 @@ import (
 )
 
 type Config struct {
+	DbPath    string
 	Port      string
 	Dest      string
+	AuthNodes []string // PubKs in hex format of the AuthNodes for the blockchain
 }
 
-var C Config
+func MustRead(c *cli.Context) (*Config, error) {
+	var config Config
 
-func MustRead(c *cli.Context) error {
 	viper.SetConfigType("yaml")
 	viper.SetConfigName("config")
 	viper.AddConfigPath(".")          // adding home directory as first search path
@@ -27,10 +29,10 @@ func MustRead(c *cli.Context) error {
 	}
 
 	if err := viper.ReadInConfig(); err != nil {
-		return err
+		return nil, err
 	}
-	if err := viper.Unmarshal(&C); err != nil {
+	if err := viper.Unmarshal(&config); err != nil {
-		return err
+		return nil, err
 	}
-	return nil
+	return &config, nil
 }