@subwallet/extension-base 0.6.7-1wr → 0.6.7-2

package/background/handlers/Extension.js

@@ -1,5 +1,6 @@
 // Copyright 2019-2022 @polkadot/extension authors & contributors
 // SPDX-License-Identifier: Apache-2.0
+
 import { ALLOWED_PATH, PASSWORD_EXPIRY_MS } from '@subwallet/extension-base/defaults';
 import { TypeRegistry } from '@polkadot/types';
 import keyring from '@polkadot/ui-keyring';
@@ -10,13 +11,13 @@ import { withErrorLog } from "./helpers.js";
 import { createSubscription, unsubscribe } from "./subscriptions.js";
 export const SEED_DEFAULT_LENGTH = 12;
 export const SEED_LENGTHS = [12, 15, 18, 21, 24];
-export const ETH_DERIVE_DEFAULT = '/m/44\'/60\'/0\'/0/0'; // a global registry to use internally
+export const ETH_DERIVE_DEFAULT = '/m/44\'/60\'/0\'/0/0';
 
+// a global registry to use internally
 const registry = new TypeRegistry();
 export function getSuri(seed, type) {
   return type === 'ethereum' ? `${seed}${ETH_DERIVE_DEFAULT}` : seed;
 }
-
 function transformAccounts(accounts) {
   return Object.values(accounts).map(({
     json: {
@@ -30,20 +31,16 @@ function transformAccounts(accounts) {
     type
   }));
 }
-
 function isJsonPayload(value) {
   return value.genesisHash !== undefined;
 }
-
 export default class Extension {
   #cachedUnlocks;
   #state;
-
   constructor(state) {
     this.#cachedUnlocks = {};
     this.#state = state;
   }
-
   accountsCreateExternal({
     address,
     genesisHash,
@@ -55,7 +52,6 @@ export default class Extension {
     });
     return true;
   }
-
   accountsCreateHardware({
     accountIndex,
     address,
@@ -72,7 +68,6 @@ export default class Extension {
     });
     return true;
   }
-
   accountsCreateSuri({
     genesisHash,
     name,
@@ -86,7 +81,6 @@ export default class Extension {
     }, type);
     return true;
   }
-
   accountsChangePassword({
     address,
     newPass,
@@ -94,33 +88,29 @@ export default class Extension {
   }) {
     const pair = keyring.getPair(address);
     assert(pair, 'Unable to find pair');
-
     try {
       if (!pair.isLocked) {
         pair.lock();
       }
-
       pair.decodePkcs8(oldPass);
     } catch (error) {
       throw new Error('oldPass is invalid');
     }
-
     keyring.encryptAccount(pair, newPass);
     return true;
   }
-
   accountsEdit({
     address,
     name
   }) {
     const pair = keyring.getPair(address);
     assert(pair, 'Unable to find pair');
-    keyring.saveAccountMeta(pair, { ...pair.meta,
+    keyring.saveAccountMeta(pair, {
+      ...pair.meta,
       name
     });
     return true;
   }
-
   accountsExport({
     address,
     password
@@ -128,60 +118,57 @@ export default class Extension {
     return {
       exportedJson: keyring.backupAccount(keyring.getPair(address), password)
     };
-  } // private async accountsBatchExport ({ addresses, password }: RequestAccountBatchExport): Promise<ResponseAccountsExport> {
+  }
+
+  // private async accountsBatchExport ({ addresses, password }: RequestAccountBatchExport): Promise<ResponseAccountsExport> {
   //   return {
   //     exportedJson: await keyring.backupAccounts(addresses, password)
   //   };
   // }
 
-
   accountsForget({
     address
   }) {
     keyring.forgetAccount(address);
     return true;
   }
-
   refreshAccountPasswordCache(pair) {
     const {
       address
     } = pair;
     const savedExpiry = this.#cachedUnlocks[address] || 0;
     const remainingTime = savedExpiry - Date.now();
-
     if (remainingTime < 0) {
       this.#cachedUnlocks[address] = 0;
       pair.lock();
       return 0;
     }
-
     return remainingTime;
   }
-
   accountsShow({
     address,
     isShowing
   }) {
     const pair = keyring.getPair(address);
     assert(pair, 'Unable to find pair');
-    keyring.saveAccountMeta(pair, { ...pair.meta,
+    keyring.saveAccountMeta(pair, {
+      ...pair.meta,
       isHidden: !isShowing
     });
     return true;
   }
-
   accountsTie({
     address,
     genesisHash
   }) {
     const pair = keyring.getPair(address);
     assert(pair, 'Unable to find pair');
-    keyring.saveAccountMeta(pair, { ...pair.meta,
+    keyring.saveAccountMeta(pair, {
+      ...pair.meta,
       genesisHash
     });
     return true;
   }
-
   accountsValidate({
     address,
     password
@@ -192,9 +179,9 @@ export default class Extension {
     } catch (e) {
       return false;
     }
-  } // FIXME This looks very much like what we have in Tabs
-
+  }
 
+  // FIXME This looks very much like what we have in Tabs
   accountsSubscribe(id, port) {
     const cb = createSubscription(id, port);
     const subscription = accountsObservable.subject.subscribe(accounts => cb(transformAccounts(accounts)));
@@ -204,7 +191,6 @@ export default class Extension {
     });
     return true;
   }
-
   authorizeApprove({
     id
   }) {
@@ -216,13 +202,11 @@ export default class Extension {
     resolve(true);
     return true;
   }
-
   getAuthList() {
     return {
       list: this.#state.authUrls
     };
   }
-
   authorizeReject({
     id
   }) {
@@ -233,9 +217,9 @@ export default class Extension {
     } = queued;
     reject(new Error('Rejected'));
     return true;
-  } // FIXME This looks very much like what we have in accounts
-
+  }
 
+  // FIXME This looks very much like what we have in accounts
   authorizeSubscribe(id, port) {
     const cb = createSubscription(id, port);
     const subscription = this.#state.authSubject.subscribe(requests => cb(requests));
@@ -245,7 +229,6 @@ export default class Extension {
     });
     return true;
   }
-
   metadataApprove({
     id
   }) {
@@ -259,15 +242,12 @@ export default class Extension {
     resolve(true);
     return true;
   }
-
   metadataGet(genesisHash) {
     return this.#state.knownMetadata.find(result => result.genesisHash === genesisHash) || null;
   }
-
   metadataList() {
     return this.#state.knownMetadata;
   }
-
   metadataReject({
     id
   }) {
@@ -279,7 +259,6 @@ export default class Extension {
     reject(new Error('Rejected'));
     return true;
   }
-
   metadataSubscribe(id, port) {
     const cb = createSubscription(id, port);
     const subscription = this.#state.metaSubject.subscribe(requests => cb(requests));
@@ -289,7 +268,6 @@ export default class Extension {
     });
     return true;
   }
-
   jsonRestore({
     file,
     password
@@ -300,7 +278,6 @@ export default class Extension {
       throw new Error(error.message);
     }
   }
-
   batchRestore({
     file,
     password
@@ -311,7 +288,6 @@ export default class Extension {
       throw new Error(error.message);
     }
   }
-
   jsonGetAccountInfo(json) {
     try {
       const {
@@ -333,20 +309,17 @@ export default class Extension {
       throw new Error(e.message);
     }
   }
-
   seedCreate({
     length = SEED_DEFAULT_LENGTH,
     seed: _seed,
     type
   }) {
     const seed = _seed || mnemonicGenerate(length);
-
     return {
       address: keyring.createFromUri(getSuri(seed, type), {}, type).address,
       seed
     };
   }
-
   seedValidate({
     suri,
     type
@@ -354,7 +327,6 @@ export default class Extension {
     const {
       phrase
     } = keyExtractSuri(suri);
-
     if (isHex(phrase)) {
       assert(isHex(phrase, 256), 'Hex seed needs to be 256-bits');
     } else {
@@ -362,13 +334,11 @@ export default class Extension {
       assert(SEED_LENGTHS.includes(phrase.split(' ').length), `Mnemonic needs to contain ${SEED_LENGTHS.join(', ')} words`);
       assert(mnemonicValidate(phrase), 'Not a valid mnemonic seed');
     }
-
     return {
       address: keyring.createFromUri(getSuri(suri, type), {}, type).address,
       suri
     };
   }
-
   signingApprovePassword({
     id,
     password,
@@ -381,59 +351,52 @@ export default class Extension {
       request,
       resolve
     } = queued;
-    const pair = keyring.getPair(queued.account.address); // unlike queued.account.address the following
+    const pair = keyring.getPair(queued.account.address);
+
+    // unlike queued.account.address the following
     // address is encoded with the default prefix
     // which what is used for password caching mapping
-
     const {
       address
     } = pair;
-
     if (!pair) {
       reject(new Error('Unable to find pair'));
       return false;
     }
+    this.refreshAccountPasswordCache(pair);
 
-    this.refreshAccountPasswordCache(pair); // if the keyring pair is locked, the password is needed
-
+    // if the keyring pair is locked, the password is needed
     if (pair.isLocked && !password) {
       reject(new Error('Password needed to unlock the account'));
     }
-
     if (pair.isLocked) {
       pair.decodePkcs8(password);
     }
-
     const {
       payload
     } = request;
-
     if (isJsonPayload(payload)) {
       // Get the metadata for the genesisHash
-      const currentMetadata = this.#state.knownMetadata.find(meta => meta.genesisHash === payload.genesisHash); // set the registry before calling the sign function
+      const currentMetadata = this.#state.knownMetadata.find(meta => meta.genesisHash === payload.genesisHash);
 
+      // set the registry before calling the sign function
       registry.setSignedExtensions(payload.signedExtensions, currentMetadata === null || currentMetadata === void 0 ? void 0 : currentMetadata.userExtensions);
-
       if (currentMetadata) {
         registry.register(currentMetadata === null || currentMetadata === void 0 ? void 0 : currentMetadata.types);
       }
     }
-
     const result = request.sign(registry, pair);
-
     if (savePass) {
       this.#cachedUnlocks[address] = Date.now() + PASSWORD_EXPIRY_MS;
     } else {
       pair.lock();
     }
-
     resolve({
       id,
       ...result
     });
     return true;
   }
-
   signingApproveSignature({
     id,
     signature
@@ -449,7 +412,6 @@ export default class Extension {
     });
     return true;
   }
-
   signingCancel({
     id
   }) {
@@ -461,7 +423,6 @@ export default class Extension {
     reject(new Error('Cancelled'));
     return true;
   }
-
   signingIsLocked({
     id
   }) {
@@ -476,7 +437,6 @@ export default class Extension {
       remainingTime
     };
   }
-
   qrIsLocked({
     address
   }) {
@@ -488,7 +448,6 @@ export default class Extension {
       remainingTime
     };
   }
-
   qrSignSubstrate({
     address,
     message,
@@ -497,11 +456,9 @@ export default class Extension {
   }) {
     const pair = keyring.getPair(address);
     assert(pair, 'Unable to find pair');
-
     if (pair.isLocked && !password) {
       throw new Error('Password needed to unlock the account');
     }
-
     if (pair.isLocked) {
       try {
         pair.decodePkcs8(password);
@@ -509,22 +466,19 @@ export default class Extension {
         throw new Error('invalid password');
       }
     }
-
     const signed = u8aToHex(pair.sign(message));
     const _address = pair.address;
-
     if (savePass) {
       this.#cachedUnlocks[_address] = Date.now() + PASSWORD_EXPIRY_MS;
     } else {
       pair.lock();
     }
-
     return {
       signature: signed
     };
-  } // FIXME This looks very much like what we have in authorization
-
+  }
 
+  // FIXME This looks very much like what we have in authorization
   signingSubscribe(id, port) {
     const cb = createSubscription(id, port);
     const subscription = this.#state.signSubject.subscribe(requests => cb(requests));
@@ -534,37 +488,30 @@ export default class Extension {
     });
     return true;
   }
-
   windowOpen(path) {
     const url = `${chrome.extension.getURL('index.html')}#${path}`;
-
     if (!ALLOWED_PATH.includes(path)) {
       console.error('Not allowed to open the url:', url);
       return false;
     }
-
     withErrorLog(() => chrome.tabs.create({
       url
     }));
     return true;
   }
-
   derive(parentAddress, suri, password, metadata) {
     const parentPair = keyring.getPair(parentAddress);
-
     try {
       parentPair.decodePkcs8(password);
     } catch (e) {
       throw new Error('invalid password');
     }
-
     try {
       return parentPair.derive(suri, metadata);
     } catch (err) {
       throw new Error(`"${suri}" is not a valid derivation path`);
     }
   }
-
   derivationValidate({
     parentAddress,
     parentPassword,
@@ -576,7 +523,6 @@ export default class Extension {
       suri
     };
   }
-
   derivationCreate({
     genesisHash,
     name,
@@ -594,136 +540,96 @@ export default class Extension {
     keyring.addPair(childPair, password);
     return true;
   }
-
   toggleAuthorization(url) {
     return {
       list: this.#state.toggleAuthorization(url)
     };
-  } // Weird thought, the eslint override is not needed in Tabs
-  // eslint-disable-next-line @typescript-eslint/require-await
-
+  }
 
+  // Weird thought, the eslint override is not needed in Tabs
+  // eslint-disable-next-line @typescript-eslint/require-await
   async handle(id, type, request, port) {
     switch (type) {
       case 'pri(authorize.approve)':
         return this.authorizeApprove(request);
-
       case 'pri(authorize.list)':
         return this.getAuthList();
-
       case 'pri(authorize.reject)':
         return this.authorizeReject(request);
-
       case 'pri(authorize.toggle)':
         return this.toggleAuthorization(request);
-
       case 'pri(authorize.requests)':
         return this.authorizeSubscribe(id, port);
-
       case 'pri(accounts.create.external)':
         return this.accountsCreateExternal(request);
-
       case 'pri(accounts.create.hardware)':
         return this.accountsCreateHardware(request);
-
       case 'pri(accounts.create.suri)':
         return this.accountsCreateSuri(request);
-
       case 'pri(accounts.changePassword)':
         return this.accountsChangePassword(request);
-
       case 'pri(accounts.edit)':
         return this.accountsEdit(request);
-
       case 'pri(accounts.export)':
         return this.accountsExport(request);
-
       case 'pri(accounts.batchExport)':
         // return this.accountsBatchExport(request as RequestAccountBatchExport);
         // Disable export all util use master password
         return null;
-
       case 'pri(accounts.forget)':
         return this.accountsForget(request);
-
       case 'pri(accounts.show)':
         return this.accountsShow(request);
-
       case 'pri(accounts.subscribe)':
         return this.accountsSubscribe(id, port);
-
       case 'pri(accounts.tie)':
         return this.accountsTie(request);
-
       case 'pri(accounts.validate)':
         return this.accountsValidate(request);
-
       case 'pri(metadata.approve)':
         return this.metadataApprove(request);
-
       case 'pri(metadata.get)':
         return this.metadataGet(request);
-
       case 'pri(metadata.list)':
         return this.metadataList();
-
       case 'pri(metadata.reject)':
         return this.metadataReject(request);
-
       case 'pri(metadata.requests)':
         return this.metadataSubscribe(id, port);
-
       case 'pri(derivation.create)':
         return this.derivationCreate(request);
-
       case 'pri(derivation.validate)':
         return this.derivationValidate(request);
-
       case 'pri(json.restore)':
         return this.jsonRestore(request);
-
       case 'pri(json.batchRestore)':
         return this.batchRestore(request);
-
       case 'pri(json.account.info)':
         return this.jsonGetAccountInfo(request);
-
       case 'pri(seed.create)':
         return this.seedCreate(request);
-
       case 'pri(seed.validate)':
         return this.seedValidate(request);
-
       case 'pri(settings.notification)':
         return this.#state.setNotification(request);
-
       case 'pri(signing.approve.password)':
         return this.signingApprovePassword(request);
-
       case 'pri(signing.approve.signature)':
         return this.signingApproveSignature(request);
-
       case 'pri(signing.cancel)':
         return this.signingCancel(request);
-
       case 'pri(signing.isLocked)':
         return this.signingIsLocked(request);
-
       case 'pri(qr.isLocked)':
         return this.qrIsLocked(request);
-
       case 'pri(qr.sign.substrate)':
         return this.qrSignSubstrate(request);
-
       case 'pri(signing.requests)':
         return this.signingSubscribe(id, port);
-
       case 'pri(window.open)':
         return this.windowOpen(request);
-
       default:
         throw new Error(`Unable to handle message of type ${type}`);
     }
   }
-
 }