nimbus-eth1/nimbus/sync/handlers/setup.nim

# Nimbus
# Copyright (c) 2018-2021 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.

{.used, push raises: [].}

import
  eth/p2p,
  ../../db/select_backend,
  ../../core/[chain, tx_pool],
  ../protocol,
  ./eth as handlers_eth

# ------------------------------------------------------------------------------
# Public functions: convenience mappings for `eth`
# ------------------------------------------------------------------------------

proc setEthHandlerNewBlocksAndHashes*(
    node: EthereumNode;
    blockHandler: NewBlockHandler;
    hashesHandler: NewBlockHashesHandler;
    arg: pointer;
      ) {.gcsafe, raises: [CatchableError].} =
  let w = EthWireRef(node.protocolState protocol.eth)
  w.setNewBlockHandler(blockHandler, arg)
  w.setNewBlockHashesHandler(hashesHandler, arg)

proc addEthHandlerCapability*(
    node: EthereumNode;
    peerPool: PeerPool;
    chain: ChainRef;
    txPool = TxPoolRef(nil);
      ) =
  ## Install `eth` handlers. Passing `txPool` as `nil` installs the handler
  ## in minimal/outbound mode.
  node.addCapability(
    protocol.eth,
    EthWireRef.new(chain, txPool, peerPool))

# ------------------------------------------------------------------------------
# Public functions: convenience mappings for `snap`
# ------------------------------------------------------------------------------

when dbBackend != select_backend.none:
  import
    ./snap as handlers_snap

  proc addSnapHandlerCapability*(
      node: EthereumNode;
      peerPool: PeerPool;
      chain = ChainRef(nil);
        ) =
    ## Install `snap` handlers,Passing `chein` as `nil` installs the handler
    ## in minimal/outbound mode.
    if chain.isNil:
      node.addCapability protocol.snap
    else:
      node.addCapability(protocol.snap, SnapWireRef.init(chain, peerPool))

# ------------------------------------------------------------------------------
# End
# ------------------------------------------------------------------------------