forked from extern/nushell
e4226def16
This commit extracts five new crates: - nu-source, which contains the core source-code handling logic in Nu, including Text, Span, and also the pretty.rs-based debug logic - nu-parser, which is the parser and expander logic - nu-protocol, which is the bulk of the types and basic conveniences used by plugins - nu-errors, which contains ShellError, ParseError and error handling conveniences - nu-textview, which is the textview plugin extracted into a crate One of the major consequences of this refactor is that it's no longer possible to `impl X for Spanned<Y>` outside of the `nu-source` crate, so a lot of types became more concrete (Value became a concrete type instead of Spanned<Value>, for example). This also turned a number of inherent methods in the main nu crate into plain functions (impl Value {} became a bunch of functions in the `value` namespace in `crate::data::value`).
56 lines
1.3 KiB
Rust
56 lines
1.3 KiB
Rust
use crate::commands::WholeStreamCommand;
|
|
use crate::data::base::property_get::get_data_by_key;
|
|
use crate::prelude::*;
|
|
use nu_errors::ShellError;
|
|
use nu_protocol::{Signature, SyntaxShape, Value};
|
|
use nu_source::Tagged;
|
|
|
|
pub struct SortBy;
|
|
|
|
#[derive(Deserialize)]
|
|
pub struct SortByArgs {
|
|
rest: Vec<Tagged<String>>,
|
|
}
|
|
|
|
impl WholeStreamCommand for SortBy {
|
|
fn name(&self) -> &str {
|
|
"sort-by"
|
|
}
|
|
|
|
fn signature(&self) -> Signature {
|
|
Signature::build("sort-by").rest(SyntaxShape::String, "the column(s) to sort by")
|
|
}
|
|
|
|
fn usage(&self) -> &str {
|
|
"Sort by the given columns."
|
|
}
|
|
|
|
fn run(
|
|
&self,
|
|
args: CommandArgs,
|
|
registry: &CommandRegistry,
|
|
) -> Result<OutputStream, ShellError> {
|
|
args.process(registry, sort_by)?.run()
|
|
}
|
|
}
|
|
|
|
fn sort_by(
|
|
SortByArgs { rest }: SortByArgs,
|
|
mut context: RunnableContext,
|
|
) -> Result<OutputStream, ShellError> {
|
|
Ok(OutputStream::new(async_stream! {
|
|
let mut vec = context.input.drain_vec().await;
|
|
|
|
let calc_key = |item: &Value| {
|
|
rest.iter()
|
|
.map(|f| get_data_by_key(item, f.borrow_spanned()).map(|i| i.clone()))
|
|
.collect::<Vec<Option<Value>>>()
|
|
};
|
|
vec.sort_by_cached_key(calc_key);
|
|
|
|
for item in vec {
|
|
yield item.into();
|
|
}
|
|
}))
|
|
}
|