Fix table expand wrap in case no header is there (#7605)

ref #7598

To be honest I was not able to obtain such results in basic mode as you
@rgwood.
But I've got it in `table -e`.

So this must fix the `table -e` wrapping.

Could you verify if it got fixed?

Signed-off-by: Maxim Zhiburt <zhiburt@gmail.com>
This commit is contained in:
Maxim Zhiburt 2022-12-27 16:44:34 +03:00 committed by GitHub
parent 38fc42d352
commit 4f812a7f34
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 96 additions and 46 deletions

View File

@ -2,6 +2,7 @@ use lscolors::{LsColors, Style};
use nu_color_config::color_from_hex; use nu_color_config::color_from_hex;
use nu_color_config::{Alignment, StyleComputer, TextStyle}; use nu_color_config::{Alignment, StyleComputer, TextStyle};
use nu_engine::{column::get_columns, env_to_string, CallExt}; use nu_engine::{column::get_columns, env_to_string, CallExt};
use nu_protocol::TrimStrategy;
use nu_protocol::{ use nu_protocol::{
ast::{Call, PathMember}, ast::{Call, PathMember},
engine::{Command, EngineState, Stack, StateWorkingSet}, engine::{Command, EngineState, Stack, StateWorkingSet},
@ -550,13 +551,13 @@ fn build_expanded_table(
style_computer, style_computer,
); );
wrap_text(failed_value.0, remaining_width) wrap_text(failed_value.0, remaining_width, config)
} }
} }
} }
val => { val => {
let text = value_to_styled_string(&val, config, style_computer).0; let text = value_to_styled_string(&val, config, style_computer).0;
wrap_text(text, remaining_width) wrap_text(text, remaining_width, config)
} }
} }
}; };
@ -1041,6 +1042,10 @@ fn convert_to_table2<'a>(
} }
if !with_header { if !with_header {
if available_width >= ADDITIONAL_CELL_SPACE {
available_width -= PADDING_SPACE;
}
for (row, item) in input.into_iter().enumerate() { for (row, item) in input.into_iter().enumerate() {
if nu_utils::ctrl_c::was_pressed(&ctrlc) { if nu_utils::ctrl_c::was_pressed(&ctrlc) {
return Ok(None); return Ok(None);
@ -1145,7 +1150,7 @@ fn convert_to_table2<'a>(
} }
let value = create_table2_entry_basic(item, &header, head, config, style_computer); let value = create_table2_entry_basic(item, &header, head, config, style_computer);
let value = wrap_nu_text(value, available_width); let value = wrap_nu_text(value, available_width, config);
let value_width = string_width(&value.0); let value_width = string_width(&value.0);
column_width = max(column_width, value_width); column_width = max(column_width, value_width);
@ -1170,7 +1175,7 @@ fn convert_to_table2<'a>(
} }
let value = create_table2_entry_basic(item, &header, head, config, style_computer); let value = create_table2_entry_basic(item, &header, head, config, style_computer);
let value = wrap_nu_text(value, OK_CELL_CONTENT_WIDTH); let value = wrap_nu_text(value, OK_CELL_CONTENT_WIDTH, config);
let value = NuTable::create_cell(value.0, value.1); let value = NuTable::create_cell(value.0, value.1);
@ -1300,7 +1305,7 @@ fn create_table2_entry(
flatten_sep, flatten_sep,
width, width,
), ),
Err(_) => wrap_nu_text(error_sign(style_computer), width), Err(_) => wrap_nu_text(error_sign(style_computer), width, config),
} }
} }
_ => convert_to_table2_entry( _ => convert_to_table2_entry(
@ -1320,21 +1325,21 @@ fn error_sign(style_computer: &StyleComputer) -> (String, TextStyle) {
make_styled_string(style_computer, String::from(""), None, 0) make_styled_string(style_computer, String::from(""), None, 0)
} }
fn wrap_nu_text(mut text: NuText, width: usize) -> NuText { fn wrap_nu_text(mut text: NuText, width: usize, config: &Config) -> NuText {
if string_width(&text.0) <= width { if string_width(&text.0) <= width {
return text; return text;
} }
text.0 = nu_table::wrap_string(&text.0, width); text.0 = nu_table::string_wrap(&text.0, width, is_cfg_trim_keep_words(config));
text text
} }
fn wrap_text(text: String, width: usize) -> String { fn wrap_text(text: String, width: usize, config: &Config) -> String {
if string_width(&text) <= width { if string_width(&text) <= width {
return text; return text;
} }
nu_table::wrap_string(&text, width) nu_table::string_wrap(&text, width, is_cfg_trim_keep_words(config))
} }
#[allow(clippy::too_many_arguments)] #[allow(clippy::too_many_arguments)]
@ -1352,13 +1357,21 @@ fn convert_to_table2_entry(
) -> NuText { ) -> NuText {
let is_limit_reached = matches!(deep, Some(0)); let is_limit_reached = matches!(deep, Some(0));
if is_limit_reached { if is_limit_reached {
return wrap_nu_text(value_to_styled_string(item, config, style_computer), width); return wrap_nu_text(
value_to_styled_string(item, config, style_computer),
width,
config,
);
} }
match &item { match &item {
Value::Record { span, cols, vals } => { Value::Record { span, cols, vals } => {
if cols.is_empty() && vals.is_empty() { if cols.is_empty() && vals.is_empty() {
wrap_nu_text(value_to_styled_string(item, config, style_computer), width) wrap_nu_text(
value_to_styled_string(item, config, style_computer),
width,
config,
)
} else { } else {
let table = convert_to_table2( let table = convert_to_table2(
0, 0,
@ -1392,7 +1405,11 @@ fn convert_to_table2_entry(
(table, TextStyle::default()) (table, TextStyle::default())
} else { } else {
// error so back down to the default // error so back down to the default
wrap_nu_text(value_to_styled_string(item, config, style_computer), width) wrap_nu_text(
value_to_styled_string(item, config, style_computer),
width,
config,
)
} }
} }
} }
@ -1405,6 +1422,7 @@ fn convert_to_table2_entry(
wrap_nu_text( wrap_nu_text(
convert_value_list_to_string(vals, config, style_computer, flatten_sep), convert_value_list_to_string(vals, config, style_computer, flatten_sep),
width, width,
config,
) )
} else { } else {
let table = convert_to_table2( let table = convert_to_table2(
@ -1440,13 +1458,17 @@ fn convert_to_table2_entry(
} else { } else {
// error so back down to the default // error so back down to the default
wrap_nu_text(value_to_styled_string(item, config, style_computer), width) wrap_nu_text(
value_to_styled_string(item, config, style_computer),
width,
config,
)
} }
} }
} }
_ => { _ => {
let text = value_to_styled_string(item, config, style_computer); let text = value_to_styled_string(item, config, style_computer);
wrap_nu_text(text, width) wrap_nu_text(text, width, config)
} // unknown type. } // unknown type.
} }
} }
@ -1540,6 +1562,15 @@ fn convert_with_precision(val: &str, precision: usize) -> Result<String, ShellEr
Ok(format!("{:.prec$}", val_float, prec = precision)) Ok(format!("{:.prec$}", val_float, prec = precision))
} }
fn is_cfg_trim_keep_words(config: &Config) -> bool {
matches!(
config.trim_strategy,
TrimStrategy::Wrap {
try_to_keep_words: true
}
)
}
struct PagingTableCreator { struct PagingTableCreator {
head: Span, head: Span,
stream: ListStream, stream: ListStream,

View File

@ -1,7 +1,7 @@
use nu_color_config::{Alignment, StyleComputer, TextStyle}; use nu_color_config::{Alignment, StyleComputer, TextStyle};
use nu_engine::column::get_columns; use nu_engine::column::get_columns;
use nu_protocol::FooterMode;
use nu_protocol::{ast::PathMember, Config, ShellError, Span, TableIndexMode, Value}; use nu_protocol::{ast::PathMember, Config, ShellError, Span, TableIndexMode, Value};
use nu_protocol::{FooterMode, TrimStrategy};
use nu_table::{string_width, Table as NuTable, TableConfig, TableTheme}; use nu_table::{string_width, Table as NuTable, TableConfig, TableTheme};
use std::sync::Arc; use std::sync::Arc;
use std::{ use std::{
@ -225,13 +225,13 @@ fn build_expanded_table(
style_computer, style_computer,
); );
nu_table::wrap_string(&failed_value.0, remaining_width) wrap_nu_text(failed_value, remaining_width, config).0
} }
} }
} }
val => { val => {
let text = value_to_styled_string(&val, config, style_computer).0; let text = value_to_styled_string(&val, config, style_computer).0;
nu_table::wrap_string(&text, remaining_width) wrap_nu_text((text, TextStyle::default()), remaining_width, config).0
} }
} }
}; };
@ -489,7 +489,7 @@ fn convert_to_table2<'a>(
} }
let value = create_table2_entry_basic(item, &header, head, config, style_computer); let value = create_table2_entry_basic(item, &header, head, config, style_computer);
let value = wrap_nu_text(value, available_width); let value = wrap_nu_text(value, available_width, config);
let value_width = string_width(&value.0); let value_width = string_width(&value.0);
column_width = max(column_width, value_width); column_width = max(column_width, value_width);
@ -516,7 +516,7 @@ fn convert_to_table2<'a>(
} }
let value = create_table2_entry_basic(item, &header, head, config, style_computer); let value = create_table2_entry_basic(item, &header, head, config, style_computer);
let value = wrap_nu_text(value, OK_CELL_CONTENT_WIDTH); let value = wrap_nu_text(value, OK_CELL_CONTENT_WIDTH, config);
let value = NuTable::create_cell(value.0, value.1); let value = NuTable::create_cell(value.0, value.1);
@ -646,7 +646,7 @@ fn create_table2_entry(
flatten_sep, flatten_sep,
width, width,
), ),
Err(_) => wrap_nu_text(error_sign(style_computer), width), Err(_) => wrap_nu_text(error_sign(style_computer), width, config),
} }
} }
_ => convert_to_table2_entry( _ => convert_to_table2_entry(
@ -666,8 +666,8 @@ fn error_sign(style_computer: &StyleComputer) -> (String, TextStyle) {
make_styled_string(style_computer, String::from(""), None, 0) make_styled_string(style_computer, String::from(""), None, 0)
} }
fn wrap_nu_text(mut text: NuText, width: usize) -> NuText { fn wrap_nu_text(mut text: NuText, width: usize, config: &Config) -> NuText {
text.0 = nu_table::wrap_string(&text.0, width); text.0 = nu_table::string_wrap(&text.0, width, is_cfg_trim_keep_words(config));
text text
} }
@ -686,13 +686,21 @@ fn convert_to_table2_entry(
) -> NuText { ) -> NuText {
let is_limit_reached = matches!(deep, Some(0)); let is_limit_reached = matches!(deep, Some(0));
if is_limit_reached { if is_limit_reached {
return wrap_nu_text(value_to_styled_string(item, config, style_computer), width); return wrap_nu_text(
value_to_styled_string(item, config, style_computer),
width,
config,
);
} }
match &item { match &item {
Value::Record { span, cols, vals } => { Value::Record { span, cols, vals } => {
if cols.is_empty() && vals.is_empty() { if cols.is_empty() && vals.is_empty() {
wrap_nu_text(value_to_styled_string(item, config, style_computer), width) wrap_nu_text(
value_to_styled_string(item, config, style_computer),
width,
config,
)
} else { } else {
let table = convert_to_table2( let table = convert_to_table2(
0, 0,
@ -726,7 +734,11 @@ fn convert_to_table2_entry(
(table, TextStyle::default()) (table, TextStyle::default())
} else { } else {
// error so back down to the default // error so back down to the default
wrap_nu_text(value_to_styled_string(item, config, style_computer), width) wrap_nu_text(
value_to_styled_string(item, config, style_computer),
width,
config,
)
} }
} }
} }
@ -739,6 +751,7 @@ fn convert_to_table2_entry(
wrap_nu_text( wrap_nu_text(
convert_value_list_to_string(vals, config, style_computer, flatten_sep), convert_value_list_to_string(vals, config, style_computer, flatten_sep),
width, width,
config,
) )
} else { } else {
let table = convert_to_table2( let table = convert_to_table2(
@ -774,11 +787,19 @@ fn convert_to_table2_entry(
} else { } else {
// error so back down to the default // error so back down to the default
wrap_nu_text(value_to_styled_string(item, config, style_computer), width) wrap_nu_text(
value_to_styled_string(item, config, style_computer),
width,
config,
)
} }
} }
} }
_ => wrap_nu_text(value_to_styled_string(item, config, style_computer), width), // unknown type. _ => wrap_nu_text(
value_to_styled_string(item, config, style_computer),
width,
config,
), // unknown type.
} }
} }
@ -921,3 +942,12 @@ fn need_footer(config: &Config, count_records: u64) -> bool {
matches!(config.footer_mode, FooterMode::RowCount(limit) if count_records > limit) matches!(config.footer_mode, FooterMode::RowCount(limit) if count_records > limit)
|| matches!(config.footer_mode, FooterMode::Always) || matches!(config.footer_mode, FooterMode::Always)
} }
fn is_cfg_trim_keep_words(config: &Config) -> bool {
matches!(
config.trim_strategy,
TrimStrategy::Wrap {
try_to_keep_words: true
}
)
}

View File

@ -4,7 +4,7 @@ pub fn string_width(text: &str) -> usize {
tabled::papergrid::util::string_width_multiline_tab(text, 4) tabled::papergrid::util::string_width_multiline_tab(text, 4)
} }
pub fn wrap_string(text: &str, width: usize) -> String { pub fn string_wrap(text: &str, width: usize, keep_words: bool) -> String {
// todo: change me... // todo: change me...
// //
// well... it's not effitient to build a table to wrap a string, // well... it's not effitient to build a table to wrap a string,
@ -14,13 +14,17 @@ pub fn wrap_string(text: &str, width: usize) -> String {
return String::new(); return String::new();
} }
let wrap = if keep_words {
Width::wrap(width).keep_words()
} else {
Width::wrap(width)
};
Builder::from_iter([[text]]) Builder::from_iter([[text]])
.build() .build()
.with(Style::empty()) .with(Style::empty())
.with(Padding::zero()) .with(Padding::zero())
.with(Modify::new(Cell(0, 0)).with(Width::wrap(width))) .with(Modify::new(Cell(0, 0)).with(wrap))
.to_string()
.trim_end()
.to_string() .to_string()
} }
@ -39,18 +43,3 @@ pub fn string_truncate(text: &str, width: usize) -> String {
.with(Width::truncate(width)) .with(Width::truncate(width))
.to_string() .to_string()
} }
pub fn string_wrap(text: &str, width: usize) -> String {
// todo: change me...
if text.is_empty() {
return String::new();
}
Builder::from_iter([[text]])
.build()
.with(Style::empty())
.with(Padding::zero())
.with(Width::wrap(width))
.to_string()
}