animiassumenda/app/components/Wizard/helpers/decreyptKeyStores.ts

Blox live, desktop app for managing staking accounts

import * as fs from 'fs';
import EthereumKeyStore from 'eth2-keystore-js';
/**
 * Extract not extracted keystores using password
 * @param decryptedKeyStores
 * @param keyStoresFiles
 * @param password
 * @param callBack
 */
export const extractKeyStores = async (decryptedKeyStores: any[], keyStoresFiles: File[], password: string, callBack: any, hashExistingPublicKeys: string[], isCreation?: boolean) => {
  const files = await readAllFiles(keyStoresFiles);
  const keyStores = [];
  // eslint-disable-next-line guard-for-in,no-restricted-syntax
  for (const i in files) {
    try {
      // @ts-ignore
      const keyStore = new EthereumKeyStore(files[i].fileTextPlain);
      const keyStorePublicKey = keyStore.getPublicKey();
      let keyStoreAlreadyDecrypted = false;
      for (let keyStoreIndex = 0; keyStoreIndex < decryptedKeyStores.length; keyStoreIndex += 1) {
        if (decryptedKeyStores[keyStoreIndex].publicKey === keyStorePublicKey) {
          keyStoreAlreadyDecrypted = true;
          break;
        }
      }

      if (isCreation && hashExistingPublicKeys[`0x${keyStorePublicKey}`]) {
        throw Error(`Keystore: ${files[i].fileName} (${`0x${keyStorePublicKey.substr(0, 4)}...${keyStorePublicKey.substr(keyStorePublicKey.length - 4, 4)}`}) already exists in your account.`);
      }
      if (!keyStoreAlreadyDecrypted) {
        keyStores.push({
          // @ts-ignore
          fileName: files[i].fileName,
          // eslint-disable-next-line no-await-in-loop
          privateKey: await keyStore.getPrivateKey(password),
          publicKey: keyStorePublicKey,
          deposited: null
        });
      }
      callBack(parseInt(i, 10) + 1);
    } catch (err) {
      if (err && err.message !== 'Invalid keystore file password.') {
        throw (err);
      }
      console.error(err);
      // @ts-ignore
      throw Error(`Invalid keystore file password (${files[i].fileName}) - only files with the same password are supported. You can always upload the rest later on.`);
    }
  }
  return keyStores;
};

const readAllFiles = async (AllFiles: File[]) => {
  return await Promise.all(AllFiles.map(async (file: File) => {
    return await handleFileChosen(file);
  }));
};

const handleFileChosen = async (file: File) => {
  return new Promise((resolve, reject) => {
    fs.readFile(file.path, 'utf8', (err: any | null, data: string) => {
      if (err) {
        reject(err);
      } else {
        resolve({ fileTextPlain: data, fileName: file.name });
      }
    });
  });
};