50 lines
1.8 KiB
Nim
50 lines
1.8 KiB
Nim
# Nimbus - Portal Network
|
|
# Copyright (c) 2022-2023 Status Research & Development GmbH
|
|
# Licensed and distributed under either of
|
|
# * MIT license (license terms in the root directory or at https://opensource.org/licenses/MIT).
|
|
# * Apache v2 license (license terms in the root directory or at https://www.apache.org/licenses/LICENSE-2.0).
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
import
|
|
chronos,
|
|
eth/p2p/discoveryv5/protocol as discv5_protocol,
|
|
beacon_chain/spec/forks,
|
|
../../network/wire/[portal_protocol, portal_stream],
|
|
../../network/beacon_light_client/[
|
|
beacon_light_client_init_loader,
|
|
beacon_light_client_network
|
|
],
|
|
../test_helpers
|
|
|
|
type LightClientNode* = ref object
|
|
discoveryProtocol*: discv5_protocol.Protocol
|
|
lightClientNetwork*: LightClientNetwork
|
|
|
|
proc newLCNode*(
|
|
rng: ref HmacDrbgContext,
|
|
port: int,
|
|
networkData: NetworkInitData): LightClientNode =
|
|
let
|
|
node = initDiscoveryNode(rng, PrivateKey.random(rng[]), localAddress(port))
|
|
db = LightClientDb.new(networkData, "", inMemory = true)
|
|
streamManager = StreamManager.new(node)
|
|
network = LightClientNetwork.new(node, db, streamManager, networkData.forks)
|
|
|
|
return LightClientNode(discoveryProtocol: node, lightClientNetwork: network)
|
|
|
|
func portalProtocol*(n: LightClientNode): PortalProtocol =
|
|
n.lightClientNetwork.portalProtocol
|
|
|
|
func localNode*(n: LightClientNode): Node =
|
|
n.discoveryProtocol.localNode
|
|
|
|
proc start*(n: LightClientNode) =
|
|
n.lightClientNetwork.start()
|
|
|
|
proc stop*(n: LightClientNode) {.async.} =
|
|
n.lightClientNetwork.stop()
|
|
await n.discoveryProtocol.closeWait()
|
|
|
|
proc containsId*(n: LightClientNode, contentId: ContentId): bool =
|
|
n.lightClientNetwork.lightClientDb.get(contentId).isSome()
|