1 files changed, 39 insertions, 4 deletions

diff --git a/src/lib.rs b/src/lib.rs
index c74aadf..7c6520f 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,11 +1,11 @@
 use crate::models::search::{DatabaseQuery, SearchRequest};
-use crate::models::{Database, DatabaseId, ListResponse, Object, Page};
+use crate::models::{Database, DatabaseId, ListResponse, Object, Page, Block, BlockId};
 use reqwest::header::{HeaderMap, HeaderValue};
 use reqwest::{header, Client, ClientBuilder, RequestBuilder};
 use serde::de::DeserializeOwned;
 use snafu::{ResultExt, Snafu};
 
-mod models;
+pub mod models;
 
 const NOTION_API_VERSION: &str = "2021-05-13";
 
@@ -127,6 +127,18 @@ impl NotionApi {
         )
         .await?)
     }
+
+    pub async fn get_block_children<T: Identifiable<Type = BlockId>>(
+        &self,
+        block_id: T
+    ) -> Result<ListResponse<Block>, NotionApiClientError> {
+        Ok(NotionApi::make_json_request(
+            self.client.get(&format!(
+                "https://api.notion.com/v1/blocks/{block_id}/children",
+                block_id = block_id.id()
+            ))
+        ).await?)
+    }
 }
 
 #[cfg(test)]
@@ -135,8 +147,8 @@ mod tests {
     use crate::models::search::{
         DatabaseQuery, FilterCondition, FilterProperty, FilterValue, NotionSearch, TextCondition,
     };
-    use crate::models::Object;
-    use crate::NotionApi;
+    use crate::models::{Object, BlockId};
+    use crate::{NotionApi, Identifiable};
 
     fn test_token() -> String {
         let token = {
@@ -227,6 +239,29 @@ mod tests {
     }
 
     #[tokio::test]
+    async fn get_block_children() -> Result<(), Box<dyn std::error::Error>> {
+        let api = test_client();
+
+        let search_response = api
+            .search(NotionSearch::Filter {
+                value: FilterValue::Page,
+                property: FilterProperty::Object,
+            })
+            .await?;
+
+        println!("{:?}", search_response.results.len());
+
+        for object in search_response.results {
+            let _block = match object {
+                Object::Page { page } => api.get_block_children(BlockId::from(page.id())).await.unwrap(),
+                _ => panic!("Should not have received anything but pages!")
+            };
+        }
+
+        Ok(())
+    }
+
+    #[tokio::test]
     async fn query_database() -> Result<(), Box<dyn std::error::Error>> {
         let api = test_client();