Merge pull request #245 from Psychedelic/feat/import-from-secretKey

feat: implement importAccounFromSecretKey

Author: tomiir

package.json

@@ -58,7 +58,7 @@
     "typescript": "^4.5"
   },
   "name": "@psychedelic/plug-controller",
-  "version": "0.24.9",
+  "version": "0.24.10",
   "description": "Internet Computer Plug wallet's controller",
   "main": "dist/index.js",
   "scripts": {

src/PlugKeyRing/index.ts

@@ -36,10 +36,12 @@ import {
   ImportMnemonicOptions,
   ImportFromPemOptions,
   GetPrincipalFromPem,
-  ValidatePemResponse
+  ValidatePemResponse,
+  ImportFromSecretKey
 } from './interfaces';
 import { WALLET_METHODS, MAIN_WALLET_METHODS } from './constants';
 import { getIdentityFromPem } from './../utils/identity/parsePem'
+import Secp256k1KeyIdentity from '../utils/identity/secpk256k1/identity';
 
 class PlugKeyRing {
   // state
@@ -276,6 +278,41 @@ class PlugKeyRing {
     return wallet;
   };
 
+  public importAccountFromPrivateKey = async ({
+    icon,
+    name,
+    secretKey,
+  }: ImportFromSecretKey
+  ): Promise<PlugWallet> => {
+    await this.checkInitialized();
+    this.checkUnlocked();
+    const walletId = uuid(); 
+    const orderNumber = Object.keys(this.state.wallets).length;
+    const buffSecretKey = Buffer.from(secretKey, 'hex');
+    const identity = Secp256k1KeyIdentity.fromSecretKey(buffSecretKey);
+    const wallet = new PlugWallet({
+      icon,
+      name,
+      walletId,
+      orderNumber,
+      fetch: this.fetch,
+      network: this.networkModule.network,
+      type: Types.secretKey256k1,
+      identity,
+    });
+
+    if (this.checkRepeatedAccount(wallet.principal)) {
+      throw new Error(ERRORS.INVALID_ACCOUNT);
+    }
+    
+    const wallets = { ...this.state.wallets, [walletId]: wallet };
+    this.state.wallets = wallets;
+    await this.saveEncryptedState({ wallets }, this.state.password);
+    return wallet;
+  };
+
+
+
   public getPrincipalFromPem = async ({
     pem,
   }: GetPrincipalFromPem

src/PlugKeyRing/interfaces.ts

@@ -26,6 +26,10 @@ export interface ImportFromPemOptions extends CreatePrincipalOptions {
   pem: string;
 }
 
+export interface ImportFromSecretKey extends CreatePrincipalOptions {
+  secretKey: string;
+}
+
 export interface GetPrincipalFromPem {
   pem: string;
 }

src/constants/version.ts

@@ -1 +1 @@
-export const PLUG_CONTROLLER_VERSION = "0.24.9";
+export const PLUG_CONTROLLER_VERSION = "0.24.10";

src/utils/account/constants.ts

@@ -13,4 +13,5 @@ export enum Types {
     mnemonic = "MNEMONIC",
     pem256k1 = "PEM_256k1",
     pem25519 = "PEM_25519",
+    secretKey256k1 = "SECRET_KEY_256k1",
 }

src/utils/identity/identityFactory.ts

@@ -7,6 +7,7 @@ import { ERRORS } from '../../errors';
 export class IdentityFactory {
     public static createIdentity(type: string, secretKey: string): GenericSignIdentity {
         switch (type) {
+            case Types.secretKey256k1:
             case Types.pem256k1:
             case Types.mnemonic:
                 return Secp256k1KeyIdentity.fromJSON(secretKey);