1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

//! Types for EPP domain check request

use std::fmt;

use instant_xml::{FromXml, Serializer, ToXml};

use super::XMLNS;
use crate::common::{NoExtension, EPP_XMLNS};
use crate::request::{Command, Transaction};

impl<'a> Transaction<NoExtension> for DomainCheck<'a> {}

impl<'a> Command for DomainCheck<'a> {
    type Response = CheckData;
    const COMMAND: &'static str = "check";
}

// Request

#[derive(Debug, ToXml)]
#[xml(rename = "check", ns(XMLNS))]
struct DomainList<'a> {
    #[xml(rename = "name", ns(XMLNS))]
    domains: &'a [&'a str],
}

fn serialize_domains<W: fmt::Write + ?Sized>(
    domains: &[&str],
    serializer: &mut Serializer<W>,
) -> Result<(), instant_xml::Error> {
    DomainList { domains }.serialize(None, serializer)
}

#[derive(ToXml, Debug)]
#[xml(rename = "check", ns(EPP_XMLNS))]
pub struct DomainCheck<'a> {
    /// The list of domains to be checked for availability
    #[xml(serialize_with = "serialize_domains")]
    pub domains: &'a [&'a str],
}

// Response

#[derive(Debug, FromXml)]
#[xml(rename = "name", ns(XMLNS))]
pub struct Checked {
    #[xml(attribute, rename = "avail")]
    pub available: bool,
    #[xml(attribute)]
    pub reason: Option<String>,
    #[xml(direct)]
    pub id: String,
}

#[derive(Debug, FromXml)]
#[xml(rename = "cd", ns(XMLNS))]
pub struct CheckedDomain {
    /// Data under the `<cd>` tag
    #[xml(rename = "cd")]
    pub inner: Checked,
}

/// Type that represents the `<chkData>` tag for host check response
#[derive(Debug, FromXml)]
#[xml(rename = "chkData", ns(XMLNS))]
pub struct CheckData {
    pub list: Vec<CheckedDomain>,
}

#[cfg(test)]
mod tests {
    use super::DomainCheck;
    use crate::response::ResultCode;
    use crate::tests::{assert_serialized, response_from_file, CLTRID, SUCCESS_MSG, SVTRID};

    #[test]
    fn command() {
        let object = DomainCheck {
            domains: &["eppdev.com", "eppdev.net"],
        };
        assert_serialized("request/domain/check.xml", &object);
    }

    #[test]
    fn response() {
        let object = response_from_file::<DomainCheck>("response/domain/check.xml");
        let result = dbg!(&object).res_data().unwrap();

        assert_eq!(object.result.code, ResultCode::CommandCompletedSuccessfully);
        assert_eq!(object.result.message, SUCCESS_MSG);
        assert_eq!(result.list[0].inner.id, "eppdev.com");
        assert!(result.list[0].inner.available);
        assert_eq!(result.list[1].inner.id, "eppdev.net");
        assert!(!result.list[1].inner.available);
        assert_eq!(object.tr_ids.client_tr_id.unwrap(), CLTRID);
        assert_eq!(object.tr_ids.server_tr_id, SVTRID);
    }
}