forked from extern/nushell
028fc9b9cd
Refactored out most of internal work for summarizing data opening the door for generating charts from it. A model is introduced to hold information needed for a summary, Histogram command is an example of a partial usage. This is the beginning. Removed implicit arithmetic traits on Value and Primitive to avoid mixed types panics. The std operations traits can't fail and we can't guarantee that. We can handle gracefully now since compute_values was introduced after the parser changes four months ago. The handling logic should be taken care of either explicitly or in compute_values. The zero identity trait was also removed (and implementing this forced us to also implement Add, Mult, etc) Also: the `math` operations now remove in the output if a given column is not computable: ``` > ls | math sum ──────┬────────── size │ 150.9 KB ──────┴────────── ```
73 lines
2.4 KiB
Rust
73 lines
2.4 KiB
Rust
use crate::prelude::*;
|
|
use nu_errors::ShellError;
|
|
use nu_protocol::{Dictionary, ReturnSuccess, UntaggedValue, Value};
|
|
|
|
use indexmap::map::IndexMap;
|
|
|
|
pub type MathFunction = fn(values: &[Value], tag: &Tag) -> Result<Value, ShellError>;
|
|
|
|
pub async fn run_with_function(
|
|
RunnableContext {
|
|
mut input, name, ..
|
|
}: RunnableContext,
|
|
mf: MathFunction,
|
|
) -> Result<OutputStream, ShellError> {
|
|
let values: Vec<Value> = input.drain_vec().await;
|
|
|
|
let res = calculate(&values, &name, mf);
|
|
match res {
|
|
Ok(v) => {
|
|
if v.value.is_table() {
|
|
Ok(OutputStream::from(
|
|
v.table_entries()
|
|
.map(|v| ReturnSuccess::value(v.clone()))
|
|
.collect::<Vec<_>>(),
|
|
))
|
|
} else {
|
|
Ok(OutputStream::one(ReturnSuccess::value(v)))
|
|
}
|
|
}
|
|
Err(e) => Err(e),
|
|
}
|
|
}
|
|
|
|
pub fn calculate(values: &[Value], name: &Tag, mf: MathFunction) -> Result<Value, ShellError> {
|
|
if values.iter().all(|v| v.is_primitive()) {
|
|
mf(&values, &name)
|
|
} else {
|
|
// If we are not dealing with Primitives, then perhaps we are dealing with a table
|
|
// Create a key for each column name
|
|
let mut column_values = IndexMap::new();
|
|
for value in values {
|
|
if let UntaggedValue::Row(row_dict) = &value.value {
|
|
for (key, value) in row_dict.entries.iter() {
|
|
column_values
|
|
.entry(key.clone())
|
|
.and_modify(|v: &mut Vec<Value>| v.push(value.clone()))
|
|
.or_insert(vec![value.clone()]);
|
|
}
|
|
}
|
|
}
|
|
// The mathematical function operates over the columns of the table
|
|
let mut column_totals = IndexMap::new();
|
|
for (col_name, col_vals) in column_values {
|
|
if let Ok(out) = mf(&col_vals, &name) {
|
|
column_totals.insert(col_name, out);
|
|
}
|
|
}
|
|
|
|
if column_totals.keys().len() == 0 {
|
|
return Err(ShellError::labeled_error(
|
|
"Attempted to compute values that can't be operated on",
|
|
"value appears here",
|
|
name.span,
|
|
));
|
|
}
|
|
|
|
Ok(UntaggedValue::Row(Dictionary {
|
|
entries: column_totals,
|
|
})
|
|
.into_untagged_value())
|
|
}
|
|
}
|