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
63
64
65
66
67
68
69

use crate::asset_canister::batch::{commit_batch, create_batch};
use crate::asset_canister::list::list_assets;
use crate::asset_canister::protocol::{AssetDetails, BatchOperationKind};
use crate::operations::{
    create_new_assets, delete_incompatible_assets, set_encodings, unset_obsolete_encodings,
};
use crate::params::CanisterCallParams;
use crate::plumbing::{make_project_assets, AssetLocation, ProjectAsset};
use ic_utils::Canister;
use std::collections::HashMap;
use std::path::PathBuf;
use std::time::Duration;

/// Upload the specified files
pub async fn upload(
    canister: &Canister<'_>,
    timeout: Duration,
    files: HashMap<String, PathBuf>,
) -> anyhow::Result<()> {
    let asset_locations: Vec<AssetLocation> = files
        .iter()
        .map(|x| AssetLocation {
            source: x.1.clone(),
            key: x.0.clone(),
        })
        .collect();

    let canister_call_params = CanisterCallParams { canister, timeout };

    let container_assets = list_assets(&canister_call_params).await?;

    println!("Starting batch.");

    let batch_id = create_batch(&canister_call_params).await?;

    println!("Staging contents of new and changed assets:");

    let project_assets = make_project_assets(
        &canister_call_params,
        &batch_id,
        asset_locations,
        &container_assets,
    )
    .await?;

    let operations = assemble_upload_operations(project_assets, container_assets);

    println!("Committing batch.");

    commit_batch(&canister_call_params, &batch_id, operations).await?;

    Ok(())
}

fn assemble_upload_operations(
    project_assets: HashMap<String, ProjectAsset>,
    container_assets: HashMap<String, AssetDetails>,
) -> Vec<BatchOperationKind> {
    let mut container_assets = container_assets;

    let mut operations = vec![];

    delete_incompatible_assets(&mut operations, &project_assets, &mut container_assets);
    create_new_assets(&mut operations, &project_assets, &container_assets);
    unset_obsolete_encodings(&mut operations, &project_assets, &container_assets);
    set_encodings(&mut operations, project_assets);

    operations
}