use serde::{Deserialize, Serialize, de::DeserializeOwned};
use notionrs_types::prelude::*;
#[derive(Debug, Default, Clone, notionrs_macro::Setter)]
pub struct QueryDataSourceClient {
pub(crate) reqwest_client: reqwest::Client,
pub(crate) data_source_id: Option<String>,
pub(crate) filter: Option<notionrs_types::object::request::filter::Filter>,
pub(crate) sorts: Vec<notionrs_types::object::request::sort::Sort>,
pub(crate) start_cursor: Option<String>,
pub(crate) page_size: Option<u32>,
}
crate::impl_paginate!(
QueryDataSourceClient,
notionrs_types::object::page::PageResponse
);
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct QueryDataSourceRequestBody {
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) filter: Option<notionrs_types::object::request::filter::Filter>,
#[serde(skip_serializing_if = "Vec::is_empty")]
pub(crate) sorts: Vec<notionrs_types::object::request::sort::Sort>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) start_cursor: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) page_size: Option<u32>,
}
impl QueryDataSourceClient {
pub async fn send<T>(
self,
) -> Result<ListResponse<notionrs_types::object::page::PageResponse<T>>, crate::error::Error>
where
T: Serialize + DeserializeOwned + Clone + Send + Sync + 'static,
{
match self.data_source_id {
Some(id) => {
let url = format!("https://api.notion.com/v1/data_sources/{}/query", id);
let request_body = serde_json::to_string(&QueryDataSourceRequestBody {
filter: self.filter.clone(),
sorts: self.sorts.clone(),
start_cursor: self.start_cursor.clone(),
page_size: self.page_size,
})?;
let request = self
.reqwest_client
.post(url)
.header("Content-Type", "application/json")
.body(request_body);
let response = request
.send()
.await
.map_err(|e| crate::error::Error::Network(e.to_string()))?;
if !response.status().is_success() {
return Err(crate::error::Error::try_from_response_async(response).await);
}
let body = response
.bytes()
.await
.map_err(|e| crate::error::Error::BodyParse(e.to_string()))?;
let pages = serde_json::from_slice::<
ListResponse<notionrs_types::object::page::PageResponse<T>>,
>(&body)?;
Ok(pages)
}
None => Err(crate::error::Error::RequestParameter(
"`database_id` is not set".to_string(),
)),
}
}
}