mackerel_client 0.6.6

use chrono::{DateTime, Utc};
use http::Method;
use serde_derive::{Deserialize, Serialize};
use serde_with::{DeserializeFromStr, SerializeDisplay};
use std::borrow::Borrow;
use strum::{Display, EnumString};
use typed_builder::TypedBuilder;

use crate::client::*;
use crate::entity::{Entity, Id};
use crate::error::Result;
use crate::monitor::MonitorId;
use crate::role::RoleFullname;
use crate::service::ServiceName;

/// A downtime entity
pub type Downtime = Entity<DowntimeValue>;

/// A downtime id
pub type DowntimeId = Id<DowntimeValue>;

/// A downtime value
#[derive(PartialEq, Clone, Debug, TypedBuilder, Serialize, Deserialize)]
#[builder(field_defaults(setter(into)))]
#[serde(rename_all = "camelCase")]
pub struct DowntimeValue {
    pub name: String,
    #[builder(default)]
    #[serde(default, skip_serializing_if = "String::is_empty")]
    pub memo: String,
    #[serde(with = "chrono::serde::ts_seconds")]
    pub start: DateTime<Utc>,
    #[builder(setter(!into))]
    pub duration: u64,
    #[builder(default, setter(strip_option))]
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub recurrence: Option<DowntimeRecurrence>,
    #[builder(
        default,
        setter(transform = |service_names: impl IntoIterator<Item = impl Into<ServiceName>>| service_names
            .into_iter().map(Into::into).collect::<Vec<_>>()),
    )]
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub service_scopes: Vec<ServiceName>,
    #[builder(
        default,
        setter(transform = |service_names: impl IntoIterator<Item = impl Into<ServiceName>>| service_names
            .into_iter().map(Into::into).collect::<Vec<_>>()),
    )]
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub service_exclude_scopes: Vec<ServiceName>,
    #[builder(
        default,
        setter(transform = |role_fullnames: impl IntoIterator<Item = impl Into<RoleFullname>>| role_fullnames
            .into_iter().map(Into::into).collect::<Vec<_>>()),
    )]
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub role_scopes: Vec<RoleFullname>,
    #[builder(
        default,
        setter(transform = |role_fullnames: impl IntoIterator<Item = impl Into<RoleFullname>>| role_fullnames
            .into_iter().map(Into::into).collect::<Vec<_>>()),
    )]
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub role_exclude_scopes: Vec<RoleFullname>,
    #[builder(
        default,
        setter(transform = |monitor_ids: impl IntoIterator<Item = impl Into<MonitorId>>| monitor_ids
            .into_iter().map(Into::into).collect::<Vec<_>>()),
    )]
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub monitor_scopes: Vec<MonitorId>,
    #[builder(
        default,
        setter(transform = |monitor_ids: impl IntoIterator<Item = impl Into<MonitorId>>| monitor_ids
            .into_iter().map(Into::into).collect::<Vec<_>>()),
    )]
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub monitor_exclude_scopes: Vec<MonitorId>,
}

/// A downtime recurrence setting
#[derive(PartialEq, Clone, Debug, TypedBuilder, Serialize, Deserialize)]
#[builder(field_defaults(setter(into)))]
pub struct DowntimeRecurrence {
    #[serde(rename = "type")]
    pub recurrence_type: DowntimeRecurrenceType,
    #[builder(setter(!into))]
    pub interval: u64,
    #[builder(default)]
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub weekdays: Vec<DowntimeRecurrenceWeekday>,
    #[builder(default, setter(strip_option))]
    #[serde(default, with = "chrono::serde::ts_seconds_option")]
    pub until: Option<DateTime<Utc>>,
}

/// A downtime recurrence type
#[derive(
    PartialEq, Eq, Copy, Clone, Debug, Display, EnumString, SerializeDisplay, DeserializeFromStr,
)]
#[strum(serialize_all = "lowercase")]
pub enum DowntimeRecurrenceType {
    Hourly,
    Daily,
    Weekly,
    Monthly,
    Yearly,
}

/// A downtime recurrence weekday
#[derive(
    PartialEq, Eq, Copy, Clone, Debug, Display, EnumString, SerializeDisplay, DeserializeFromStr,
)]
pub enum DowntimeRecurrenceWeekday {
    Sunday,
    Monday,
    Tuesday,
    Wednesday,
    Thursday,
    Friday,
    Saturday,
}

#[cfg(test)]
mod tests {
    use super::*;
    use rstest::rstest;
    use serde_json::json;

    fn downtime_example1() -> Downtime {
        Downtime::builder()
            .id("downtime1")
            .value(
                DowntimeValue::builder()
                    .name("Example downtime")
                    .memo("This is a downtime memo.")
                    .start(DateTime::from_timestamp(1700000000, 0).unwrap())
                    .duration(60)
                    .build(),
            )
            .build()
    }

    fn json_example1() -> serde_json::Value {
        json!({
            "id": "downtime1",
            "name": "Example downtime",
            "memo": "This is a downtime memo.",
            "start": 1700000000,
            "duration": 60,
        })
    }

    fn downtime_example2() -> Downtime {
        Downtime::builder()
            .id("downtime2")
            .value(
                DowntimeValue::builder()
                    .name("Example downtime")
                    .start(DateTime::from_timestamp(1700000000, 0).unwrap())
                    .duration(60)
                    .recurrence(
                        DowntimeRecurrence::builder()
                            .recurrence_type(DowntimeRecurrenceType::Weekly)
                            .interval(30)
                            .weekdays([
                                DowntimeRecurrenceWeekday::Sunday,
                                DowntimeRecurrenceWeekday::Tuesday,
                                DowntimeRecurrenceWeekday::Wednesday,
                            ])
                            .until(DateTime::from_timestamp(1710000000, 0).unwrap())
                            .build(),
                    )
                    .service_scopes(["service0"])
                    .service_exclude_scopes(["service1"])
                    .role_scopes(["service0:role0"])
                    .role_exclude_scopes(["service1:role1"])
                    .monitor_scopes(["monitor0"])
                    .monitor_exclude_scopes(["monitor1"])
                    .build(),
            )
            .build()
    }

    fn json_example2() -> serde_json::Value {
        json!({
            "id": "downtime2",
            "name": "Example downtime",
            "start": 1700000000,
            "duration": 60,
            "recurrence": {
                "type": "weekly",
                "interval": 30,
                "weekdays": ["Sunday", "Tuesday", "Wednesday"],
                "until": 1710000000,
            },
            "serviceScopes": ["service0"],
            "serviceExcludeScopes": ["service1"],
            "roleScopes": ["service0:role0"],
            "roleExcludeScopes": ["service1:role1"],
            "monitorScopes": ["monitor0"],
            "monitorExcludeScopes": ["monitor1"],
        })
    }

    #[rstest]
    #[case(downtime_example1(), json_example1())]
    #[case(downtime_example2(), json_example2())]
    fn test_downtime_json(#[case] downtime: Downtime, #[case] json: serde_json::Value) {
        assert_eq!(serde_json::to_value(&downtime).unwrap(), json);
        assert_eq!(downtime, serde_json::from_value(json).unwrap());
    }

    #[rstest]
    #[case(DowntimeRecurrenceType::Hourly, "hourly")]
    #[case(DowntimeRecurrenceType::Daily, "daily")]
    #[case(DowntimeRecurrenceType::Weekly, "weekly")]
    #[case(DowntimeRecurrenceType::Monthly, "monthly")]
    #[case(DowntimeRecurrenceType::Yearly, "yearly")]
    fn test_downtime_recurrence_type(
        #[case] downtime_recurrence_type: DowntimeRecurrenceType,
        #[case] downtime_recurrence_type_str: &str,
    ) {
        assert_eq!(
            downtime_recurrence_type.to_string(),
            downtime_recurrence_type_str
        );
        assert_eq!(
            downtime_recurrence_type,
            downtime_recurrence_type_str.parse().unwrap()
        );
        assert_eq!(
            downtime_recurrence_type,
            serde_json::from_value(downtime_recurrence_type_str.into()).unwrap()
        );
        assert_eq!(
            serde_json::to_value(downtime_recurrence_type).unwrap(),
            downtime_recurrence_type_str
        );
    }

    #[rstest]
    #[case(DowntimeRecurrenceWeekday::Sunday, "Sunday")]
    #[case(DowntimeRecurrenceWeekday::Monday, "Monday")]
    #[case(DowntimeRecurrenceWeekday::Tuesday, "Tuesday")]
    #[case(DowntimeRecurrenceWeekday::Wednesday, "Wednesday")]
    #[case(DowntimeRecurrenceWeekday::Thursday, "Thursday")]
    #[case(DowntimeRecurrenceWeekday::Friday, "Friday")]
    #[case(DowntimeRecurrenceWeekday::Saturday, "Saturday")]
    fn test_downtime_recurrence_weekday(
        #[case] downtime_recurrence_weekday: DowntimeRecurrenceWeekday,
        #[case] downtime_recurrence_weekday_str: &str,
    ) {
        assert_eq!(
            downtime_recurrence_weekday.to_string(),
            downtime_recurrence_weekday_str
        );
        assert_eq!(
            downtime_recurrence_weekday,
            downtime_recurrence_weekday_str.parse().unwrap()
        );
        assert_eq!(
            downtime_recurrence_weekday,
            serde_json::from_value(downtime_recurrence_weekday_str.into()).unwrap()
        );
        assert_eq!(
            serde_json::to_value(downtime_recurrence_weekday).unwrap(),
            downtime_recurrence_weekday_str
        );
    }
}

impl Client {
    /// Fetches all the downtimes.
    ///
    /// See <https://mackerel.io/api-docs/entry/downtimes#list>.
    pub async fn list_downtimes(&self) -> Result<Vec<Downtime>> {
        self.request(
            Method::GET,
            "/api/v0/downtimes",
            query_params![],
            request_body![],
            response_body! { downtimes: Vec<Downtime> },
        )
        .await
    }

    /// Creates a new downtime.
    ///
    /// See <https://mackerel.io/api-docs/entry/downtimes#create>.
    pub async fn create_downtime(
        &self,
        downtime_value: impl Borrow<DowntimeValue>,
    ) -> Result<Downtime> {
        self.request(
            Method::POST,
            "/api/v0/downtimes",
            query_params![],
            request_body!(downtime_value.borrow()),
            response_body!(..),
        )
        .await
    }

    /// Updates a downtime.
    ///
    /// See <https://mackerel.io/api-docs/entry/downtimes#update>.
    pub async fn update_downtime(
        &self,
        downtime_id: impl Into<DowntimeId>,
        downtime_value: impl Borrow<DowntimeValue>,
    ) -> Result<Downtime> {
        self.request(
            Method::PUT,
            format_url!("/api/v0/downtimes/{}", downtime_id),
            query_params![],
            request_body!(downtime_value.borrow()),
            response_body!(..),
        )
        .await
    }

    /// Deletes a downtime.
    ///
    /// See <https://mackerel.io/api-docs/entry/downtimes#delete>.
    pub async fn delete_downtime(&self, downtime_id: impl Into<DowntimeId>) -> Result<Downtime> {
        self.request(
            Method::DELETE,
            format_url!("/api/v0/downtimes/{}", downtime_id),
            query_params![],
            request_body![],
            response_body!(..),
        )
        .await
    }
}

#[cfg(test)]
mod client_tests {
    use serde_json::json;

    use crate::downtime::*;
    use crate::tests::*;

    fn value_example() -> DowntimeValue {
        DowntimeValue::builder()
            .name("Example downtime")
            .memo("This is a downtime memo.")
            .start(DateTime::from_timestamp(1698890400, 0).unwrap())
            .duration(86400)
            .service_scopes(["service0"])
            .role_scopes(["service1:role1"])
            .monitor_scopes(["monitor2"])
            .build()
    }

    fn entity_example() -> Downtime {
        Downtime {
            id: DowntimeId::from("downtime0"),
            value: value_example(),
        }
    }

    fn value_json_example() -> serde_json::Value {
        json!({
            "name": "Example downtime",
            "memo": "This is a downtime memo.",
            "start": 1698890400,
            "duration": 86400,
            "serviceScopes": ["service0"],
            "roleScopes": ["service1:role1"],
            "monitorScopes": ["monitor2"],
        })
    }

    fn entity_json_example() -> serde_json::Value {
        let mut json = value_json_example();
        json["id"] = json!("downtime0");
        json
    }

    #[async_std::test]
    async fn list_downtimes() {
        let server = test_server! {
            method = GET,
            path = "/api/v0/downtimes",
            response = json!({
                "downtimes": [entity_json_example()],
            }),
        };
        assert_eq!(
            test_client!(server).list_downtimes().await,
            Ok(vec![entity_example()]),
        );
    }

    #[async_std::test]
    async fn create_downtime() {
        let server = test_server! {
            method = POST,
            path = "/api/v0/downtimes",
            request = value_json_example(),
            response = entity_json_example(),
        };
        assert_eq!(
            test_client!(server).create_downtime(value_example()).await,
            Ok(entity_example()),
        );
        assert_eq!(
            test_client!(server).create_downtime(&value_example()).await,
            Ok(entity_example()),
        );
    }

    #[async_std::test]
    async fn update_downtime() {
        let server = test_server! {
            method = PUT,
            path = "/api/v0/downtimes/downtime0",
            request = value_json_example(),
            response = entity_json_example(),
        };
        assert_eq!(
            test_client!(server)
                .update_downtime("downtime0", value_example())
                .await,
            Ok(entity_example()),
        );
        assert_eq!(
            test_client!(server)
                .update_downtime(DowntimeId::from("downtime0"), &value_example())
                .await,
            Ok(entity_example()),
        );
    }

    #[async_std::test]
    async fn delete_downtime() {
        let server = test_server! {
            method = DELETE,
            path = "/api/v0/downtimes/downtime0",
            response = entity_json_example(),
        };
        assert_eq!(
            test_client!(server).delete_downtime("downtime0").await,
            Ok(entity_example()),
        );
        assert_eq!(
            test_client!(server)
                .delete_downtime(DowntimeId::from("downtime0"))
                .await,
            Ok(entity_example()),
        );
    }
}