cloud-image-download 0.5.2

Downloads cloud images (qcow2 qemu files for instance) to a directory
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

/* Configuration management */
use crate::cli::Cli;
use crate::website::WebSite;
use config::Config;
use log::error;
use serde::Deserialize;
use std::process::exit;

/// Stores settings read from a configuration file.
#[derive(Debug, Deserialize)]
pub struct Settings {
    pub db_path: Option<String>,
    pub sites: Vec<WebSite>,
}

impl Settings {
    /// Deserializes (if possible) the whole configuration file that
    /// may have been specified in the command line
    #[must_use]
    pub fn from_config(cli: &Cli) -> Self {
        let config_filename = match shellexpand::full(&cli.config) {
            Ok(conf) => conf,
            Err(e) => {
                error!("Error expanding {}: {e}", cli.config);
                exit(1);
            }
        };

        let config = match Config::builder()
            .add_source(config::File::with_name(&config_filename).required(false))
            .add_source(config::Environment::with_prefix("CID_"))
            .build()
        {
            Ok(conf) => conf,
            Err(e) => {
                error!("Error: {e}");
                exit(1);
            }
        };

        let mut settings = match config.try_deserialize::<Settings>() {
            Ok(settings) => settings,
            Err(e) => {
                error!("Error deserializing: {e}");
                exit(1);
            }
        };

        // To give the command line option the latest word:
        // overwrite db_path setting if one was provided with
        // the cli
        if let Some(db_path) = &cli.db_path {
            settings.db_path = Some(db_path.clone());
        }

        settings
    }
}

// Tests that the test configuration has been correctly parsed
#[test]
fn test_settings_from_config() {
    use clap_verbosity_flag::Verbosity;

    let cli = Cli {
        db_path: None,
        config: "test_data/cloud-image-download.toml".to_string(),
        verbose: Verbosity::new(0, 0),
        concurrent_downloads: 2,
        verify_skipped: false,
    };

    let settings = Settings::from_config(&cli);

    assert_eq!(settings.sites.len(), 4);
    assert_eq!(settings.db_path, Some("~/.cache/".to_string()));
}

// Tests cli precedence
#[test]
fn test_db_path_settings_from_config() {
    use clap_verbosity_flag::Verbosity;

    let cli = Cli {
        db_path: Some("/var/lib/cid".to_string()),
        config: "test_data/cloud-image-download.toml".to_string(),
        verbose: Verbosity::new(0, 0),
        concurrent_downloads: 2,
        verify_skipped: false,
    };

    let settings = Settings::from_config(&cli);
    assert_eq!(settings.db_path, Some("/var/lib/cid".to_string()));
}

// Tests default
#[test]
fn test_db_path_settings_default() {
    use clap_verbosity_flag::Verbosity;

    let cli = Cli {
        db_path: None,
        config: "test_data/no_db_path.toml".to_string(),
        verbose: Verbosity::new(0, 0),
        concurrent_downloads: 2,
        verify_skipped: false,
    };

    let settings = Settings::from_config(&cli);
    assert_eq!(settings.db_path, None);
}