nu_data/utils/
group.rs

1use indexmap::IndexMap;
2use nu_errors::ShellError;
3use nu_protocol::{TaggedDictBuilder, UntaggedValue, Value};
4use nu_source::Tag;
5use nu_value_ext::as_string;
6
7#[allow(clippy::type_complexity)]
8pub fn group(
9    values: &Value,
10    grouper: &Option<Box<dyn Fn(usize, &Value) -> Result<String, ShellError> + Send>>,
11    tag: impl Into<Tag>,
12) -> Result<Value, ShellError> {
13    let tag = tag.into();
14
15    let mut groups: IndexMap<String, Vec<Value>> = IndexMap::new();
16
17    for (idx, value) in values.table_entries().enumerate() {
18        let group_key = if let Some(ref grouper) = grouper {
19            grouper(idx, value)
20        } else {
21            as_string(value)
22        };
23
24        let group = groups.entry(group_key?).or_insert(vec![]);
25        group.push((*value).clone());
26    }
27
28    let mut out = TaggedDictBuilder::new(&tag);
29
30    for (k, v) in &groups {
31        out.insert_untagged(k, UntaggedValue::table(v));
32    }
33
34    Ok(out.into_value())
35}
nu_data/utils/group.rs

nu_data/utils/
group.rs
