httm 0.50.1

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
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

//       ___           ___           ___           ___
//      /\__\         /\  \         /\  \         /\__\
//     /:/  /         \:\  \        \:\  \       /::|  |
//    /:/__/           \:\  \        \:\  \     /:|:|  |
//   /::\  \ ___       /::\  \       /::\  \   /:/|:|__|__
//  /:/\:\  /\__\     /:/\:\__\     /:/\:\__\ /:/ |::::\__\
//  \/__\:\/:/  /    /:/  \/__/    /:/  \/__/ \/__/~~/:/  /
//       \::/  /    /:/  /        /:/  /            /:/  /
//       /:/  /     \/__/         \/__/            /:/  /
//      /:/  /                                    /:/  /
//      \/__/                                     \/__/
//
// Copyright (c) 2023, 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 crate::interactive::preview::PreviewSelection;
use crate::library::results::HttmError;
use crate::{
    GLOBAL_CONFIG,
    HttmResult,
    exit_success,
};
use skim::prelude::*;
use std::io::Cursor;

pub enum ViewMode {
    Browse,
    Select(Option<String>),
    Restore,
    Prune,
}

pub enum MultiSelect {
    On,
    Off,
}

pub const TIEBREAK: &[RankCriteria] = &[
    RankCriteria::Score,
    RankCriteria::PathName,
    RankCriteria::Index,
    RankCriteria::NegLength,
    RankCriteria::Begin,
    RankCriteria::End,
];

impl ViewMode {
    pub fn print_header(&self) -> String {
        format!(
            "──────────────────────────────────────────────────────────────────────────────\n\
            EXIT:       esc      | SELECT:       enter      | SELECT, MULTIPLE: shift+tab\n\
            PREVIEW UP: shift+up | PREVIEW DOWN: shift+down | {}\n\
            PAGE UP:    page up  | PAGE DOWN:    page down \n",
            self.print_mode()
        )
    }

    fn print_mode(&self) -> &str {
        match self {
            ViewMode::Browse => "====> [ Browse Mode ] <====",
            ViewMode::Select(_) => "====> [ Select Mode ] <====",
            ViewMode::Restore => "====> [ Restore Mode ] <====",
            ViewMode::Prune => "====> [ Prune Mode ] <====",
        }
    }

    pub fn view_buffer(&self, buffer: &str, opt_multi: MultiSelect) -> HttmResult<Vec<String>> {
        let preview_selection = PreviewSelection::new(&self)?;

        let header = self.print_header();

        let opt_multi = match opt_multi {
            MultiSelect::On => true,
            MultiSelect::Off => false,
        };

        let mut opts = SkimOptionsBuilder::default();

        // build our browse view - less to do than before - no previews, looking through one 'lil buffer
        if let Some(command) = preview_selection.opt_preview_command() {
            opts.preview_window(preview_selection.opt_preview_window())
                .preview(command);
        };

        let skim_opts = opts
            .disabled(true)
            .tac(true)
            .no_sort(true)
            .tabstop(4usize)
            .exact(true)
            .multi(opt_multi)
            .regex(false)
            .tiebreak(TIEBREAK.to_vec())
            .header(header)
            .build()
            .expect("Could not initialized skim options for select_restore_view");

        let item_reader_opts = SkimItemReaderOption::default().ansi(true);
        let item_reader = SkimItemReader::new(item_reader_opts);

        let items = item_reader.of_bufread(Box::new(Cursor::new(buffer.to_owned())));

        // run_with() reads and shows items from the thread stream created above
        let res = match skim::Skim::run_with(skim_opts, Some(items)) {
            Ok(output) if output.is_abort => {
                eprintln!("httm select/restore/prune session was aborted.  Quitting.");
                exit_success();
            }
            Ok(output) => output
                .selected_items
                .iter()
                .map(|i| i.output().into_owned())
                .collect(),
            Err(_) => {
                return HttmError::new("httm select/restore/prune session failed.").into();
            }
        };

        if GLOBAL_CONFIG.opt_debug {
            if let Some(preview_command) = preview_selection.opt_preview_command() {
                eprintln!("DEBUG: Preview command executed: {}", preview_command)
            }
        }

        Ok(res)
    }
}