jcc-ethereum-tool/src/jcc-ethereum-keystore.js

jccdex jcc ethereum tool

const fs = require("fs");
const utils = require("jcc-ethereum-utils");
const { Wallet } = require("@ethereumjs/wallet");
var readlineSync = require("readline-sync");
const { hexToBytes } = require("@ethereumjs/util");

var Writable = require("stream").Writable;

var mutableStdout = new Writable({
  write: function(chunk, encoding, callback) {
    if (!this.muted) {
      process.stdout.write(chunk, encoding);
      console.log("-");
    }
    callback();
  }
});
mutableStdout.muted = false;

function processCreateWallet() {
  var w = utils.Ethereum.createWallet();
  console.log(JSON.stringify(w, null, 2));

  process.exit(0);
}

function resolvePassword(explicit) {
  if (explicit) return explicit;
  if (process.env.JCC_KEYSTORE_PASSWORD)
    return process.env.JCC_KEYSTORE_PASSWORD;
  return readlineSync.question("Password:", { hideEchoBack: true });
}

function saveKeystore() {
  var w = Wallet.generate();
  var f = w.getV3Filename();

  var password = resolvePassword(null);

  w.toV3(password).then(v3 => {
    fs.writeFileSync(f, JSON.stringify(v3, null, 2), "utf-8");
    console.log("\n", f, "saved");
    return;
  });
}

function importToKeystore() {
  var secret = readlineSync.question("MOAC wallet secret:", {
    hideEchoBack: true
  });
  if (secret.indexOf("0x") == -1) {
    secret = "0x" + secret;
  }
  if (!utils.Ethereum.isValidSecret(secret)) {
    console.log("secret invalid, abort.");
    return;
  }
  var w = Wallet.fromPrivateKey(hexToBytes(secret));
  var f = w.getV3Filename();
  var password = resolvePassword(null);
  w.toV3(password).then(v3 => {
    fs.writeFileSync(f, JSON.stringify(v3, null, 2), "utf-8");
    console.log("\n", f, "saved");
    return;
  });
}

async function getWalletFromKeystore(_file, _password) {
  if (fs.existsSync(_file)) {
    try {
      var ks = JSON.parse(fs.readFileSync(_file, "utf-8"));
      var password = resolvePassword(_password);
      const w = await Wallet.fromV3(ks, password);
      return {
        address: w.getAddressString(),
        secret: w.getPrivateKeyString()
      };
    } catch (error) {
      console.log("Parse keystore file fail, check and correct it", error);
      process.exit();
    }
  } else {
    console.log("Can not find", _file, "abort!");
    process.exit();
  }
}
module.exports = {
  processCreateWallet,
  saveKeystore,
  importToKeystore,
  getWalletFromKeystore
};