From fc700cbbdcd21ee7131301b6522fee551e7a11bf Mon Sep 17 00:00:00 2001
From: Alex Vanin <alexey@nspcc.ru>
Date: Mon, 12 Oct 2020 16:57:37 +0300
Subject: [PATCH] [#78] Add private key global argument in CLI

Signed-off-by: Alex Vanin <alexey@nspcc.ru>
---
 cmd/neofs-cli/modules/root.go | 29 +++++++++++++++++++++++++----
 1 file changed, 25 insertions(+), 4 deletions(-)

diff --git a/cmd/neofs-cli/modules/root.go b/cmd/neofs-cli/modules/root.go
index 9d8fdab1..e75202ae 100644
--- a/cmd/neofs-cli/modules/root.go
+++ b/cmd/neofs-cli/modules/root.go
@@ -1,16 +1,22 @@
 package cmd
 
 import (
+	"crypto/ecdsa"
+	"errors"
 	"fmt"
 	"os"
 
+	"github.com/mitchellh/go-homedir"
+	crypto "github.com/nspcc-dev/neofs-crypto"
 	"github.com/spf13/cobra"
-
-	homedir "github.com/mitchellh/go-homedir"
 	"github.com/spf13/viper"
 )
 
-var cfgFile string
+// Global scope flags.
+var (
+	cfgFile    string
+	privateKey string
+)
 
 // rootCmd represents the base command when called without any subcommands
 var rootCmd = &cobra.Command{
@@ -24,6 +30,10 @@ notation, managing container access through protocol gates, querying network map
 and much more!`,
 }
 
+var (
+	errInvalidKey = errors.New("provided key is incorrect")
+)
+
 // Execute adds all child commands to the root command and sets flags appropriately.
 // This is called by main.main(). It only needs to happen once to the rootCmd.
 func Execute() {
@@ -40,7 +50,8 @@ func init() {
 	// Cobra supports persistent flags, which, if defined here,
 	// will be global for your application.
 
-	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.config/neofs-cli/config.yaml)")
+	rootCmd.PersistentFlags().StringVarP(&cfgFile, "config", "c", "", "config file (default is $HOME/.config/neofs-cli/config.yaml)")
+	rootCmd.PersistentFlags().StringVarP(&privateKey, "key", "k", "", "private key in hex, WIF or filepath")
 
 	// Cobra also supports local flags, which will only run
 	// when this action is called directly.
@@ -72,3 +83,13 @@ func initConfig() {
 		fmt.Println("Using config file:", viper.ConfigFileUsed())
 	}
 }
+
+// getKey returns private key that was provided in global arguments.
+func getKey() (*ecdsa.PrivateKey, error) {
+	key, err := crypto.LoadPrivateKey(privateKey)
+	if err != nil {
+		return nil, errInvalidKey
+	}
+
+	return key, nil
+}