abigail-command-test/src/utils/wallet.js

A CLI tools for abigail

import AElf from 'aelf-sdk';
import fs from 'fs';
import path from 'path';
import { mkdirpSync } from 'mkdirp';
import inquirer from 'inquirer';
import Registry from '../rc/index.js';
import { passwordPrompts } from './constants.js';
import BaseSubCommand from '../command/baseSubCommand.js';

/**
 * Retrieves a wallet based on the provided command root, address, and password.
 * @param {string} commandRoot - The root directory of the command.
 * @param {string} address - The address of the wallet.
 * @param {string} password - The password for the wallet.
 * @returns {any} - The wallet instance.
 */
function getWallet(commandRoot, address, password) {
  const keyStoreFile = path.resolve(commandRoot, `keys/${address}.json`);
  const keyStore = JSON.parse(Registry.getFileOrNot(keyStoreFile, '{}').toString());
  if (Object.keys(keyStore).length === 0) {
    throw new Error('Make sure you entered the correct account address');
  }
  try {
    const { privateKey, mnemonic } = AElf.wallet.keyStore.unlockKeystore(keyStore, password);
    return { ...AElf.wallet.getWalletByPrivateKey(privateKey), mnemonic };
  } catch (e) {
    throw new Error(e.message || 'Make sure you entered the correct password');
  }
}

/**
 * Saves the wallet keystore to the specified directory.
 * @param {any} wallet - The wallet instance to be saved.
 * @param {string} datadir - The directory to save the wallet keystore.
 * @param {string} [cipher] - Optional cipher to be used for encryption.
 * @returns {Promise<string>} - A promise that resolves to the path of the saved keystore file.
 */
async function saveKeyStore(wallet, datadir, cipher = 'aes-128-ctr') {
  const { password, confirmPassword } = BaseSubCommand.normalizeConfig(await inquirer.prompt(passwordPrompts));
  if (password !== confirmPassword) {
    throw new Error('Passwords are different');
  }
  if (password.length <= 6) {
    throw new Error('password is too short');
  }
  const keyStore = AElf.wallet.keyStore.getKeystore(wallet, password, {
    cipher
  });
  const keyStorePath = path.resolve(datadir, `keys/${wallet.address}.json`);
  if (!fs.existsSync(path.resolve(datadir, 'keys'))) {
    mkdirpSync(path.resolve(datadir, 'keys'));
  }
  fs.writeFileSync(keyStorePath, JSON.stringify(keyStore, null, 2));
  return keyStorePath;
}

export { getWallet, saveKeyStore };