esdiag 0.16.4

Elastic Stack diagnostic collector and processor
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
55

// Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
// or more contributor license agreements. Licensed under the Elastic License 2.0;
// you may not use this file except in compliance with the Elastic License 2.0.

use eyre::{Result, eyre};
use url::Url;

use crate::receiver::archive::ArchiveBytesReceiver;

#[derive(Clone)]
pub struct UploadServiceDownloader {
    token: String,
    url: Url,
}

impl UploadServiceDownloader {
    /// Downloads a file from the Elastic Uploader service given a URL and token
    /// The URL format of `https://upload.elastic.co/...` will have been validated previously.
    pub fn download(self) -> Result<ArchiveBytesReceiver> {
        // Using block_in_place allows a synchronous file download inside an async runtime
        tokio::task::block_in_place(|| {
            let client = reqwest::blocking::Client::new();
            let mut headers = reqwest::header::HeaderMap::new();
            headers.insert("Authorization", reqwest::header::HeaderValue::from_str(&self.token)?);
            let request = client.get(self.url.clone()).headers(headers);
            let response = request.send()?;
            let bytes = response.bytes()?;
            tracing::debug!("Downloaded archive size: {} bytes", bytes.len());
            match bytes.len() {
                0 => Err(eyre!("Downloaded empty file, check upload link expiration")),
                _ => Ok(ArchiveBytesReceiver::try_from(bytes)?),
            }
        })
    }
}

impl TryFrom<Url> for UploadServiceDownloader {
    type Error = eyre::Report;

    fn try_from(url: Url) -> Result<Self> {
        let mut url = url.clone();
        let token = url.password().ok_or_else(|| eyre!("No token provided"))?.to_string();
        // Since token authentication is by header, clear provided username and password from the URL
        url.set_username("").ok();
        url.set_password(None).ok();
        tracing::info!("Downloading archive from {url}");
        Ok(Self { token, url })
    }
}

impl std::fmt::Display for UploadServiceDownloader {
    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        write!(f, "Elastic Uploader {}", self.url)
    }
}