import Common from '@ethereumjs/common'
import { Address, BN } from 'ethereumjs-util'
import { EIP2929StateManager } from '../../state/interface'
import { RunState } from './../interpreter'

/**
 * Adds address to accessedAddresses set if not already included.
 * Adjusts cost incurred for executing opcode based on whether address read
 * is warm/cold. (EIP 2929)
 * @param {RunState} runState
 * @param {BN}       address
 * @param {Common}   common
 * @param {Boolean}  chargeGas (default: true)
 * @param {Boolean}  isSelfdestruct (default: false)
 */
export function accessAddressEIP2929(
  runState: RunState,
  address: Address,
  common: Common,
  chargeGas = true,
  isSelfdestruct = false
): BN {
  if (!common.isActivatedEIP(2929)) return new BN(0)

  const stateManager = runState.stateManager as EIP2929StateManager
  const addressStr = address.buf

  // Cold
  if (!stateManager.isWarmedAddress(addressStr)) {
    stateManager.addWarmedAddress(addressStr)

    // CREATE, CREATE2 opcodes have the address warmed for free.
    // selfdestruct beneficiary address reads are charged an *additional* cold access
    if (chargeGas) {
      return new BN(common.param('gasPrices', 'coldaccountaccess'))
    }
    // Warm: (selfdestruct beneficiary address reads are not charged when warm)
  } else if (chargeGas && !isSelfdestruct) {
    return new BN(common.param('gasPrices', 'warmstorageread'))
  }
  return new BN(0)
}

/**
 * Adds (address, key) to accessedStorage tuple set if not already included.
 * Adjusts cost incurred for executing opcode based on whether storage read
 * is warm/cold. (EIP 2929)
 * @param {RunState} runState
 * @param {Buffer} key (to storage slot)
 * @param {Common} common
 */
export function accessStorageEIP2929(
  runState: RunState,
  key: Buffer,
  isSstore: boolean,
  common: Common
): BN {
  if (!common.isActivatedEIP(2929)) return new BN(0)

  const stateManager = runState.stateManager as EIP2929StateManager
  const address = runState.eei.getAddress().buf
  const slotIsCold = !stateManager.isWarmedStorage(address, key)

  // Cold (SLOAD and SSTORE)
  if (slotIsCold) {
    stateManager.addWarmedStorage(address, key)
    return new BN(common.param('gasPrices', 'coldsload'))
  } else if (!isSstore) {
    return new BN(common.param('gasPrices', 'warmstorageread'))
  }
  return new BN(0)
}

/**
 * Adjusts cost of SSTORE_RESET_GAS or SLOAD (aka sstorenoop) (EIP-2200) downward when storage
 * location is already warm
 * @param  {RunState} runState
 * @param  {Buffer}   key          storage slot
 * @param  {BN}       defaultCost  SSTORE_RESET_GAS / SLOAD
 * @param  {string}   costName     parameter name ('noop')
 * @param  {Common}   common
 * @return {BN}                    adjusted cost
 */
export function adjustSstoreGasEIP2929(
  runState: RunState,
  key: Buffer,
  defaultCost: BN,
  costName: string,
  common: Common
): BN {
  if (!common.isActivatedEIP(2929)) return defaultCost

  const stateManager = runState.stateManager as EIP2929StateManager
  const address = runState.eei.getAddress().buf
  const warmRead = new BN(common.param('gasPrices', 'warmstorageread'))
  const coldSload = new BN(common.param('gasPrices', 'coldsload'))

  if (stateManager.isWarmedStorage(address, key)) {
    switch (costName) {
      case 'noop':
        return warmRead
      case 'initRefund':
        return new BN(common.param('gasPrices', 'sstoreInitGasEIP2200')).sub(warmRead)
      case 'cleanRefund':
        return new BN(common.param('gasPrices', 'sstoreReset')).sub(coldSload).sub(warmRead)
    }
  }

  return defaultCost
}