2019-05-15 20:14:51 +02:00
|
|
|
use crate::errors::ShellError;
|
2019-05-22 09:12:03 +02:00
|
|
|
use crate::object::base::select_fields;
|
2019-05-16 00:58:44 +02:00
|
|
|
use crate::object::Value;
|
2019-05-15 20:14:51 +02:00
|
|
|
use crate::prelude::*;
|
|
|
|
|
2019-05-28 07:05:14 +02:00
|
|
|
pub fn column(args: CommandArgs) -> Result<OutputStream, ShellError> {
|
2019-06-01 07:50:16 +02:00
|
|
|
if args.positional.is_empty() {
|
2019-05-22 09:12:03 +02:00
|
|
|
return Err(ShellError::string("select requires a field"));
|
|
|
|
}
|
2019-05-15 20:14:51 +02:00
|
|
|
|
2019-06-01 07:50:16 +02:00
|
|
|
let fields: Result<Vec<String>, _> = args.positional.iter().map(|a| a.as_string()).collect();
|
2019-05-22 09:12:03 +02:00
|
|
|
let fields = fields?;
|
2019-05-15 20:14:51 +02:00
|
|
|
|
2019-05-22 09:12:03 +02:00
|
|
|
let objects = args
|
|
|
|
.input
|
2019-05-23 09:23:06 +02:00
|
|
|
.map(move |item| Value::Object(select_fields(&item, &fields)))
|
|
|
|
.map(|item| ReturnValue::Value(item));
|
2019-05-15 20:14:51 +02:00
|
|
|
|
2019-05-23 09:23:06 +02:00
|
|
|
let stream = Pin::new(Box::new(objects));
|
|
|
|
Ok(stream)
|
2019-05-15 20:14:51 +02:00
|
|
|
}
|