extern/nushell
1
0
Fork 1
mirror of https://github.com/nushell/nushell.git synced 2025-04-22 20:28:22 +02:00
Code Issues Packages Projects Releases Wiki Activity

Refactoring

Browse Source
This commit is contained in:
Sam Hedin 2020-03-28 00:42:26 +01:00
parent 9a676d23af
commit 319aac505e
1 changed files with 5 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
crates/nu-cli/src/commands/headers.rs
View File

@ -6,8 +6,6 @@ use crate::prelude::*;
use futures::stream::StreamExt; use futures::stream::StreamExt;
use nu_errors::ShellError; use nu_errors::ShellError;
use nu_protocol::{ReturnSuccess, Signature, UntaggedValue, Value}; use nu_protocol::{ReturnSuccess, Signature, UntaggedValue, Value};
use std::fs::File;
use std::io::Write;
pub struct Headers; pub struct Headers;
#[derive(Deserialize)] #[derive(Deserialize)]
@ -41,19 +39,15 @@ pub fn count(
RunnableContext { input, name, .. }: RunnableContext, RunnableContext { input, name, .. }: RunnableContext,
) -> Result<OutputStream, ShellError> { ) -> Result<OutputStream, ShellError> {
let stream = async_stream! { let stream = async_stream! {
let rows: Vec<Value> = input.values.collect().await; let rows: Vec<Value> = input.values.collect().await;
let mut headers = vec![]; let headers: Vec<String> = match &rows[0].value {
match &rows[0].value {
UntaggedValue::Row(d) => { UntaggedValue::Row(d) => {
for (_, v) in d.entries.iter() { d.entries.iter().map(|(_, v)| v.as_string().unwrap()).collect()
headers.push(v.as_string().unwrap());
}
}
_ => ()
} }
_ => vec![]
};
let mut newrows: Vec<Value> = vec![]; let mut newrows: Vec<Value> = vec![];
for r in rows.iter().skip(1) { for r in rows.iter().skip(1) {
@ -72,11 +66,7 @@ pub fn count(
} }
} }
let mut file = File::create("headout").unwrap(); yield ReturnSuccess::value(UntaggedValue::table(&newrows).into_value(name))
write!(file, "args: {:#?}", newrows).unwrap();
// yield ReturnSuccess::value(UntaggedValue::int(rows.len()).into_value(name))
yield ReturnSuccess::value(UntaggedValue::Table(newrows).into_value(name))
}; };
Ok(stream.to_output_stream()) Ok(stream.to_output_stream())