extern/nushell
1
0
Fork 1
mirror of https://github.com/nushell/nushell.git synced 2025-08-10 14:48:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1049 from andrasio/embed-list

Browse Source 
embed as column when embedding a list
This commit is contained in:
Andrés N. Robalino
2019-12-03 02:51:58 -05:00
committed by GitHub
parent 49f9253ca2 e5405d7f5c
commit 4526d757b6
2 changed files with 71 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
src/plugins/embed.rs
View File

@ -11,18 +11,40 @@ use nu_source::Tag;
struct Embed {
field: Option<String>,
are_all_rows: bool,
values: Vec<Value>,
}
impl Embed {
fn new() -> Embed {
Embed {
field: None,
are_all_rows: true,
values: Vec::new(),
}
}
fn embed(&mut self, value: Value) -> Result<(), ShellError> {
self.values.push(value);
match &value {
Value {
value: UntaggedValue::Row(_),
..
} => {
self.values.push(value);
}
_ => {
self.are_all_rows = false;
self.values.push(
value::row(indexmap! {
match &self.field {
Some(key) => key.clone(),
None => "Column".into()
} => value
})
.into_value(Tag::unknown()),
);
}
}
Ok(())
}
}
@ -58,15 +80,23 @@ impl Plugin for Embed {
}
fn end_filter(&mut self) -> Result<Vec<ReturnValue>, ShellError> {
let row = value::row(indexmap! {
match &self.field {
Some(key) => key.clone(),
None => "root".into(),
} => value::table(&self.values).into_value(Tag::unknown()),
})
.into_untagged_value();
if self.are_all_rows {
let row = value::row(indexmap! {
match &self.field {
Some(key) => key.clone(),
None => "Column".into(),
} => value::table(&self.values).into_value(Tag::unknown()),
})
.into_untagged_value();
Ok(vec![ReturnSuccess::value(row)])
Ok(vec![ReturnSuccess::value(row)])
} else {
Ok(self
.values
.iter()
.map(|row| ReturnSuccess::value(row.clone()))
.collect::<Vec<_>>())
}
}
}