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

Return iter from from vcf (#2137)

Browse Source
This commit is contained in:
Joseph T. Lyons 2020-07-08 09:20:57 -04:00 committed by GitHub
parent 0fdb9ac5e2
commit b004236927
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
crates/nu-cli/src/commands/from_vcf.rs
View File

@ -5,7 +5,6 @@ use ical::parser::vcard::component::*;
use ical::property::Property; use ical::property::Property;
use nu_errors::ShellError; use nu_errors::ShellError;
use nu_protocol::{Primitive, ReturnSuccess, Signature, TaggedDictBuilder, UntaggedValue, Value}; use nu_protocol::{Primitive, ReturnSuccess, Signature, TaggedDictBuilder, UntaggedValue, Value};
use std::io::BufReader;
pub struct FromVcf; pub struct FromVcf;
@ -42,28 +41,20 @@ async fn from_vcf(
let input = args.input; let input = args.input;
let input_string = input.collect_string(tag.clone()).await?.item; let input_string = input.collect_string(tag.clone()).await?.item;
let input_bytes = input_string.as_bytes(); let input_bytes = input_string.into_bytes();
let buf_reader = BufReader::new(input_bytes); let buf_reader = std::io::Cursor::new(input_bytes);
let parser = ical::VcardParser::new(buf_reader); let parser = ical::VcardParser::new(buf_reader);
let mut values_vec_deque = VecDeque::new(); let iter = parser.map(move |contact| match contact {
Ok(c) => ReturnSuccess::value(contact_to_value(c, tag.clone())),
for contact in parser { Err(_) => Err(ShellError::labeled_error(
match contact {
Ok(c) => {
values_vec_deque.push_back(ReturnSuccess::value(contact_to_value(c, tag.clone())))
}
Err(_) => {
return Err(ShellError::labeled_error(
"Could not parse as .vcf", "Could not parse as .vcf",
"input cannot be parsed as .vcf", "input cannot be parsed as .vcf",
tag.clone(), tag.clone(),
)) )),
} });
}
}
Ok(futures::stream::iter(values_vec_deque).to_output_stream()) Ok(futures::stream::iter(iter).to_output_stream())
} }
fn contact_to_value(contact: VcardContact, tag: Tag) -> Value { fn contact_to_value(contact: VcardContact, tag: Tag) -> Value {