augent 0.6.6

Lean package manager for various AI coding platforms
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

//! Progress bar display for installations

use indicatif::{ProgressBar, ProgressStyle};

/// Progress display for installations
pub struct ProgressDisplay {
    /// Main progress bar for bundle installation
    bundle_pb: ProgressBar,
    /// Optional file progress bar (shown when installing files)
    file_pb: Option<ProgressBar>,
}

impl ProgressDisplay {
    /// Create a new progress display with total bundle count
    pub fn new(total_bundles: u64) -> Self {
        let bundle_style = ProgressStyle::default_bar()
            .template("[{bar:40.cyan/blue}] {pos}/{len} {msg}")
            .unwrap()
            .progress_chars("#>-");

        let bundle_pb = ProgressBar::new(total_bundles);
        bundle_pb.set_style(bundle_style);

        Self {
            bundle_pb,
            file_pb: None,
        }
    }

    /// Initialize file progress bar with total file count
    pub fn init_file_progress(&mut self, total_files: u64) {
        let file_style = ProgressStyle::default_bar()
            .template("  [{bar:40.green/yellow}] {pos}/{len} files {msg}")
            .unwrap()
            .progress_chars("█▉▊▋▌▍▎▏  ");

        let file_pb = ProgressBar::new(total_files);
        file_pb.set_style(file_style);
        self.file_pb = Some(file_pb);
    }

    /// Update to show current bundle being installed
    pub fn update_bundle(&self, bundle_name: &str, current: usize, total: usize) {
        let msg = format!("({}/{}) {}", current, total, bundle_name);
        self.bundle_pb.set_message(msg);
    }

    /// Increment bundle progress
    pub fn inc_bundle(&self) {
        self.bundle_pb.inc(1);
    }

    /// Update file progress
    pub fn update_file(&self, file_path: &str) {
        if let Some(ref file_pb) = self.file_pb {
            // Truncate long paths for display
            let display_path = if file_path.len() > 50 {
                format!("...{}", &file_path[file_path.len() - 47..])
            } else {
                file_path.to_string()
            };
            file_pb.set_message(display_path);
            file_pb.inc(1);
        }
    }

    /// Finish file progress
    pub fn finish_files(&self) {
        if let Some(ref file_pb) = self.file_pb {
            file_pb.finish();
        }
        self.bundle_pb.finish();
    }

    /// Abandon on error
    pub fn abandon(&self) {
        if let Some(ref file_pb) = self.file_pb {
            file_pb.abandon();
        }
        self.bundle_pb.abandon();
    }
}