use crate::text_value::sanitize_text;
use unicode_width::UnicodeWidthStr;
const COLUMN_GAP: &str = " ";
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum ColumnAlign {
Left,
Right,
}
#[must_use]
pub fn render_table<const N: usize>(
headers: &[&str; N],
rows: &[[String; N]],
alignments: &[ColumnAlign; N],
) -> String {
let widths = table_widths(headers, rows);
let mut lines = Vec::with_capacity(rows.len() + 2);
lines.push(render_table_row(headers, &widths, alignments));
lines.push(render_separator(&widths));
lines.extend(
rows.iter()
.map(|row| render_table_row(row, &widths, alignments)),
);
lines.join("\n")
}
#[must_use]
pub fn table_widths<const N: usize>(headers: &[&str; N], rows: &[[String; N]]) -> [usize; N] {
let mut widths = headers.map(|header| UnicodeWidthStr::width(sanitize_text(header).as_str()));
for row in rows {
for (index, cell) in row.iter().enumerate() {
widths[index] = widths[index].max(UnicodeWidthStr::width(sanitize_text(cell).as_str()));
}
}
widths
}
#[must_use]
pub fn render_table_row<const N: usize>(
row: &[impl AsRef<str>],
widths: &[usize; N],
alignments: &[ColumnAlign; N],
) -> String {
widths
.iter()
.zip(alignments)
.enumerate()
.map(|(index, (width, alignment))| {
let value = sanitize_text(row.get(index).map_or("", AsRef::as_ref));
pad_cell(&value, *width, *alignment)
})
.collect::<Vec<_>>()
.join(COLUMN_GAP)
.trim_end()
.to_string()
}
#[must_use]
pub fn render_separator<const N: usize>(widths: &[usize; N]) -> String {
widths
.iter()
.map(|width| "-".repeat(*width))
.collect::<Vec<_>>()
.join(COLUMN_GAP)
}
fn pad_cell(value: &str, width: usize, alignment: ColumnAlign) -> String {
let padding = " ".repeat(width.saturating_sub(UnicodeWidthStr::width(value)));
match alignment {
ColumnAlign::Left => format!("{value}{padding}"),
ColumnAlign::Right => format!("{padding}{value}"),
}
}