nushell/src/prelude.rs

90 lines
2.6 KiB
Rust
Raw Normal View History

#[macro_export]
macro_rules! stream {
($($expr:expr),*) => {{
let mut v = VecDeque::new();
$(
v.push_back($expr);
)*
v
}}
}
#[macro_export]
macro_rules! trace_stream {
2019-07-12 21:22:08 +02:00
(target: $target:tt, $desc:tt = $expr:expr) => {{
if log::log_enabled!(target: $target, log::Level::Trace) {
use futures::stream::StreamExt;
// Blocking is generally quite bad, but this is for debugging
// let mut local = futures::executor::LocalPool::new();
// let objects = local.run_until($expr.into_vec());
// let objects: Vec<_> = futures::executor::block_on($expr.into_vec());
let objects = $expr.values.inspect(|o| {
2019-07-12 21:22:08 +02:00
trace!(target: $target, "{} = {:#?}", $desc, o.debug());
});
$crate::stream::InputStream::from_stream(objects.boxed())
} else {
$expr
}
}};
}
2019-05-23 06:30:43 +02:00
crate use crate::cli::MaybeOwned;
crate use crate::commands::command::{
2019-08-02 21:15:07 +02:00
CommandAction, CommandArgs, ReturnSuccess, ReturnValue, RunnableContext,
};
2019-07-24 00:22:11 +02:00
crate use crate::context::{CommandRegistry, Context};
2019-05-24 21:35:22 +02:00
crate use crate::env::host::handle_unexpected;
crate use crate::env::{Environment, Host};
crate use crate::errors::ShellError;
2019-08-02 21:15:07 +02:00
crate use crate::object::base as value;
crate use crate::object::types::ExtractType;
2019-06-27 18:47:24 +02:00
crate use crate::object::{Primitive, Value};
2019-08-02 21:15:07 +02:00
crate use crate::parser::registry::Signature;
crate use crate::parser::{hir::SyntaxType, Span, Spanned, SpannedItem};
crate use crate::stream::{InputStream, OutputStream};
2019-07-24 00:22:11 +02:00
crate use crate::traits::{HasSpan, ToDebug};
2019-06-22 22:46:16 +02:00
crate use crate::Text;
crate use futures::stream::BoxStream;
2019-07-24 00:22:11 +02:00
crate use futures::{FutureExt, Stream, StreamExt};
2019-08-03 04:17:28 +02:00
crate use futures_async_stream::async_stream_block;
2019-08-06 18:26:33 +02:00
#[allow(unused)]
2019-08-02 21:15:07 +02:00
crate use serde::{Deserialize, Serialize};
crate use std::collections::VecDeque;
crate use std::future::Future;
crate use std::sync::{Arc, Mutex};
pub trait FromInputStream {
fn from_input_stream(self) -> OutputStream;
}
impl<T> FromInputStream for T
where
2019-07-08 18:44:53 +02:00
T: Stream<Item = Spanned<Value>> + Send + 'static,
{
fn from_input_stream(self) -> OutputStream {
OutputStream {
values: self.map(ReturnSuccess::value).boxed(),
}
}
}
pub trait ToOutputStream {
fn to_output_stream(self) -> OutputStream;
}
impl<T, U> ToOutputStream for T
where
T: Stream<Item = U> + Send + 'static,
U: Into<ReturnValue>,
{
fn to_output_stream(self) -> OutputStream {
OutputStream {
values: self.map(|item| item.into()).boxed(),
}
}
}