Merge pull request #4 from bruno-kakele/add-account-by-index

Makes it possible to add an account to a wallet based on its index.

cmd/add.go

@@ -19,12 +19,17 @@ var addCmd = &cobra.Command{
 			wi := wallets[walletIndex]
 			wi.init()
 			var a *wallet.Account
-			for {
-				var err error
-				a, err = wi.w.NewAccount(nil)
+			var err error
+			if (walletAccountIndex > 0) {
+				a, err = wi.w.NewAccount(&walletAccountIndex)
 				fatalIf(err)
-				if _, ok := wi.Accounts[a.Address()]; !ok {
-					break
+			} else {
+				for {
+					a, err = wi.w.NewAccount(nil)
+					fatalIf(err)
+					if _, ok := wi.Accounts[a.Address()]; !ok {
+						break
+					}
 				}
 			}
 			wi.Accounts[a.Address()] = a.Index()

cmd/root.go

@@ -45,6 +45,7 @@ func init() {
 	rootCmd.PersistentFlags().StringVarP(&walletAccount, "account", "a", "", "Account to operate on")
 	rootCmd.PersistentFlags().StringVarP(&rpcURL, "rpc", "r", "https://mynano.ninja/api/node", "RPC endpoint URL")
 	rootCmd.PersistentFlags().StringVarP(&rpcWorkURL, "rpc-work", "s", "http://[::1]:7076", "RPC endpoint URL for work generation")
+	rootCmd.PersistentFlags().Uint32VarP(&walletAccountIndex, "account_index", "i", 0, "Index of the account within the wallet to use. Not all operations support it yet")	
 }
 
 // initConfig reads in config file and ENV variables if set.

cmd/wallet.go

@@ -20,6 +20,7 @@ type walletInfo struct {
 var wallets []*walletInfo
 var walletIndex int
 var walletAccount string
+var walletAccountIndex uint32
 
 func initWallets() {
 	v := viper.GetStringMap("wallets")