pub mod search_database;
pub mod search_page;
pub use search_database::*;
pub use search_page::*;
use serde::{Deserialize, Serialize};
use notionrs_types::object::response::ListResponse;
#[derive(Debug, Default)]
pub struct SearchClient {
pub(crate) reqwest_client: reqwest::Client,
pub(crate) body: SearchRequestBody,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct SearchRequestBody {
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) query: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) filter: Option<notionrs_types::object::request::search::SearchFilter>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) sort: Option<notionrs_types::object::request::search::SearchSort>,
#[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 SearchClient {
pub async fn send(
self,
) -> Result<
ListResponse<notionrs_types::object::response::SearchResultItem>,
crate::error::Error,
> {
let url = String::from("https://api.notion.com/v1/search");
let request_body = serde_json::to_string(&self.body)?;
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::response::SearchResultItem>,
>(&body)?;
Ok(pages)
}
pub fn query<T: AsRef<str>>(mut self, query: T) -> Self {
self.body.query = Some(query.as_ref().to_string());
self
}
pub fn page_size(mut self, page_size: u32) -> Self {
self.body.page_size = Some(page_size);
self
}
pub fn start_cursor<T: AsRef<str>>(mut self, start_cursor: T) -> Self {
self.body.start_cursor = Some(start_cursor.as_ref().to_string());
self
}
pub fn sort_timestamp_asc(self) -> Self {
self.sort(notionrs_types::object::request::search::SearchSort::asc())
}
pub fn sort_timestamp_desc(self) -> Self {
self.sort(notionrs_types::object::request::search::SearchSort::desc())
}
pub fn filter_database(self) -> Self {
self.filter(notionrs_types::object::request::search::SearchFilter::database())
}
pub fn filter_page(self) -> Self {
self.filter(notionrs_types::object::request::search::SearchFilter::page())
}
fn filter(mut self, filter: notionrs_types::object::request::search::SearchFilter) -> Self {
self.body.filter = Some(filter);
self
}
fn sort(mut self, sort: notionrs_types::object::request::search::SearchSort) -> Self {
self.body.sort = Some(sort);
self
}
}