notion-client 1.1.1

use crate::{
    endpoints::{
        databases::query::{request::QueryDatabaseRequest, response::QueryDatabaseResponse},
        parse_response, NOTION_URI,
    },
    NotionClientError,
};

use super::DataSourcesEndpoint;

impl DataSourcesEndpoint {
    pub async fn query_a_data_source(
        &self,
        data_source_id: &str,
        request: QueryDatabaseRequest,
    ) -> Result<QueryDatabaseResponse, NotionClientError> {
        let json = serde_json::to_string(&request)
            .map_err(|e| NotionClientError::FailedToSerialize { source: e })?;

        let result = self
            .client
            .post(format!(
                "{notion_uri}/data_sources/{data_source_id}/query",
                notion_uri = NOTION_URI,
                data_source_id = data_source_id
            ))
            .body(json)
            .send()
            .await
            .map_err(|e| NotionClientError::FailedToRequest { source: e })?;

        let status = result.status();
        let body = result
            .text()
            .await
            .map_err(|e| NotionClientError::FailedToText { source: e })?;

        parse_response(status, body)
    }
}