feat(network): add cbor decoder for HardForkQuery (#335)

txpipe · Nov 20, 2023 · 9b57b0f · 9b57b0f
1 parent 155139e
commit 9b57b0f
Showing 1 changed file with 22 additions and 4 deletions.
diff --git a/pallas-network/src/miniprotocols/localstate/queries_v16/codec.rs b/pallas-network/src/miniprotocols/localstate/queries_v16/codec.rs
@@ -172,8 +172,14 @@ impl Encode<()> for HardForkQuery {
 }
 
 impl<'b> Decode<'b, ()> for HardForkQuery {
-    fn decode(_d: &mut Decoder<'b>, _: &mut ()) -> Result<Self, decode::Error> {
-        todo!()
+    fn decode(d: &mut Decoder<'b>, _: &mut ()) -> Result<Self, decode::Error> {
+        d.array()?;
+        let tag = d.u16()?;
+        match tag {
+            0 => Ok(Self::GetInterpreter),
+            1 => Ok(Self::GetCurrentEra),
+            _ => Err(decode::Error::message("invalid tag")),
+        }
     }
 }
 
@@ -197,8 +203,20 @@ impl Encode<()> for LedgerQuery {
 }
 
 impl<'b> Decode<'b, ()> for LedgerQuery {
-    fn decode(_d: &mut Decoder<'b>, _: &mut ()) -> Result<Self, decode::Error> {
-        todo!()
+    fn decode(d: &mut Decoder<'b>, _: &mut ()) -> Result<Self, decode::Error> {
+        d.array()?;
+        let tag = d.u16()?;
+        match tag {
+            0 => {
+                let (era, q) = d.decode()?;
+                Ok(Self::BlockQuery(era, q))
+            }
+            2 => {
+                let q = d.decode()?;
+                Ok(Self::HardForkQuery(q))
+            }
+            _ => Err(decode::Error::message("invalid tag")),
+        }
     }
 }