shell-cell 1.7.1

mod loading;
mod stopping;
mod ui;

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

use self::{loading::LoadingState, stopping::StoppingState};
use crate::{
    buildkit::BuildKitD,
    cli::{MIN_FPS, terminal::Terminal},
};

pub enum App {
    Loading(LoadingState),
    Stopping(StoppingState),
    Exit,
}

impl App {
    pub fn run(
        buildkit: &BuildKitD,
        mut terminal: Option<Terminal>,
    ) -> color_eyre::Result<()> {
        // Define a timeout if only terminal is Some(_)
        let timeout = terminal.as_ref().map(|_| MIN_FPS);

        // First step
        let mut app = LoadingState::load(buildkit.clone());
        loop {
            // Check for state transitions
            if let App::Loading(state) = app {
                app = state.try_recv(timeout)?;
            }

            if let App::Stopping(ref mut state) = app
                && state.try_update(timeout)
            {
                app = App::Exit;
            }

            if matches!(app, App::Exit) {
                return Ok(());
            }

            // If terminal is None (non-interactive mode), use plain stdout prints
            // instead of the TUI renderer to report progress to the user.
            if let Some(terminal) = &mut terminal {
                terminal.draw(|f| {
                    f.render_widget(&app, f.area());
                })?;
                app = app.handle_key_event()?;
            }
        }
    }

    fn handle_key_event(mut self) -> color_eyre::Result<Self> {
        if event::poll(MIN_FPS)?
            && let Event::Key(key) = event::read()?
            && key.kind == KeyEventKind::Press
            && let KeyCode::Char('c' | 'd') = key.code
            && key.modifiers.contains(event::KeyModifiers::CONTROL)
        {
            self = App::Exit;
        }

        Ok(self)
    }
}