Refactor host check models

This commit is contained in:
Nicholas Rempel 2021-11-26 10:27:03 -08:00 committed by masalachai
parent 18097c14f0
commit fa032804a6
9 changed files with 68 additions and 69 deletions

View File

@ -5,7 +5,6 @@ pub mod request;
pub mod response; pub mod response;
pub mod xml; pub mod xml;
pub use request::host::check::*;
pub use request::host::create::*; pub use request::host::create::*;
pub use request::host::delete::*; pub use request::host::delete::*;
pub use request::host::info::*; pub use request::host::info::*;
@ -13,7 +12,6 @@ pub use request::host::update::*;
pub use request::message::ack::*; pub use request::message::ack::*;
pub use request::message::poll::*; pub use request::message::poll::*;
pub use response::host::check::*;
pub use response::host::create::*; pub use response::host::create::*;
pub use response::host::delete::*; pub use response::host::delete::*;
pub use response::host::info::*; pub use response::host::info::*;

View File

@ -1,6 +1,5 @@
//! Types for EPP host requests //! Types for EPP host requests
pub mod check;
pub mod create; pub mod create;
pub mod delete; pub mod delete;
pub mod info; pub mod info;

View File

@ -1,6 +1,5 @@
//! Types for EPP host responses //! Types for EPP host responses
pub mod check;
pub mod create; pub mod create;
pub mod delete; pub mod delete;
pub mod info; pub mod info;

View File

@ -1,49 +0,0 @@
//! Types for EPP host check response
use serde::{Deserialize, Serialize};
use crate::epp::object::{EppObject, StringValue};
use crate::epp::response::CommandResponse;
/// Type that represents the <epp> tag for the EPP XML host check response
pub type EppHostCheckResponse = EppObject<CommandResponse<HostCheckResult>>;
/// Type that represents the &lt;name&gt; tag for host check response
#[derive(Serialize, Deserialize, Debug)]
pub struct HostCheck {
/// The host name
#[serde(rename = "$value")]
pub name: StringValue,
/// The host (un)availability
#[serde(rename = "avail")]
pub available: u16,
}
/// Type that represents the &lt;cd&gt; tag for host check response
#[derive(Serialize, Deserialize, Debug)]
pub struct HostCheckDataItem {
/// Data under the &lt;name&gt; tag
#[serde(rename = "name")]
pub host: HostCheck,
/// The reason for (un)availability
pub reason: Option<StringValue>,
}
/// Type that represents the &lt;chkData&gt; tag for host check response
#[derive(Serialize, Deserialize, Debug)]
pub struct HostCheckData {
/// XML namespace for host response data
#[serde(rename = "xmlns:host")]
xmlns: String,
/// Data under the &lt;cd&gt; tag
#[serde(rename = "cd")]
pub host_list: Vec<HostCheckDataItem>,
}
/// Type that represents the &lt;resData&gt; tag for host check response
#[derive(Serialize, Deserialize, Debug)]
pub struct HostCheckResult {
/// Data under the &lt;chkData&gt; tag
#[serde(rename = "chkData")]
pub check_data: HostCheckData,
}

1
epp-client/src/host.rs Normal file
View File

@ -0,0 +1 @@
pub mod check;

View File

@ -4,6 +4,7 @@ use epp_client_macros::*;
use crate::epp::object::{ElementName, EppObject, StringValue}; use crate::epp::object::{ElementName, EppObject, StringValue};
use crate::epp::request::Command; use crate::epp::request::Command;
use crate::epp::response::CommandResponse;
use crate::epp::xml::EPP_HOST_XMLNS; use crate::epp::xml::EPP_HOST_XMLNS;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
@ -16,7 +17,7 @@ use serde::{Deserialize, Serialize};
/// ///
/// use epp_client::config::{EppClientConfig, EppClientConnection}; /// use epp_client::config::{EppClientConfig, EppClientConnection};
/// use epp_client::EppClient; /// use epp_client::EppClient;
/// use epp_client::epp::{EppHostCheck, EppHostCheckResponse}; /// use epp_client::host::check::{EppHostCheck, EppHostCheckResponse};
/// use epp_client::epp::generate_client_tr_id; /// use epp_client::epp::generate_client_tr_id;
/// ///
/// #[tokio::main] /// #[tokio::main]
@ -56,7 +57,28 @@ use serde::{Deserialize, Serialize};
/// client.logout().await.unwrap(); /// client.logout().await.unwrap();
/// } /// }
/// ``` /// ```
pub type EppHostCheck = EppObject<Command<HostCheck>>; pub type EppHostCheck = EppObject<Command<HostCheckRequest>>;
impl EppHostCheck {
/// Creates a new EppObject for host check corresponding to the &lt;epp&gt; tag in EPP XML
pub fn new(hosts: &[&str], client_tr_id: &str) -> EppHostCheck {
let hosts = hosts.iter().map(|&d| d.into()).collect();
let host_check = HostCheckRequest {
list: HostList {
xmlns: EPP_HOST_XMLNS.to_string(),
hosts,
},
};
EppObject::build(Command::<HostCheckRequest>::new(host_check, client_tr_id))
}
}
/// Type that represents the &lt;epp&gt; tag for the EPP XML host check response
pub type EppHostCheckResponse = EppObject<CommandResponse<HostCheckResult>>;
// Request
/// Type for data under the host &lt;check&gt; tag /// Type for data under the host &lt;check&gt; tag
#[derive(Serialize, Deserialize, Debug)] #[derive(Serialize, Deserialize, Debug)]
@ -72,24 +94,50 @@ pub struct HostList {
#[derive(Serialize, Deserialize, Debug, ElementName)] #[derive(Serialize, Deserialize, Debug, ElementName)]
#[element_name(name = "check")] #[element_name(name = "check")]
/// Type for EPP XML &lt;check&gt; command for hosts /// Type for EPP XML &lt;check&gt; command for hosts
pub struct HostCheck { pub struct HostCheckRequest {
/// The instance holding the list of hosts to be checked /// The instance holding the list of hosts to be checked
#[serde(rename = "host:check", alias = "check")] #[serde(rename = "host:check", alias = "check")]
list: HostList, list: HostList,
} }
impl EppHostCheck { // Response
/// Creates a new EppObject for host check corresponding to the &lt;epp&gt; tag in EPP XML
pub fn new(hosts: &[&str], client_tr_id: &str) -> EppHostCheck {
let hosts = hosts.iter().map(|&d| d.into()).collect();
let host_check = HostCheck { /// Type that represents the &lt;name&gt; tag for host check response
list: HostList { #[derive(Serialize, Deserialize, Debug)]
xmlns: EPP_HOST_XMLNS.to_string(), pub struct HostCheck {
hosts, /// The host name
}, #[serde(rename = "$value")]
}; pub name: StringValue,
/// The host (un)availability
EppObject::build(Command::<HostCheck>::new(host_check, client_tr_id)) #[serde(rename = "avail")]
} pub available: u16,
}
/// Type that represents the &lt;cd&gt; tag for host check response
#[derive(Serialize, Deserialize, Debug)]
pub struct HostCheckDataItem {
/// Data under the &lt;name&gt; tag
#[serde(rename = "name")]
pub host: HostCheck,
/// The reason for (un)availability
pub reason: Option<StringValue>,
}
/// Type that represents the &lt;chkData&gt; tag for host check response
#[derive(Serialize, Deserialize, Debug)]
pub struct HostCheckData {
/// XML namespace for host response data
#[serde(rename = "xmlns:host")]
xmlns: String,
/// Data under the &lt;cd&gt; tag
#[serde(rename = "cd")]
pub host_list: Vec<HostCheckDataItem>,
}
/// Type that represents the &lt;resData&gt; tag for host check response
#[derive(Serialize, Deserialize, Debug)]
pub struct HostCheckResult {
/// Data under the &lt;chkData&gt; tag
#[serde(rename = "chkData")]
pub check_data: HostCheckData,
} }

View File

@ -106,6 +106,7 @@ pub mod contact;
pub mod domain; pub mod domain;
pub mod epp; pub mod epp;
pub mod error; pub mod error;
pub mod host;
pub use connection::client::EppClient; pub use connection::client::EppClient;
#[cfg(test)] #[cfg(test)]

View File

@ -27,6 +27,7 @@ mod response {
}; };
use crate::epp::xml::EppXml; use crate::epp::xml::EppXml;
use crate::epp::*; use crate::epp::*;
use crate::host::check::EppHostCheckResponse;
const SVTRID: &str = "RO-6879-1627224678242975"; const SVTRID: &str = "RO-6879-1627224678242975";
const SUCCESS_MSG: &str = "Command completed successfully"; const SUCCESS_MSG: &str = "Command completed successfully";

View File

@ -30,6 +30,7 @@ mod request {
use crate::epp::request::{EppHello, EppLogin, EppLogout}; use crate::epp::request::{EppHello, EppLogin, EppLogout};
use crate::epp::xml::EppXml; use crate::epp::xml::EppXml;
use crate::epp::*; use crate::epp::*;
use crate::host::check::EppHostCheck;
use chrono::{DateTime, NaiveDate}; use chrono::{DateTime, NaiveDate};
use std::str::FromStr; use std::str::FromStr;