wreq-proto 0.2.3

//! Pieces pertaining to the HTTP message protocol.

mod headers;

pub mod http1;
pub mod http2;

/// A request line of an incoming request message.
#[derive(Debug, Default, PartialEq)]
pub(crate) struct RequestLine(http::Method, http::Uri);

/// An Incoming Message head. Includes request/status line, and headers.
#[derive(Debug, Default)]
pub(crate) struct MessageHead<S> {
    /// HTTP version of the message.
    version: http::Version,
    /// Subject (request line or status line) of Incoming message.
    subject: S,
    /// Headers of the Incoming message.
    headers: http::HeaderMap,
    /// Extensions.
    extensions: http::Extensions,
}

/// An incoming request message.
type RequestHead = MessageHead<RequestLine>;

/// An incoming response message.
type ResponseHead = MessageHead<http::StatusCode>;

#[derive(Debug)]
pub(crate) enum BodyLength {
    /// Content-Length
    Known(u64),
    /// Transfer-Encoding: chunked (if h1)
    Unknown,
}

/// Status of when a Dispatcher future completes.
pub(crate) enum Dispatched {
    /// Dispatcher completely shutdown connection.
    Shutdown,
    /// Dispatcher has pending upgrade, and so did not shutdown.
    Upgrade(crate::upgrade::Pending),
}

impl MessageHead<http::StatusCode> {
    fn into_response<B>(self, body: B) -> http::Response<B> {
        let mut res = http::Response::new(body);
        *res.status_mut() = self.subject;
        *res.headers_mut() = self.headers;
        *res.version_mut() = self.version;
        *res.extensions_mut() = self.extensions;
        res
    }
}