glim-tui 0.1.0

A TUI for monitoring GitLab CI/CD pipelines and projects
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

use std::{io, panic};


use crossterm::{
    event::{DisableMouseCapture, EnableMouseCapture},
    terminal::{self, EnterAlternateScreen, LeaveAlternateScreen},
};
use ratatui::Frame;
use ratatui::layout::Size;
use crate::event::{GlimEvent, EventHandler};
use crate::result::GlimError;
use crate::result::GlimError::GeneralError;

pub type CrosstermTerminal =
    ratatui::Terminal<ratatui::backend::CrosstermBackend<io::Stdout>>;

/// Representation of a terminal user interface.
///
/// It is responsible for setting up the terminal,
/// initializing the interface and handling the draw events.
pub struct Tui {
    /// Interface to the Terminal.
    terminal: CrosstermTerminal,
    /// Terminal event handler.
    events: EventHandler,
}

impl Tui {
    pub fn new(terminal: CrosstermTerminal, events: EventHandler) -> Self {
        Self { terminal, events }
    }

    pub fn draw(
        &mut self,
        render_ui: impl FnOnce(&mut Frame),
    ) -> Result<(), GlimError> {
        self.terminal.draw(render_ui)
            .map_err(|_| GeneralError("failed to draw UI".to_string()))?;
        Ok(())
    }

    pub fn size(&self) -> Size {
        self.terminal.size().unwrap()
    }

    /// iterates over all currently available events; waits
    /// until at least one event is available.
    pub fn receive_events<F>(&self, mut f: F)
        where F: FnMut(GlimEvent)
    {
        let mut apply_event = |e| match e {
            GlimEvent::ReceivedProjects(p) if p.is_empty() => (),
            GlimEvent::ReceivedPipelines(p) if p.is_empty() => (),
            GlimEvent::ReceivedJobs(_, _, j) if j.is_empty() => (),
            _ => f(e)
        };

        apply_event(self.events.next().unwrap());
        while let Some(event) = self.events.try_next() { apply_event(event) }
    }

    pub fn enter(&mut self) -> Result<(), GlimError> {
        terminal::enable_raw_mode()
            .map_err(|_| GeneralError("failed to initialize raw mode".to_string()))?;
        
        crossterm::execute!(
            io::stderr(),
            EnterAlternateScreen,
            EnableMouseCapture
        ).map_err(|_| GeneralError("failed to enter alternate screen".to_string()))?;

        // Define a custom panic hook to reset the terminal properties.
        // This way, you won't have your terminal messed up if an unexpected error happens.
        let panic_hook = panic::take_hook();
        panic::set_hook(Box::new(move |panic| {
            Self::reset().expect("failed to reset the terminal");
            panic_hook(panic);
        }));

        self.terminal.hide_cursor()
            .map_err(|_| GeneralError("failed to hide cursor".to_string()))?;
        self.terminal.clear()
            .map_err(|_| GeneralError("failed to clear the screen".to_string()))?;
        Ok(())
    }

    fn reset() -> Result<(), GlimError> {
        terminal::disable_raw_mode()
            .map_err(|_| GeneralError("failed to disable raw mode".to_string()))?;
        crossterm::execute!(
            io::stderr(),
            LeaveAlternateScreen,
            DisableMouseCapture
        ).map_err(|_| GeneralError("failed to leave alternate screen".to_string()))?;
        
        Ok(())
    }

    pub fn exit(&mut self) -> Result<(), GlimError> {
        Self::reset()?;
        self.terminal.show_cursor()
            .map_err(|_| GeneralError("failed to show cursor".to_string()))?;
        Ok(())
    }
}