# Nimbus
# Copyright (c) 2018-2024 Status Research & Development GmbH
# Licensed under either of
#  * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or
#    http://www.apache.org/licenses/LICENSE-2.0)
#  * MIT license ([LICENSE-MIT](LICENSE-MIT) or
#    http://opensource.org/licenses/MIT)
# at your option. This file may not be copied, modified, or distributed except
# according to those terms.

import
  ./types,
  ./state,
  ./code_bytes,
  ./precompiles,
  ../common/evmforks,
  ../utils/utils,
  ../db/ledger

proc isCreate*(message: Message): bool =
  message.kind in {EVMC_CREATE, EVMC_CREATE2}

proc generateContractAddress*(vmState: BaseVMState,
                              kind: CallKind,
                              sender: Address,
                              salt = ZERO_CONTRACTSALT,
                              code = CodeBytesRef(nil)): Address =
  if kind == EVMC_CREATE:
    let creationNonce = vmState.readOnlyStateDB().getNonce(sender)
    generateAddress(sender, creationNonce)
  else:
    generateSafeAddress(sender, salt, code.bytes)

proc getCallCode*(vmState: BaseVMState, codeAddress: Address): CodeBytesRef =
  let isPrecompile = getPrecompile(vmState.fork, codeAddress).isSome()
  if isPrecompile:
    return CodeBytesRef(nil)

  if vmState.fork >= FkPrague:
    vmState.readOnlyStateDB.resolveCode(codeAddress)
  else:
    vmState.readOnlyStateDB.getCode(codeAddress)