Revert "Forbid sorting lists and record values of different types"
This reverts commit cab953b8bb
.
This commit is contained in:
parent
ec2e06c5a3
commit
dd44cd4d15
|
@ -162,7 +162,8 @@ impl Command for Sort {
|
||||||
}
|
}
|
||||||
Value::List { vals, .. } => {
|
Value::List { vals, .. } => {
|
||||||
let mut vec = vals.to_owned();
|
let mut vec = vals.to_owned();
|
||||||
crate::sort(&mut vec, call.head, insensitive, natural)?;
|
|
||||||
|
crate::sort(&mut vec, insensitive, natural)?;
|
||||||
|
|
||||||
if reverse {
|
if reverse {
|
||||||
vec.reverse()
|
vec.reverse()
|
||||||
|
|
|
@ -1,5 +1,4 @@
|
||||||
use alphanumeric_sort::compare_str;
|
use alphanumeric_sort::compare_str;
|
||||||
use itertools::Itertools;
|
|
||||||
use nu_engine::ClosureEval;
|
use nu_engine::ClosureEval;
|
||||||
use nu_protocol::{
|
use nu_protocol::{
|
||||||
ast::CellPath,
|
ast::CellPath,
|
||||||
|
@ -7,7 +6,7 @@ use nu_protocol::{
|
||||||
PipelineData, Record, ShellError, Span, Value,
|
PipelineData, Record, ShellError, Span, Value,
|
||||||
};
|
};
|
||||||
use nu_utils::IgnoreCaseExt;
|
use nu_utils::IgnoreCaseExt;
|
||||||
use std::{borrow::Borrow, cmp::Ordering, mem::Discriminant};
|
use std::cmp::Ordering;
|
||||||
|
|
||||||
// This module includes sorting functionality that is useful in sort-by and elsewhere.
|
// This module includes sorting functionality that is useful in sort-by and elsewhere.
|
||||||
// Eventually it would be nice to find a better home for it; sorting logic is only coupled
|
// Eventually it would be nice to find a better home for it; sorting logic is only coupled
|
||||||
|
@ -18,83 +17,23 @@ pub enum Comparator {
|
||||||
CellPath(CellPath),
|
CellPath(CellPath),
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO(rose) remove span, or explicitly ask for call.head span (depending on error impl)
|
pub fn sort(vec: &mut [Value], insensitive: bool, natural: bool) -> Result<(), ShellError> {
|
||||||
fn typecheck(
|
// to apply insensitive or natural sorting, all values must be strings
|
||||||
vals: &[Value],
|
let string_sort: bool = vec
|
||||||
span: Span,
|
|
||||||
insensitive: bool,
|
|
||||||
natural: bool,
|
|
||||||
) -> Result<(), ShellError> {
|
|
||||||
let variants: Vec<_> = vals
|
|
||||||
.iter()
|
.iter()
|
||||||
.filter(|val| !val.is_nothing())
|
.all(|value| matches!(value, &Value::String { .. }));
|
||||||
.map(|val| (val, std::mem::discriminant(val)))
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
match variants.iter().map(|(_, disc)| disc).all_equal_value() {
|
|
||||||
Ok(_) | Err(None) => (),
|
|
||||||
Err(Some((first, second))) => {
|
|
||||||
let first_val = variants
|
|
||||||
.iter()
|
|
||||||
.filter(|(_, disc)| std::ptr::eq(disc, first))
|
|
||||||
.at_most_one()
|
|
||||||
.unwrap() // TODO(rose)
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let second_val = variants
|
|
||||||
.iter()
|
|
||||||
.filter(|(_, disc)| std::ptr::eq(disc, second))
|
|
||||||
.at_most_one()
|
|
||||||
.unwrap()
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
// TODO(rose) replace with bespoke error
|
|
||||||
return Err(ShellError::OperatorMismatch {
|
|
||||||
op_span: span,
|
|
||||||
lhs_ty: first_val.0.get_type().to_string(),
|
|
||||||
lhs_span: first_val.0.span(),
|
|
||||||
rhs_ty: second_val.0.get_type().to_string(),
|
|
||||||
rhs_span: second_val.0.span(),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if insensitive || natural {
|
|
||||||
// does not seem like it is possible to get the discriminant without constructing a value :(
|
|
||||||
let string_disc: Discriminant<Value> = std::mem::discriminant(&Value::String {
|
|
||||||
val: String::new(),
|
|
||||||
internal_span: Span::unknown(),
|
|
||||||
});
|
|
||||||
if let Some((val, _)) = variants.iter().find(|(_, disc)| disc != &string_disc) {
|
|
||||||
return Err(ShellError::IncompatibleParametersSingle {
|
|
||||||
msg: "unable to use case insensitive or natural sorting with non-string values"
|
|
||||||
.to_string(),
|
|
||||||
span: val.span(),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn sort(
|
|
||||||
vals: &mut [Value],
|
|
||||||
span: Span,
|
|
||||||
insensitive: bool,
|
|
||||||
natural: bool,
|
|
||||||
) -> Result<(), ShellError> {
|
|
||||||
typecheck(vals, span, insensitive, natural)?;
|
|
||||||
|
|
||||||
// allow the comparator function to indicate error
|
// allow the comparator function to indicate error
|
||||||
// by mutating this option captured by the closure,
|
// by mutating this option captured by the closure,
|
||||||
// since sort_by closure must be infallible
|
// since sort_by closure must be infallible
|
||||||
let mut compare_err: Option<ShellError> = None;
|
let mut compare_err: Option<ShellError> = None;
|
||||||
|
|
||||||
vals.sort_by(|a, b| {
|
vec.sort_by(|a, b| {
|
||||||
crate::compare_values(a, b, insensitive, natural).unwrap_or_else(|err| {
|
crate::compare_values(a, b, insensitive && string_sort, natural && string_sort)
|
||||||
compare_err.get_or_insert(err);
|
.unwrap_or_else(|err| {
|
||||||
Ordering::Equal
|
compare_err.get_or_insert(err);
|
||||||
})
|
Ordering::Equal
|
||||||
|
})
|
||||||
});
|
});
|
||||||
|
|
||||||
if let Some(err) = compare_err {
|
if let Some(err) = compare_err {
|
||||||
|
@ -122,23 +61,17 @@ pub fn sort_by(
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
for cmp in comparators.iter() {
|
// to apply insensitive or natural sorting, all values must be strings
|
||||||
// closures shouldn't affect whether cell paths are sorted naturally/insensitively
|
let string_sort: bool = comparators.iter().all(|cmp| {
|
||||||
let Comparator::CellPath(cell_path) = cmp else {
|
let Comparator::CellPath(cell_path) = cmp else {
|
||||||
continue;
|
// closures shouldn't affect whether cell paths are sorted naturally/insensitively
|
||||||
|
return true;
|
||||||
};
|
};
|
||||||
|
vec.iter().all(|value| {
|
||||||
let follow: Vec<_> = vec
|
let inner = value.clone().follow_cell_path(&cell_path.members, false);
|
||||||
.iter()
|
matches!(inner, Ok(Value::String { .. }))
|
||||||
.filter_map(|value| {
|
})
|
||||||
value
|
});
|
||||||
.clone()
|
|
||||||
.follow_cell_path(&cell_path.members, false)
|
|
||||||
.ok() // we can ignore for now, we'll notice later if there's an error
|
|
||||||
})
|
|
||||||
.collect();
|
|
||||||
typecheck(&follow, span, insensitive, natural)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
// allow the comparator function to indicate error
|
// allow the comparator function to indicate error
|
||||||
// by mutating this option captured by the closure,
|
// by mutating this option captured by the closure,
|
||||||
|
@ -151,8 +84,8 @@ pub fn sort_by(
|
||||||
b,
|
b,
|
||||||
&comparators,
|
&comparators,
|
||||||
span,
|
span,
|
||||||
insensitive,
|
insensitive && string_sort,
|
||||||
natural,
|
natural && string_sort,
|
||||||
&mut compare_err,
|
&mut compare_err,
|
||||||
)
|
)
|
||||||
});
|
});
|
||||||
|
@ -173,17 +106,6 @@ pub fn sort_record(
|
||||||
) -> Result<Record, ShellError> {
|
) -> Result<Record, ShellError> {
|
||||||
let mut input_pairs: Vec<(String, Value)> = record.into_iter().collect();
|
let mut input_pairs: Vec<(String, Value)> = record.into_iter().collect();
|
||||||
|
|
||||||
if sort_by_value {
|
|
||||||
// TODO(rose): don't clone here?
|
|
||||||
let vals: Vec<_> = input_pairs.iter().map(|(_, val)| val.clone()).collect();
|
|
||||||
typecheck(
|
|
||||||
&vals,
|
|
||||||
Span::unknown(), // TODO(rose)
|
|
||||||
insensitive,
|
|
||||||
natural,
|
|
||||||
)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
// allow the comparator function to indicate error
|
// allow the comparator function to indicate error
|
||||||
// by mutating this option captured by the closure,
|
// by mutating this option captured by the closure,
|
||||||
// since sort_by closure must be infallible
|
// since sort_by closure must be infallible
|
||||||
|
|
Loading…
Reference in New Issue
Block a user