matchmaker/ui/

display.rs

use cba::bait::TransformExt;
use ratatui::{
    layout::Constraint,
    text::{Line, Text},
    widgets::{Cell, Paragraph, Row, Table},
};

use crate::{
    config::{DisplayConfig, RowConnectionStyle},
    utils::{
        serde::StringOrVec,
        text::{wrap_line, wrap_text, wrapping_indicator},
    },
};
pub type HeaderTable = Vec<Vec<Line<'static>>>;
#[derive(Debug)]
pub struct DisplayUI {
    width: u16,
    height: u16,
    text: Vec<Text<'static>>,
    lines: HeaderTable, // lines from input
    pub show: bool,
    pub config: DisplayConfig,
}

impl DisplayUI {
    pub fn new(config: DisplayConfig) -> Self {
        let (text, height) = match &config.content {
            Some(StringOrVec::String(s)) => {
                let text = Text::from(s.clone());
                let height = text.height() as u16;
                (vec![text], height)
            }
            Some(StringOrVec::Vec(s)) => {
                let text: Vec<_> = s.iter().map(|s| Text::from(s.clone())).collect();
                let height = text.iter().map(|t| t.height()).max().unwrap_or_default() as u16;
                (text, height)
            }
            _ => (vec![], 0),
        };

        Self {
            height,
            width: 0,
            show: config.content.is_some() || config.header_lines > 0,
            lines: Vec::new(),
            text,
            config,
        }
    }

    pub fn update_width(&mut self, width: u16) {
        let border_w = self.config.border.width();
        let new_w = width.saturating_sub(border_w);
        self.width = new_w;
    }

    pub fn height(&self) -> u16 {
        if !self.show {
            return 0;
        }
        let mut height = self.height;
        height += self.config.border.height();

        height
    }

    /// Set text (single column) and show. The base style is applied "under" the text's styling.
    pub fn set(&mut self, text: impl Into<Text<'static>>) {
        self.text = vec![text.into()];

        self.show = true;
    }

    pub fn clear(&mut self, keep_header: bool) {
        if !keep_header {
            self.lines.clear();
            self.show = false;
        } else if self.lines.is_empty() {
            self.show = false;
        }

        self.text.clear();
    }

    /// Whether this is table has just one column
    pub fn is_single_column(&self) -> bool {
        self.text.len() == 1
    }

    pub fn header_table(&mut self, table: HeaderTable) {
        self.lines = table
    }

    // lowpri: how much to be gained by caching texts to not have to always rewrap?
    pub fn make_display(
        &mut self,
        result_indentation: u16,
        mut widths: Vec<u16>,
        col_spacing: u16,
    ) -> Table<'_> {
        if self.text.is_empty() && self.lines.is_empty() || widths.is_empty() {
            return Table::default();
        }

        let block = {
            let b = self.config.border.as_block();
            if self.config.match_indent {
                let mut padding = self.config.border.padding;

                padding.left = result_indentation.saturating_sub(self.config.border.left());
                widths[0] -= result_indentation;
                b.padding(padding.0)
            } else {
                b
            }
        };

        let (cells, height) = if self.is_single_column() {
            // Single Cell (Full Width)
            let text = wrap_text(
                self.text[0].clone(),
                if self.config.wrap { self.width } else { 0 },
            )
            .0;
            let cells = vec![Cell::from(text)];
            let height = self.text[0].height() as u16;

            (cells, height)
        } else
        // Multiple (multi-line) columns
        {
            let mut height = 0;
            // wrap text according to result column widths
            let cells = self
                .text
                .iter()
                .cloned()
                .zip(widths.iter().copied())
                .map(|(text, width)| {
                    let ret = wrap_text(text, if self.config.wrap { width } else { 0 }).0;
                    height = height.max(ret.height() as u16);

                    Cell::from(ret.transform_if(
                        matches!(self.config.row_connection, RowConnectionStyle::Disjoint),
                        |t| t.style(self.config.style),
                    ))
                })
                .collect();

            (cells, height)
        };

        let row = Row::new(cells).style(self.config.style).height(height);
        let mut rows = vec![row];
        self.height = height;

        // add header_line cells
        if !self.lines.is_empty() {
            // todo: support wrapping on header lines
            rows.extend(self.lines.iter().map(|row| {
                let cells: Vec<Cell> = row
                    .iter()
                    .cloned()
                    .enumerate()
                    .map(|(i, l)| {
                        wrap_line(
                            l,
                            self.config
                                .wrap
                                .then_some(widths.get(i).cloned())
                                .flatten()
                                .unwrap_or_default(),
                            &wrapping_indicator(),
                        )
                    })
                    .map(Cell::from)
                    .collect();
                Row::new(cells)
            }));

            self.height += self.lines.len() as u16;
        }

        let widths = if self.is_single_column() {
            vec![Constraint::Percentage(100)]
        } else {
            widths.into_iter().map(Constraint::Length).collect()
        };

        Table::new(rows, widths)
            .block(block)
            .column_spacing(col_spacing)
            .transform_if(
                !matches!(self.config.row_connection, RowConnectionStyle::Disjoint),
                |t| t.style(self.config.style),
            )
    }

    /// Draw in the same area as display when self.single() to produce a full width row over the table area
    pub fn make_full_width_row(&self, result_indentation: u16) -> Paragraph<'_> {
        // Compute padding
        let left = if self.config.match_indent {
            result_indentation.saturating_sub(self.config.border.left())
        } else {
            self.config.border.left()
        };
        let top = self.config.border.top();
        let right = self.config.border.width().saturating_sub(left);
        let bottom = self.config.border.height() - top;

        let block = ratatui::widgets::Block::default().padding(ratatui::widgets::Padding {
            left,
            top,
            right,
            bottom,
        });

        Paragraph::new(self.text[0].clone())
            .block(block)
            .style(self.config.style)
    }
}