wyrd-client 0.3.0

Shared Wyrd client: transport, authentication, and composed client capabilities.
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
56
57
58
59
60
61
62

//! Private server-authorized artifact download orchestration.

use std::path::Path;
use std::sync::Arc;

use crate::WyrdClient;
use crate::storage::WyrdStorageClient;
use wyrd_spec::ids::CardUid;
use wyrd_spec::registry::RelativeArtifactPath;
use wyrd_spec::storage::{DownloadInitRequest, DownloadInitResponse};

use crate::cards::{download_progress::DownloadProgressDisplay, error::RegistryEngineError};

/// Initialize, render, and execute one verified artifact download.
///
/// # Errors
/// Returns the existing registry or storage error when initialization, transfer,
/// local writing, or verification fails. The display's bar is removed before
/// the error is returned.
///
/// # Cancellation
/// Cancellation can leave a partial destination file for the caller's staging
/// owner to discard.
pub(crate) async fn download_artifact(
    client: &WyrdClient,
    storage: &WyrdStorageClient,
    card_uid: &CardUid,
    relative_path: &RelativeArtifactPath,
    dest: &Path,
    display: Arc<DownloadProgressDisplay>,
) -> Result<(), RegistryEngineError> {
    let response: DownloadInitResponse = client
        .http
        .request_json(
            reqwest::Method::POST,
            "/v1/cards/download/init",
            Some(&DownloadInitRequest {
                card_uid: card_uid.clone(),
                relative_path: relative_path.as_str().to_owned(),
                ttl_secs: None,
            }),
        )
        .await?;
    display.start(relative_path, Some(response.size_bytes));
    let path = relative_path.clone();
    let progress_display = Arc::clone(&display);
    let result = storage
        .download_verified_with_progress(
            &response.plan,
            dest,
            &response.sha256,
            response.size_bytes,
            Arc::new(move |downloaded_bytes, total_bytes| {
                progress_display.set_position(&path, downloaded_bytes, total_bytes);
            }),
        )
        .await
        .map(|_| ())
        .map_err(RegistryEngineError::from);
    display.finish(relative_path, result.is_ok());
    result
}