httm 0.16.5

A CLI tool for viewing snapshot file versions on ZFS and btrfs datasets
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

//       ___           ___           ___           ___
//      /\__\         /\  \         /\  \         /\__\
//     /:/  /         \:\  \        \:\  \       /::|  |
//    /:/__/           \:\  \        \:\  \     /:|:|  |
//   /::\  \ ___       /::\  \       /::\  \   /:/|:|__|__
//  /:/\:\  /\__\     /:/\:\__\     /:/\:\__\ /:/ |::::\__\
//  \/__\:\/:/  /    /:/  \/__/    /:/  \/__/ \/__/~~/:/  /
//       \::/  /    /:/  /        /:/  /            /:/  /
//       /:/  /     \/__/         \/__/            /:/  /
//      /:/  /                                    /:/  /
//      \/__/                                     \/__/
//
// (c) Robert Swinford <robert.swinford<...at...>gmail.com>
//
// For the full copyright and license information, please view the LICENSE file
// that was distributed with this source code.

use std::collections::BTreeMap;

use itertools::Itertools;
use rayon::prelude::*;

use crate::config::generate::{Config, ExecMode};
use crate::data::paths::PathData;
use crate::library::results::HttmResult;
use crate::lookup::versions::select_search_datasets;
use crate::MostProximateAndOptAlts;

pub type MountsForFiles = BTreeMap<PathData, Vec<PathData>>;

pub fn get_mounts_for_files(config: &Config) -> HttmResult<MountsForFiles> {
    // we only check for phantom files in "mount for file" mode because
    // people should be able to search for deleted files in other modes
    let (non_phantom_files, phantom_files): (Vec<&PathData>, Vec<&PathData>) = config
        .paths
        .par_iter()
        .partition(|pathdata| pathdata.metadata.is_some());

    if !phantom_files.is_empty() && !matches!(config.exec_mode, ExecMode::SnapFileMount(_)) {
        eprintln!(
            "httm was unable to determine mount locations for all input files, \
        because the following files do not appear to exist: "
        );

        phantom_files
            .iter()
            .for_each(|pathdata| eprintln!("{:?}", pathdata.path_buf));
    }

    let mounts_for_files: MountsForFiles = non_phantom_files
        .into_iter()
        .map(|pathdata| {
            let datasets: Vec<MostProximateAndOptAlts> = config
                .dataset_collection
                .snaps_selected_for_search
                .get_value()
                .iter()
                .flat_map(|dataset_type| select_search_datasets(config, pathdata, dataset_type))
                .collect();
            (pathdata.clone(), datasets)
        })
        .into_group_map_by(|(pathdata, _snap_types_for_search)| pathdata.clone())
        .into_iter()
        .map(|(pathdata, vec_snap_types_for_search)| {
            let datasets: Vec<PathData> = vec_snap_types_for_search
                .into_iter()
                .flat_map(|(_proximate_mount, snap_types_for_search)| snap_types_for_search)
                .flat_map(|snap_types_for_search| snap_types_for_search.get_datasets_of_interest())
                .map(|path| PathData::from(path.as_path()))
                .rev()
                .collect();
            (pathdata, datasets)
        })
        .collect();

    Ok(mounts_for_files)
}