Add --allocate option

Author: vrde

README.md

@@ -18,6 +18,22 @@ ethnode
 
 By default `ethnode` runs `geth`. If you want to run `parity` type `ethnode parity`.
 
+# Examples
+
+## Start ethnode and store the data in a specific directory
+Every time you run `ethnode`, it creates a new temporary directory to store the data. If you want to persist the data across
+```
+ethnode --workdir=mydata
+```
+
+## Start ethnode and allocate 100ETH to one or more target addresses
+Sometimes you want to allocate Ether to some specific addresses (maybe some other accounts you have on MetaMask). This is an alternative approach to import a private key to your MetaMask extension.
+```
+ethnode --allocate=0xad7b5e515e557b2dc4d0625d206394b502412003,0xecdd5b467e38731bfad4bd75faa45c7d58e41b49
+```
+
+## Start
+
 ## FAQ
 ### Why not just running `parity --config dev`?
 Parity has a nice feature to run it as a [private development chain][parity:devchain] (aka *test RPC*).

cli.js

@@ -13,17 +13,22 @@ function getOptions(program) {
   return {
     workdir: program.workdir || fs.mkdtempSync(`${os.tmpdir()}${sep}`),
     download: program.download,
-    logging: program.logging
+    logging: program.logging,
+    allocate: program.allocate
   };
 }
 
 program
   .version(packageJson.version)
   .option("-d, --download", "Download the Ethereum client and exit.")
   .option("-w, --workdir <dir>", "Specify a working dir.")
+  .option("-l, --logging <level>", "Specify logging level (error, warn, info).")
+  .option("-w, --workdir <dir>", "Specify a working dir.")
   .option(
-    "-l, --logging <level>",
-    "Specify logging level (error, warn, info)."
+    "-a, --allocate <addresses>",
+    "Comma separated list of addresses. Allocate 100 Ethers for each address.",
+    val => val.split(","),
+    []
   );
 
 program.version(packageJson.version);

main.js

@@ -47,11 +47,11 @@ function generateGenesis(client, balances) {
   return genesis;
 }
 
-function generateBalances(keypairs, balance) {
+function generateBalances(addresses, balance) {
   balance = balance || "100000000000000000000";
   const balances = {};
-  for (var i = 0; i < keypairs.length; i++) {
-    balances[keypairs[i].address] = {
+  for (var i = 0; i < addresses.length; i++) {
+    balances[addresses[i]] = {
       balance: balance
     };
   }
@@ -69,10 +69,13 @@ function downloadClient(client, workdir) {
   }
 }
 
-function provide(client, workdir) {
+function provide(client, workdir, allocate) {
   const paths = getPaths(client, workdir);
   const keypairs = getKeypairs(KEYS_SOURCE, "password");
-  const genesis = generateGenesis(client, generateBalances(keypairs));
+  const balances = generateBalances(
+    keypairs.map(x => x.address).concat(allocate)
+  );
+  const genesis = generateGenesis(client, balances);
   let keysDest =
     client === "geth" ? paths.keys : path.join(paths.keys, genesis.name);
 
@@ -109,14 +112,14 @@ function provide(client, workdir) {
   }
 }
 
-function run(client, { download, workdir, logging }) {
+function run(client, { download, workdir, logging, allocate }) {
   const paths = getPaths(client, workdir);
   downloadClient(client, workdir);
   if (download) {
     return;
   }
   if (!fs.existsSync(paths.genesis)) {
-    provide(client, workdir);
+    provide(client, workdir, allocate);
   }
 
   const genesis = JSON.parse(fs.readFileSync(paths.genesis));
@@ -128,9 +131,18 @@ function run(client, { download, workdir, logging }) {
   console.log("Run development node using configuration in", workdir);
   console.log("Test accounts");
   console.log("#  Address                                    Private Key");
-  for (var i = 0; i < keypairs.length; i++) {
+  for (let i = 0; i < keypairs.length; i++) {
     console.log(`${i}: ${keypairs[i].address} ${keypairs[i].privateKey}`);
   }
+  if (allocate.length > 0) {
+    console.log();
+    console.log("Extra account allocations");
+    console.log("Address                                    Private Key");
+    for (let i = 0; i < allocate.length; i++) {
+      console.log(`${allocate[i]} <no private key available>`);
+    }
+  }
+  console.log();
 
   let args;
   if (client === "geth") {

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "ethnode",
-  "version": "0.0.12",
+  "version": "0.0.13",
   "description": "Run an Ethereum node for development.",
   "homepage": "https://github.com/vrde/ethnode",
   "main": "main.js",