[ETCM-411] Add key pair generation to cli

Author: KonradStaniec

src/main/scala/io/iohk/ethereum/cli/CliCommands.scala

@@ -7,11 +7,13 @@ import io.iohk.ethereum.crypto._
 import io.iohk.ethereum.domain.Address
 import io.iohk.ethereum.utils.ByteStringUtils
 import java.security.SecureRandom
+import io.iohk.ethereum.nodebuilder.SecureRandomBuilder
 import org.bouncycastle.util.encoders.Hex
 
-object CliCommands {
+object CliCommands extends SecureRandomBuilder {
 
   val generatePrivateKeyCommand = "generate-private-key"
+  val generateKeyPairsCommand = "generate-key-pairs"
   val deriveAddressCommand = "derive-address"
   val generateAllocsCommand = "generate-allocs"
   val balanceOption = "balance"
@@ -27,6 +29,22 @@ object CliCommands {
       }
     }
 
+  private val GenerateKeyPairs: Command[String] =
+    Command(name = generateKeyPairsCommand, header = "Generate key pairs private/public") {
+      val keyNumberOpts = Opts.argument[Int]("number of keys to generate").withDefault(1)
+
+      keyNumberOpts.map { numOfKeys =>
+        val keyPairs = for (_ <- 1 to numOfKeys) yield newRandomKeyPairAsStrings(secureRandom)
+
+        /**
+          * The key pairs will be printed in the format:
+          *   priv-key-hex (32 bytes)
+          *   pub-key-hex (64 bytes)
+          */
+        keyPairs.map { case (prv, pub) => s"$prv\n$pub\n" }.mkString("\n")
+      }
+    }
+
   private val DeriveAddressFromPrivateKey: Command[String] =
     Command(name = deriveAddressCommand, header = "Derive address from private key") {
 
@@ -69,6 +87,6 @@ object CliCommands {
   }
 
   val api: Command[String] = Command.apply(name = "cli", header = "Mantis CLI") {
-    Opts.subcommands(GeneratePrivateKeyCommand, DeriveAddressFromPrivateKey, GenerateAllocs)
+    Opts.subcommands(GeneratePrivateKeyCommand, DeriveAddressFromPrivateKey, GenerateAllocs, GenerateKeyPairs)
   }
 }

src/test/scala/io/iohk/ethereum/cli/CliCommandsSpec.scala

@@ -80,6 +80,22 @@ class CliCommandsSpec extends AnyFlatSpec with Matchers with EitherValues {
       .value shouldBe s""""alloc": {$address: { "balance": $requestedBalance }, $address3: { "balance": $requestedBalance }, $address2: { "balance": $requestedBalance }}"""
   }
 
+  behavior of generateKeyPairsCommand
+  it should "generate one key pair when passed no args" in {
+    val result = api.parse(Seq(generateKeyPairsCommand))
+    result shouldBe a[Right[_, _]]
+    val stringSplit = result.right.get.split("\\n\\n")
+    stringSplit.length shouldEqual 1
+  }
+  
+  it should "generate multiple key-pair when passed correct args" in {
+    val numOfKeys = "5"
+    val numOfKeysAsInt = numOfKeys.toInt
+    val result = api.parse(Seq(generateKeyPairsCommand, numOfKeys))
+    result shouldBe a[Right[_, _]]
+    val stringSplit = result.right.get.split("\\n\\n")
+    stringSplit.length shouldEqual numOfKeysAsInt
+  }
 }
 
 object Fixture {