use std::fmt::{self, Write};
use reifydb_type::{
util::unicode::UnicodeWidthStr,
value::frame::{column::FrameColumn, frame::Frame},
};
pub struct FrameRenderer;
impl FrameRenderer {
pub fn render_full(frame: &Frame) -> Result<String, fmt::Error> {
let mut output = String::new();
Self::render_full_to(frame, &mut output)?;
Ok(output)
}
pub fn render_without_row_numbers(frame: &Frame) -> Result<String, fmt::Error> {
let mut output = String::new();
Self::render_without_row_numbers_to(frame, &mut output)?;
Ok(output)
}
pub fn render_full_to(frame: &Frame, f: &mut dyn Write) -> fmt::Result {
Self::render_internal(frame, f, true)
}
pub fn render_without_row_numbers_to(frame: &Frame, f: &mut dyn Write) -> fmt::Result {
Self::render_internal(frame, f, false)
}
fn render_internal(frame: &Frame, f: &mut dyn Write, include_row_numbers: bool) -> fmt::Result {
let row_count = frame.first().map_or(0, |c| c.data.len());
let has_row_numbers = include_row_numbers && !frame.row_numbers.is_empty();
let has_created_at = !frame.created_at.is_empty();
let has_updated_at = !frame.updated_at.is_empty();
let col_count = frame.len()
+ if has_row_numbers {
1
} else {
0
} + if has_created_at {
1
} else {
0
} + if has_updated_at {
1
} else {
0
};
let column_order = Self::get_column_display_order(frame);
let mut col_widths = vec![0; col_count];
let mut sys_col_idx = 0;
if has_row_numbers {
col_widths[sys_col_idx] = Self::display_width("#rownum");
for row_num in &frame.row_numbers {
col_widths[sys_col_idx] =
col_widths[sys_col_idx].max(Self::display_width(&row_num.to_string()));
}
sys_col_idx += 1;
}
if has_created_at {
col_widths[sys_col_idx] = Self::display_width("#created_at");
for ts in &frame.created_at {
col_widths[sys_col_idx] =
col_widths[sys_col_idx].max(Self::display_width(&ts.to_string()));
}
sys_col_idx += 1;
}
if has_updated_at {
col_widths[sys_col_idx] = Self::display_width("#updated_at");
for ts in &frame.updated_at {
col_widths[sys_col_idx] =
col_widths[sys_col_idx].max(Self::display_width(&ts.to_string()));
}
sys_col_idx += 1;
}
let row_num_col_idx = sys_col_idx;
for (display_idx, &col_idx) in column_order.iter().enumerate() {
let col = &frame[col_idx];
let display_name = Self::escape_control_chars(&col.name);
col_widths[row_num_col_idx + display_idx] = Self::display_width(&display_name);
}
for row_numberx in 0..row_count {
for (display_idx, &col_idx) in column_order.iter().enumerate() {
let col = &frame[col_idx];
let s = Self::extract_string_value(col, row_numberx);
col_widths[row_num_col_idx + display_idx] =
col_widths[row_num_col_idx + display_idx].max(Self::display_width(&s));
}
}
for w in &mut col_widths {
*w += 2;
}
let sep = format!("+{}+", col_widths.iter().map(|w| "-".repeat(*w + 2)).collect::<Vec<_>>().join("+"));
writeln!(f, "{}", sep)?;
let mut header = Vec::new();
let mut sys_idx = 0;
if has_row_numbers {
let w = col_widths[sys_idx];
let name = "#rownum";
let pad = w - Self::display_width(name);
let l = pad / 2;
let r = pad - l;
header.push(format!(" {:left$}{}{:right$} ", "", name, "", left = l, right = r));
sys_idx += 1;
}
if has_created_at {
let w = col_widths[sys_idx];
let name = "#created_at";
let pad = w - Self::display_width(name);
let l = pad / 2;
let r = pad - l;
header.push(format!(" {:left$}{}{:right$} ", "", name, "", left = l, right = r));
sys_idx += 1;
}
if has_updated_at {
let w = col_widths[sys_idx];
let name = "#updated_at";
let pad = w - Self::display_width(name);
let l = pad / 2;
let r = pad - l;
header.push(format!(" {:left$}{}{:right$} ", "", name, "", left = l, right = r));
sys_idx += 1;
}
let _ = sys_idx;
for (display_idx, &col_idx) in column_order.iter().enumerate() {
let col = &frame[col_idx];
let w = col_widths[row_num_col_idx + display_idx];
let name = Self::escape_control_chars(&col.name);
let pad = w - Self::display_width(&name);
let l = pad / 2;
let r = pad - l;
header.push(format!(" {:left$}{}{:right$} ", "", name, "", left = l, right = r));
}
writeln!(f, "|{}|", header.join("|"))?;
writeln!(f, "{}", sep)?;
for row_numberx in 0..row_count {
let mut row = Vec::new();
let mut sys_idx = 0;
if has_row_numbers {
let w = col_widths[sys_idx];
let s = frame.row_numbers[row_numberx].to_string();
let pad = w - Self::display_width(&s);
let l = pad / 2;
let r = pad - l;
row.push(format!(" {:left$}{}{:right$} ", "", s, "", left = l, right = r));
sys_idx += 1;
}
if has_created_at {
let w = col_widths[sys_idx];
let s = frame.created_at[row_numberx].to_string();
let pad = w - Self::display_width(&s);
let l = pad / 2;
let r = pad - l;
row.push(format!(" {:left$}{}{:right$} ", "", s, "", left = l, right = r));
sys_idx += 1;
}
if has_updated_at {
let w = col_widths[sys_idx];
let s = frame.updated_at[row_numberx].to_string();
let pad = w - Self::display_width(&s);
let l = pad / 2;
let r = pad - l;
row.push(format!(" {:left$}{}{:right$} ", "", s, "", left = l, right = r));
sys_idx += 1;
}
let _ = sys_idx;
for (display_idx, &col_idx) in column_order.iter().enumerate() {
let col = &frame[col_idx];
let w = col_widths[row_num_col_idx + display_idx];
let s = Self::extract_string_value(col, row_numberx);
let pad = w - Self::display_width(&s);
let l = pad / 2;
let r = pad - l;
row.push(format!(" {:left$}{}{:right$} ", "", s, "", left = l, right = r));
}
writeln!(f, "|{}|", row.join("|"))?;
}
writeln!(f, "{}", sep)
}
fn display_width(s: &str) -> usize {
if s.contains('\n') {
s.lines().map(|line| line.width()).max().unwrap_or(0)
} else {
s.width()
}
}
fn escape_control_chars(s: &str) -> String {
s.replace('\n', "\\n").replace('\t', "\\t")
}
fn get_column_display_order(frame: &Frame) -> Vec<usize> {
(0..frame.len()).collect()
}
fn extract_string_value(col: &FrameColumn, row_numberx: usize) -> String {
let s = col.data.as_string(row_numberx);
Self::escape_control_chars(&s)
}
}