From 5f9215a3717c83d1d8a62a967ef1d6292a19318d Mon Sep 17 00:00:00 2001
From: Mark Spanbroek <mark@spanbroek.net>
Date: Thu, 8 Apr 2021 09:45:41 +0200
Subject: [PATCH] Add handler for pricing messages

---
 dagger/bitswap/network.nim           | 14 ++++++++++++++
 tests/dagger/bitswap/testnetwork.nim | 16 ++++++++++++++++
 tests/dagger/examples.nim            |  8 ++++++++
 3 files changed, 38 insertions(+)

diff --git a/dagger/bitswap/network.nim b/dagger/bitswap/network.nim
index ac4c48bb..659da963 100644
--- a/dagger/bitswap/network.nim
+++ b/dagger/bitswap/network.nim
@@ -17,10 +17,12 @@ import pkg/nitro
 
 import ../blocktype as bt
 import ./protobuf/bitswap as pb
+import ./protobuf/payments
 import ./networkpeer
 
 export pb, networkpeer
 export nitro
+export payments
 
 logScope:
   topics = "dagger bitswap network"
@@ -31,11 +33,13 @@ type
   WantListHandler* = proc(peer: PeerID, wantList: WantList) {.gcsafe.}
   BlocksHandler* = proc(peer: PeerID, blocks: seq[bt.Block]) {.gcsafe.}
   BlockPresenceHandler* = proc(peer: PeerID, precense: seq[BlockPresence]) {.gcsafe.}
+  PricingHandler* = proc(peer: PeerID, pricing: Pricing) {.gcsafe.}
 
   BitswapHandlers* = object
     onWantList*: WantListHandler
     onBlocks*: BlocksHandler
     onPresence*: BlockPresenceHandler
+    onPricing*: PricingHandler
 
   WantListBroadcaster* = proc(
     id: PeerID,
@@ -195,6 +199,13 @@ proc broadcastBlockPresence*(
   trace "Sending presence to peer", peer = id
   asyncSpawn b.peers[id].send(Message(blockPresences: presence))
 
+proc handlePricing(network: BitswapNetwork,
+                   peer: NetworkPeer,
+                   pricing: Pricing) =
+  if network.handlers.onPricing.isNil:
+    return
+  network.handlers.onPricing(peer.id, pricing)
+
 proc rpcHandler(b: BitswapNetwork, peer: NetworkPeer, msg: Message) {.async.} =
   try:
     if msg.wantlist.entries.len > 0:
@@ -209,6 +220,9 @@ proc rpcHandler(b: BitswapNetwork, peer: NetworkPeer, msg: Message) {.async.} =
     if msg.blockPresences.len > 0:
       b.handleBlockPresence(peer, msg.blockPresences)
 
+    if pricing =? Pricing.init(msg.pricing):
+      b.handlePricing(peer, pricing)
+
   except CatchableError as exc:
     trace "Exception in bitswap rpc handler", exc = exc.msg
 
diff --git a/tests/dagger/bitswap/testnetwork.nim b/tests/dagger/bitswap/testnetwork.nim
index be894e58..903319bb 100644
--- a/tests/dagger/bitswap/testnetwork.nim
+++ b/tests/dagger/bitswap/testnetwork.nim
@@ -9,11 +9,13 @@ import pkg/protobuf_serialization
 
 import pkg/dagger/stores/memorystore
 import pkg/dagger/bitswap/network
+import pkg/dagger/bitswap/protobuf/payments
 import pkg/dagger/p2p/rng
 import pkg/dagger/chunker
 import pkg/dagger/blocktype as bt
 
 import ../helpers
+import ../examples
 
 suite "Bitswap network":
   let
@@ -102,6 +104,20 @@ suite "Bitswap network":
 
     await done.wait(500.millis)
 
+  test "handles pricing messages":
+    let pricing = Pricing.example
+
+    proc handlePricing(peer: PeerID, received: Pricing) =
+      check received == pricing
+      done.complete()
+
+    network.handlers.onPricing = handlePricing
+
+    let message = Message(pricing: PricingMessage.init(pricing))
+    await buffer.pushData(lenPrefix(Protobuf.encode(message)))
+
+    await done.wait(100.millis)
+
 suite "Bitswap Network - e2e":
   let
     chunker = newRandomChunker(Rng.instance(), size = 1024, chunkSize = 256)
diff --git a/tests/dagger/examples.nim b/tests/dagger/examples.nim
index e5d0cae2..3927ce05 100644
--- a/tests/dagger/examples.nim
+++ b/tests/dagger/examples.nim
@@ -1,5 +1,6 @@
 import std/random
 import pkg/nitro
+import pkg/dagger/bitswap/protobuf/payments
 
 proc example*(_: type EthAddress): EthAddress =
   EthPrivateKey.random().toPublicKey.toAddress
@@ -21,3 +22,10 @@ proc example*(_: type SignedState): SignedState =
   let nonce = UInt48.example
   let channel = wallet.openLedgerChannel(hub, chainId, nonce, asset, amount).get
   wallet.pay(channel, asset, receiver, amount).get
+
+proc example*(_: type Pricing): Pricing =
+  Pricing(
+    address: EthAddress.example,
+    asset: EthAddress.example,
+    price: UInt256.example()
+  )