feat: Add Display impl for PeerMessages (#140)

Author: artrixdotdev

crates/libtortillas/src/peer/actor.rs

@@ -293,7 +293,7 @@ impl Actor for PeerActor {
 impl Message<PeerMessages> for PeerActor {
    type Reply = ();
 
-   #[instrument(skip(self), fields(addr = %self.stream, id = %self.peer.id.unwrap()))]
+   #[instrument(skip(self, msg), fields(addr = %self.stream, id = %self.peer.id.unwrap(), message = %msg))]
    async fn handle(
       &mut self, msg: PeerMessages, _: &mut KameoContext<Self, Self::Reply>,
    ) -> Self::Reply {

crates/libtortillas/src/protocol/messages.rs

@@ -1,5 +1,6 @@
 use std::{
    collections::HashMap,
+   fmt::Display,
    net::{IpAddr, Ipv4Addr, Ipv6Addr},
    sync::{
       Arc,
@@ -124,6 +125,42 @@ pub enum PeerMessages {
    KeepAlive,
 }
 
+impl Display for PeerMessages {
+   fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+      match self {
+         PeerMessages::Handshake(handshake) => write!(f, "Handshake({})", handshake.peer_id),
+         PeerMessages::Choke => write!(f, "Choke"),
+         PeerMessages::Unchoke => write!(f, "Unchoke"),
+         PeerMessages::Interested => write!(f, "Interested"),
+         PeerMessages::NotInterested => write!(f, "NotInterested"),
+         PeerMessages::Have(index) => write!(f, "Have({})", index),
+         PeerMessages::Bitfield(bitfield) => write!(
+            f,
+            "Bitfield(true: {}, false: {})",
+            bitfield.count_ones(),
+            bitfield.count_zeros()
+         ),
+         PeerMessages::Request(index, begin, length) => {
+            write!(f, "Request({}, {}, {})", index, begin, length)
+         }
+         PeerMessages::Piece(index, begin, data) => {
+            write!(f, "Piece({}, {}, {})", index, begin, data.len())
+         }
+         PeerMessages::Cancel(index, begin, length) => {
+            write!(f, "Cancel({}, {}, {})", index, begin, length)
+         }
+         PeerMessages::Extended(extended_id, handshake_message, metadata) => write!(
+            f,
+            "Extended({}, {:?}, {:?})",
+            extended_id,
+            handshake_message,
+            metadata.as_ref().map(|m| m.len())
+         ),
+         PeerMessages::KeepAlive => write!(f, "KeepAlive"),
+      }
+   }
+}
+
 impl PeerMessages {
    pub fn to_bytes(&self) -> Result<Bytes, PeerActorError> {
       Ok(match self {

crates/libtortillas/src/torrent/messages.rs

@@ -67,13 +67,14 @@ impl fmt::Debug for TorrentMessage {
          TorrentMessage::KillTracker(tracker) => write!(f, "KillTracker({tracker:?})"),
          TorrentMessage::AddPeer(peer) => write!(f, "AddPeer({peer:?})"),
          TorrentMessage::IncomingPiece(index, offset, data) => {
-            write!(f, "IncomingPiece({index}, {offset}, {data:?})")
+            write!(f, "IncomingPiece({index}, {offset}, {})", data.len())
          }
          TorrentMessage::Announce(peers) => write!(f, "Announce({peers:?})"),
          _ => write!(f, "TorrentMessage"), // Add more later,
       }
    }
 }
+
 actor_request_response!(
    #[allow(dead_code)]
    pub(crate) TorrentRequest,