chore(network): Port Types to Alloy

src/network/service/discovery.rs

@@ -1,16 +1,17 @@
+//! Module handles Discv5 discovery
+
 use std::{str::FromStr, time::Duration};
 
+use super::enr::OpStackEnrData;
 use discv5::{
     enr::{CombinedKey, Enr, EnrBuilder, NodeId},
     Discv5, Discv5Config,
 };
-use ethers::utils::rlp;
 use eyre::Result;
 use tokio::{
     sync::mpsc::{self, Receiver},
     time::sleep,
 };
-use unsigned_varint::{decode, encode};
 
 use super::types::{NetworkAddress, Peer};
 
@@ -81,43 +82,6 @@ fn create_disc(chain_id: u64) -> Result<Discv5> {
     Discv5::new(enr, key, config).map_err(|_| eyre::eyre!("could not create disc service"))
 }
 
-/// The unique L2 network identifier
-#[derive(Debug)]
-struct OpStackEnrData {
-    /// Chain ID
-    chain_id: u64,
-    /// The version. Always set to 0.
-    version: u64,
-}
-
-impl TryFrom<&[u8]> for OpStackEnrData {
-    type Error = eyre::Report;
-
-    /// Converts a slice of RLP encoded bytes to [OpStackEnrData]
-    fn try_from(value: &[u8]) -> Result<Self> {
-        let bytes: Vec<u8> = rlp::decode(value)?;
-        let (chain_id, rest) = decode::u64(&bytes)?;
-        let (version, _) = decode::u64(rest)?;
-
-        Ok(Self { chain_id, version })
-    }
-}
-
-impl From<OpStackEnrData> for Vec<u8> {
-    /// Converts [OpStackEnrData] to a vector of bytes.
-    fn from(value: OpStackEnrData) -> Vec<u8> {
-        let mut chain_id_buf = encode::u128_buffer();
-        let chain_id_slice = encode::u128(value.chain_id as u128, &mut chain_id_buf);
-
-        let mut version_buf = encode::u128_buffer();
-        let version_slice = encode::u128(value.version as u128, &mut version_buf);
-
-        let opstack = [chain_id_slice, version_slice].concat();
-
-        rlp::encode(&opstack).to_vec()
-    }
-}
-
 /// Default bootnodes to use. Currently consists of 2 Base bootnodes & 1 Optimism bootnode.
 fn bootnodes() -> Vec<Enr<CombinedKey>> {
     let bootnodes = [

src/network/service/enr.rs

@@ -0,0 +1,47 @@
+//! Contains the [OpStackEnrData] struct.
+
+use alloy_rlp::{Decodable, Encodable, RlpEncodable, RlpDecodable};
+use eyre::Result;
+
+#[derive(Debug, RlpEncodable, RlpDecodable, PartialEq, Clone)]
+pub struct OpStackEnrData {
+    /// Chain ID
+    pub chain_id: u64,
+    /// The version. Always set to 0.
+    pub version: u64,
+}
+
+impl TryFrom<&[u8]> for OpStackEnrData {
+    type Error = eyre::Report;
+
+    /// Converts a slice of RLP encoded bytes to [OpStackEnrData]
+    fn try_from(mut value: &[u8]) -> Result<Self> {
+        Ok(OpStackEnrData::decode(&mut value)?)
+    }
+}
+
+impl From<OpStackEnrData> for Vec<u8> {
+    /// Converts [OpStackEnrData] to a vector of bytes.
+    fn from(value: OpStackEnrData) -> Vec<u8> {
+        let mut bytes = Vec::new();
+        value.encode(&mut bytes);
+        bytes
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_round_trip_opstack() {
+        let data = OpStackEnrData {
+            chain_id: 123,
+            version: 1,
+        };
+        let bytes: Vec<u8> = data.into();
+        let decoded = OpStackEnrData::try_from(bytes.as_slice()).unwrap();
+        assert_eq!(decoded.chain_id, 123);
+        assert_eq!(decoded.version, 1);
+    }
+}

src/network/service/mod.rs

@@ -14,9 +14,8 @@ use openssl::sha::sha256;
 
 use super::{handlers::Handler, service::types::NetworkAddress};
 
-/// A module to handle peer discovery
 mod discovery;
-/// A module to handle commonly used types in the p2p system.
+mod enr;
 mod types;
 
 /// Responsible for management of the `Discv5` & `libp2p` services.

src/network/service/types.rs

@@ -1,3 +1,4 @@
+/// Module contains commonly used types in the p2p system.
 use std::net::{IpAddr, Ipv4Addr, SocketAddr};
 
 use discv5::enr::{CombinedKey, Enr};