ragegun 0.2.2

use std::env;
use std::io::Cursor;


use http_req::request;

const DOWNLOAD_BASE_URL: &str = "https://raw.githubusercontent.com/ragegun/ragegun-assets/master/";

const TOKENIZER: [&str; 1] = [
    "en_tokenizer.bin"
];

pub fn download_asset(asset: &str, out_path: &str) -> Result<(), http_req::error::Error> {
    let mut response = Vec::new();
    let mut writer = Cursor::new(&mut response);

    let url = format!("{}{}", DOWNLOAD_BASE_URL, asset);

    request::get(&url, &mut writer)?;

    // write file to outbuf
    std::fs::write(out_path, response)?;

    Ok(())
}

fn main() {
    let mut lexica = Vec::<&str>::new();

    #[cfg(feature = "age")]
    lexica.push("age.csv");

    #[cfg(feature = "affection")]
    lexica.push("affection.csv");

    #[cfg(feature = "distress")]
    lexica.push("distress.csv");

    #[cfg(feature = "emolex_all_languages")]
    lexica.push("emolex-full.csv");

    #[cfg(feature = "emolex_english")]
    lexica.push("emolex.csv");

    #[cfg(feature = "gender")]
    lexica.push("gender.csv");

    #[cfg(feature = "perma")]
    lexica.push("perma.csv");

    #[cfg(feature = "temporal")]
    lexica.push("temporal.csv");

    let dir = env::var("OUT_DIR").unwrap();

    for file in lexica.iter().chain(TOKENIZER.iter()) {
        let out_path = format!("{}/{}", &dir, file);

        // check if file exists
        if std::path::Path::new(&out_path).exists() {
            continue;
        }

        download_asset(file, &out_path).unwrap();
    }

    println!("cargo:rustc-env=ASSET_DIR={}", &dir);
}