Merge pull request #1522 from 0chain/feature/cli-key-file-raw-param

Feature: add raw key file param

Author: dabasov

code/go/0chain.net/blobber/flags.go

@@ -8,6 +8,7 @@ import (
 var (
 	deploymentMode int
 	keysFile       string
+	keysFileRaw    string
 	mountPoint     string
 	metadataDB     string
 	logDir         string
@@ -24,6 +25,7 @@ var (
 func init() {
 	flag.IntVar(&deploymentMode, "deployment_mode", 2, "deployment mode: 0=dev,1=test, 2=mainnet")
 	flag.StringVar(&keysFile, "keys_file", "", "keys_file")
+	flag.StringVar(&keysFileRaw, "keys_file_raw", "", "keys_file_raw")
 	flag.StringVar(&mountPoint, "files_dir", "", "Mounted partition where all files will be stored")
 	flag.StringVar(&metadataDB, "db_dir", "", "db_dir")
 	flag.StringVar(&logDir, "log_dir", "", "log_dir")

code/go/0chain.net/blobber/node.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"bytes"
 	"errors"
 	"fmt"
 	"os"
@@ -18,15 +19,23 @@ var publicKey, privateKey string
 func setupNode() error {
 	fmt.Println("> setup blobber")
 
-	err := readKeysFromAws()
-	if err != nil {
-		err = readKeysFromFile(&keysFile)
+	var err error
+
+	if keysFileRaw != "" {
+		err = readKeysFromString(&keysFileRaw)
+
+		fmt.Println("using blobber keys from local string")
+	} else {
+		err = readKeysFromAws()
 		if err != nil {
-			panic(err)
+			err = readKeysFromFile(&keysFile)
+			if err != nil {
+				panic(err)
+			}
+			fmt.Println("using blobber keys from local")
+		} else {
+			fmt.Println("using blobber keys from aws")
 		}
-		fmt.Println("using blobber keys from local")
-	} else {
-		fmt.Println("using blobber keys from aws")
 	}
 
 	node.Self.SetKeys(publicKey, privateKey)
@@ -72,6 +81,12 @@ func readKeysFromAws() error {
 	return nil
 }
 
+func readKeysFromString(keyFileRaw *string) error {
+	publicKey, privateKey, _, _ = encryption.ReadKeys(
+		bytes.NewBufferString(*keyFileRaw))
+	return nil
+}
+
 func readKeysFromFile(keysFile *string) error {
 	reader, err := os.Open(*keysFile)
 	if err != nil {
@@ -80,4 +95,4 @@ func readKeysFromFile(keysFile *string) error {
 	defer reader.Close()
 	publicKey, privateKey, _, _ = encryption.ReadKeys(reader)
 	return nil
-}
+}