git-barber 0.1.0

Trim stale merged git branches (classic + squash merges), with a TUI
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

mod app;
mod ui;

use std::process::ExitCode;
use std::time::Duration;

use anyhow::Result;
use ratatui::crossterm::event::{self, Event, KeyCode, KeyEventKind, KeyModifiers};

use crate::git::Git;
use crate::ops;
use crate::scan::Scan;
use app::{Action, App};

/// Run the interactive TUI. `ratatui::try_init` installs a panic hook that
/// restores the terminal before any panic message prints.
pub fn run(git: &dyn Git, scan: Scan, preselect_remote: bool, now_unix: i64) -> Result<ExitCode> {
    let mut app = App::new(scan, preselect_remote, now_unix);
    let mut terminal = match ratatui::try_init() {
        Ok(terminal) => terminal,
        Err(err) => {
            // try_init can fail AFTER enabling raw mode (alternate screen or
            // size query); restore is idempotent and saves the user's shell.
            ratatui::restore();
            return Err(err.into());
        }
    };
    let result = event_loop(&mut terminal, &mut app, git);
    ratatui::restore();
    // The alternate screen is gone the moment we restore — replay the
    // outcomes (and their undo commands) onto the real terminal so they
    // survive the TUI. This also covers the error path.
    if !app.results.is_empty() {
        print!(
            "{}",
            crate::output::human_execute(&app.base.name, &app.warnings, &app.results)
        );
    }
    result
}

fn event_loop(
    terminal: &mut ratatui::DefaultTerminal,
    app: &mut App,
    git: &dyn Git,
) -> Result<ExitCode> {
    loop {
        terminal.draw(|f| ui::render(f, app))?;
        if event::poll(Duration::from_millis(200))?
            && let Event::Key(key) = event::read()?
            && key.kind == KeyEventKind::Press // Windows also reports Release/Repeat
            && let Some(Action::Execute(plans)) = app.update(key)
        {
            let base = app.base.clone();
            let current = ops::current_branch(git);
            for plan in plans {
                app.in_flight = Some(plan.candidate.name.clone());
                // Draw errors must not abort the batch: results (and undo
                // hints) matter more than one bad frame.
                let _ = terminal.draw(|f| ui::render(f, app));
                let result = ops::delete_one(git, &base, current.as_deref(), &plan);
                app.apply_result(result);
                app.in_flight = None;
                let _ = terminal.draw(|f| ui::render(f, app));
            }
            app.finish();
            // Keys typed impatiently during the batch must not dismiss the
            // results screen the instant it appears — but a Ctrl+C hammered
            // during a slow push still means "get me out".
            while event::poll(Duration::ZERO)? {
                if let Event::Key(key) = event::read()?
                    && key.kind == KeyEventKind::Press
                    && key.modifiers.contains(KeyModifiers::CONTROL)
                    && matches!(key.code, KeyCode::Char('c') | KeyCode::Char('C'))
                {
                    app.should_quit = true;
                }
            }
        }
        if app.should_quit {
            return Ok(if app.any_failed() {
                ExitCode::from(1)
            } else {
                ExitCode::SUCCESS
            });
        }
    }
}