use notionrs_types::prelude::*;
use serde::{Deserialize, Serialize};
#[derive(Debug, Default, Clone, notionrs_macro::Setter)]
pub struct CreateDataSourceClient {
pub(crate) reqwest_client: reqwest::Client,
pub(crate) database_id: Option<String>,
pub(crate) properties:
std::collections::HashMap<String, notionrs_types::object::data_source::DataSourceProperty>,
pub(crate) title: Vec<RichText>,
pub(crate) icon: Option<notionrs_types::object::emoji_and_icon::EmojiAndIcon>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct CreateDataSourceRequestBody {
pub(crate) parent: DatabaseParent,
pub(crate) properties:
std::collections::HashMap<String, notionrs_types::object::data_source::DataSourceProperty>,
#[serde(skip_serializing_if = "Vec::is_empty")]
pub(crate) title: Vec<RichText>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) icon: Option<notionrs_types::object::emoji_and_icon::EmojiAndIcon>,
}
impl CreateDataSourceClient {
pub async fn send(
self,
) -> Result<notionrs_types::object::data_source::DataSourceResponse, crate::error::Error> {
let url = "https://api.notion.com/v1/data_sources";
let request_body = if let Some(database_id) = self.database_id {
let req = CreateDataSourceRequestBody {
parent: DatabaseParent {
r#type: "database".to_owned(),
database_id,
},
properties: self.properties,
title: self.title,
icon: self.icon,
};
let string = serde_json::to_string(&req)?;
Ok(string)
} else {
Err(crate::error::Error::RequestParameter(
"database_id is not set.".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 data_source = serde_json::from_slice::<
notionrs_types::object::data_source::DataSourceResponse,
>(&body)?;
Ok(data_source)
}
}