      1
      2
      3
      4
      5
      6
      7
      8
      9
     10
     11
     12
     13
     14
     15
     16
     17
     18
     19
     20
     21
     22
     23
     24
     25
     26
     27
     28
     29
     30
     31
     32
     33
     34
     35
     36
     37
     38
     39
     40
     41
     42
     43
     44
     45
     46
     47
     48
     49
     50
     51
     52
     53
     54
     55
     56
     57
     58
     59
     60
     61
     62
     63
     64
     65
     66
     67
     68
     69
     70
     71
     72
     73
     74
     75
     76
     77
     78
     79
     80
     81
     82
     83
     84
     85
     86
     87
     88
     89
     90
     91
     92
     93
     94
     95
     96
     97
     98
     99
    100
    101
    102
    103
    104
    105
    106
    107
    108
    109
    110
    111

    use std::fmt;
    use std::io;
    use std::io::IoSlice;
    use std::pin::Pin;
    use std::task::{Context, Poll};

    use hyper::client::connect::{Connected, Connection};
    use tokio::io::{AsyncRead, AsyncWrite, ReadBuf};
    pub use tokio_native_tls::TlsStream;

    /// A stream that might be protected with TLS.
    pub enum MaybeHttpsStream<T> {
        /// A stream over plain text.
        Http(T),
        /// A stream protected with TLS.
        Https(TlsStream<T>),
    }

    // ===== impl MaybeHttpsStream =====

    impl<T: fmt::Debug> fmt::Debug for MaybeHttpsStream<T> {
        fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
            match self {
                MaybeHttpsStream::Http(s) => f.debug_tuple("Http").field(s).finish(),
                MaybeHttpsStream::Https(s) => f.debug_tuple("Https").field(s).finish(),
            }
        }
    }

    impl<T> From<T> for MaybeHttpsStream<T> {
        fn from(inner: T) -> Self {
            MaybeHttpsStream::Http(inner)
        }
    }

    impl<T> From<TlsStream<T>> for MaybeHttpsStream<T> {
        fn from(inner: TlsStream<T>) -> Self {
            MaybeHttpsStream::Https(inner)
        }
    }

    impl<T: AsyncRead + AsyncWrite + Unpin> AsyncRead for MaybeHttpsStream<T> {
        #[inline]
        fn poll_read(
            self: Pin<&mut Self>,
            cx: &mut Context,
            buf: &mut ReadBuf,
        ) -> Poll<Result<(), io::Error>> {
            match Pin::get_mut(self) {
                MaybeHttpsStream::Http(s) => Pin::new(s).poll_read(cx, buf),
                MaybeHttpsStream::Https(s) => Pin::new(s).poll_read(cx, buf),
            }
        }
    }

    impl<T: AsyncWrite + AsyncRead + Unpin> AsyncWrite for MaybeHttpsStream<T> {
        #[inline]
        fn poll_write(
            self: Pin<&mut Self>,
            cx: &mut Context<'_>,
            buf: &[u8],
        ) -> Poll<Result<usize, io::Error>> {
            match Pin::get_mut(self) {
                MaybeHttpsStream::Http(s) => Pin::new(s).poll_write(cx, buf),
                MaybeHttpsStream::Https(s) => Pin::new(s).poll_write(cx, buf),
            }
        }

        fn poll_write_vectored(
            self: Pin<&mut Self>,
            cx: &mut Context<'_>,
            bufs: &[IoSlice<'_>],
        ) -> Poll<Result<usize, io::Error>> {
            match Pin::get_mut(self) {
                MaybeHttpsStream::Http(s) => Pin::new(s).poll_write_vectored(cx, bufs),
                MaybeHttpsStream::Https(s) => Pin::new(s).poll_write_vectored(cx, bufs),
            }
        }

        fn is_write_vectored(&self) -> bool {
            match self {
                MaybeHttpsStream::Http(s) => s.is_write_vectored(),
                MaybeHttpsStream::Https(s) => s.is_write_vectored(),
            }
        }

        #[inline]
        fn poll_flush(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), io::Error>> {
            match Pin::get_mut(self) {
                MaybeHttpsStream::Http(s) => Pin::new(s).poll_flush(cx),
                MaybeHttpsStream::Https(s) => Pin::new(s).poll_flush(cx),
            }
        }

        #[inline]
        fn poll_shutdown(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), io::Error>> {
            match Pin::get_mut(self) {
                MaybeHttpsStream::Http(s) => Pin::new(s).poll_shutdown(cx),
                MaybeHttpsStream::Https(s) => Pin::new(s).poll_shutdown(cx),
            }
        }
    }

    impl<T: AsyncRead + AsyncWrite + Connection + Unpin> Connection for MaybeHttpsStream<T> {
        fn connected(&self) -> Connected {
            match self {
                MaybeHttpsStream::Http(s) => s.connected(),
                MaybeHttpsStream::Https(s) => s.get_ref().get_ref().get_ref().connected(),
            }
        }
    }