downloader.rs

use super::download::Download;
use hyper::client::HttpConnector;
use hyper::rt::Future;
use hyper::{Body, Client};
use hyper_tls::HttpsConnector;
use native_tls::TlsConnector;
use tokio::runtime::Runtime;

pub type HttpClient = Client<HttpsConnector<HttpConnector>>;

pub struct Downloader {
    client: HttpClient,
    runtime: Runtime,
    downloads: Vec<Download>,
}

impl Downloader {
    pub fn new() -> Downloader {
        let runtime = Runtime::new().expect("Could not create tokio runtime");
        let tls = TlsConnector::new().unwrap();
        let http_connector = HttpConnector::new_with_handle(4, runtime.reactor().clone());
        let https_connector = HttpsConnector::from((http_connector, tls));
        let client = Client::builder()
            .http1_writev(false)
            .build::<_, Body>(https_connector);

        Downloader {
            client,
            runtime,
            downloads: Vec::new(),
        }
    }

    pub fn download(&mut self, uri: &str) {
        let download = Download::new(uri);
        self.downloads.push(download);
        let download = self.downloads.last_mut().unwrap();
        //             ^^^^^^^^^^^^^^^
        //             cannot infer an appropriate lifetime for lifetime
        //             parameter in function call due to conflicting
        //             requirements
        let future = download.make_future(&self.client);
        self.runtime.spawn(future);
    }

    pub fn shutdown(self) {
        self.runtime.shutdown_on_idle().wait().unwrap()
    }
}