import { reduce } from 'lodash';
import { isAddress, keccak256 } from 'web3-utils';
import { pubToAddress } from 'ethereumjs-util';
import { ETH_ZERO_ADDR, ETH_ZERO_ADDR_SHORT } from 'lib/constants';

export const stripHexPrefix = (hex: string) => {
  return hex.startsWith('0x') ? hex.slice(2) : hex;
};

export const addHexPrefix = (hex: string) => {
  return hex.startsWith('0x') ? hex : `0x${hex}`;
};

export const isValidAddress = (a: string) => '0x0' === a || isAddress(a);

export const abbreviateAddress = (address: string) =>
  `${address.slice(0, 6)}…${address.slice(-4)}`;

export const isZeroAddress = (a: string) =>
  a === ETH_ZERO_ADDR || a === ETH_ZERO_ADDR_SHORT;

export const toChecksumAddress = (address: string) => {
  const addr = stripHexPrefix(address).toLowerCase();
  const hash = keccak256(addr).toString('hex');

  return reduce(
    addr,
    (acc, char, idx) =>
      parseInt(hash[idx], 16) >= 8 ? acc + char.toUpperCase() : acc + char,
    '0x'
  );
};

export const eqAddr = (addr0: string, addr1: string) =>
  !addr0 || !addr1
    ? false
    : toChecksumAddress(addr0) === toChecksumAddress(addr1);

export const publicToAddress = (publicKey: Buffer): string => {
  return '0x' + pubToAddress(Buffer.from(publicKey), true).toString('hex');
};