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
40
41
42
43
44
45
46
47
48
49
50
51

use color_eyre::eyre::{eyre, Result};
use reqwest::blocking::{RequestBuilder, Response};
use reqwest::IntoUrl;
use std::fs::File;
use std::path::Path;

pub struct Client {
    reqwest: reqwest::blocking::Client,
}

impl Client {
    pub fn new() -> Result<Self> {
        let reqwest = reqwest::blocking::ClientBuilder::new()
            .user_agent(format!("rtx/{}", env!("CARGO_PKG_VERSION")))
            .build()?;
        Ok(Self { reqwest })
    }

    pub fn get<U: IntoUrl>(&self, url: U) -> RequestBuilder {
        let url = url.into_url().unwrap();
        debug!("GET {}", url);
        self.reqwest.get(url)
    }

    pub fn get_text<U: IntoUrl>(&self, url: U) -> Result<String> {
        let url = url.into_url().unwrap();
        debug!("GET.txt {}", url);
        let resp = self.get(url).send()?;
        self.ensure_success(&resp)?;
        let text = resp.text()?;
        Ok(text)
    }

    pub fn download_file<U: IntoUrl>(&self, url: U, path: &Path) -> Result<()> {
        let url = url.into_url()?;
        debug!("Downloading {} to {}", &url, path.display());
        let mut resp = self.get(url).send()?;
        self.ensure_success(&resp)?;
        let mut file = File::create(path)?;
        resp.copy_to(&mut file)?;
        Ok(())
    }

    pub fn ensure_success(&self, resp: &Response) -> Result<()> {
        if resp.status().is_success() {
            Ok(())
        } else {
            Err(eyre!("HTTP error: {} on {}", resp.status(), resp.url()))
        }
    }
}