dirscan 1.1.0

A high performance tool for summarizing large directories or drives
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

use crate::walker::ClientState;
use console::style;
use indicatif::{HumanBytes, ProgressBar, ProgressStyle};
use jwalk::DirEntry;

use prettytable::{cell, row, table};
use std::fmt;
use std::path::PathBuf;
use std::time::{Duration, Instant};

pub struct WalkProgress {
    errors: u64,
    total: u64,
    total_size: u64,

    root: PathBuf,
    update_frequency: Duration,
    started: Instant,
    last_update: Instant,
}

impl WalkProgress {
    pub fn new(root: PathBuf) -> WalkProgress {
        let update_frequency = Duration::from_millis(500);
        let started = Instant::now();

        WalkProgress {
            errors: 0,
            total: 0,
            total_size: 0,

            root,
            update_frequency,
            started,
            last_update: started,
        }
    }

    pub fn create_progress_bar(&self) -> ProgressBar {
        let progress_bar = ProgressBar::new_spinner();
        progress_bar.set_style(
            ProgressStyle::default_spinner()
                .template("[{elapsed_precise}] Per sec: {per_sec:.cyan/blue} | {msg}"),
        );
        progress_bar
    }

    pub fn should_update(&self) -> bool {
        self.last_update.elapsed() > self.update_frequency
    }

    pub fn update(&mut self, progress_bar: &ProgressBar) {
        self.last_update = Instant::now();
        progress_bar.set_position(self.total);
        progress_bar.set_message(
            format!(
                "Files: {} | Size: {} | Errors: {}",
                style(self.total).green(),
                style(HumanBytes(self.total_size)).green(),
                style(self.errors).red(),
            )
            .as_ref(),
        );
    }

    pub fn record_progress(&mut self, item: &Result<DirEntry<((), ClientState)>, jwalk::Error>) {
        self.total += 1;
        match item {
            Err(_) => self.errors += 1,
            Ok(dir_entry) => match &dir_entry.client_state {
                Some(metadata) => {
                    self.total_size += metadata.size;
                }
                None => self.errors += 1,
            },
        }
    }
}

impl std::fmt::Display for WalkProgress {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> std::fmt::Result {
        let elapsed = chrono::Duration::from_std(self.started.elapsed()).unwrap();
        let runtime = chrono_humanize::HumanTime::from(elapsed);
        let runtime_text = runtime.to_text_en(
            chrono_humanize::Accuracy::Precise,
            chrono_humanize::Tense::Present,
        );
        let errors = if self.errors > 0 {
            style(self.errors).red()
        } else {
            style(self.errors).green()
        };
        let mut table = table!(
            ["Root", style(self.root.display()).blue()],
            [
                "Total Size",
                style(indicatif::HumanBytes(self.total_size)).green()
            ],
            ["Files", style(self.total).green()],
            ["Duration", style(runtime_text).green()],
            ["Errors", errors]
        );
        table.set_format(*prettytable::format::consts::FORMAT_CLEAN);
        write!(f, "{}", table.to_string())
    }
}