rpk 0.2.2

A lightweight, cross-platform cli package manager.
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
52
53
54

use std::{
    io::{BufWriter, Write},
    path::Path,
};

use ureq::{Middleware, MiddlewareNext, Request, Response};
use url::Url;

use super::temp::TempFile;

pub struct BearerAuthMiddleware(pub Option<String>);

impl Middleware for BearerAuthMiddleware {
    fn handle(&self, request: Request, next: MiddlewareNext) -> Result<Response, ureq::Error> {
        let req = match &self.0 {
            Some(token) => request.set("Authorization", format!("Bearer {}", token).as_str()),
            None => request,
        };

        next.handle(req)
    }
}

pub trait UreqExt {
    fn download(&self, url: &Url, path: impl AsRef<Path>) -> anyhow::Result<()>;
}

impl UreqExt for ureq::Agent {
    fn download(&self, url: &Url, path: impl AsRef<Path>) -> anyhow::Result<()> {
        let mut reader = self.get(url.as_str()).call()?.into_reader();
        let mut tmp_file = TempFile::new_force(path.as_ref())?;
        {
            let mut writer = BufWriter::new(tmp_file.file());
            let mut buf = [0; 4 * 1024];
            loop {
                let nread = reader.read(&mut buf)?;
                // pb.inc(nread as u64);
                if nread == 0 {
                    break;
                }
                writer.write_all(&buf[..nread])?;
            }
            writer.flush()?;
        }
        tmp_file.persist()?;

        Ok(())
    }
}

pub fn http_get(url: Url) -> anyhow::Result<String> {
    let resp = ureq::get(url.as_str()).call()?;
    Ok(resp.into_string()?)
}