initial commit

src/stream.rs

@@ -0,0 +1,34 @@
+use rustls;
+use std::io::Read;
+use std::io::Result;
+use std::io::Write;
+use std::net::TcpStream;
+
+pub enum Stream {
+    Http(TcpStream),
+    Https(rustls::ClientSession, TcpStream),
+}
+
+impl Read for Stream {
+    fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
+        match self {
+            Stream::Http(sock) => sock.read(buf),
+            Stream::Https(sess, sock) => rustls::Stream::new(sess, sock).read(buf),
+        }
+    }
+}
+
+impl Write for Stream {
+    fn write(&mut self, buf: &[u8]) -> Result<usize> {
+        match self {
+            Stream::Http(sock) => sock.write(buf),
+            Stream::Https(sess, sock) => rustls::Stream::new(sess, sock).write(buf),
+        }
+    }
+    fn flush(&mut self) -> Result<()> {
+        match self {
+            Stream::Http(sock) => sock.flush(),
+            Stream::Https(sess, sock) => rustls::Stream::new(sess, sock).flush(),
+        }
+    }
+}