wreq-rb 0.5.0 → 0.5.1

data/vendor/wreq/src/client/core/conn/http1.rs

@@ -1,342 +0,0 @@
-//! HTTP/1 client connections
-
-use std::{
-    fmt,
-    future::Future,
-    pin::Pin,
-    task::{Context, Poll, ready},
-};
-
-use bytes::Bytes;
-use http::{Request, Response};
-use http_body::Body;
-use httparse::ParserConfig;
-use tokio::io::{AsyncRead, AsyncWrite};
-
-use crate::client::core::{
-    Error, Result,
-    body::Incoming as IncomingBody,
-    dispatch::{self, TrySendError},
-    error::BoxError,
-    http1::Http1Options,
-    proto,
-};
-
-type Dispatcher<T, B> =
-    proto::dispatch::Dispatcher<proto::dispatch::Client<B>, B, T, proto::h1::ClientTransaction>;
-
-/// The sender side of an established connection.
-pub struct SendRequest<B> {
-    dispatch: dispatch::Sender<Request<B>, Response<IncomingBody>>,
-}
-
-/// Deconstructed parts of a `Connection`.
-///
-/// This allows taking apart a `Connection` at a later time, in order to
-/// reclaim the IO object, and additional related pieces.
-#[derive(Debug)]
-#[non_exhaustive]
-pub struct Parts<T> {
-    /// The original IO object used in the handshake.
-    pub io: T,
-    /// A buffer of bytes that have been read but not processed as HTTP.
-    ///
-    /// For instance, if the `Connection` is used for an HTTP upgrade request,
-    /// it is possible the server sent back the first bytes of the new protocol
-    /// along with the response upgrade.
-    ///
-    /// You will want to check for any existing bytes if you plan to continue
-    /// communicating on the IO object.
-    pub read_buf: Bytes,
-}
-
-/// A future that processes all HTTP state for the IO object.
-///
-/// In most cases, this should just be spawned into an executor, so that it
-/// can process incoming and outgoing messages, notice hangups, and the like.
-#[must_use = "futures do nothing unless polled"]
-pub struct Connection<T, B>
-where
-    T: AsyncRead + AsyncWrite,
-    B: Body + 'static,
-{
-    inner: Dispatcher<T, B>,
-}
-
-impl<T, B> Connection<T, B>
-where
-    T: AsyncRead + AsyncWrite + Unpin,
-    B: Body + 'static,
-    B::Error: Into<BoxError>,
-{
-    /// Return the inner IO object, and additional information.
-    ///
-    /// Only works for HTTP/1 connections. HTTP/2 connections will panic.
-    pub fn into_parts(self) -> Parts<T> {
-        let (io, read_buf, _) = self.inner.into_inner();
-        Parts { io, read_buf }
-    }
-}
-
-/// A builder to configure an HTTP connection.
-///
-/// After setting options, the builder is used to create a handshake future.
-///
-/// **Note**: The default values of options are *not considered stable*. They
-/// are subject to change at any time.
-#[derive(Clone, Debug)]
-pub struct Builder {
-    opts: Http1Options,
-}
-
-// ===== impl SendRequest
-
-impl<B> SendRequest<B> {
-    /// Polls to determine whether this sender can be used yet for a request.
-    ///
-    /// If the associated connection is closed, this returns an Error.
-    pub fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<()>> {
-        self.dispatch.poll_ready(cx)
-    }
-
-    /// Waits until the dispatcher is ready
-    ///
-    /// If the associated connection is closed, this returns an Error.
-    pub async fn ready(&mut self) -> Result<()> {
-        std::future::poll_fn(|cx| self.poll_ready(cx)).await
-    }
-
-    /// Checks if the connection is currently ready to send a request.
-    ///
-    /// # Note
-    ///
-    /// This is mostly a hint. Due to inherent latency of networks, it is
-    /// possible that even after checking this is ready, sending a request
-    /// may still fail because the connection was closed in the meantime.
-    pub fn is_ready(&self) -> bool {
-        self.dispatch.is_ready()
-    }
-}
-
-impl<B> SendRequest<B>
-where
-    B: Body + 'static,
-{
-    /// Sends a `Request` on the associated connection.
-    ///
-    /// Returns a future that if successful, yields the `Response`.
-    ///
-    /// # Error
-    ///
-    /// If there was an error before trying to serialize the request to the
-    /// connection, the message will be returned as part of this error.
-    pub fn try_send_request(
-        &mut self,
-        req: Request<B>,
-    ) -> impl Future<Output = std::result::Result<Response<IncomingBody>, TrySendError<Request<B>>>>
-    {
-        let sent = self.dispatch.try_send(req);
-        async move {
-            match sent {
-                Ok(rx) => match rx.await {
-                    Ok(Ok(res)) => Ok(res),
-                    Ok(Err(err)) => Err(err),
-                    // this is definite bug if it happens, but it shouldn't happen!
-                    Err(_) => panic!("dispatch dropped without returning error"),
-                },
-                Err(req) => {
-                    debug!("connection was not ready");
-                    let error = Error::new_canceled().with("connection was not ready");
-                    Err(TrySendError {
-                        error,
-                        message: Some(req),
-                    })
-                }
-            }
-        }
-    }
-}
-
-impl<B> fmt::Debug for SendRequest<B> {
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        f.debug_struct("SendRequest").finish()
-    }
-}
-
-// ===== impl Connection
-
-impl<T, B> Connection<T, B>
-where
-    T: AsyncRead + AsyncWrite + Unpin + Send,
-    B: Body + 'static,
-    B::Error: Into<BoxError>,
-{
-    /// Enable this connection to support higher-level HTTP upgrades.
-    pub fn with_upgrades(self) -> upgrades::UpgradeableConnection<T, B> {
-        upgrades::UpgradeableConnection { inner: Some(self) }
-    }
-}
-
-impl<T, B> fmt::Debug for Connection<T, B>
-where
-    T: AsyncRead + AsyncWrite + fmt::Debug,
-    B: Body + 'static,
-{
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        f.debug_struct("Connection").finish()
-    }
-}
-
-impl<T, B> Future for Connection<T, B>
-where
-    T: AsyncRead + AsyncWrite + Unpin,
-    B: Body + 'static,
-    B::Data: Send,
-    B::Error: Into<BoxError>,
-{
-    type Output = Result<()>;
-
-    fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
-        match ready!(Pin::new(&mut self.inner).poll(cx))? {
-            proto::Dispatched::Shutdown => Poll::Ready(Ok(())),
-            proto::Dispatched::Upgrade(pending) => {
-                // With no `Send` bound on `I`, we can't try to do
-                // upgrades here. In case a user was trying to use
-                // `upgrade` with this API, send a special
-                // error letting them know about that.
-                pending.manual();
-                Poll::Ready(Ok(()))
-            }
-        }
-    }
-}
-
-// ===== impl Builder
-
-impl Default for Builder {
-    fn default() -> Self {
-        Self::new()
-    }
-}
-
-impl Builder {
-    /// Creates a new connection builder.
-    #[inline]
-    pub fn new() -> Builder {
-        Builder {
-            opts: Default::default(),
-        }
-    }
-
-    /// Provide a options configuration for the HTTP/1 connection.
-    #[inline]
-    pub fn options(&mut self, opts: Http1Options) {
-        self.opts = opts;
-    }
-
-    /// Constructs a connection with the configured options and IO.
-    ///
-    /// Note, if [`Connection`] is not `await`-ed, [`SendRequest`] will
-    /// do nothing.
-    pub async fn handshake<T, B>(self, io: T) -> Result<(SendRequest<B>, Connection<T, B>)>
-    where
-        T: AsyncRead + AsyncWrite + Unpin,
-        B: Body + 'static,
-        B::Data: Send,
-        B::Error: Into<BoxError>,
-    {
-        trace!("client handshake HTTP/1");
-
-        let (tx, rx) = dispatch::channel();
-        let mut conn = proto::Conn::new(io);
-
-        // Set the HTTP/1 parser configuration
-        let h1_parser_config = {
-            let mut h1_parser_config = ParserConfig::default();
-            h1_parser_config
-                .ignore_invalid_headers_in_responses(self.opts.ignore_invalid_headers_in_responses)
-                .allow_spaces_after_header_name_in_responses(
-                    self.opts.allow_spaces_after_header_name_in_responses,
-                )
-                .allow_obsolete_multiline_headers_in_responses(
-                    self.opts.allow_obsolete_multiline_headers_in_responses,
-                );
-            h1_parser_config
-        };
-        conn.set_h1_parser_config(h1_parser_config);
-
-        // Set the h1 write strategy
-        if let Some(writev) = self.opts.h1_writev {
-            if writev {
-                conn.set_write_strategy_queue();
-            } else {
-                conn.set_write_strategy_flatten();
-            }
-        }
-
-        // Set the maximum size of the request headers
-        if let Some(max_headers) = self.opts.h1_max_headers {
-            conn.set_http1_max_headers(max_headers);
-        }
-
-        // Enable HTTP/0.9 responses if requested
-        if self.opts.h09_responses {
-            conn.set_h09_responses();
-        }
-
-        // Set the read buffer size if specified
-        if let Some(sz) = self.opts.h1_read_buf_exact_size {
-            conn.set_read_buf_exact_size(sz);
-        }
-
-        // Set the maximum buffer size for HTTP/1 connections
-        if let Some(max) = self.opts.h1_max_buf_size {
-            conn.set_max_buf_size(max);
-        }
-
-        let cd = proto::h1::dispatch::Client::new(rx);
-        let proto = proto::h1::Dispatcher::new(cd, conn);
-
-        Ok((SendRequest { dispatch: tx }, Connection { inner: proto }))
-    }
-}
-
-mod upgrades {
-    use super::*;
-    use crate::client::core::upgrade::Upgraded;
-
-    // A future binding a connection with a Service with Upgrade support.
-    //
-    // This type is unnameable outside the crate.
-    #[must_use = "futures do nothing unless polled"]
-    pub struct UpgradeableConnection<T, B>
-    where
-        T: AsyncRead + AsyncWrite + Unpin + Send + 'static,
-        B: Body + 'static,
-        B::Error: Into<BoxError>,
-    {
-        pub(super) inner: Option<Connection<T, B>>,
-    }
-
-    impl<I, B> Future for UpgradeableConnection<I, B>
-    where
-        I: AsyncRead + AsyncWrite + Unpin + Send + 'static,
-        B: Body + 'static,
-        B::Data: Send,
-        B::Error: Into<BoxError>,
-    {
-        type Output = Result<()>;
-
-        fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
-            match ready!(Pin::new(&mut self.inner.as_mut().unwrap().inner).poll(cx)) {
-                Ok(proto::Dispatched::Shutdown) => Poll::Ready(Ok(())),
-                Ok(proto::Dispatched::Upgrade(pending)) => {
-                    let Parts { io, read_buf } = self.inner.take().unwrap().into_parts();
-                    pending.fulfill(Upgraded::new(io, read_buf));
-                    Poll::Ready(Ok(()))
-                }
-                Err(e) => Poll::Ready(Err(e)),
-            }
-        }
-    }
-}

data/vendor/wreq/src/client/core/conn/http2.rs

@@ -1,307 +0,0 @@
-//! HTTP/2 client connections
-
-use std::{
-    fmt,
-    future::Future,
-    marker::PhantomData,
-    pin::Pin,
-    task::{Context, Poll, ready},
-};
-
-use http::{Request, Response};
-use http_body::Body;
-use tokio::io::{AsyncRead, AsyncWrite};
-
-use crate::{
-    client::core::{
-        Result,
-        body::Incoming as IncomingBody,
-        dispatch::{self, TrySendError},
-        error::{BoxError, Error},
-        proto::{self, h2::ping},
-        rt::{ArcTimer, Time, Timer, bounds::Http2ClientConnExec},
-    },
-    http2::Http2Options,
-};
-
-/// The sender side of an established connection.
-pub struct SendRequest<B> {
-    dispatch: dispatch::UnboundedSender<Request<B>, Response<IncomingBody>>,
-}
-
-impl<B> Clone for SendRequest<B> {
-    fn clone(&self) -> SendRequest<B> {
-        SendRequest {
-            dispatch: self.dispatch.clone(),
-        }
-    }
-}
-
-/// A future that processes all HTTP state for the IO object.
-///
-/// In most cases, this should just be spawned into an executor, so that it
-/// can process incoming and outgoing messages, notice hangups, and the like.
-#[must_use = "futures do nothing unless polled"]
-pub struct Connection<T, B, E>
-where
-    T: AsyncRead + AsyncWrite + Unpin,
-    B: Body + 'static,
-    E: Http2ClientConnExec<B, T> + Unpin,
-    B::Error: Into<BoxError>,
-{
-    inner: (PhantomData<T>, proto::h2::ClientTask<B, E, T>),
-}
-
-/// A builder to configure an HTTP connection.
-///
-/// After setting options, the builder is used to create a handshake future.
-///
-/// **Note**: The default values of options are *not considered stable*. They
-/// are subject to change at any time.
-#[derive(Clone)]
-pub struct Builder<Ex> {
-    exec: Ex,
-    timer: Time,
-    opts: Http2Options,
-}
-
-// ===== impl SendRequest
-
-impl<B> SendRequest<B> {
-    /// Polls to determine whether this sender can be used yet for a request.
-    ///
-    /// If the associated connection is closed, this returns an Error.
-    pub fn poll_ready(&mut self, _cx: &mut Context<'_>) -> Poll<Result<()>> {
-        if self.is_closed() {
-            Poll::Ready(Err(Error::new_closed()))
-        } else {
-            Poll::Ready(Ok(()))
-        }
-    }
-
-    /// Waits until the dispatcher is ready
-    ///
-    /// If the associated connection is closed, this returns an Error.
-    pub async fn ready(&mut self) -> Result<()> {
-        std::future::poll_fn(|cx| self.poll_ready(cx)).await
-    }
-
-    /// Checks if the connection is currently ready to send a request.
-    ///
-    /// # Note
-    ///
-    /// This is mostly a hint. Due to inherent latency of networks, it is
-    /// possible that even after checking this is ready, sending a request
-    /// may still fail because the connection was closed in the meantime.
-    pub fn is_ready(&self) -> bool {
-        self.dispatch.is_ready()
-    }
-
-    /// Checks if the connection side has been closed.
-    pub fn is_closed(&self) -> bool {
-        self.dispatch.is_closed()
-    }
-}
-
-impl<B> SendRequest<B>
-where
-    B: Body + 'static,
-{
-    /// Sends a `Request` on the associated connection.
-    ///
-    /// Returns a future that if successful, yields the `Response`.
-    ///
-    /// # Error
-    ///
-    /// If there was an error before trying to serialize the request to the
-    /// connection, the message will be returned as part of this error.
-    pub fn try_send_request(
-        &mut self,
-        req: Request<B>,
-    ) -> impl Future<Output = std::result::Result<Response<IncomingBody>, TrySendError<Request<B>>>>
-    {
-        let sent = self.dispatch.try_send(req);
-        async move {
-            match sent {
-                Ok(rx) => match rx.await {
-                    Ok(Ok(res)) => Ok(res),
-                    Ok(Err(err)) => Err(err),
-                    // this is definite bug if it happens, but it shouldn't happen!
-                    Err(_) => panic!("dispatch dropped without returning error"),
-                },
-                Err(req) => {
-                    debug!("connection was not ready");
-                    let error = Error::new_canceled().with("connection was not ready");
-                    Err(TrySendError {
-                        error,
-                        message: Some(req),
-                    })
-                }
-            }
-        }
-    }
-}
-
-impl<B> fmt::Debug for SendRequest<B> {
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        f.debug_struct("SendRequest").finish()
-    }
-}
-
-// ===== impl Connection
-
-impl<T, B, E> fmt::Debug for Connection<T, B, E>
-where
-    T: AsyncRead + AsyncWrite + fmt::Debug + 'static + Unpin,
-    B: Body + 'static,
-    E: Http2ClientConnExec<B, T> + Unpin,
-    B::Error: Into<BoxError>,
-{
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        f.debug_struct("Connection").finish()
-    }
-}
-
-impl<T, B, E> Future for Connection<T, B, E>
-where
-    T: AsyncRead + AsyncWrite + Unpin + 'static,
-    B: Body + 'static + Unpin,
-    B::Data: Send,
-    E: Unpin,
-    B::Error: Into<BoxError>,
-    E: Http2ClientConnExec<B, T> + Unpin,
-{
-    type Output = Result<()>;
-
-    fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
-        match ready!(Pin::new(&mut self.inner.1).poll(cx))? {
-            proto::Dispatched::Shutdown => Poll::Ready(Ok(())),
-            proto::Dispatched::Upgrade(_pending) => unreachable!("http2 cannot upgrade"),
-        }
-    }
-}
-
-// ===== impl Builder
-
-impl<Ex> Builder<Ex>
-where
-    Ex: Clone,
-{
-    /// Creates a new connection builder.
-    #[inline]
-    pub fn new(exec: Ex) -> Builder<Ex> {
-        Builder {
-            exec,
-            timer: Time::Empty,
-            opts: Default::default(),
-        }
-    }
-
-    /// Provide a timer to execute background HTTP2 tasks.
-    #[inline]
-    pub fn timer<M>(&mut self, timer: M)
-    where
-        M: Timer + Send + Sync + 'static,
-    {
-        self.timer = Time::Timer(ArcTimer::new(timer));
-    }
-
-    /// Provide a options configuration for the HTTP/2 connection.
-    #[inline]
-    pub fn options(&mut self, opts: Http2Options) {
-        self.opts = opts;
-    }
-
-    /// Constructs a connection with the configured options and IO.
-    ///
-    /// Note, if [`Connection`] is not `await`-ed, [`SendRequest`] will
-    /// do nothing.
-    pub async fn handshake<T, B>(self, io: T) -> Result<(SendRequest<B>, Connection<T, B, Ex>)>
-    where
-        T: AsyncRead + AsyncWrite + Unpin,
-        B: Body + 'static,
-        B::Data: Send,
-        B::Error: Into<BoxError>,
-        Ex: Http2ClientConnExec<B, T> + Unpin,
-    {
-        trace!("client handshake HTTP/2");
-
-        // Crate the HTTP/2 client with the provided options.
-        let builder = {
-            let mut builder = http2::client::Builder::default();
-            builder
-                .initial_max_send_streams(self.opts.initial_max_send_streams)
-                .initial_window_size(self.opts.initial_window_size)
-                .initial_connection_window_size(self.opts.initial_conn_window_size)
-                .max_send_buffer_size(self.opts.max_send_buffer_size);
-            if let Some(id) = self.opts.initial_stream_id {
-                builder.initial_stream_id(id);
-            }
-            if let Some(max) = self.opts.max_pending_accept_reset_streams {
-                builder.max_pending_accept_reset_streams(max);
-            }
-            if let Some(max) = self.opts.max_concurrent_reset_streams {
-                builder.max_concurrent_reset_streams(max);
-            }
-            if let Some(max) = self.opts.max_concurrent_streams {
-                builder.max_concurrent_streams(max);
-            }
-            if let Some(max) = self.opts.max_header_list_size {
-                builder.max_header_list_size(max);
-            }
-            if let Some(opt) = self.opts.enable_push {
-                builder.enable_push(opt);
-            }
-            if let Some(max) = self.opts.max_frame_size {
-                builder.max_frame_size(max);
-            }
-            if let Some(max) = self.opts.header_table_size {
-                builder.header_table_size(max);
-            }
-            if let Some(v) = self.opts.enable_connect_protocol {
-                builder.enable_connect_protocol(v);
-            }
-            if let Some(v) = self.opts.no_rfc7540_priorities {
-                builder.no_rfc7540_priorities(v);
-            }
-            if let Some(order) = self.opts.settings_order {
-                builder.settings_order(order);
-            }
-            if let Some(experimental_settings) = self.opts.experimental_settings {
-                builder.experimental_settings(experimental_settings);
-            }
-            if let Some(stream_dependency) = self.opts.headers_stream_dependency {
-                builder.headers_stream_dependency(stream_dependency);
-            }
-            if let Some(order) = self.opts.headers_pseudo_order {
-                builder.headers_pseudo_order(order);
-            }
-            if let Some(priority) = self.opts.priorities {
-                builder.priorities(priority);
-            }
-
-            builder
-        };
-
-        // Create the ping configuration for the connection.
-        let ping_config = ping::Config::new(
-            self.opts.adaptive_window,
-            self.opts.initial_window_size,
-            self.opts.keep_alive_interval,
-            self.opts.keep_alive_timeout,
-            self.opts.keep_alive_while_idle,
-        );
-
-        let (tx, rx) = dispatch::channel();
-        let h2 = proto::h2::client::handshake(io, rx, builder, ping_config, self.exec, self.timer)
-            .await?;
-        Ok((
-            SendRequest {
-                dispatch: tx.unbound(),
-            },
-            Connection {
-                inner: (PhantomData, h2),
-            },
-        ))
-    }
-}

data/vendor/wreq/src/client/core/conn.rs

@@ -1,11 +0,0 @@
-//! Lower-level client connection API.
-//!
-//! The types in this module are to provide a lower-level API based around a
-//! single connection. Connecting to a host, pooling connections, and the like
-//! are not handled at this level. This module provides the building blocks to
-//! customize those things externally.
-
-pub mod http1;
-pub mod http2;
-
-pub use super::dispatch::TrySendError;