From 95435e6a54d94f607e30a6cbb4a1804edf68ad41 Mon Sep 17 00:00:00 2001
From: Alexandru Vasile <alexandru.vasile@parity.io>
Date: Mon, 9 Jan 2023 17:46:40 +0000
Subject: [PATCH] blocks: Fetch the block's body

Signed-off-by: Alexandru Vasile <alexandru.vasile@parity.io>
---
 subxt/src/blocks/block_types.rs | 43 +++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/subxt/src/blocks/block_types.rs b/subxt/src/blocks/block_types.rs
index 09e913ace4..1f20b29fb5 100644
--- a/subxt/src/blocks/block_types.rs
+++ b/subxt/src/blocks/block_types.rs
@@ -73,6 +73,12 @@ pub enum ChainHeadError {
     Other(String),
 }
 
+impl From<Error> for ChainHeadError {
+    fn from(error: Error) -> Self {
+        ChainHeadError::Other(error.to_string())
+    }
+}
+
 impl TryFrom<ChainHeadEvent<String>> for Vec<u8> {
     type Error = ChainHeadError;
 
@@ -92,6 +98,43 @@ impl TryFrom<ChainHeadEvent<String>> for Vec<u8> {
     }
 }
 
+impl<T, C> ChainHeadBlock<T, C>
+where
+    T: Config,
+    C: OnlineClientT<T>,
+{
+    /// Wrapper to fetch the block's body from the `chainHead_body` subscription.
+    async fn fetch_body(
+        &self,
+        subscription_id: String,
+        hash: T::Hash,
+    ) -> Result<Vec<Vec<u8>>, ChainHeadError> {
+        let mut sub = self
+            .client
+            .rpc()
+            .subscribe_chainhead_body(subscription_id, hash)
+            .await?;
+
+        if let Some(event) = sub.next().await {
+            let event = event?;
+
+            let bytes = Vec::<u8>::try_from(event)?;
+
+            let extrinsics: Vec<Vec<u8>> =
+                Decode::decode(&mut &bytes[..]).map_err(Into::<Error>::into)?;
+            return Ok(extrinsics)
+        }
+
+        Err(Error::Other("Failed to fetch the block body".into()).into())
+    }
+
+    /// Fetch the body (vector of extrinsics) of this block.
+    pub async fn body(&self) -> Result<Vec<Vec<u8>>, ChainHeadError> {
+        self.fetch_body(self.subscription_id.clone(), self.hash)
+            .await
+    }
+}
+
 /// A representation of a block.
 pub struct Block<T: Config, C> {
     header: T::Header,