Add import command

Author: kseo

src/command/import.ts

@@ -0,0 +1,30 @@
+import { CCKey, SecretStorage } from "codechain-keystore";
+
+import { H256 } from "codechain-sdk/lib/core/classes";
+import { blake256, getAccountIdFromPublic } from "codechain-sdk/lib/utils";
+import * as _ from "lodash";
+
+import { AccountType } from "../types";
+import { getAddressFromPublic } from "../util";
+
+export async function importKey(
+    cckey: CCKey,
+    accountType: AccountType,
+    secret: SecretStorage,
+    passphrase: string
+): Promise<void> {
+    const publicKey = await cckey[accountType].importKey({
+        secret,
+        passphrase
+    });
+    if (accountType === "platform") {
+        const accountId = getAccountIdFromPublic(publicKey);
+        cckey.mapping.add({ key: accountId, value: publicKey });
+    }
+    if (accountType === "asset") {
+        const hash = H256.ensure(blake256(publicKey)).value;
+        cckey.mapping.add({ key: hash, value: publicKey });
+    }
+
+    console.log(getAddressFromPublic(accountType, publicKey));
+}

src/index.ts

@@ -2,14 +2,22 @@
 
 import { CCKey } from "codechain-keystore";
 import * as program from "commander";
+import * as fs from "fs";
 import * as _ from "lodash";
 import * as process from "process";
 
 import { createKey } from "./command/create";
 import { deleteKey } from "./command/delete";
+import { importKey } from "./command/import";
 import { listKeys } from "./command/list";
 import { CLIError, CLIErrorType } from "./error";
-import { AccountType, CreateOption, DeleteOption, ListOption } from "./types";
+import {
+    AccountType,
+    CreateOption,
+    DeleteOption,
+    ImportOption,
+    ListOption
+} from "./types";
 
 const VERSION = "0.1.1";
 
@@ -47,12 +55,23 @@ program
     .option("-a, --address <address>", "address")
     .action(handleError(deleteCommand));
 
+program
+    .command("import <path>")
+    .description("import a key")
+    .option(
+        "-t, --account-type <accountType>",
+        "'platform' or 'asset'. The type of the key",
+        "platform"
+    )
+    .option("-p, --passphrase <passphrase>", "passphrase")
+    .action(handleError(importCommand));
+
 function handleError(
-    f: (option: any) => Promise<void>
-): (option: any) => Promise<void> {
-    return async (option: any) => {
+    f: (...args: any[]) => Promise<void>
+): (...args: any[]) => Promise<void> {
+    return async (...args: any[]) => {
         try {
-            await f(option);
+            await f(...args);
         } catch (err) {
             console.error(err.toString());
             process.exit(1);
@@ -80,6 +99,14 @@ async function deleteCommand(option: DeleteOption) {
     await deleteKey(cckey, accountType, address);
 }
 
+async function importCommand(path: string, option: ImportOption) {
+    const cckey = await CCKey.create();
+    const accountType = parseAccountType(option.accountType);
+    const passphrase = parsePassphrase(option.passphrase);
+    const contents = fs.readFileSync(path, { encoding: "utf8" });
+    await importKey(cckey, accountType, JSON.parse(contents), passphrase);
+}
+
 program.on("--help", () => {
     console.log(`  Examples:
 

src/types.ts

@@ -14,3 +14,8 @@ export interface DeleteOption {
     accountType: string;
     address: string;
 }
+
+export interface ImportOption {
+    accountType: string;
+    passphrase: string;
+}