use hyper::client::{Client, Response};
use hyper::error;
use hyper::error::Result as HResult;
use hyper::header::Location;
use hyper::net::HttpsConnector;
use hyper::status::StatusCode;
use hyper::Url;
use hyper_native_tls::NativeTlsClient;
use time;
use std::error::Error;
use std::fmt;
use std::fs::File;
use std::io::{self, ErrorKind, Read};
use crate::message::error::ParseError;
use crate::message::job_status::{PageState, ScanJobStatus};
use crate::message::scan_job::{Format, ScanJob};
use crate::message::scan_status::ScanStatus;
#[derive(Debug)]
pub enum ScannerError {
Io(io::Error),
Parse(ParseError),
AdfEmpty,
Busy,
NotAvailable(io::Error),
Other(String),
}
impl From<ParseError> for ScannerError {
fn from(err: ParseError) -> Self {
ScannerError::Parse(err)
}
}
impl From<error::Error> for ScannerError {
fn from(err: error::Error) -> Self {
if let error::Error::Io(io) = err {
ScannerError::from(io)
} else {
ScannerError::Other(err.to_string())
}
}
}
impl From<String> for ScannerError {
fn from(err: String) -> Self {
ScannerError::Other(err)
}
}
impl From<io::Error> for ScannerError {
fn from(err: io::Error) -> Self {
let not_available = match err.kind() {
ErrorKind::ConnectionRefused => true,
ErrorKind::Other if err.description().contains("Name or service not known") => true,
_ => match err.raw_os_error() {
Some(111) | Some(113) => true,
_ => false,
},
};
if not_available {
ScannerError::NotAvailable(err)
} else {
ScannerError::Io(err)
}
}
}
impl From<error::ParseError> for ScannerError {
fn from(err: error::ParseError) -> Self {
ScannerError::Other(err.description().to_string())
}
}
impl fmt::Display for ScannerError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match *self {
ScannerError::Io(ref err) => write!(f, "{}", err),
ScannerError::Parse(ref err) => write!(f, "{}", err),
ScannerError::AdfEmpty => write!(f, "Adf is empty"),
ScannerError::Busy => write!(f, "Scanner is busy"),
ScannerError::NotAvailable(ref err) => write!(
f,
"Scanner is not available. Is it turned off? Cause: {}",
err
),
ScannerError::Other(ref err) => write!(f, "{}", err),
}
}
}
#[derive(Debug)]
pub struct Scanner {
base_url: Url,
client: Client,
}
#[derive(Debug)]
pub struct Job<'a> {
scanner: &'a Scanner,
location: Url,
binary_url: Option<String>,
}
impl Scanner {
pub fn new(host: &str, use_tls: bool) -> Scanner {
let client = if use_tls {
let ssl = NativeTlsClient::new().unwrap();
let connector = HttpsConnector::new(ssl);
Client::with_connector(connector)
} else {
Client::new()
};
let base_url_string = if use_tls {
format!("https://{}", host)
} else {
format!("http://{}", host)
};
let base_url = Url::parse(&base_url_string).unwrap();
Scanner { client, base_url }
}
pub fn host(&self) -> &str {
self.base_url.host_str().unwrap()
}
pub fn get_scan_status(&self) -> Result<ScanStatus, ScannerError> {
let resp = self.retrieve_scan_status()?;
let status = ScanStatus::read_xml(resp)?;
Ok(status)
}
fn retrieve_scan_status(&self) -> HResult<Response> {
let url = self.base_url.join("/Scan/Status")?;
self.client.get(url).send()
}
pub fn start_job(&self, job: ScanJob) -> Result<Job<'_>, ScannerError> {
let mut target: Vec<u8> = Vec::new();
job.write_xml(&mut target).unwrap();
let result = String::from_utf8(target).unwrap();
println!("{}", result);
let url = self.base_url.join("/Scan/Jobs")?;
let response = self.client.post(url).body(&result).send()?;
if response.status != StatusCode::Created {
return Err(ScannerError::Other(format!(
"Received status {}",
response.status
)));
}
let location: &Location = response.headers.get().unwrap();
let loc_url = Url::parse(location)?;
let loc_url_rebase = self.base_url.join(loc_url.path())?;
println!("{}", loc_url_rebase);
Ok(Job::new(self, loc_url_rebase))
}
fn get_job_status(&self, job: &Job<'_>) -> Result<ScanJobStatus, ScannerError> {
let response = self.client.get(job.location.clone()).send()?;
let status = ScanJobStatus::read_xml(response)?;
Ok(status)
}
fn download_reader(&self, binary_url: &str) -> Result<Box<dyn Read + Send>, ScannerError> {
let url = self.base_url.join(binary_url)?;
let response = self.client.get(url).send()?;
Ok(Box::new(response))
}
}
impl<'a> Job<'a> {
fn new(scanner: &Scanner, location: Url) -> Job<'_> {
Job {
scanner,
location,
binary_url: None,
}
}
pub fn retrieve_status(&mut self) -> Result<bool, ScannerError> {
let status = self.scanner.get_job_status(self)?;
let page = status.pages().get(0).unwrap();
let page_state = page.state();
if page_state == PageState::ReadyToUpload {
self.binary_url = Some(page.binary_url().unwrap().to_owned());
Ok(true)
} else {
Ok(false)
}
}
pub fn download_reader(self) -> Result<Box<dyn Read + Send>, ScannerError> {
self.scanner.download_reader(&self.binary_url.unwrap())
}
pub fn download_to_file(self, target: &str) -> Result<(), ScannerError> {
let mut reader = self.download_reader()?;
let mut file = File::create(target)?;
io::copy(&mut reader, &mut file)?;
Ok(())
}
}
pub fn output_file_name(format: Format, time: &time::Tm) -> String {
let extension = match format {
Format::Pdf => "pdf",
Format::Jpeg => "jpeg",
};
let ts = time::strftime("%Y%m%d_%H%M%S", time).unwrap();
format!("scan_{}.{}", ts, extension)
}
#[cfg(test)]
mod test {
use super::*;
#[test]
fn check_output_file_name() {
let time = time::at_utc(time::Timespec::new(1486905545, 0));
assert_eq!(
"scan_20170212_131905.pdf",
output_file_name(Format::Pdf, &time)
);
assert_eq!(
"scan_20170212_131905.jpeg",
output_file_name(Format::Jpeg, &time)
);
}
}