import Common from '@ethereumjs/common'
import { BN } from 'ethereumjs-util'
import { RunState } from './../interpreter'

/**
 * Adjusts gas usage and refunds of SStore ops per EIP-1283 (Constantinople)
 *
 * @param {RunState} runState
 * @param {Buffer}   currentStorage
 * @param {Buffer}   originalStorage
 * @param {Buffer}   value
 * @param {Common}   common
 */
export function updateSstoreGasEIP1283(
  runState: RunState,
  currentStorage: Buffer,
  originalStorage: Buffer,
  value: Buffer,
  common: Common
) {
  if (currentStorage.equals(value)) {
    // If current value equals new value (this is a no-op), 200 gas is deducted.
    return new BN(common.param('gasPrices', 'netSstoreNoopGas'))
  }
  // If current value does not equal new value
  if (originalStorage.equals(currentStorage)) {
    // If original value equals current value (this storage slot has not been changed by the current execution context)
    if (originalStorage.length === 0) {
      // If original value is 0, 20000 gas is deducted.
      return new BN(common.param('gasPrices', 'netSstoreInitGas'))
    }
    if (value.length === 0) {
      // If new value is 0, add 15000 gas to refund counter.
      runState.eei.refundGas(
        new BN(common.param('gasPrices', 'netSstoreClearRefund')),
        'EIP-1283 -> netSstoreClearRefund'
      )
    }
    // Otherwise, 5000 gas is deducted.
    return new BN(common.param('gasPrices', 'netSstoreCleanGas'))
  }
  // If original value does not equal current value (this storage slot is dirty), 200 gas is deducted. Apply both of the following clauses.
  if (originalStorage.length !== 0) {
    // If original value is not 0
    if (currentStorage.length === 0) {
      // If current value is 0 (also means that new value is not 0), remove 15000 gas from refund counter. We can prove that refund counter will never go below 0.
      runState.eei.subRefund(
        new BN(common.param('gasPrices', 'netSstoreClearRefund')),
        'EIP-1283 -> netSstoreClearRefund'
      )
    } else if (value.length === 0) {
      // If new value is 0 (also means that current value is not 0), add 15000 gas to refund counter.
      runState.eei.refundGas(
        new BN(common.param('gasPrices', 'netSstoreClearRefund')),
        'EIP-1283 -> netSstoreClearRefund'
      )
    }
  }
  if (originalStorage.equals(value)) {
    // If original value equals new value (this storage slot is reset)
    if (originalStorage.length === 0) {
      // If original value is 0, add 19800 gas to refund counter.
      runState.eei.refundGas(
        new BN(common.param('gasPrices', 'netSstoreResetClearRefund')),
        'EIP-1283 -> netSstoreResetClearRefund'
      )
    } else {
      // Otherwise, add 4800 gas to refund counter.
      runState.eei.refundGas(
        new BN(common.param('gasPrices', 'netSstoreResetRefund')),
        'EIP-1283 -> netSstoreResetRefund'
      )
    }
  }
  return new BN(common.param('gasPrices', 'netSstoreDirtyGas'))
}