use std::marker::PhantomData;
use serde::{Deserialize, Serialize, de::DeserializeOwned};
#[derive(Debug, notionrs_macro::Setter)]
pub struct CreatePageClient<
T = std::collections::HashMap<String, notionrs_types::object::page::PageProperty>,
> {
pub(crate) reqwest_client: reqwest::Client,
pub(crate) page_id: Option<String>,
pub(crate) data_source_id: Option<String>,
pub(crate) properties:
std::collections::HashMap<String, notionrs_types::object::page::PageProperty>,
pub(crate) children: Option<Vec<notionrs_types::object::block::Block>>,
pub(crate) markdown: Option<String>,
pub(crate) icon: Option<notionrs_types::object::emoji_and_icon::EmojiAndIcon>,
pub(crate) cover: Option<notionrs_types::object::file::File>,
pub(crate) template: Option<CreatePageTemplate>,
pub(crate) position: Option<CreatePageTemplatePosition>,
#[skip]
pub(crate) _phantom: PhantomData<T>,
}
impl<T> Default for CreatePageClient<T> {
fn default() -> Self {
Self {
reqwest_client: reqwest::Client::default(),
page_id: None,
data_source_id: None,
properties: std::collections::HashMap::new(),
children: None,
markdown: None,
icon: None,
cover: None,
template: None,
position: None,
_phantom: PhantomData,
}
}
}
#[derive(Debug, Serialize, Deserialize)]
pub struct CreatePageRequestBody {
pub(crate) parent: notionrs_types::object::parent::Parent,
pub(crate) properties:
std::collections::HashMap<String, notionrs_types::object::page::PageProperty>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) children: Option<Vec<notionrs_types::object::block::Block>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) markdown: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) icon: Option<notionrs_types::object::emoji_and_icon::EmojiAndIcon>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) cover: Option<notionrs_types::object::file::File>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) template: Option<CreatePageTemplate>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) position: Option<CreatePageTemplatePosition>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct CreatePageTemplate {
pub(crate) r#type: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) template_id: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) timezone: Option<String>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct CreatePageTemplatePosition {
r#type: CreatePageTemplatePositionType,
#[serde(skip_serializing_if = "Option::is_none")]
after_block: Option<CreatePageTemplatePositionAfterBlockPayload>,
}
#[derive(Debug, Serialize, Deserialize, Copy, Clone)]
#[serde(rename_all = "snake_case")]
pub enum CreatePageTemplatePositionType {
AfterBlock,
PageStart,
PageEnd,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct CreatePageTemplatePositionAfterBlockPayload {
id: String,
}
impl<T> CreatePageClient<T> {
pub fn typed<U>(self) -> CreatePageClient<U> {
CreatePageClient {
reqwest_client: self.reqwest_client,
page_id: self.page_id,
data_source_id: self.data_source_id,
properties: self.properties,
children: self.children,
markdown: self.markdown,
icon: self.icon,
cover: self.cover,
template: self.template,
position: self.position,
_phantom: PhantomData,
}
}
pub fn template_id(mut self, template_id: String) -> Self {
self.template = Some(CreatePageTemplate {
r#type: "template_id".to_string(),
template_id: Some(template_id),
timezone: None,
});
self
}
pub fn template_id_with_timezone(mut self, template_id: String, timezone: String) -> Self {
self.template = Some(CreatePageTemplate {
r#type: "template_id".to_string(),
template_id: Some(template_id),
timezone: Some(timezone),
});
self
}
pub fn template_default(mut self) -> Self {
self.template = Some(CreatePageTemplate {
r#type: "default".to_string(),
template_id: None,
timezone: None,
});
self
}
pub fn template_default_with_timezone(mut self, timezone: String) -> Self {
self.template = Some(CreatePageTemplate {
r#type: "default".to_string(),
template_id: None,
timezone: Some(timezone),
});
self
}
pub fn template_position_after_block(mut self, block_id: String) -> Self {
self.position = Some(CreatePageTemplatePosition {
r#type: CreatePageTemplatePositionType::AfterBlock,
after_block: Some(CreatePageTemplatePositionAfterBlockPayload { id: block_id }),
});
self
}
pub fn template_position_page_start(mut self) -> Self {
self.position = Some(CreatePageTemplatePosition {
r#type: CreatePageTemplatePositionType::PageStart,
after_block: None,
});
self
}
pub fn template_position_page_end(mut self) -> Self {
self.position = Some(CreatePageTemplatePosition {
r#type: CreatePageTemplatePositionType::PageEnd,
after_block: None,
});
self
}
pub async fn send(
self,
) -> Result<notionrs_types::object::page::PageResponse<T>, crate::error::Error>
where
T: DeserializeOwned + Clone + Send + Sync + 'static,
{
let mut parent: Option<notionrs_types::object::parent::Parent> = None;
if let Some(page_id) = self.page_id {
parent = Some(notionrs_types::object::parent::Parent::PageParent(
notionrs_types::object::parent::PageParent::from(page_id),
));
}
if let Some(data_source_id) = self.data_source_id {
parent = Some(notionrs_types::object::parent::Parent::DataSourceParent(
notionrs_types::object::parent::DataSourceParent::from(data_source_id),
));
}
let parent = parent.ok_or_else(|| {
crate::error::Error::RequestParameter(
"Either `page_id` or `data_source_id` must be set.".to_string(),
)
})?;
if self.children.is_some() && self.markdown.is_some() {
return Err(crate::error::Error::RequestParameter(
"`children` and `markdown` are mutually exclusive. Please set only one of them."
.to_string(),
));
}
let request_body_struct = CreatePageRequestBody {
parent,
properties: self.properties,
children: self.children,
markdown: self.markdown,
icon: self.icon,
cover: self.cover,
template: self.template,
position: self.position,
};
let request_body = serde_json::to_string(&request_body_struct)?;
let url = "https://api.notion.com/v1/pages".to_string();
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 page =
serde_json::from_slice::<notionrs_types::object::page::PageResponse<T>>(&body)?;
Ok(page)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn serialize_create_page_request_body_with_markdown() {
let request_body = CreatePageRequestBody {
parent: notionrs_types::object::parent::Parent::PageParent(
notionrs_types::object::parent::PageParent::from("page-id-123"),
),
properties: std::collections::HashMap::new(),
children: None,
markdown: Some("# Hello World\n\nThis is a test.".to_string()),
icon: None,
cover: None,
template: None,
position: None,
};
let json = serde_json::to_value(&request_body).expect("Failed to serialize");
assert_eq!(
json["markdown"],
"# Hello World\n\nThis is a test."
);
assert!(json.get("children").is_none());
}
#[test]
fn serialize_create_page_request_body_without_markdown() {
let request_body = CreatePageRequestBody {
parent: notionrs_types::object::parent::Parent::PageParent(
notionrs_types::object::parent::PageParent::from("page-id-123"),
),
properties: std::collections::HashMap::new(),
children: None,
markdown: None,
icon: None,
cover: None,
template: None,
position: None,
};
let json = serde_json::to_value(&request_body).expect("Failed to serialize");
assert!(json.get("markdown").is_none());
assert!(json.get("children").is_none());
}
#[test]
fn create_page_client_markdown_setter() {
let client = CreatePageClient::<std::collections::HashMap<
String,
notionrs_types::object::page::PageProperty,
>>::default()
.page_id("page-id-123")
.markdown("# Hello World");
assert_eq!(
client.markdown,
Some("# Hello World".to_string())
);
}
#[tokio::test]
async fn create_page_client_rejects_children_and_markdown() {
let client = CreatePageClient::<std::collections::HashMap<
String,
notionrs_types::object::page::PageProperty,
>>::default()
.page_id("page-id-123")
.children(vec![])
.markdown("# Hello World");
let result = client.send().await;
assert!(result.is_err());
let err = result.unwrap_err();
match err {
crate::error::Error::RequestParameter(msg) => {
assert!(msg.contains("mutually exclusive"));
}
_ => panic!("Expected RequestParameter error, got: {:?}", err),
}
}
}