oxkerclone/src/ui/mod.rs

use anyhow::Result;
use crossterm::{
    event::{self, DisableMouseCapture, EnableMouseCapture, Event},
    execute,
    terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
};
use parking_lot::Mutex;
use std::{
    io,
    sync::{atomic::Ordering, Arc},
};
use std::{
    sync::atomic::AtomicBool,
    time::{Duration, Instant},
};
use tokio::sync::mpsc::Sender;
use tui::{
    backend::{Backend, CrosstermBackend},
    layout::{Constraint, Direction, Layout},
    Frame, Terminal,
};

mod color_match;
mod draw_blocks;
mod gui_state;

pub use self::color_match::*;
pub use self::gui_state::{GuiState, SelectablePanel};
use crate::{
    app_data::AppData, app_error::AppError, docker_data::DockerMessage,
    input_handler::InputMessages,
};
use draw_blocks::*;

/// Take control of the terminal in order to draw gui
pub async fn create_ui(
    app_data: Arc<Mutex<AppData>>,
    sender: Sender<InputMessages>,
    is_running: Arc<AtomicBool>,
    gui_state: Arc<Mutex<GuiState>>,
    docker_sx: Sender<DockerMessage>,
    update_duration: Duration,
) -> Result<()> {
    enable_raw_mode()?;
    let mut stdout = io::stdout();
    execute!(stdout, EnterAlternateScreen, EnableMouseCapture)?;
    let backend = CrosstermBackend::new(stdout);
    let mut terminal = Terminal::new(backend)?;

    let res = run_app(
        &mut terminal,
        app_data,
        sender,
        is_running,
        gui_state,
        docker_sx,
        update_duration,
    )
    .await;

    disable_raw_mode().unwrap();
    execute!(
        terminal.backend_mut(),
        LeaveAlternateScreen,
        DisableMouseCapture
    )?;
    terminal.show_cursor().unwrap();

    if let Err(err) = res {
        println!("{}", err);
    }
    Ok(())
}

/// Run a loop to draw the gui
async fn run_app<B: Backend>(
    terminal: &mut Terminal<B>,
    app_data: Arc<Mutex<AppData>>,
    sender: Sender<InputMessages>,
    is_running: Arc<AtomicBool>,
    gui_state: Arc<Mutex<GuiState>>,
    docker_sx: Sender<DockerMessage>,
    update_duration: Duration,
) -> Result<(), AppError> {
    let input_poll_rate = std::time::Duration::from_millis(75);

    // Check for docker connect errors before attempting to draw the gui
    let e = app_data.lock().get_error();
    if let Some(error) = e {
        if let AppError::DockerConnect = error {
            let mut seconds = 5;
            loop {
                if seconds < 1 {
                    is_running.store(false, Ordering::SeqCst);
                    break;
                }
                terminal
                    .draw(|f| draw_error(f, AppError::DockerConnect, Some(seconds)))
                    .unwrap();
                tokio::time::sleep(std::time::Duration::from_secs(1)).await;
                seconds -= 1;
            }
        }
    } else {
        let mut now = Instant::now();
        loop {
            terminal.draw(|f| ui(f, &app_data, &gui_state)).unwrap();
            if crossterm::event::poll(input_poll_rate).unwrap() {
                let event = event::read().unwrap();
                if let Event::Key(key) = event {
                    sender
                        .send(InputMessages::ButtonPress(key.code))
                        .await
                        .unwrap_or(());
                } else if let Event::Mouse(m) = event {
                    sender
                        .send(InputMessages::MouseEvent(m))
                        .await
                        .unwrap_or(());
                } else if let Event::Resize(_, _) = event {
                    gui_state.lock().clear_area_map();
                    terminal.autoresize().unwrap_or(());
                }
            }

            if now.elapsed() >= update_duration {
                docker_sx.send(DockerMessage::Update).await.unwrap();
                now = Instant::now();
            }

            if !is_running.load(Ordering::SeqCst) {
                break;
            }
        }
    }
    Ok(())
}

fn ui<B: Backend>(
    f: &mut Frame<'_, B>,
    app_data: &Arc<Mutex<AppData>>,
    gui_state: &Arc<Mutex<GuiState>>,
) {
    // set max height for container section, needs +4 to deal with docker commands list and borders
    let mut height = app_data.lock().get_container_len();
    if height < 12 {
        height += 4;
    } else {
        height = 12
    }

    let column_widths = app_data.lock().get_width();
    let has_containers = !app_data.lock().containers.items.is_empty();
    let has_error = app_data.lock().get_error();
    let log_index = app_data.lock().get_selected_log_index();
    let show_help = gui_state.lock().show_help;
    let info_text = gui_state.lock().info_box_text.clone();
    let loading_icon = gui_state.lock().get_loading();

    let whole_layout = Layout::default()
        .direction(Direction::Vertical)
        .constraints([Constraint::Min(1), Constraint::Min(100)].as_ref())
        .split(f.size());

    // Split into 3, containers+controls, logs, then graphs
    let upper_main = Layout::default()
        .direction(Direction::Vertical)
        .constraints([Constraint::Max(height as u16), Constraint::Percentage(50)].as_ref())
        .split(whole_layout[1]);

    let top_split = if has_containers {
        vec![Constraint::Percentage(90), Constraint::Percentage(10)]
    } else {
        vec![Constraint::Percentage(100)]
    };
    // Containers + docker commands
    let top_panel = Layout::default()
        .direction(Direction::Horizontal)
        .constraints(top_split.as_ref())
        .split(upper_main[0]);

    let lower_split = if has_containers {
        vec![Constraint::Percentage(75), Constraint::Percentage(25)]
    } else {
        vec![Constraint::Percentage(100)]
    };

    // Split into 3, containers+controls, logs, then graphs
    let lower_main = Layout::default()
        .direction(Direction::Vertical)
        .constraints(lower_split.as_ref())
        .split(upper_main[1]);

    draw_containers(
        app_data,
        top_panel[0],
        f,
        gui_state,
        &column_widths,
    );

    if has_containers {
        draw_commands(
            app_data,
            top_panel[1],
            f,
            gui_state,
            log_index,
        );
    }

    draw_logs(
        app_data,
        lower_main[0],
        f,
        gui_state,
        log_index,
        loading_icon.to_owned(),
    );

    draw_heading_bar(
        whole_layout[0],
        &column_widths,
        f,
        has_containers,
        loading_icon,
        show_help,
    );

    // only draw charts if there are containers
    if has_containers {
        draw_chart(f, lower_main[1], app_data, log_index);
    }

    if let Some(info) = info_text {
        draw_info(f, info);
    }

    // Check if error, and show popup if so
    if show_help {
        draw_help_box(f);
    }

    if let Some(error) = has_error {
        app_data.lock().show_error = true;
        draw_error(f, error, None);
    }
}