liftinstall/src/http.rs

//! http.rs
//!
//! A simple wrapper around Hyper's HTTP client.

use reqwest::header::CONTENT_LENGTH;

use std::io::Read;
use std::time::Duration;

use reqwest::async::Client as AsyncClient;
use reqwest::Client;
use reqwest::StatusCode;

/// Asserts that a URL is valid HTTPS, else returns an error.
pub fn assert_ssl(url: &str) -> Result<(), String> {
    if url.starts_with("https://") {
        Ok(())
    } else {
        Err("Specified URL was not https".to_string())
    }
}

/// Builds a customised HTTP client.
pub fn build_client() -> Result<Client, String> {
    Client::builder()
        .timeout(Duration::from_secs(8))
        .build()
        .map_err(|x| format!("Unable to build client: {:?}", x))
}

/// Builds a customised async HTTP client.
pub fn build_async_client() -> Result<AsyncClient, String> {
    AsyncClient::builder()
        .timeout(Duration::from_secs(8))
        .build()
        .map_err(|x| format!("Unable to build client: {:?}", x))
}

/// Streams a file from a HTTP server.
pub fn stream_file<F>(
    url: &str,
    authorization: Option<String>,
    mut callback: F,
) -> Result<(), String>
where
    F: FnMut(Vec<u8>, u64) -> (),
{
    assert_ssl(url)?;

    let mut client = build_client()?.get(url);

    if let Some(auth) = authorization {
        client = client.header("Authorization", format!("Bearer {}", auth));
    }

    let mut client = client
        .send()
        .map_err(|x| format!("Failed to GET resource: {:?}", x))?;

    match client.status() {
        StatusCode::OK => {}
        StatusCode::TOO_MANY_REQUESTS => {
            return Err(
                "Your token has exceeded the number of daily allowable IP addresses. \
                 Please wait 24 hours and try again."
                    .to_string(),
            );
        }
        x => {
            return Err(format!("Bad status code: {:?}.", x));
        }
    }

    let size = match client.headers().get(CONTENT_LENGTH) {
        Some(ref v) => v
            .to_str()
            .map_err(|x| format!("Content length header was invalid: {:?}", x))?
            .parse()
            .map_err(|x| format!("Failed to parse content length: {:?}", x))?,
        None => 0,
    };

    let mut buf = [0 as u8; 8192];
    loop {
        let len = client
            .read(&mut buf)
            .map_err(|x| format!("Failed to read resource: {:?}", x))?;

        if len == 0 {
            break;
        }

        let buf_copy = &buf[0..len];
        let buf_copy = buf_copy.to_vec();

        callback(buf_copy, size);
    }

    Ok(())
}