Prompt an address dialog when --address option is omitted

Author: kakao-jun-e

package.json

@@ -40,6 +40,7 @@
     "enquirer": "^1.0.3",
     "lodash": "^4.17.10",
     "prompt-confirm": "^2.0.4",
-    "prompt-password": "^1.2.0"
+    "prompt-password": "^1.2.0",
+    "prompt-list": "^3.2.0"
   }
 }

src/index.ts

@@ -22,6 +22,7 @@ import {
     ImportOption,
     ListOption
 } from "./types";
+import { getAddressFromKey } from "./util";
 
 const VERSION = "0.3.0";
 
@@ -134,8 +135,11 @@ async function createCommand(args: any[], option: CreateOption) {
 async function deleteCommand(args: any[], option: DeleteOption) {
     const cckey = await CCKey.create({ dbPath: option.parent.keysPath });
     const accountType = parseAccountType(option.parent.accountType);
-    const address = parseAddress(accountType, option.address);
     const networkId = option.parent.networkId;
+    if (_.isUndefined(option.address) && process.stdout.isTTY) {
+        option.address = await selectAddress(cckey, networkId, accountType);
+    }
+    const address = parseAddress(accountType, option.address);
     await deleteKey(
         {
             cckey,
@@ -182,9 +186,12 @@ async function importRawCommand([privateKey]: any[], option: ImportOption) {
 async function exportCommand(args: any[], option: ExportOption) {
     const cckey = await CCKey.create({ dbPath: option.parent.keysPath });
     const accountType = parseAccountType(option.parent.accountType);
+    const networkId = option.parent.networkId;
+    if (_.isUndefined(option.address) && process.stdout.isTTY) {
+        option.address = await selectAddress(cckey, networkId, accountType);
+    }
     const address = parseAddress(accountType, option.address);
     const passphrase = await parsePassphrase(option.passphrase);
-    const networkId = option.parent.networkId;
     const secret = await exportKey(
         {
             cckey,
@@ -274,3 +281,24 @@ async function parsePassphrase(passphrase: string): Promise<string> {
     }
     return answers.passphrase;
 }
+
+async function selectAddress(
+    cckey: CCKey,
+    networkId: string,
+    accountType: AccountType
+): Promise<string> {
+    const Enquirer = require("enquirer");
+    const enquirer = new Enquirer();
+    enquirer.register("list", require("prompt-list"));
+
+    let keys = await cckey[accountType].getKeys();
+    keys = _.map(keys, key => getAddressFromKey(accountType, key, networkId));
+    const questions = {
+        type: "list",
+        name: "address",
+        message: "Select your address please",
+        choices: keys
+    };
+    const answers = await enquirer.ask(questions);
+    return answers.address;
+}

yarn.lock

@@ -1579,6 +1579,14 @@ prompt-base@^4.0.1, prompt-base@^4.0.2:
     readline-utils "^2.2.3"
     static-extend "^0.1.2"
 
+prompt-checkbox@^2.2.0:
+  version "2.2.0"
+  resolved "https://registry.yarnpkg.com/prompt-checkbox/-/prompt-checkbox-2.2.0.tgz#a1b086bac7e3422a3f1c34f66d6da52486472690"
+  dependencies:
+    ansi-cyan "^0.1.1"
+    debug "^2.6.8"
+    prompt-base "^4.0.2"
+
 prompt-choices@^3.0.3:
   version "3.0.6"
   resolved "https://registry.yarnpkg.com/prompt-choices/-/prompt-choices-3.0.6.tgz#5c35c3d99ee17edfd9563c1fb81046b0849ef8ff"
@@ -1636,6 +1644,14 @@ prompt-input@^3.0.0:
     debug "^2.6.8"
     prompt-base "^4.0.2"
 
+prompt-list@^3.2.0:
+  version "3.2.0"
+  resolved "https://registry.yarnpkg.com/prompt-list/-/prompt-list-3.2.0.tgz#b33472e01677a5751f59094fa7fe20b09da9db94"
+  dependencies:
+    ansi-cyan "^0.1.1"
+    ansi-dim "^0.1.1"
+    prompt-radio "^1.2.1"
+
 prompt-password@^1.2.0:
   version "1.2.0"
   resolved "https://registry.yarnpkg.com/prompt-password/-/prompt-password-1.2.0.tgz#87dba4afd5c34954e511c2c069fb922dab60b718"
@@ -1667,6 +1683,13 @@ prompt-question@^5.0.1:
     koalas "^1.0.2"
     prompt-choices "^4.0.5"
 
+prompt-radio@^1.2.1:
+  version "1.2.1"
+  resolved "https://registry.yarnpkg.com/prompt-radio/-/prompt-radio-1.2.1.tgz#66a4cb2b189d3373e7f9ddbbc10700186c8959ff"
+  dependencies:
+    debug "^2.6.8"
+    prompt-checkbox "^2.2.0"
+
 punycode@^1.4.1:
   version "1.4.1"
   resolved "https://registry.yarnpkg.com/punycode/-/punycode-1.4.1.tgz#c0d5a63b2718800ad8e1eb0fa5269c84dd41845e"