mirror of https://github.com/waku-org/nwaku.git
152 lines
4.1 KiB
Nim
152 lines
4.1 KiB
Nim
## Waku on libp2p
|
|
##
|
|
## This file should eventually correspond to waku_protocol as RLPx subprotocol.
|
|
## Instead, it should likely be on top of GossipSub with a similar interface.
|
|
|
|
import strutils
|
|
import chronos, chronicles
|
|
import libp2p/protocols/pubsub/pubsub,
|
|
libp2p/protocols/pubsub/pubsubpeer,
|
|
libp2p/protocols/pubsub/floodsub,
|
|
libp2p/protocols/pubsub/gossipsub,
|
|
libp2p/protocols/pubsub/rpc/[messages],
|
|
libp2p/stream/connection
|
|
|
|
import metrics
|
|
|
|
declarePublicGauge connected_peers, "number of peers in the pool" # XXX
|
|
declarePublicGauge total_messages, "number of messages received"
|
|
|
|
logScope:
|
|
topic = "WakuSub"
|
|
|
|
const WakuSubCodec* = "/wakusub/2.0.0-alpha1"
|
|
|
|
type
|
|
WakuSub* = ref object of GossipSub
|
|
# XXX: just playing
|
|
text*: string
|
|
gossip_enabled*: bool
|
|
|
|
method init(w: WakuSub) =
|
|
debug "init"
|
|
proc handler(conn: Connection, proto: string) {.async.} =
|
|
## main protocol handler that gets triggered on every
|
|
## connection for a protocol string
|
|
## e.g. ``/wakusub/0.0.1``, etc...
|
|
##
|
|
|
|
debug "Incoming WakuSub connection"
|
|
# XXX: Increment connectedPeers counter, unclear if this is the right place tho
|
|
# Where is the disconnect event?
|
|
connected_peers.inc()
|
|
await w.handleConn(conn, proto)
|
|
|
|
# XXX: Handler hijack GossipSub here?
|
|
w.handler = handler
|
|
w.codec = WakuSubCodec
|
|
|
|
method initPubSub*(w: WakuSub) =
|
|
debug "initWakuSub"
|
|
w.text = "Foobar"
|
|
debug "w.text", text = w.text
|
|
|
|
# Using GossipSub
|
|
w.gossip_enabled = true
|
|
|
|
if w.gossip_enabled:
|
|
procCall GossipSub(w).initPubSub()
|
|
else:
|
|
procCall FloodSub(w).initPubSub()
|
|
|
|
w.init()
|
|
|
|
method subscribe*(w: WakuSub,
|
|
topic: string,
|
|
handler: TopicHandler) {.async.} =
|
|
debug "subscribe", topic=topic
|
|
# XXX: Pubsub really
|
|
|
|
# XXX: This is what is called, I think
|
|
if w.gossip_enabled:
|
|
await procCall GossipSub(w).subscribe(topic, handler)
|
|
else:
|
|
await procCall FloodSub(w).subscribe(topic, handler)
|
|
|
|
|
|
# Subscribing a peer to a specified topic
|
|
method subscribeTopic*(w: WakuSub,
|
|
topic: string,
|
|
subscribe: bool,
|
|
peerId: string) {.async, gcsafe.} =
|
|
proc handler(topic: string, data: seq[byte]) {.async, gcsafe.} =
|
|
info "Hit NOOP handler", topic
|
|
|
|
debug "subscribeTopic", topic=topic, subscribe=subscribe, peerId=peerId
|
|
|
|
if w.gossip_enabled:
|
|
await procCall GossipSub(w).subscribeTopic(topic, subscribe, peerId)
|
|
else:
|
|
await procCall FloodSub(w).subscribeTopic(topic, subscribe, peerId)
|
|
|
|
# XXX: This should distingish light and etc node
|
|
# NOTE: Relay subscription
|
|
# TODO: If peer is light node
|
|
info "about to call subscribe"
|
|
await w.subscribe(topic, handler)
|
|
|
|
|
|
|
|
|
|
# TODO: Fix decrement connected peers here or somewhere else
|
|
method handleDisconnect*(w: WakuSub, peer: PubSubPeer) {.async.} =
|
|
debug "handleDisconnect (NYI)"
|
|
#connected_peers.dec()
|
|
|
|
method rpcHandler*(w: WakuSub,
|
|
peer: PubSubPeer,
|
|
rpcMsgs: seq[RPCMsg]) {.async.} =
|
|
debug "rpcHandler"
|
|
|
|
# XXX: Right place?
|
|
total_messages.inc()
|
|
|
|
if w.gossip_enabled:
|
|
await procCall GossipSub(w).rpcHandler(peer, rpcMsgs)
|
|
else:
|
|
await procCall FloodSub(w).rpcHandler(peer, rpcMsgs)
|
|
# XXX: here
|
|
|
|
method publish*(w: WakuSub,
|
|
topic: string,
|
|
data: seq[byte]): Future[int] {.async.} =
|
|
debug "publish", topic=topic
|
|
|
|
if w.gossip_enabled:
|
|
return await procCall GossipSub(w).publish(topic, data)
|
|
else:
|
|
return await procCall FloodSub(w).publish(topic, data)
|
|
|
|
method unsubscribe*(w: WakuSub,
|
|
topics: seq[TopicPair]) {.async.} =
|
|
debug "unsubscribe"
|
|
if w.gossip_enabled:
|
|
await procCall GossipSub(w).unsubscribe(topics)
|
|
else:
|
|
await procCall FloodSub(w).unsubscribe(topics)
|
|
|
|
# GossipSub specific methods
|
|
method start*(w: WakuSub) {.async.} =
|
|
debug "start"
|
|
if w.gossip_enabled:
|
|
await procCall GossipSub(w).start()
|
|
else:
|
|
await procCall FloodSub(w).start()
|
|
|
|
method stop*(w: WakuSub) {.async.} =
|
|
debug "stop"
|
|
if w.gossip_enabled:
|
|
await procCall GossipSub(w).stop()
|
|
else:
|
|
await procCall FloodSub(w).stop()
|