liftinstall/src/http.rs

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

use hyper::header::ContentLength;

use reqwest;

use std::io::Read;

/// Downloads a text file from the specified URL.
pub fn download_text(url: &str) -> Result<String, String> {
    // TODO: Decrease check time
    let mut client = match reqwest::get(url) {
        Ok(v) => v,
        Err(v) => return Err(format!("Failed to GET resource: {:?}", v)),
    };

    client
        .text()
        .map_err(|v| format!("Failed to get text from resource: {:?}", v))
}

/// Streams a file from a HTTP server.
pub fn stream_file<F>(url: &str, mut callback: F) -> Result<(), String>
where
    F: FnMut(Vec<u8>, u64) -> (),
{
    // TODO: Decrease check time
    let mut client = match reqwest::get(url) {
        Ok(v) => v,
        Err(v) => return Err(format!("Failed to GET resource: {:?}", v)),
    };

    let size = {
        let size: Option<&ContentLength> = client.headers().get();
        match size {
            Some(&ContentLength(v)) => v,
            None => 0,
        }
    };

    let mut buf = [0 as u8; 8192];
    loop {
        let len = client.read(&mut buf);
        let len = match len {
            Ok(v) => v,
            Err(v) => return Err(format!("Failed to read resource: {:?}", v)),
        };

        if len == 0 {
            break;
        }

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

        callback(buf_copy, size);
    }

    Ok(())
}