65 lines
2.2 KiB
Nim
65 lines
2.2 KiB
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.
|
|
|
|
import
|
|
eth/p2p,
|
|
../../core/[chain, tx_pool],
|
|
../protocol,
|
|
./eth as handlers_eth,
|
|
./snap as handlers_snap
|
|
|
|
{.used, push raises: [Defect].}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Public functions: convenience mappings for `eth`
|
|
# ------------------------------------------------------------------------------
|
|
|
|
proc setEthHandlerNewBlocksAndHashes*(
|
|
node: var EthereumNode;
|
|
blockHandler: NewBlockHandler;
|
|
hashesHandler: NewBlockHashesHandler;
|
|
arg: pointer;
|
|
) {.gcsafe, raises: [Defect,CatchableError].} =
|
|
let w = EthWireRef(node.protocolState protocol.eth)
|
|
w.setNewBlockHandler(blockHandler, arg)
|
|
w.setNewBlockHashesHandler(hashesHandler, arg)
|
|
|
|
proc addEthHandlerCapability*(
|
|
node: var 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`
|
|
# ------------------------------------------------------------------------------
|
|
|
|
proc addSnapHandlerCapability*(
|
|
node: var 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
|
|
# ------------------------------------------------------------------------------
|