Fix find -v command on tables (issue #9043) (#9159)

# Description
This PR fixes issue #9043 where find -v was returning empty tables
and/or wrong output.
It also refactors some big code chunks with repetitions into it's own
functions.

# User-Facing Changes

# Tests + Formatting
Unit tests added for asserting changes.

# After Submitting
This commit is contained in:
juanPabloMiceli 2023-05-11 15:39:21 -03:00 committed by GitHub
parent 39e51f1953
commit e735d0c561
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 286 additions and 298 deletions

View File

@ -124,7 +124,7 @@ impl Command for Find {
}), }),
}, },
Example { Example {
description: "Find value in records", description: "Find value in records using regex",
example: r#"[[version name]; ['0.1.0' nushell] ['0.1.1' fish] ['0.2.0' zsh]] | find -r "nu""#, example: r#"[[version name]; ['0.1.0' nushell] ['0.1.1' fish] ['0.2.0' zsh]] | find -r "nu""#,
result: Some(Value::List { result: Some(Value::List {
vals: vec![Value::test_record( vals: vec![Value::test_record(
@ -137,6 +137,51 @@ impl Command for Find {
span: Span::test_data(), span: Span::test_data(),
}), }),
}, },
Example {
description: "Find inverted values in records using regex",
example: r#"[[version name]; ['0.1.0' nushell] ['0.1.1' fish] ['0.2.0' zsh]] | find -r "nu" --invert"#,
result: Some(Value::List {
vals: vec![
Value::test_record(
vec!["version", "name"],
vec![
Value::test_string("0.1.1"),
Value::test_string("fish".to_string()),
],
),
Value::test_record(
vec!["version", "name"],
vec![
Value::test_string("0.2.0"),
Value::test_string("zsh".to_string()),
],
),
],
span: Span::test_data(),
}),
},
Example {
description: "Find value in list using regex",
example: r#"[["Larry", "Moe"], ["Victor", "Marina"]] | find -r "rr""#,
result: Some(Value::List {
vals: vec![Value::List {
vals: vec![Value::test_string("Larry"), Value::test_string("Moe")],
span: Span::test_data(),
}],
span: Span::test_data(),
}),
},
Example {
description: "Find inverted values in records using regex",
example: r#"[["Larry", "Moe"], ["Victor", "Marina"]] | find -r "rr" --invert"#,
result: Some(Value::List {
vals: vec![Value::List {
vals: vec![Value::test_string("Victor"), Value::test_string("Marina")],
span: Span::test_data(),
}],
span: Span::test_data(),
}),
},
Example { Example {
description: "Remove ANSI sequences from result", description: "Remove ANSI sequences from result",
example: "[[foo bar]; [abc 123] [def 456]] | find 123 | get bar | ansi strip", example: "[[foo bar]; [abc 123] [def 456]] | find 123 | get bar | ansi strip",
@ -144,29 +189,23 @@ impl Command for Find {
}, },
Example { Example {
description: "Find and highlight text in specific columns", description: "Find and highlight text in specific columns",
example: "[[col1 col2 col3]; [moe larry curly] [larry curly moe]] | find moe -c [col1 col3]", example:
"[[col1 col2 col3]; [moe larry curly] [larry curly moe]] | find moe -c [col1]",
result: Some(Value::List { result: Some(Value::List {
vals: vec![ vals: vec![Value::test_record(
Value::test_record(
vec!["col1".to_string(), "col2".to_string(), "col3".to_string()], vec!["col1".to_string(), "col2".to_string(), "col3".to_string()],
vec![ vec![
Value::test_string("\u{1b}[37m\u{1b}[0m\u{1b}[41;37mmoe\u{1b}[0m\u{1b}[37m\u{1b}[0m".to_string()), Value::test_string(
"\u{1b}[37m\u{1b}[0m\u{1b}[41;37mmoe\u{1b}[0m\u{1b}[37m\u{1b}[0m"
.to_string(),
),
Value::test_string("larry".to_string()), Value::test_string("larry".to_string()),
Value::test_string("curly".to_string()), Value::test_string("curly".to_string()),
]
),
Value::test_record(
vec!["col1".to_string(), "col2".to_string(), "col3".to_string()],
vec![
Value::test_string("larry".to_string()),
Value::test_string("curly".to_string()),
Value::test_string("\u{1b}[37m\u{1b}[0m\u{1b}[41;37mmoe\u{1b}[0m\u{1b}[37m\u{1b}[0m".to_string()),
]
),
], ],
)],
span: Span::test_data(), span: Span::test_data(),
}), }),
} },
] ]
} }
@ -229,27 +268,8 @@ fn find_with_regex(
input.filter( input.filter(
move |value| match value { move |value| match value {
Value::String { val, .. } => re.is_match(val.as_str()).unwrap_or(false) != invert, Value::String { val, .. } => re.is_match(val.as_str()).unwrap_or(false) != invert,
Value::Record { cols: _, vals, .. } => { Value::Record { vals, .. } | Value::List { vals, .. } => {
let matches: Vec<bool> = vals values_match_find(vals, &re, &config, invert)
.iter()
.map(|v| {
re.is_match(v.into_string(" ", &config).as_str())
.unwrap_or(false)
!= invert
})
.collect();
matches.iter().any(|b| *b)
}
Value::List { vals, .. } => {
let matches: Vec<bool> = vals
.iter()
.map(|v| {
re.is_match(v.into_string(" ", &config).as_str())
.unwrap_or(false)
!= invert
})
.collect();
matches.iter().any(|b| *b)
} }
_ => false, _ => false,
}, },
@ -257,6 +277,20 @@ fn find_with_regex(
) )
} }
fn values_match_find(values: &[Value], re: &Regex, config: &Config, invert: bool) -> bool {
match invert {
true => !record_matches_regex(values, re, config),
false => record_matches_regex(values, re, config),
}
}
fn record_matches_regex(values: &[Value], re: &Regex, config: &Config) -> bool {
values.iter().any(|v| {
re.is_match(v.into_string(" ", config).as_str())
.unwrap_or(false)
})
}
#[allow(clippy::too_many_arguments)] #[allow(clippy::too_many_arguments)]
fn highlight_terms_in_record_with_search_columns( fn highlight_terms_in_record_with_search_columns(
search_cols: &Vec<String>, search_cols: &Vec<String>,
@ -274,61 +308,39 @@ fn highlight_terms_in_record_with_search_columns(
search_cols.to_vec() search_cols.to_vec()
}; };
let mut output = vec![]; let mut output = vec![];
let mut potential_output = vec![];
let mut found_a_hit = false; // We iterate every column in the record and every search term for matches
for (cur_col, val) in cols.iter().zip(vals) { for (cur_col, val) in cols.iter().zip(vals) {
let val_str = val.into_string("", config); let val_str = val.into_string("", config);
let lower_val = val.into_string("", config).to_lowercase();
let mut term_added_to_output = false;
for term in terms { for term in terms {
let term_str = term.into_string("", config); let term_str = term.into_string("", config);
let lower_term = term.into_string("", config).to_lowercase(); let output_value =
if lower_val.contains(&lower_term) && cols_to_search.contains(cur_col) { if contains_ignore_case(&val_str, &term_str) && cols_to_search.contains(cur_col) {
found_a_hit = true; let (value_to_highlight, highlight_string_style) = if config.use_ls_colors {
term_added_to_output = true;
if config.use_ls_colors {
// Get the original LS_COLORS color // Get the original LS_COLORS color
let style = ls_colors.style_for_path(val_str.clone()); get_colored_value_and_string_style(ls_colors, &val_str, &string_style)
let ansi_style = style
.map(LsColors_Style::to_nu_ansi_term_style)
.unwrap_or_default();
let ls_colored_val = ansi_style.paint(&val_str).to_string();
let ansi_term_style = style
.map(to_nu_ansi_term_style)
.unwrap_or_else(|| string_style);
let hi =
match highlight_search_string(&ls_colored_val, &term_str, &ansi_term_style)
{
Ok(hi) => hi,
Err(_) => string_style.paint(term_str.to_string()).to_string(),
};
potential_output.push(Value::String {
val: hi,
span: *span,
});
} else { } else {
// No LS_COLORS support, so just use the original value // No LS_COLORS support, so just use the original value
let hi = match highlight_search_string(&val_str, &term_str, &string_style) { (val_str.clone(), string_style)
Ok(hi) => hi,
Err(_) => string_style.paint(term_str.to_string()).to_string(),
}; };
output.push(Value::String {
val: hi,
span: *span,
});
}
}
}
if !term_added_to_output {
potential_output.push(val.clone());
}
}
if found_a_hit { let highlighted_str = match highlight_search_string(
output.append(&mut potential_output); &value_to_highlight,
&term_str,
&highlight_string_style,
) {
Ok(highlighted_str) => highlighted_str,
Err(_) => string_style.paint(term_str).to_string(),
};
Value::String {
val: highlighted_str,
span: *span,
}
} else {
val.clone()
};
output.push(output_value);
}
} }
Value::Record { Value::Record {
@ -338,6 +350,28 @@ fn highlight_terms_in_record_with_search_columns(
} }
} }
fn get_colored_value_and_string_style(
ls_colors: &LsColors,
val_str: &str,
string_style: &Style,
) -> (String, Style) {
let style = ls_colors.style_for_path(val_str);
let ansi_style = style
.map(LsColors_Style::to_nu_ansi_term_style)
.unwrap_or_default();
let ls_colored_val = ansi_style.paint(val_str).to_string();
let ansi_term_style = style
.map(to_nu_ansi_term_style)
.unwrap_or_else(|| *string_style);
(ls_colored_val, ansi_term_style)
}
fn contains_ignore_case(string: &str, substring: &str) -> bool {
string.to_lowercase().contains(&substring.to_lowercase())
}
fn find_with_rest_and_highlight( fn find_with_rest_and_highlight(
engine_state: &EngineState, engine_state: &EngineState,
stack: &mut Stack, stack: &mut Stack,
@ -361,7 +395,6 @@ fn find_with_rest_and_highlight(
} }
}) })
.collect::<Vec<Value>>(); .collect::<Vec<Value>>();
let columns_to_search: Option<Vec<String>> = call.get_flag(&engine_state, stack, "columns")?;
let style_computer = StyleComputer::from_config(&engine_state, stack); let style_computer = StyleComputer::from_config(&engine_state, stack);
// Currently, search results all use the same style. // Currently, search results all use the same style.
@ -375,11 +408,13 @@ fn find_with_rest_and_highlight(
}; };
let ls_colors = get_ls_colors(ls_colors_env_str); let ls_colors = get_ls_colors(ls_colors_env_str);
let cols_to_search = match columns_to_search { let cols_to_search_in_map = match call.get_flag(&engine_state, stack, "columns")? {
Some(cols) => cols, Some(cols) => cols,
None => vec![], None => vec![],
}; };
let cols_to_search_in_filter = cols_to_search_in_map.clone();
match input { match input {
PipelineData::Empty => Ok(PipelineData::Empty), PipelineData::Empty => Ok(PipelineData::Empty),
PipelineData::Value(_, _) => input PipelineData::Value(_, _) => input
@ -387,7 +422,7 @@ fn find_with_rest_and_highlight(
move |mut x| match &mut x { move |mut x| match &mut x {
Value::Record { cols, vals, span } => { Value::Record { cols, vals, span } => {
highlight_terms_in_record_with_search_columns( highlight_terms_in_record_with_search_columns(
&cols_to_search, &cols_to_search_in_map,
cols, cols,
vals, vals,
span, span,
@ -403,69 +438,14 @@ fn find_with_rest_and_highlight(
)? )?
.filter( .filter(
move |value| { move |value| {
let lower_value = if let Ok(span) = value.span() { value_should_be_printed(
Value::string(value.into_string("", &filter_config).to_lowercase(), span) value,
} else { &filter_config,
value.clone() &lower_terms,
}; &span,
&cols_to_search_in_filter,
lower_terms.iter().any(|term| match value { invert,
Value::Bool { .. } )
| Value::Int { .. }
| Value::Filesize { .. }
| Value::Duration { .. }
| Value::Date { .. }
| Value::Range { .. }
| Value::Float { .. }
| Value::Block { .. }
| Value::Closure { .. }
| Value::Nothing { .. }
| Value::Error { .. } => lower_value
.eq(span, term, span)
.map_or(false, |val| val.is_true()),
Value::String { .. }
| Value::List { .. }
| Value::CellPath { .. }
| Value::CustomValue { .. } => term
.r#in(span, &lower_value, span)
.map_or(false, |val| val.is_true()),
Value::Record { vals, .. } => vals.iter().any(|val| {
if let Ok(span) = val.span() {
let lower_val = Value::string(
val.into_string("", &filter_config).to_lowercase(),
Span::test_data(),
);
term.r#in(span, &lower_val, span)
.map_or(false, |aval| aval.is_true())
} else {
term.r#in(span, val, span)
.map_or(false, |aval| aval.is_true())
}
}),
Value::LazyRecord { val, .. } => match val.collect() {
Ok(val) => match val {
Value::Record { vals, .. } => vals.iter().any(|val| {
if let Ok(span) = val.span() {
let lower_val = Value::string(
val.into_string("", &filter_config).to_lowercase(),
Span::test_data(),
);
term.r#in(span, &lower_val, span)
.map_or(false, |aval| aval.is_true())
} else {
term.r#in(span, val, span)
.map_or(false, |aval| aval.is_true())
}
}),
_ => false,
},
Err(_) => false,
},
Value::Binary { .. } => false,
Value::MatchPattern { .. } => false,
}) != invert
}, },
ctrlc, ctrlc,
), ),
@ -474,7 +454,7 @@ fn find_with_rest_and_highlight(
.map(move |mut x| match &mut x { .map(move |mut x| match &mut x {
Value::Record { cols, vals, span } => { Value::Record { cols, vals, span } => {
highlight_terms_in_record_with_search_columns( highlight_terms_in_record_with_search_columns(
&cols_to_search, &cols_to_search_in_map,
cols, cols,
vals, vals,
span, span,
@ -487,69 +467,14 @@ fn find_with_rest_and_highlight(
_ => x, _ => x,
}) })
.filter(move |value| { .filter(move |value| {
let lower_value = if let Ok(span) = value.span() { value_should_be_printed(
Value::string(value.into_string("", &filter_config).to_lowercase(), span) value,
} else { &filter_config,
value.clone() &lower_terms,
}; &span,
&cols_to_search_in_filter,
lower_terms.iter().any(|term| match value { invert,
Value::Bool { .. } )
| Value::Int { .. }
| Value::Filesize { .. }
| Value::Duration { .. }
| Value::Date { .. }
| Value::Range { .. }
| Value::Float { .. }
| Value::Block { .. }
| Value::Closure { .. }
| Value::Nothing { .. }
| Value::Error { .. } => lower_value
.eq(span, term, span)
.map_or(false, |value| value.is_true()),
Value::String { .. }
| Value::List { .. }
| Value::CellPath { .. }
| Value::CustomValue { .. } => term
.r#in(span, &lower_value, span)
.map_or(false, |value| value.is_true()),
Value::Record { vals, .. } => vals.iter().any(|val| {
if let Ok(span) = val.span() {
let lower_val = Value::string(
val.into_string("", &filter_config).to_lowercase(),
Span::test_data(),
);
term.r#in(span, &lower_val, span)
.map_or(false, |value| value.is_true())
} else {
term.r#in(span, val, span)
.map_or(false, |value| value.is_true())
}
}),
Value::LazyRecord { val, .. } => match val.collect() {
Ok(val) => match val {
Value::Record { vals, .. } => vals.iter().any(|val| {
if let Ok(span) = val.span() {
let lower_val = Value::string(
val.into_string("", &filter_config).to_lowercase(),
Span::test_data(),
);
term.r#in(span, &lower_val, span)
.map_or(false, |value| value.is_true())
} else {
term.r#in(span, val, span)
.map_or(false, |value| value.is_true())
}
}),
_ => false,
},
Err(_) => false,
},
Value::Binary { .. } => false,
Value::MatchPattern { .. } => false,
}) != invert
}), }),
ctrlc.clone(), ctrlc.clone(),
) )
@ -607,6 +532,96 @@ fn find_with_rest_and_highlight(
} }
} }
fn value_should_be_printed(
value: &Value,
filter_config: &Config,
lower_terms: &[Value],
span: &Span,
columns_to_search: &Vec<String>,
invert: bool,
) -> bool {
let lower_value = if let Ok(span) = value.span() {
Value::string(value.into_string("", filter_config).to_lowercase(), span)
} else {
value.clone()
};
let mut match_found = lower_terms.iter().any(|term| match value {
Value::Bool { .. }
| Value::Int { .. }
| Value::Filesize { .. }
| Value::Duration { .. }
| Value::Date { .. }
| Value::Range { .. }
| Value::Float { .. }
| Value::Block { .. }
| Value::Closure { .. }
| Value::Nothing { .. }
| Value::Error { .. } => term_equals_value(term, &lower_value, span),
Value::String { .. }
| Value::List { .. }
| Value::CellPath { .. }
| Value::CustomValue { .. } => term_contains_value(term, &lower_value, span),
Value::Record { cols, vals, .. } => {
record_matches_term(cols, vals, columns_to_search, filter_config, term, span)
}
Value::LazyRecord { val, .. } => match val.collect() {
Ok(val) => match val {
Value::Record { cols, vals, .. } => {
record_matches_term(&cols, &vals, columns_to_search, filter_config, term, span)
}
_ => false,
},
Err(_) => false,
},
Value::Binary { .. } => false,
Value::MatchPattern { .. } => false,
});
if invert {
match_found = !match_found;
}
match_found
}
fn term_contains_value(term: &Value, value: &Value, span: &Span) -> bool {
term.r#in(*span, value, *span)
.map_or(false, |value| value.is_true())
}
fn term_equals_value(term: &Value, value: &Value, span: &Span) -> bool {
term.eq(*span, value, *span)
.map_or(false, |value| value.is_true())
}
fn record_matches_term(
cols: &[String],
vals: &[Value],
columns_to_search: &Vec<String>,
filter_config: &Config,
term: &Value,
span: &Span,
) -> bool {
let cols_to_search = if columns_to_search.is_empty() {
cols.to_vec()
} else {
columns_to_search.to_vec()
};
cols.iter().zip(vals).any(|(col, val)| {
if !cols_to_search.contains(col) {
return false;
}
let lower_val = if val.span().is_ok() {
Value::string(
val.into_string("", filter_config).to_lowercase(),
Span::test_data(),
)
} else {
(*val).clone()
};
term_contains_value(term, &lower_val, span)
})
}
fn to_nu_ansi_term_style(style: &LsColors_Style) -> Style { fn to_nu_ansi_term_style(style: &LsColors_Style) -> Style {
fn to_nu_ansi_term_color(color: &LsColors_Color) -> Color { fn to_nu_ansi_term_color(color: &LsColors_Color) -> Color {
match *color { match *color {

View File

@ -1,123 +1,96 @@
use nu_test_support::fs::Stub::EmptyFile; use nu_test_support::nu;
use nu_test_support::playground::Playground;
use nu_test_support::{nu, pipeline};
#[test] #[test]
fn find_with_list_search_with_string() { fn find_with_list_search_with_string() {
let actual = nu!( let actual = nu!("[moe larry curly] | find moe | get 0");
cwd: ".", pipeline(
r#"
[moe larry curly] | find moe | get 0
"#
));
assert_eq!(actual.out, "moe"); assert_eq!(actual.out, "moe");
} }
#[test] #[test]
fn find_with_list_search_with_char() { fn find_with_list_search_with_char() {
let actual = nu!( let actual = nu!("[moe larry curly] | find l | to json -r");
cwd: ".", pipeline(
r#"
[moe larry curly] | find l | to json -r
"#
));
assert_eq!(actual.out, r#"["larry","curly"]"#); assert_eq!(actual.out, r#"["larry","curly"]"#);
} }
#[test] #[test]
fn find_with_list_search_with_number() { fn find_with_list_search_with_number() {
let actual = nu!( let actual = nu!("[1 2 3 4 5] | find 3 | get 0");
cwd: ".", pipeline(
r#"
[1 2 3 4 5] | find 3 | get 0
"#
));
assert_eq!(actual.out, "3"); assert_eq!(actual.out, "3");
} }
#[test] #[test]
fn find_with_string_search_with_string() { fn find_with_string_search_with_string() {
let actual = nu!( let actual = nu!("echo Cargo.toml | find toml");
cwd: ".", pipeline(
r#"
echo Cargo.toml | find toml
"#
));
assert_eq!(actual.out, "Cargo.toml"); assert_eq!(actual.out, "Cargo.toml");
} }
#[test] #[test]
fn find_with_string_search_with_string_not_found() { fn find_with_string_search_with_string_not_found() {
let actual = nu!( let actual = nu!("[moe larry curly] | find shemp | is-empty");
cwd: ".", pipeline(
r#"
[moe larry curly] | find shemp | is-empty
"#
));
assert_eq!(actual.out, "true"); assert_eq!(actual.out, "true");
} }
#[test] #[test]
fn find_with_filepath_search_with_string() { fn find_with_filepath_search_with_string() {
Playground::setup("filepath_test_1", |dirs, sandbox| { let actual =
sandbox.with_files(vec![ nu!(r#"["amigos.txt","arepas.clu","los.txt","tres.txt"] | find arep | to json -r"#);
EmptyFile("amigos.txt"),
EmptyFile("arepas.clu"),
EmptyFile("los.txt"),
EmptyFile("tres.txt"),
]);
let actual = nu!(
cwd: dirs.test(), pipeline(
r#"
ls
| get name
| find arep
| to json -r
"#
));
assert_eq!(actual.out, r#"["arepas.clu"]"#); assert_eq!(actual.out, r#"["arepas.clu"]"#);
})
} }
#[test] #[test]
fn find_with_filepath_search_with_multiple_patterns() { fn find_with_filepath_search_with_multiple_patterns() {
Playground::setup("filepath_test_2", |dirs, sandbox| { let actual =
sandbox.with_files(vec![ nu!(r#"["amigos.txt","arepas.clu","los.txt","tres.txt"] | find arep ami | to json -r"#);
EmptyFile("amigos.txt"),
EmptyFile("arepas.clu"),
EmptyFile("los.txt"),
EmptyFile("tres.txt"),
]);
let actual = nu!(
cwd: dirs.test(), pipeline(
r#"
ls
| get name
| find arep ami
| to json -r
"#
));
assert_eq!(actual.out, r#"["amigos.txt","arepas.clu"]"#); assert_eq!(actual.out, r#"["amigos.txt","arepas.clu"]"#);
})
} }
#[test] #[test]
fn find_takes_into_account_linebreaks_in_string() { fn find_takes_into_account_linebreaks_in_string() {
let actual = nu!( let actual = nu!(r#""atest\nanothertest\nnohit\n" | find a | length"#);
cwd: ".", pipeline(
r#"
"atest\nanothertest\nnohit\n" | find a | length
"#
));
assert_eq!(actual.out, "2"); assert_eq!(actual.out, "2");
} }
#[test]
fn find_with_regex_in_table_keeps_row_if_one_column_matches() {
let actual = nu!(
"[[name nickname]; [Maurice moe] [Laurence larry]] | find --regex ce | get name | to json -r"
);
assert_eq!(actual.out, r#"["Maurice","Laurence"]"#);
}
#[test]
fn inverted_find_with_regex_in_table_keeps_row_if_none_of_the_columns_matches() {
let actual = nu!(
"[[name nickname]; [Maurice moe] [Laurence larry]] | find --regex moe --invert | get name | to json -r"
);
assert_eq!(actual.out, r#"["Laurence"]"#);
}
#[test]
fn find_in_table_only_keep_rows_with_matches_on_selected_columns() {
let actual = nu!(
"[[name nickname]; [Maurice moe] [Laurence larry]] | find r --columns [nickname] | get name | to json -r"
);
assert!(actual.out.contains("Laurence"));
assert!(!actual.out.contains("Maurice"));
}
#[test]
fn inverted_find_in_table_keeps_row_if_none_of_the_selected_columns_matches() {
let actual = nu!(
"[[name nickname]; [Maurice moe] [Laurence larry]] | find r --columns [nickname] --invert | get name | to json -r"
);
assert_eq!(actual.out, r#"["Maurice"]"#);
}