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
|
//! The `show-generation` subcommand.
use crate::chunk::ClientTrust;
use crate::client::BackupClient;
use crate::config::ClientConfig;
use crate::error::ObnamError;
use crate::fsentry::FilesystemKind;
use indicatif::HumanBytes;
use structopt::StructOpt;
use tempfile::NamedTempFile;
use tokio::runtime::Runtime;
/// Show information about a generation.
#[derive(Debug, StructOpt)]
pub struct ShowGeneration {
/// Reference to the generation. Defaults to latest.
#[structopt(default_value = "latest")]
gen_id: String,
}
impl ShowGeneration {
/// Run the command.
pub fn run(&self, config: &ClientConfig) -> Result<(), ObnamError> {
let rt = Runtime::new()?;
rt.block_on(self.run_async(config))
}
async fn run_async(&self, config: &ClientConfig) -> Result<(), ObnamError> {
let temp = NamedTempFile::new()?;
let client = BackupClient::new(config)?;
let trust = client
.get_client_trust()
.await?
.or_else(|| Some(ClientTrust::new("FIXME", None, "".to_string(), vec![])))
.unwrap();
let genlist = client.list_generations(&trust);
let gen_id = genlist.resolve(&self.gen_id)?;
let gen = client.fetch_generation(&gen_id, temp.path()).await?;
let mut files = gen.files()?;
let mut files = files.iter()?;
let total_bytes = files.try_fold(0, |acc, file| {
file.map(|(_, e, _, _)| {
if e.kind() == FilesystemKind::Regular {
acc + e.len()
} else {
acc
}
})
});
let total_bytes = total_bytes?;
println!("generation-id: {}", gen_id);
println!("file-count: {}", gen.file_count()?);
println!("file-bytes: {}", HumanBytes(total_bytes));
println!("file-bytes-raw: {}", total_bytes);
Ok(())
}
}
|
