rightkit-http 0.3.5

use std::fmt;
use std::net::IpAddr;
use std::time::Duration;

use crate::netpolicy::{AddressClass, AddressDenial};

/// Maximum error-body bytes retained on a [`HttpError::Status`].
pub const MAX_ERROR_BODY_BYTES: usize = 64 * 1024;

/// Where in a call an error happened. Carried by [`HttpError::Timeout`] and
/// [`HttpError::Transport`] so apps can keep their own per-phase messages.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[non_exhaustive]
pub enum Phase {
    /// Name lookup, TCP connect or TLS handshake.
    Connect,
    /// Sending the request and waiting for the response headers.
    Headers,
    /// Reading the response body.
    Body,
    /// Any phase of a redirect hop after the first request, or an exhausted redirect limit.
    Redirect,
}

impl Phase {
    pub fn as_str(self) -> &'static str {
        match self {
            Phase::Connect => "connect",
            Phase::Headers => "headers",
            Phase::Body => "body",
            Phase::Redirect => "redirect",
        }
    }
}

#[derive(Debug)]
#[non_exhaustive]
pub enum HttpError {
    /// URL, header, or request construction failed before anything was sent.
    InvalidRequest(String),
    /// Connection, TLS, socket failure, or exhausted redirect limit.
    /// `connect_phase` is true when nothing
    /// can have reached the server (safe to retry any method); it is true exactly when
    /// `phase` is [`Phase::Connect`].
    Transport {
        message: String,
        connect_phase: bool,
        phase: Phase,
    },
    /// A configured timeout elapsed. `message` names the elapsed timeout.
    Timeout { message: String, phase: Phase },
    /// Non-2xx response (after retries were exhausted or not applicable).
    Status {
        status: u16,
        body: String,
        retry_after: Option<Duration>,
        attempts: u32,
    },
    /// Response body exceeded the configured limit.
    BodyTooLarge { limit: usize },
    /// Response was not valid JSON for the requested type.
    Json(String),
    /// Streaming protocol violation (oversized SSE line, bad UTF-8, ...).
    Stream(String),
    /// The call was cancelled through a [`crate::CancelToken`].
    Cancelled,
    /// [`ProxyMode::System`](crate::ProxyMode::System): the Windows proxy settings hold
    /// something this version cannot follow exactly, named in `element`: different proxy
    /// routes for HTTP and HTTPS, a bypass entry that is not a plain host (`<local>`, an entry
    /// limited to one scheme, an inner wildcard), or a proxy auto-config script. The client
    /// sends nothing: it never drops a bypass entry while still using the proxy, and never
    /// sends one scheme to the other's proxy. "Automatically detect settings" alone is not
    /// refused (since 0.3.1; see `ProxyModeUsed`). Returned by
    /// `Client::try_new` and by every request of a client built with `Client::new`. An app
    /// catches it and uses the mode it used before (`ProxyMode::Environment`, or
    /// `ProxyMode::Explicit`). Never retried.
    SystemProxyUnsupported { element: String },
    /// The outbound address policy refused the destination (see [`crate::netpolicy`]).
    /// `addr` and `class` are `None` when the denial is about the name or scheme (no
    /// address, a metadata host name, or a host outside a first-party list). `reason`
    /// carries the detail, including custom `deny_if` text. Never retried.
    AddressDenied {
        host: String,
        addr: Option<IpAddr>,
        class: Option<AddressClass>,
        reason: String,
    },
    /// [`crate::ClientConfig::https_only`] is set and the URL (or a redirect target when
    /// `redirect` is true) is not `https`. `scheme` is lowercase, `host` has no
    /// credentials. Nothing was sent to that URL. Never retried.
    HttpsRequired {
        host: String,
        scheme: String,
        redirect: bool,
    },
}

impl From<AddressDenial> for HttpError {
    fn from(d: AddressDenial) -> Self {
        HttpError::AddressDenied {
            host: d.host,
            addr: d.addr,
            class: d.class,
            reason: d.reason,
        }
    }
}

/// Coarse class of an [`HttpError`], from [`HttpError::class`]. Apps map their own error
/// classes and retry sets from this one table instead of matching statuses themselves.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[non_exhaustive]
pub enum ErrorClass {
    /// HTTP 401, 403 or 407.
    Auth,
    /// HTTP 429. `retry_after` is the parsed `Retry-After` delay, when the server sent one.
    RateLimit { retry_after: Option<Duration> },
    /// HTTP 5xx.
    Server,
    /// Any other status: 4xx not listed above, or an unfollowed 3xx.
    Client,
    /// A configured timeout elapsed, or the server answered HTTP 408 Request Timeout.
    Timeout,
    /// Connection, TLS or socket failure, or an exhausted redirect limit.
    Transport,
    /// The call was cancelled.
    Cancelled,
    /// The outbound address policy refused the destination.
    AddressDenied,
    /// The request could not be built, or the response body was malformed or oversized
    /// (`InvalidRequest`, `Json`, `Stream`, `BodyTooLarge`), or `https_only` refused a URL
    /// (`HttpsRequired`).
    Invalid,
}

impl HttpError {
    pub(crate) fn timeout(message: impl Into<String>, phase: Phase) -> Self {
        HttpError::Timeout {
            message: message.into(),
            phase,
        }
    }

    pub(crate) fn transport(message: impl Into<String>, phase: Phase) -> Self {
        HttpError::Transport {
            message: message.into(),
            connect_phase: phase == Phase::Connect,
            phase,
        }
    }

    /// The [`Phase`] of a `Timeout` or `Transport` error; `None` for every other kind.
    pub fn phase(&self) -> Option<Phase> {
        match self {
            HttpError::Timeout { phase, .. } | HttpError::Transport { phase, .. } => Some(*phase),
            _ => None,
        }
    }

    /// Re-label a `Timeout` or `Transport` error as happening during a redirect hop. A
    /// transport error is no longer a pure connect failure, because earlier hops reached servers.
    pub(crate) fn into_redirect_phase(self) -> Self {
        match self {
            HttpError::Timeout { message, .. } => HttpError::timeout(message, Phase::Redirect),
            HttpError::Transport { message, .. } => HttpError::Transport {
                message,
                connect_phase: false,
                phase: Phase::Redirect,
            },
            other => other,
        }
    }

    pub fn status(&self) -> Option<u16> {
        match self {
            HttpError::Status { status, .. } => Some(*status),
            _ => None,
        }
    }

    /// The error's [`ErrorClass`]. See the README for the full table.
    pub fn class(&self) -> ErrorClass {
        match self {
            HttpError::Status {
                status,
                retry_after,
                ..
            } => match *status {
                401 | 403 | 407 => ErrorClass::Auth,
                408 => ErrorClass::Timeout,
                429 => ErrorClass::RateLimit {
                    retry_after: *retry_after,
                },
                500..=599 => ErrorClass::Server,
                _ => ErrorClass::Client,
            },
            HttpError::Timeout { .. } => ErrorClass::Timeout,
            HttpError::Transport { .. } => ErrorClass::Transport,
            HttpError::Cancelled => ErrorClass::Cancelled,
            HttpError::SystemProxyUnsupported { .. } => ErrorClass::Invalid,
            HttpError::AddressDenied { .. } => ErrorClass::AddressDenied,
            HttpError::HttpsRequired { .. } => ErrorClass::Invalid,
            HttpError::InvalidRequest(_)
            | HttpError::Json(_)
            | HttpError::Stream(_)
            | HttpError::BodyTooLarge { .. } => ErrorClass::Invalid,
        }
    }
}

impl fmt::Display for HttpError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            HttpError::InvalidRequest(m) => write!(f, "invalid request: {m}"),
            HttpError::Transport { message, .. } => write!(f, "transport error: {message}"),
            HttpError::Timeout { message, .. } => write!(f, "timeout: {message}"),
            HttpError::Status {
                status, attempts, ..
            } => {
                write!(f, "HTTP {status} after {attempts} attempt(s)")
            }
            HttpError::BodyTooLarge { limit } => write!(f, "response body exceeds {limit} bytes"),
            HttpError::Json(m) => write!(f, "invalid JSON: {m}"),
            HttpError::Stream(m) => write!(f, "stream error: {m}"),
            HttpError::Cancelled => write!(f, "cancelled"),
            HttpError::SystemProxyUnsupported { element } => write!(
                f,
                "the Windows proxy settings cannot be followed exactly by this version \
                 ({element}); nothing was sent"
            ),
            HttpError::AddressDenied { host, reason, .. } => {
                write!(f, "address denied for host '{host}': {reason}")
            }
            HttpError::HttpsRequired {
                host,
                scheme,
                redirect,
            } => {
                let what = if *redirect {
                    "redirect to"
                } else {
                    "request to"
                };
                write!(
                    f,
                    "https required: {what} '{scheme}' URL for host '{host}' refused"
                )
            }
        }
    }
}

impl std::error::Error for HttpError {}

#[cfg(test)]
mod tests {
    use super::*;

    fn status(code: u16, retry_after: Option<Duration>) -> HttpError {
        HttpError::Status {
            status: code,
            body: String::new(),
            retry_after,
            attempts: 1,
        }
    }

    #[test]
    fn statuses_map_to_their_classes() {
        for code in [401, 403, 407] {
            assert_eq!(status(code, None).class(), ErrorClass::Auth, "{code}");
        }
        for code in [500, 502, 503, 504, 599] {
            assert_eq!(status(code, None).class(), ErrorClass::Server, "{code}");
        }
        for code in [400, 402, 404, 409, 410, 422, 451, 499, 301, 302] {
            assert_eq!(status(code, None).class(), ErrorClass::Client, "{code}");
        }
    }

    #[test]
    fn http_408_is_a_timeout_not_a_client_error() {
        assert_eq!(status(408, None).class(), ErrorClass::Timeout);
        assert_eq!(status(408, None).status(), Some(408));
    }

    #[test]
    fn rate_limit_carries_the_parsed_retry_after() {
        assert_eq!(
            status(429, Some(Duration::from_secs(7))).class(),
            ErrorClass::RateLimit {
                retry_after: Some(Duration::from_secs(7))
            }
        );
        assert_eq!(
            status(429, None).class(),
            ErrorClass::RateLimit { retry_after: None }
        );
    }

    #[test]
    fn non_status_errors_map_to_their_classes() {
        assert_eq!(
            HttpError::timeout("x", Phase::Body).class(),
            ErrorClass::Timeout
        );
        for phase in [Phase::Connect, Phase::Headers, Phase::Body, Phase::Redirect] {
            assert_eq!(
                HttpError::transport("x", phase).class(),
                ErrorClass::Transport
            );
        }
        assert_eq!(HttpError::Cancelled.class(), ErrorClass::Cancelled);
        assert_eq!(
            HttpError::AddressDenied {
                host: "h".into(),
                addr: None,
                class: None,
                reason: "r".into()
            }
            .class(),
            ErrorClass::AddressDenied
        );
        assert_eq!(
            HttpError::InvalidRequest("x".into()).class(),
            ErrorClass::Invalid
        );
        assert_eq!(HttpError::Json("x".into()).class(), ErrorClass::Invalid);
        assert_eq!(HttpError::Stream("x".into()).class(), ErrorClass::Invalid);
        assert_eq!(
            HttpError::BodyTooLarge { limit: 1 }.class(),
            ErrorClass::Invalid
        );
    }

    #[test]
    fn phase_is_carried_by_timeout_and_transport_only() {
        for phase in [Phase::Connect, Phase::Headers, Phase::Body, Phase::Redirect] {
            assert_eq!(HttpError::timeout("t", phase).phase(), Some(phase));
            assert_eq!(HttpError::transport("t", phase).phase(), Some(phase));
        }
        assert_eq!(HttpError::Cancelled.phase(), None);
        assert_eq!(status(500, None).phase(), None);
        assert_eq!(HttpError::InvalidRequest("x".into()).phase(), None);
    }

    #[test]
    fn connect_phase_is_true_exactly_for_the_connect_phase() {
        for (phase, want) in [
            (Phase::Connect, true),
            (Phase::Headers, false),
            (Phase::Body, false),
            (Phase::Redirect, false),
        ] {
            match HttpError::transport("t", phase) {
                HttpError::Transport { connect_phase, .. } => assert_eq!(connect_phase, want),
                other => panic!("{other:?}"),
            }
        }
    }

    #[test]
    fn redirect_relabel_clears_connect_phase() {
        match HttpError::transport("t", Phase::Connect).into_redirect_phase() {
            HttpError::Transport {
                connect_phase,
                phase,
                ..
            } => {
                assert!(!connect_phase);
                assert_eq!(phase, Phase::Redirect);
            }
            other => panic!("{other:?}"),
        }
        assert_eq!(
            HttpError::timeout("t", Phase::Headers)
                .into_redirect_phase()
                .phase(),
            Some(Phase::Redirect)
        );
        assert!(matches!(
            HttpError::Cancelled.into_redirect_phase(),
            HttpError::Cancelled
        ));
    }

    #[test]
    fn phase_names() {
        assert_eq!(Phase::Connect.as_str(), "connect");
        assert_eq!(Phase::Headers.as_str(), "headers");
        assert_eq!(Phase::Body.as_str(), "body");
        assert_eq!(Phase::Redirect.as_str(), "redirect");
    }
}