/* eslint-env worker */

const ethUtils = require('ethereumjs-util');
const randomBytes = require('randombytes');

const step = 500;

/**
 * Create a wallet from a random private key
 * @returns {{address: string, privKey: string}}
 */
const getRandomWallet = () => {
	const randbytes = randomBytes(32);
	return {
		address: '0x' + ethUtils.privateToAddress(randbytes).toString('hex'),
		privKey: randbytes.toString('hex')
	};
};

/**
 * Check if a wallet respects the input constraints
 * @param wallet
 * @param input
 * @param isChecksum
 * @returns {boolean}
 */
const isValidVanityWallet = (wallet, input, isChecksum) => {
	if (!isChecksum) {
		return wallet.address.substr(2, input.length) === input;
	}
	const address = ethUtils.stripHexPrefix(wallet.address).toLowerCase();
	const hash = ethUtils.sha3(address).toString('hex');

	for (let i = 0; i < input.length; i++) {
		if (parseInt(hash[i], 16) >= 8) {
			if (address[i].toUpperCase() !== input[i]) {
				return false;
			}
		} else if (address[i] !== input[i]) {
			return false;
		}
	}
	return true;
};

/**
 * Generate a lot of wallets until one satisfies the input constraints
 * @param input
 * @param isChecksum
 * @param cb - Callback called after x attempts, or when an address if found
 * @returns
 */
const getVanityWallet = (input, isChecksum, cb) => {
	input = isChecksum ? input : input.toLowerCase();
	let wallet = getRandomWallet();
	let attempts = 1;

	while (!isValidVanityWallet(wallet, input, isChecksum)) {
		if (attempts >= step) {
			cb({attempts});
			attempts = 0;
		}
		wallet = getRandomWallet();
		attempts++;
	}

	cb({address: ethUtils.toChecksumAddress(wallet.address), attempts});
};

onmessage = function (event) {
	const input = event.data;
	try {
		getVanityWallet(input.prefix, input.checksum, message => postMessage(message));
	} catch (err) {
		postMessage({error: err.toString()});
	}
};

module.exports = {
	onmessage
};