matchmaker/ui/

mod.rs

mod display;
mod input;
mod overlay;
mod preview;
mod results;
pub use display::*;
pub use input::*;
pub use overlay::*;
pub use preview::*;
pub use results::*;

pub use ratatui::{
    Frame,
    layout::{Constraint, Direction, Layout, Rect},
    widgets::Table,
}; // reexport for convenience

use crate::{
    SSS, Selection, Selector,
    config::{
        DisplayConfig, InputConfig, PreviewLayout, RenderConfig, ResultsConfig, StatusConfig,
        TerminalLayoutSettings, UiConfig,
    },
    nucleo::Worker,
    preview::Preview,
    render::Click,
    tui::Tui,
};
// UI
pub struct UI {
    pub layout: Option<TerminalLayoutSettings>,
    area: Rect, // unused
    pub config: UiConfig,
}

// requires columns > 1
impl UI {
    pub fn new<'a, T: SSS, S: Selection, W: std::io::Write>(
        mut config: RenderConfig,
        matcher: &'a mut nucleo::Matcher,
        worker: Worker<T>,
        selection_set: Selector<T, S>,
        view: Option<Preview>,
        tui: &mut Tui<W>,
        hidden_columns: Vec<bool>,
    ) -> (Self, PickerUI<'a, T, S>, DisplayUI, Option<PreviewUI>) {
        assert!(!worker.columns.is_empty());

        if config.results.reverse.is_none() {
            config.results.reverse = (
                tui.is_fullscreen() && tui.area.y < tui.area.height / 2
                // reverse if fullscreen + cursor is in lower half of the screen
            )
            .into()
        }

        let ui_area = [
            tui.area.width.saturating_sub(config.ui.border.width()),
            tui.area.height.saturating_sub(config.ui.border.height()),
        ];

        let area = Rect {
            x: tui.area.x + config.ui.border.left(),
            y: tui.area.y + config.ui.border.top(),
            width: ui_area[0],
            height: ui_area[1],
        };

        let ui = Self {
            layout: tui.config.layout.clone(),
            area,
            config: config.ui,
        };

        let mut picker = PickerUI::new(
            config.results,
            config.status,
            config.input,
            config.header,
            matcher,
            worker,
            selection_set,
        );
        picker.results.hidden_columns(hidden_columns);

        let preview = if let Some(view) = view {
            Some(PreviewUI::new(view, config.preview, ui_area))
        } else {
            None
        };

        let footer = DisplayUI::new(config.footer);

        (ui, picker, footer, preview)
    }

    pub fn update_dimensions(&mut self, area: Rect) {
        let border = &self.config.border;

        self.area = Rect {
            x: area.x + border.left(),
            y: area.y + border.top(),
            width: area.width.saturating_sub(border.width()),
            height: area.height.saturating_sub(border.height()),
        };
    }

    pub fn make_ui(&self) -> ratatui::widgets::Block<'_> {
        self.config.border.as_block()
    }

    pub fn area(&self) -> Rect {
        self.area
    }

    pub fn compute_area(&self, area: &Rect) -> Rect {
        Rect {
            x: area.x + self.config.border.left(),
            y: area.y + self.config.border.top(),
            width: area.width.saturating_sub(self.config.border.width()),
            height: area.height.saturating_sub(self.config.border.height()),
        }
    }

    pub fn full_area(&self) -> Rect {
        Rect {
            x: self.area.x - self.config.border.left(),
            y: self.area.y - self.config.border.top(),
            width: self.area.width + self.config.border.width(),
            height: self.area.height + self.config.border.height(),
        }
    }
}

pub struct PickerUI<'a, T: SSS, S: Selection> {
    pub results: ResultsUI,
    pub input: InputUI,
    pub header: DisplayUI,
    pub matcher: &'a mut nucleo::Matcher,
    pub selector: Selector<T, S>,
    pub worker: Worker<T>,
}

impl<'a, T: SSS, S: Selection> PickerUI<'a, T, S> {
    pub fn new(
        results_config: ResultsConfig,
        status_config: StatusConfig,
        input_config: InputConfig,
        header_config: DisplayConfig,
        matcher: &'a mut nucleo::Matcher,
        worker: Worker<T>,
        selections: Selector<T, S>,
    ) -> Self {
        Self {
            results: ResultsUI::new(results_config, status_config),
            input: InputUI::new(input_config),
            header: DisplayUI::new(header_config),
            matcher,
            selector: selections,
            worker,
        }
    }

    pub fn layout(&self, area: Rect) -> [Rect; 4] {
        let PickerUI {
            input,
            header,
            results,
            ..
        } = self;

        let mut constraints = [
            Constraint::Length(1 + input.config.border.height()), // input
            Constraint::Length(results.status_config.show as u16), // status
            Constraint::Length(header.height()),
            Constraint::Fill(1), // results
        ];

        if self.reverse() {
            constraints.reverse();
        }

        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .constraints(constraints)
            .split(area);

        std::array::from_fn(|i| {
            chunks[if self.reverse() {
                chunks.len() - i - 1
            } else {
                i
            }]
        })
    }
}

impl<'a, T: SSS, O: Selection> PickerUI<'a, T, O> {
    pub fn make_table(&mut self, click: &mut Click) -> (Table<'_>, u16) {
        let cursor_byte = self.input.byte_index(self.input.cursor() as usize);
        let active_column = self.worker.query.active_column_index(cursor_byte);

        let table = self.results.make_table(
            active_column,
            &mut self.worker,
            &mut self.selector,
            self.matcher,
            click,
        );
        let width = self.results.table_width();
        (table, width)
    }

    pub fn update(&mut self) {
        self.worker.find(&self.input.input);
    }
    pub fn tick(&mut self) {
        self.worker.find(&self.input.input);
    }

    // creation from UI ensures Some
    pub fn reverse(&self) -> bool {
        self.results.reverse()
    }
}

impl PreviewLayout {
    pub fn split(&self, area: Rect) -> [Rect; 2] {
        use crate::config::Side;
        use ratatui::layout::{Constraint, Direction, Layout};

        let direction = match self.side {
            Side::Left | Side::Right => Direction::Horizontal,
            Side::Top | Side::Bottom => Direction::Vertical,
        };

        let side_first = matches!(self.side, Side::Left | Side::Top);

        let total = if matches!(direction, Direction::Horizontal) {
            area.width
        } else {
            area.height
        };

        let min = if self.min < 0 {
            total.saturating_sub((-self.min) as u16)
        } else {
            self.min as u16
        };

        let max = if self.max < 0 {
            total.saturating_sub((-self.max) as u16)
        } else {
            self.max as u16
        };

        let side_size = if min <= max {
            self.percentage.compute_clamped(total, min, max)
        } else {
            log::error!("PreviewLayout min > max: {min} > {max}. Ignoring max.");
            self.percentage.compute_clamped(total, min, 0)
        };

        let side_constraint = Constraint::Length(side_size);

        let constraints = if side_first {
            [side_constraint, Constraint::Min(0)]
        } else {
            [Constraint::Min(0), side_constraint]
        };

        let chunks = Layout::default()
            .direction(direction)
            .constraints(constraints)
            .split(area);

        if side_first {
            [chunks[0], chunks[1]]
        } else {
            [chunks[1], chunks[0]]
        }
    }
}