// Copyright (c) 2016-2017 Nikita Pekin and the wolfram_alpha_rs contributors
// See the README.md file at the top-level directory of this distribution.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

use hyper::error::UriError;
use serde_xml;
use std::error::Error as StdError;
use std::fmt;
use std::io;
use std::result::Result as StdResult;
use std::str::Utf8Error;

/// A convenient alias type for results for `wolfram_alpha`.
pub type Result<T> = StdResult<T, Error>;

/// Represents errors which occur while using the Wolfram|Alpha API.
#[derive(Debug)]
pub enum Error {
    /// Error sending a HTTP request to Wolfram|Alpha.
    HttpRequest(HttpRequestError),
    /// An IO error was encountered.
    Io(io::Error),
    /// Error while serializing or deserializing XML.
    Xml(serde_xml::Error),
}

impl fmt::Display for Error {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match *self {
            Error::HttpRequest(ref e) => e.fmt(f),
            Error::Io(ref e) => e.fmt(f),
            Error::Xml(ref e) => e.fmt(f),
        }
    }
}

impl StdError for Error {
    fn description(&self) -> &str {
        match *self {
            Error::HttpRequest(ref e) => e.description(),
            Error::Io(ref e) => e.description(),
            Error::Xml(ref e) => e.description(),
        }
    }

    fn cause(&self) -> Option<&StdError> {
        match *self {
            Error::HttpRequest(ref e) => e.cause(),
            Error::Io(ref e) => e.cause(),
            Error::Xml(ref e) => e.cause(),
        }
    }
}

impl From<HttpRequestError> for Error {
    fn from(error: HttpRequestError) -> Error {
        Error::HttpRequest(error)
    }
}

impl From<io::Error> for Error {
    fn from(error: io::Error) -> Error {
        Error::Io(error)
    }
}

impl From<serde_xml::Error> for Error {
    fn from(error: serde_xml::Error) -> Error {
        Error::Xml(error)
    }
}

// Implement `PartialEq` manually, since `std::io::Error` does not implement it.
impl PartialEq<Error> for Error {
    fn eq(&self, other: &Error) -> bool {
        use self::Error::*;

        match (self, other) {
            (&HttpRequest(_), &HttpRequest(_)) |
            (&Io(_), &Io(_)) |
            (&Xml(_), &Xml(_)) => true,
            _ => false,
        }
    }
}

/// Represents errors which occur when sending an HTTP request to Wolfram|Alpha.
#[derive(Debug)]
pub enum HttpRequestError {
    /// Error parsing a URL string into a `hyper::Uri`.
    UriError(UriError),
    /// Error parsing a bytes into a UTF-8 string.
    Utf8Error(Utf8Error),
    /// An error occuring during network IO operations.
    Io(io::Error),
    /// Any other error occuring during an HTTP request.
    Other(Box<StdError>),
}

impl fmt::Display for HttpRequestError {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match *self {
            HttpRequestError::Io(ref e) => e.fmt(f),
            HttpRequestError::UriError(ref e) => e.fmt(f),
            HttpRequestError::Utf8Error(ref e) => e.fmt(f),
            HttpRequestError::Other(ref e) => e.fmt(f),
        }
    }
}

impl StdError for HttpRequestError {
    fn description(&self) -> &str {
        match *self {
            HttpRequestError::Io(ref e) => e.description(),
            HttpRequestError::UriError(ref e) => e.description(),
            HttpRequestError::Utf8Error(ref e) => e.description(),
            HttpRequestError::Other(ref e) => e.description(),
        }
    }

    fn cause(&self) -> Option<&StdError> {
        match *self {
            HttpRequestError::Io(ref e) => e.cause(),
            HttpRequestError::UriError(ref e) => e.cause(),
            HttpRequestError::Utf8Error(ref e) => e.cause(),
            HttpRequestError::Other(ref e) => e.cause(),
        }
    }
}

impl From<UriError> for HttpRequestError {
    fn from(error: UriError) -> HttpRequestError {
        HttpRequestError::UriError(error)
    }
}

impl From<Utf8Error> for HttpRequestError {
    fn from(error: Utf8Error) -> HttpRequestError {
        HttpRequestError::Utf8Error(error)
    }
}

impl From<io::Error> for HttpRequestError {
    fn from(error: io::Error) -> HttpRequestError {
        HttpRequestError::Io(error)
    }
}
