All files / src/command load.js

100% Statements 25/25
100% Branches 8/8
100% Functions 2/2
100% Lines 25/25

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85            3x                                   10x                                     5x 5x 5x 5x 5x   5x 5x 2x   1x 1x   1x   1x   3x   4x   4x   4x   4x 4x 2x 1x   2x     1x   1x            
import AElf from 'aelf-sdk';
import BaseSubCommand from './baseSubCommand.js';
import { commonGlobalOptionValidatorDesc, loadCommandParameters, loadCommandUsage } from '../utils/constants.js';
import { saveKeyStore } from '../utils/wallet.js';
import { logger } from '../utils/myLogger.js';
 
const loadCommandValidatorDesc = {
  ...commonGlobalOptionValidatorDesc,
  endpoint: {
    ...commonGlobalOptionValidatorDesc.endpoint,
    required: false
  }
};
 
/**
 * @typedef {import('commander').Command} Command
 * @typedef {import('../../types/rc/index.js').default} Registry
 */
class LoadCommand extends BaseSubCommand {
  /**
   * Constructs a new LoadCommand instance.
   * @param {Registry} rc - The registry instance.
   */
  constructor(rc) {
    super(
      'load',
      loadCommandParameters,
      'Load wallet from a private key or mnemonic',
      [],
      loadCommandUsage,
      rc,
      loadCommandValidatorDesc
    );
  }
 
  /**
   * Executes the load command.
   * @param {Command} commander - The commander instance.
   * @param {...any} args - Additional arguments.
   * @returns {Promise<void>} A promise that resolves when the command execution is complete.
   */
  async run(commander, ...args) {
    // @ts-ignore
    const { options, subOptions } = await super.run(commander, ...args);
    const { datadir } = options;
    const { privateKey, saveToFile, createdByOld } = subOptions;
    try {
      let wallet = null;
      // @ts-ignore
      logger.info('Your wallet info is :');
      if (privateKey.trim().split(' ').length > 1) {
        if (createdByOld) {
          // old version sdk
          this.oraInstance.fail('Please install older versions of aelf-command before v1.0.0!');
          return;
        }
        wallet = AElf.wallet.getWalletByMnemonic(privateKey.trim());
        // @ts-ignore
        logger.info(`Mnemonic            : ${wallet.mnemonic}`);
      } else {
        wallet = AElf.wallet.getWalletByPrivateKey(privateKey.trim());
      }
      wallet.publicKey = wallet.keyPair.getPublic().encode('hex');
      // @ts-ignore
      logger.info(`Private Key         : ${wallet.privateKey}`);
      // @ts-ignore
      logger.info(`Public Key          : ${wallet.publicKey}`);
      // @ts-ignore
      logger.info(`Address             : ${wallet.address}`);
      if (saveToFile === true || saveToFile === 'true') {
        const keyStorePath = await saveKeyStore(wallet, datadir);
        this.oraInstance.succeed(`Account info has been saved to \"${keyStorePath}\"`);
      } else {
        this.oraInstance.succeed('Succeed!');
      }
    } catch (e) {
      this.oraInstance.fail('Failed!');
      // @ts-ignore
      logger.error(e);
    }
  }
}
 
export default LoadCommand;