From 49318b91f3f17567abd59a84eba0437f54d4f1e5 Mon Sep 17 00:00:00 2001
From: Maxim Zhiburt <zhiburt@gmail.com>
Date: Tue, 3 Dec 2024 01:35:08 +0300
Subject: [PATCH] update

---
 crates/nu-table/src/common.rs             |   2 +
 crates/nu-table/src/types/collapse.rs     |  21 ++-
 crates/nu-table/src/types/general.rs      |   2 +-
 crates/nu-table/src/unstructured_table.rs | 189 +++++++++++-----------
 4 files changed, 108 insertions(+), 106 deletions(-)

diff --git a/crates/nu-table/src/common.rs b/crates/nu-table/src/common.rs
index 1fe8eed0c1..e5733b98ba 100644
--- a/crates/nu-table/src/common.rs
+++ b/crates/nu-table/src/common.rs
@@ -64,6 +64,8 @@ pub fn nu_value_to_string(val: &Value, cfg: &Config, style: &StyleComputer) -> N
     make_styled_value(text, val, float_precision, style)
 }
 
+// todo: Expose a method which returns just style
+
 pub fn nu_value_to_string_clean(val: &Value, cfg: &Config, style_comp: &StyleComputer) -> NuText {
     let (text, style) = nu_value_to_string(val, cfg, style_comp);
     let mut text = clean_charset(&text);
diff --git a/crates/nu-table/src/types/collapse.rs b/crates/nu-table/src/types/collapse.rs
index 26102ac9af..9674ea2264 100644
--- a/crates/nu-table/src/types/collapse.rs
+++ b/crates/nu-table/src/types/collapse.rs
@@ -1,3 +1,4 @@
+use nu_ansi_term::Style;
 use nu_color_config::StyleComputer;
 use nu_protocol::{Config, Record, Value};
 use nu_utils::SharedCow;
@@ -46,10 +47,8 @@ fn colorize_value(value: &mut Value, config: &Config, style_computer: &StyleComp
                     .into_iter()
                     .map(|(mut header, mut val)| {
                         colorize_value(&mut val, config, style_computer);
+                        header = colorize_text(&header, style.color_style).unwrap_or(header);
 
-                        if let Some(color) = style.color_style {
-                            header = color.paint(header).to_string();
-                        }
                         (header, val)
                     })
                     .collect::<Record>(),
@@ -62,11 +61,19 @@ fn colorize_value(value: &mut Value, config: &Config, style_computer: &StyleComp
         }
         value => {
             let (text, style) = nu_value_to_string_clean(value, config, style_computer);
-            if let Some(color) = style.color_style {
-                let text = color.paint(text).to_string();
-                let span = value.span();
-                *value = Value::string(text, span);
+            if let Some(text) = colorize_text(&text, style.color_style) {
+                *value = Value::string(text, value.span());
             }
         }
     }
 }
+
+fn colorize_text(text: &str, color: Option<Style>) -> Option<String> {
+    if let Some(color) = color {
+        if !color.is_plain() {
+            return Some(color.paint(text).to_string());
+        }
+    }
+
+    None
+}
diff --git a/crates/nu-table/src/types/general.rs b/crates/nu-table/src/types/general.rs
index 9412d883e5..29f6d0467d 100644
--- a/crates/nu-table/src/types/general.rs
+++ b/crates/nu-table/src/types/general.rs
@@ -2,13 +2,13 @@ use nu_color_config::TextStyle;
 use nu_engine::column::get_columns;
 use nu_protocol::{Config, Record, ShellError, Value};
 
-use super::has_index;
 use crate::{
     clean_charset, colorize_space,
     common::{
         check_value, create_nu_table_config, get_empty_style, get_header_style, get_index_style,
         get_value_style, nu_value_to_string_colored, NuText, INDEX_COLUMN_NAME,
     },
+    types::has_index,
     NuRecordsValue, NuTable, StringResult, TableOpts, TableOutput, TableResult,
 };
 
diff --git a/crates/nu-table/src/unstructured_table.rs b/crates/nu-table/src/unstructured_table.rs
index fa9ace66dc..53e67e16c2 100644
--- a/crates/nu-table/src/unstructured_table.rs
+++ b/crates/nu-table/src/unstructured_table.rs
@@ -4,10 +4,10 @@ use nu_protocol::{Config, Record, Span, TableIndent, Value};
 use tabled::{
     grid::{
         ansi::ANSIStr,
-        config::{AlignmentHorizontal, Borders, CompactMultilineConfig},
+        config::{Borders, CompactMultilineConfig},
         dimension::{DimensionPriority, PoolTableDimension},
     },
-    settings::{Color, Padding, TableOption, Theme},
+    settings::{Alignment, Color, Padding, TableOption},
     tables::{PoolTable, TableValue},
 };
 
@@ -53,8 +53,8 @@ fn build_table(
     theme.set_horizontal_lines(Default::default());
 
     table.with(Padding::new(indent.left, indent.right, 0, 0));
-    table.with(SetRawStyle(theme));
-    table.with(SetAlignment(AlignmentHorizontal::Left));
+    table.with(*theme.get_borders());
+    table.with(Alignment::left());
     table.with(PoolTableDimension::new(
         DimensionPriority::Last,
         DimensionPriority::Last,
@@ -72,14 +72,6 @@ fn build_table(
     table.to_string()
 }
 
-fn build_table_with_border_color(mut table: PoolTable, color: Color) -> String {
-    // NOTE: We have this function presizely because of color_into_ansistr internals
-    // color must be alive  why we build table
-
-    table.with(SetBorderColor(color_into_ansistr(&color)));
-    table.to_string()
-}
-
 fn convert_nu_value_to_table_value(value: Value, config: &Config) -> TableValue {
     match value {
         Value::Record { val, .. } => build_vertical_map(val.into_owned(), config),
@@ -91,76 +83,74 @@ fn convert_nu_value_to_table_value(value: Value, config: &Config) -> TableValue
                 build_vertical_array(vals, config)
             }
         }
-        value => {
-            let mut text = value.to_abbreviated_string(config);
-            if string_width(&text) > 50 {
-                text = string_wrap(&text, 30, false);
-            }
-
-            TableValue::Cell(text)
-        }
+        value => build_string_value(value, config),
     }
 }
 
-fn build_vertical_map(record: Record, config: &Config) -> TableValue {
-    let mut rows = Vec::with_capacity(record.len());
-    for (key, value) in record {
-        let val = convert_nu_value_to_table_value(value, config);
-        let row = TableValue::Row(vec![TableValue::Cell(key), val]);
-        rows.push(row);
+fn build_string_value(value: Value, config: &Config) -> TableValue {
+    const MAX_STRING_WIDTH: usize = 50;
+    const WRAP_STRING_WIDTH: usize = 30;
+
+    let mut text = value.to_abbreviated_string(config);
+    if string_width(&text) > MAX_STRING_WIDTH {
+        text = string_wrap(&text, WRAP_STRING_WIDTH, false);
     }
 
-    let max_key_width = rows
+    TableValue::Cell(text)
+}
+
+fn build_vertical_map(record: Record, config: &Config) -> TableValue {
+    let max_key_width = record
         .iter()
-        .map(|row| match row {
-            TableValue::Row(list) => match &list[0] {
-                TableValue::Cell(key) => string_width(key),
-                _ => unreachable!(),
-            },
-            _ => unreachable!(),
-        })
+        .map(|(k, _)| string_width(k))
         .max()
         .unwrap_or(0);
 
-    rows.iter_mut().for_each(|row| {
-        match row {
-            TableValue::Row(list) => match &mut list[0] {
-                TableValue::Cell(key) => {
-                    let width = string_width(key);
-                    let rest = max_key_width - width;
-                    key.extend(std::iter::repeat(' ').take(rest));
-                }
-                _ => unreachable!(),
-            },
-            _ => unreachable!(),
-        };
-    });
+    let mut rows = Vec::with_capacity(record.len());
+    for (mut key, value) in record {
+        string_append_to_width(&mut key, max_key_width);
+
+        let value = convert_nu_value_to_table_value(value, config);
+
+        let row = TableValue::Row(vec![TableValue::Cell(key), value]);
+        rows.push(row);
+    }
 
     TableValue::Column(rows)
 }
 
+fn string_append_to_width(key: &mut String, max: usize) {
+    let width = string_width(key);
+    let rest = max - width;
+    key.extend(std::iter::repeat(' ').take(rest));
+}
+
 fn build_vertical_array(vals: Vec<Value>, config: &Config) -> TableValue {
     let map = vals
         .into_iter()
         .map(|val| convert_nu_value_to_table_value(val, config))
-        .collect::<Vec<_>>();
+        .collect();
 
     TableValue::Column(map)
 }
 
 fn is_valid_record(vals: &[Value]) -> bool {
-    let mut first_record: Option<&Record> = None;
-    for val in vals {
+    if vals.is_empty() {
+        return true;
+    }
+
+    let first_value = match &vals[0] {
+        Value::Record { val, .. } => val,
+        _ => return false,
+    };
+
+    for val in &vals[1..] {
         match val {
             Value::Record { val, .. } => {
-                if let Some(known) = first_record {
-                    let equal = known.columns().eq(val.columns());
-                    if !equal {
-                        return false;
-                    }
-                } else {
-                    first_record = Some(val)
-                };
+                let equal = val.columns().eq(first_value.columns());
+                if !equal {
+                    return false;
+                }
             }
             _ => return false,
         }
@@ -177,30 +167,41 @@ fn count_columns_in_record(vals: &[Value]) -> usize {
 }
 
 fn build_map_from_record(vals: Vec<Value>, config: &Config) -> TableValue {
-    let mut list = vec![];
+    // assumes that we have a valid record structure (checked by is_valid_record)
 
     let head = get_columns_in_record(&vals);
-    let count_columns = head.len();
+    let mut list = Vec::with_capacity(head.len());
     for col in head {
-        list.push(vec![TableValue::Cell(col)]);
+        list.push(TableValue::Column(vec![TableValue::Cell(col)]));
     }
 
     for val in vals {
-        match val {
-            Value::Record { val, .. } => {
-                for (i, (_key, val)) in val.into_owned().into_iter().take(count_columns).enumerate()
-                {
-                    let cell = convert_nu_value_to_table_value(val, config);
-                    list[i].push(cell);
-                }
-            }
-            _ => unreachable!(),
+        let val = get_as_record(val);
+        for (i, (_, val)) in val.into_owned().into_iter().enumerate() {
+            let value = convert_nu_value_to_table_value(val, config);
+            let list = get_table_value_column_mut(&mut list[i]);
+
+            list.push(value);
         }
     }
 
-    let columns = list.into_iter().map(TableValue::Column).collect::<Vec<_>>();
+    TableValue::Row(list)
+}
 
-    TableValue::Row(columns)
+fn get_table_value_column_mut(val: &mut TableValue) -> &mut Vec<TableValue> {
+    match val {
+        TableValue::Column(row) => row,
+        _ => {
+            unreachable!();
+        }
+    }
+}
+
+fn get_as_record(val: Value) -> nu_utils::SharedCow<Record> {
+    match val {
+        Value::Record { val, .. } => val,
+        _ => unreachable!(),
+    }
 }
 
 fn get_columns_in_record(vals: &[Value]) -> Vec<String> {
@@ -210,32 +211,6 @@ fn get_columns_in_record(vals: &[Value]) -> Vec<String> {
     }
 }
 
-struct SetRawStyle(Theme);
-
-impl<R, D> TableOption<R, CompactMultilineConfig, D> for SetRawStyle {
-    fn change(self, _: &mut R, cfg: &mut CompactMultilineConfig, _: &mut D) {
-        let borders = *self.0.get_borders();
-        cfg.set_borders(borders);
-    }
-}
-
-struct SetBorderColor(ANSIStr<'static>);
-
-impl<R, D> TableOption<R, CompactMultilineConfig, D> for SetBorderColor {
-    fn change(self, _: &mut R, cfg: &mut CompactMultilineConfig, _: &mut D) {
-        let borders = Borders::filled(self.0);
-        cfg.set_borders_color(borders);
-    }
-}
-
-struct SetAlignment(AlignmentHorizontal);
-
-impl<R, D> TableOption<R, CompactMultilineConfig, D> for SetAlignment {
-    fn change(self, _: &mut R, cfg: &mut CompactMultilineConfig, _: &mut D) {
-        cfg.set_alignment_horizontal(self.0);
-    }
-}
-
 fn truncate_table_value(
     value: &mut TableValue,
     has_vertical: bool,
@@ -341,6 +316,15 @@ fn truncate_table_value(
     }
 }
 
+fn build_table_with_border_color(mut table: PoolTable, color: Color) -> String {
+    // NOTE: We have this function presizely because of color_into_ansistr internals
+    // color must be alive  why we build table
+
+    let color = color_into_ansistr(&color);
+    table.with(SetBorderColor(color));
+    table.to_string()
+}
+
 fn color_into_ansistr(color: &Color) -> ANSIStr<'static> {
     // # SAFETY
     //
@@ -355,3 +339,12 @@ fn color_into_ansistr(color: &Color) -> ANSIStr<'static> {
 
     ANSIStr::new(prefix, suffix)
 }
+
+struct SetBorderColor(ANSIStr<'static>);
+
+impl<R, D> TableOption<R, CompactMultilineConfig, D> for SetBorderColor {
+    fn change(self, _: &mut R, cfg: &mut CompactMultilineConfig, _: &mut D) {
+        let borders = Borders::filled(self.0);
+        cfg.set_borders_color(borders);
+    }
+}