use std::convert::From;
use std::error::Error;
use std::fmt;
pub type Result<T> = ::std::result::Result<T, FtpError>;
#[derive(Debug)]
pub enum FtpError {
ConnectionError(::std::io::Error),
SecureError(String),
InvalidResponse(String),
InvalidArgument(String),
InvalidAddress(::std::net::AddrParseError),
}
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
pub enum FormatControl {
Default,
NonPrint,
Telnet,
Asa,
}
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
pub enum FileType {
Ascii(FormatControl),
Ebcdic(FormatControl),
Image,
Binary,
Local(u8),
}
impl ToString for FormatControl {
fn to_string(&self) -> String {
match self {
&FormatControl::Default | &FormatControl::NonPrint => String::from("N"),
&FormatControl::Telnet => String::from("T"),
&FormatControl::Asa => String::from("C"),
}
}
}
impl ToString for FileType {
fn to_string(&self) -> String {
match self {
&FileType::Ascii(ref fc) => format!("A {}", fc.to_string()),
&FileType::Ebcdic(ref fc) => format!("E {}", fc.to_string()),
&FileType::Image | &FileType::Binary => String::from("I"),
&FileType::Local(ref bits) => format!("L {}", bits),
}
}
}
impl fmt::Display for FtpError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match *self {
FtpError::ConnectionError(ref ioerr) => write!(f, "FTP ConnectionError: {}", ioerr),
FtpError::SecureError(ref desc) => write!(f, "FTP SecureError: {}", desc.clone()),
FtpError::InvalidResponse(ref desc) => {
write!(f, "FTP InvalidResponse: {}", desc.clone())
}
FtpError::InvalidArgument(ref desc) => {
write!(f, "FTP InvalidArgument: {}", desc.clone())
}
FtpError::InvalidAddress(ref perr) => write!(f, "FTP InvalidAddress: {}", perr),
}
}
}
impl Error for FtpError {
fn source(&self) -> Option<&(dyn Error + 'static)> {
match *self {
FtpError::ConnectionError(ref ioerr) => Some(ioerr),
FtpError::SecureError(_) => None,
FtpError::InvalidResponse(_) => None,
FtpError::InvalidArgument(_) => None,
FtpError::InvalidAddress(ref perr) => Some(perr),
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn format_control_str() {
assert_eq!(FormatControl::Default.to_string(), "N");
assert_eq!(FormatControl::NonPrint.to_string(), "N");
assert_eq!(FormatControl::Telnet.to_string(), "T");
assert_eq!(FormatControl::Asa.to_string(), "C");
}
#[test]
fn file_type_str() {
assert_eq!(FileType::Ascii(FormatControl::Default).to_string(), "A N");
assert_eq!(FileType::Ebcdic(FormatControl::Asa).to_string(), "E C");
assert_eq!(FileType::Image.to_string(), "I");
assert_eq!(FileType::Binary.to_string(), "I");
assert_eq!(FileType::Local(6).to_string(), "L 6");
}
}