upstream-rs 1.18.0

Fetch package updates directly from the source.
Documentation 
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
70
71
72
73
74
75

use crate::{
    application::operations::import_operation::{ImportKind, ImportOperation},
    application::output,
    services::{packaging::OperationProgressEvent, storage::package_storage::PackageStorage},
    utils::static_paths::UpstreamPaths,
};
use anyhow::Result;
use indicatif::{ProgressBar, ProgressDrawTarget, ProgressStyle};
use std::path::PathBuf;
use std::time::Duration;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ImportKindArg {
    Keys,
    Manifest,
    Snapshot,
}

impl From<ImportKindArg> for ImportKind {
    fn from(value: ImportKindArg) -> Self {
        match value {
            ImportKindArg::Keys => ImportKind::Keys,
            ImportKindArg::Manifest => ImportKind::Manifest,
            ImportKindArg::Snapshot => ImportKind::Snapshot,
        }
    }
}

fn render_import_progress(event: OperationProgressEvent) -> String {
    match event {
        OperationProgressEvent::Phase(phase) => phase.label().to_string(),
        OperationProgressEvent::Count { done, total } => format!("Importing ... {done}/{total}"),
        OperationProgressEvent::Warning(message) | OperationProgressEvent::Detail(message) => {
            message
        }
    }
}

pub async fn run_import(
    path: PathBuf,
    skip_failed: bool,
    import_as: Option<ImportKindArg>,
) -> Result<()> {
    let paths = UpstreamPaths::new()?;
    let mut package_storage = PackageStorage::new(&paths.config.packages_file)?;
    let mut import_op = ImportOperation::new(&mut package_storage, &paths);

    println!("{}", output::title("Import"));
    output::action_note(format!("Source: {}", path.display()));

    let pb = ProgressBar::new(0);
    pb.set_draw_target(ProgressDrawTarget::stderr_with_hz(10));
    pb.set_style(ProgressStyle::with_template("{spinner:.green} {msg}")?);
    pb.enable_steady_tick(Duration::from_millis(120));
    pb.set_message("Importing ...");

    let progress_pb = pb.clone();
    let mut progress_callback = Some(move |event: OperationProgressEvent| {
        progress_pb.set_message(render_import_progress(event));
    });

    import_op
        .import(
            &path,
            skip_failed,
            import_as.map(Into::into),
            &mut progress_callback,
        )
        .await?;

    pb.finish_and_clear();
    println!("{}", output::success("Import complete."));

    Ok(())
}