1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

use crate::dependency::Dependency;
use crate::source::Source;
use crate::sources::file_source::FileSource;
use log::*;
use std::fs::File;
use std::path::Path;

#[derive(Clone, Debug)]
pub struct UrlSource {
    pub url: String,
}

impl Source for UrlSource {
    fn install(&self, dependency: &Dependency, destination: &Path) -> std::io::Result<()> {
        trace!("Downloading Url from {}", self.url);
        let file_name = format!("{}.zip", dependency.clone().name);
        let cached = crate::smaug::cache_dir().join(file_name);

        if cached.exists() {
            std::fs::remove_file(cached.clone())?;
        }

        trace!("Downloading package to {}", cached.display());
        std::fs::create_dir_all(cached.parent().unwrap())?;
        let mut file = File::create(cached.clone())?;
        let response = reqwest::blocking::get(self.url.as_str());

        match response {
            Err(..) => Err(std::io::Error::new(
                std::io::ErrorKind::NotFound,
                "Couldn't download file.",
            )),
            Ok(mut response) => {
                std::io::copy(&mut response, &mut file)?;
                FileSource { path: cached }.install(dependency, destination)
            }
        }
    }
}