hyperdriver 0.12.3

use std::future::Future;
use std::ops::Deref as _;
use std::task::{Context, Poll};

use http::{Request, Response};
use http_body::Body as HttpBody;

use crate::BoxError;
pub use crate::info::HttpProtocol;
use chateau::client::conn::Connection;
use chateau::client::pool::{PoolableConnection, Pooled};

/// An asynchronous function from `Request` to `Response`.
pub trait HttpService<ReqBody> {
    /// The `HttpBody` body of the `http::Response`.
    type ResBody: HttpBody;

    /// The error type that can occur within this `Service`.
    ///
    /// Note: Returning an `Error` to a hyper server will cause the connection
    /// to be abruptly aborted. In most cases, it is better to return a `Response`
    /// with a 4xx or 5xx status code.
    type Error: Into<BoxError>;

    /// The `Future` returned by this `Service`.
    type Future: Future<Output = Result<Response<Self::ResBody>, Self::Error>>;

    #[doc(hidden)]
    fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>>;

    #[doc(hidden)]
    fn call(&mut self, req: Request<ReqBody>) -> Self::Future;
}

impl<T, BIn, BOut> HttpService<BIn> for T
where
    T: tower::Service<Request<BIn>, Response = Response<BOut>>,
    BOut: HttpBody,
    T::Error: Into<BoxError>,
{
    type ResBody = BOut;

    type Error = T::Error;
    type Future = T::Future;

    fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
        tower::Service::poll_ready(self, cx)
    }

    fn call(&mut self, req: Request<BIn>) -> Self::Future {
        tower::Service::call(self, req)
    }
}

/// A trait to specialize connection info for HTTP connections
pub trait HttpConnectionInfo<B>: Connection<http::Request<B>> {
    /// Return the protocol version for this connection.
    fn version(&self) -> HttpProtocol;
}

impl<C, B> HttpConnectionInfo<B> for Pooled<C, http::Request<B>>
where
    C: HttpConnectionInfo<B> + PoolableConnection<http::Request<B>>,
    B: Send,
{
    fn version(&self) -> HttpProtocol {
        self.deref().version()
    }
}

#[cfg(feature = "client")]
pub(super) mod http1 {

    use std::fmt;
    use std::task::{Context, Poll};

    use ::http;
    use http::Uri;
    use http::uri::Scheme;
    use tower::ServiceExt;
    use tower::util::MapRequest;

    use crate::service::http::HttpProtocol;

    use super::HttpConnectionInfo;

    type PreprocessFn<C, B> = fn((C, http::Request<B>)) -> (C, http::Request<B>);

    /// A service that checks if the request is HTTP/1.1 compatible.
    #[derive(Debug)]
    pub struct Http1ChecksService<S, C, B>
    where
        S: tower::Service<(C, http::Request<B>)>,
        C: HttpConnectionInfo<B>,
    {
        inner: MapRequest<S, PreprocessFn<C, B>>,
    }

    impl<S, C, B> tower::Service<(C, http::Request<B>)> for Http1ChecksService<S, C, B>
    where
        S: tower::Service<(C, http::Request<B>)>,
        C: HttpConnectionInfo<B>,
    {
        type Response = S::Response;

        type Error = S::Error;

        type Future = S::Future;

        fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
            self.inner.poll_ready(cx)
        }

        fn call(&mut self, req: (C, http::Request<B>)) -> Self::Future {
            self.inner.call(req)
        }
    }

    impl<S, C, B> Clone for Http1ChecksService<S, C, B>
    where
        S: tower::Service<(C, http::Request<B>)> + Clone,
        C: HttpConnectionInfo<B>,
    {
        fn clone(&self) -> Self {
            Self {
                inner: self.inner.clone(),
            }
        }
    }

    impl<S, C, B> Http1ChecksService<S, C, B>
    where
        S: tower::Service<(C, http::Request<B>)>,
        C: HttpConnectionInfo<B>,
    {
        /// Create a new `Http1ChecksService`.
        pub fn new(service: S) -> Self {
            Self {
                inner: service.map_request(check_http1_request),
            }
        }
    }

    /// A layer that checks if the request is HTTP/1.1 compatible.
    pub struct Http1ChecksLayer<C, B> {
        processor: std::marker::PhantomData<fn(C, B)>,
    }

    impl<C, B> Http1ChecksLayer<C, B> {
        /// Create a new `Http1ChecksLayer`.
        pub fn new() -> Self {
            Self {
                processor: std::marker::PhantomData,
            }
        }
    }

    impl<C, B> Default for Http1ChecksLayer<C, B> {
        fn default() -> Self {
            Self::new()
        }
    }

    impl<C, B> Clone for Http1ChecksLayer<C, B> {
        fn clone(&self) -> Self {
            Self::new()
        }
    }

    impl<C, B> fmt::Debug for Http1ChecksLayer<C, B> {
        fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
            f.debug_struct("Http1ChecksLayer").finish()
        }
    }

    impl<C, B, S> tower::layer::Layer<S> for Http1ChecksLayer<C, B>
    where
        S: tower::Service<(C, http::Request<B>)>,
        C: HttpConnectionInfo<B>,
    {
        type Service = Http1ChecksService<S, C, B>;

        fn layer(&self, service: S) -> Self::Service {
            Http1ChecksService::new(service)
        }
    }

    fn check_http1_request<C, B>((conn, mut req): (C, http::Request<B>)) -> (C, http::Request<B>)
    where
        C: HttpConnectionInfo<B>,
    {
        if conn.version() != HttpProtocol::Http1 {
            return (conn, req);
        }

        if req.method() == http::Method::CONNECT {
            authority_form(req.uri_mut());

            // If the URI is to HTTPS, and the connector claimed to be a proxy,
            // then it *should* have tunneled, and so we don't want to send
            // absolute-form in that case.
            if req.uri().scheme() == Some(&Scheme::HTTPS) {
                origin_form(req.uri_mut());
            }
        } else if req.uri().scheme().is_none() || req.uri().authority().is_none() {
            absolute_form(req.uri_mut());
        } else {
            origin_form(req.uri_mut());
        }

        (conn, req)
    }

    /// Convert the URI to authority-form, if it is not already.
    ///
    /// This is the form of the URI with just the authority and a default
    /// path and scheme. This is used in HTTP/1 CONNECT requests.
    fn authority_form(uri: &mut Uri) {
        *uri = match uri.authority() {
            Some(auth) => {
                let mut parts = ::http::uri::Parts::default();
                parts.authority = Some(auth.clone());
                Uri::from_parts(parts).expect("authority is valid")
            }
            None => {
                unreachable!("authority_form with relative uri");
            }
        };
    }

    fn absolute_form(uri: &mut Uri) {
        debug_assert!(uri.scheme().is_some(), "absolute_form needs a scheme");
        debug_assert!(
            uri.authority().is_some(),
            "absolute_form needs an authority"
        );
    }

    /// Convert the URI to origin-form, if it is not already.
    ///
    /// This form of the URI has no scheme or authority, and contains just
    /// the path, usually used in HTTP/1 requests.
    fn origin_form(uri: &mut Uri) {
        let path = match uri.path_and_query() {
            Some(path) if path.as_str() != "/" => {
                let mut parts = ::http::uri::Parts::default();
                parts.path_and_query = Some(path.clone());
                Uri::from_parts(parts).expect("path is valid uri")
            }
            _none_or_just_slash => {
                debug_assert!(Uri::default() == "/");
                Uri::default()
            }
        };
        *uri = path
    }

    #[cfg(test)]
    mod tests {

        use super::*;

        #[test]
        fn test_origin_form() {
            let mut uri = "http://example.com".parse().unwrap();
            origin_form(&mut uri);
            assert_eq!(uri, "/");

            let mut uri = "/some/path/here".parse().unwrap();
            origin_form(&mut uri);
            assert_eq!(uri, "/some/path/here");

            let mut uri = "http://example.com:8080/some/path?query#fragment"
                .parse()
                .unwrap();
            origin_form(&mut uri);
            assert_eq!(uri, "/some/path?query");

            let mut uri = "/".parse().unwrap();
            origin_form(&mut uri);
            assert_eq!(uri, "/");
        }

        #[test]
        fn test_absolute_form() {
            let mut uri = "http://example.com".parse().unwrap();
            absolute_form(&mut uri);
            assert_eq!(uri, "http://example.com");

            let mut uri = "http://example.com:8080".parse().unwrap();
            absolute_form(&mut uri);
            assert_eq!(uri, "http://example.com:8080");

            let mut uri = "https://example.com/some/path?query".parse().unwrap();
            absolute_form(&mut uri);
            assert_eq!(uri, "https://example.com/some/path?query");

            let mut uri = "https://example.com:8443".parse().unwrap();
            absolute_form(&mut uri);
            assert_eq!(uri, "https://example.com:8443");

            let mut uri = "http://example.com:443".parse().unwrap();
            absolute_form(&mut uri);
            assert_eq!(uri, "http://example.com:443");

            let mut uri = "https://example.com:80".parse().unwrap();
            absolute_form(&mut uri);
            assert_eq!(uri, "https://example.com:80");
        }
    }
}

#[cfg(feature = "client")]
pub(super) mod http2 {
    use std::fmt;
    use std::task::{Context, Poll};

    use ::http;

    use crate::service::http::HttpProtocol;

    use super::HttpConnectionInfo;

    const CONNECTION_HEADERS: [http::HeaderName; 5] = [
        http::header::CONNECTION,
        http::HeaderName::from_static("proxy-connection"),
        http::HeaderName::from_static("keep-alive"),
        http::header::TRANSFER_ENCODING,
        http::header::UPGRADE,
    ];

    #[derive(Debug, thiserror::Error)]
    pub enum HttpRequestError<E> {
        #[error("Invalid HTTP method for HTTP/2: {0}")]
        InvalidMethod(http::Method),

        #[error(transparent)]
        Connection(E),
    }

    /// A service that checks if the request is HTTP/2 compatible.
    #[derive(Debug, Clone)]
    pub struct Http2ChecksService<S> {
        inner: S,
    }

    impl<S> Http2ChecksService<S> {
        /// Create a new `Http2ChecksService`.
        pub fn new(inner: S) -> Self {
            Self { inner }
        }
    }

    fn check_http2_request<C, B, E>(
        (conn, mut req): (C, http::Request<B>),
    ) -> Result<(C, http::Request<B>), HttpRequestError<E>>
    where
        C: HttpConnectionInfo<B>,
    {
        if conn.version() == HttpProtocol::Http2 {
            if req.method() == http::Method::CONNECT {
                tracing::warn!("CONNECT method not allowed on HTTP/2");
                return Err(HttpRequestError::InvalidMethod(http::Method::CONNECT));
            }

            *req.version_mut() = http::Version::HTTP_2;

            for connection_header in &CONNECTION_HEADERS {
                if req.headers_mut().remove(connection_header).is_some() {
                    tracing::warn!(
                        "removed illegal connection header {:?} from HTTP/2 request",
                        connection_header
                    );
                };
            }

            if req.headers_mut().remove(http::header::HOST).is_some() {
                tracing::warn!("removed illegal header `host` from HTTP/2 request");
            }
        }
        Ok((conn, req))
    }

    impl<S, C, B> tower::Service<(C, http::Request<B>)> for Http2ChecksService<S>
    where
        S: tower::Service<(C, http::Request<B>)>,
        C: HttpConnectionInfo<B>,
    {
        type Response = S::Response;

        type Error = HttpRequestError<S::Error>;

        type Future = self::future::Http2ChecksFuture<S, C, B>;

        #[inline]
        fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
            self.inner
                .poll_ready(cx)
                .map_err(HttpRequestError::Connection)
        }

        #[inline]
        fn call(&mut self, req: (C, http::Request<B>)) -> Self::Future {
            match check_http2_request(req) {
                Ok(req) => self::future::Http2ChecksFuture::new(self.inner.call(req)),
                Err(error) => self::future::Http2ChecksFuture::error(error),
            }
        }
    }

    mod future {
        use std::{
            fmt,
            future::Future,
            pin::Pin,
            task::{Context, Poll, ready},
        };

        use super::HttpRequestError;
        use pin_project::pin_project;

        #[pin_project(project=Http2ChecksStateProject)]
        enum Http2ChecksState<S, C, B>
        where
            S: tower::Service<(C, http::Request<B>)>,
        {
            Service(#[pin] S::Future),
            Error(Option<HttpRequestError<S::Error>>),
        }

        /// Future returned when applying checks for HTTP/2 connections and requests.
        #[pin_project]
        pub struct Http2ChecksFuture<S, C, B>
        where
            S: tower::Service<(C, http::Request<B>)>,
        {
            #[pin]
            state: Http2ChecksState<S, C, B>,
        }

        impl<S, C, B> fmt::Debug for Http2ChecksFuture<S, C, B>
        where
            S: tower::Service<(C, http::Request<B>)>,
        {
            fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
                f.debug_struct("Http2ChecksFuture").finish()
            }
        }

        impl<S, C, B> Http2ChecksFuture<S, C, B>
        where
            S: tower::Service<(C, http::Request<B>)>,
        {
            pub(super) fn new(future: S::Future) -> Self {
                Self {
                    state: Http2ChecksState::Service(future),
                }
            }

            pub(super) fn error(error: HttpRequestError<S::Error>) -> Self {
                Self {
                    state: Http2ChecksState::Error(Some(error)),
                }
            }
        }

        impl<S, C, B> Future for Http2ChecksFuture<S, C, B>
        where
            S: tower::Service<(C, http::Request<B>)>,
        {
            type Output = Result<S::Response, HttpRequestError<S::Error>>;

            fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
                let this = self.project();
                match this.state.project() {
                    Http2ChecksStateProject::Service(future) => {
                        Poll::Ready(ready!(future.poll(cx)).map_err(HttpRequestError::Connection))
                    }
                    Http2ChecksStateProject::Error(error) => Poll::Ready(Err(error
                        .take()
                        .expect("Http2ChecksFuture Error polled after completion"))),
                }
            }
        }
    }

    /// A `Layer` that applies HTTP/2 checks to requests.
    #[derive(Default, Clone)]
    pub struct Http2ChecksLayer {
        _marker: (),
    }

    impl Http2ChecksLayer {
        /// Create a new `Http2ChecksLayer`.
        pub fn new() -> Self {
            Self { _marker: () }
        }
    }

    impl fmt::Debug for Http2ChecksLayer {
        fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
            f.debug_struct("Http2ChecksLayer").finish()
        }
    }

    impl<S> tower::layer::Layer<S> for Http2ChecksLayer {
        type Service = Http2ChecksService<S>;

        fn layer(&self, inner: S) -> Self::Service {
            Http2ChecksService::new(inner)
        }
    }
}

#[cfg(test)]
#[allow(dead_code)]
mod tests {
    use super::*;
    use bytes::Bytes;
    use http_body_util::Empty;
    use std::{convert::Infallible, future::Ready};

    struct Svc;

    impl tower::Service<http::Request<Empty<Bytes>>> for Svc {
        type Response = http::Response<Empty<Bytes>>;
        type Error = Infallible;
        type Future = Ready<Result<Self::Response, Self::Error>>;

        fn poll_ready(&mut self, _cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
            Poll::Ready(Ok(()))
        }

        fn call(&mut self, req: http::Request<Empty<Bytes>>) -> Self::Future {
            assert_eq!(req.version(), http::Version::HTTP_11);
            std::future::ready(Ok(http::Response::new(Empty::new())))
        }
    }

    static_assertions::assert_impl_all!(Svc: HttpService<Empty<Bytes>, ResBody=Empty<Bytes>, Error=Infallible>);

    struct NotASvc;

    impl tower::Service<http::Request<()>> for Svc {
        type Response = http::Response<()>;
        type Error = Infallible;
        type Future = Ready<Result<Self::Response, Self::Error>>;

        fn poll_ready(&mut self, _cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
            Poll::Ready(Ok(()))
        }

        fn call(&mut self, req: http::Request<()>) -> Self::Future {
            assert_eq!(req.version(), http::Version::HTTP_11);
            std::future::ready(Ok(http::Response::new(())))
        }
    }

    static_assertions::assert_not_impl_all!(NotASvc: HttpService<(), ResBody=(), Error=Infallible>);
}