2024-03-23 19:29:54 +01:00
|
|
|
//! Test support for [Nushell](https://nushell.sh) plugins.
|
|
|
|
//!
|
|
|
|
//! # Example
|
|
|
|
//!
|
|
|
|
//! ```rust
|
|
|
|
//! use std::sync::Arc;
|
|
|
|
//!
|
|
|
|
//! use nu_plugin::*;
|
|
|
|
//! use nu_plugin_test_support::PluginTest;
|
2024-03-27 11:59:57 +01:00
|
|
|
//! use nu_protocol::{
|
2024-07-08 00:29:01 +02:00
|
|
|
//! Example, IntoInterruptiblePipelineData, LabeledError, PipelineData, ShellError, Signals,
|
|
|
|
//! Signature, Span, Type, Value,
|
2024-03-27 11:59:57 +01:00
|
|
|
//! };
|
2024-03-23 19:29:54 +01:00
|
|
|
//!
|
|
|
|
//! struct LowercasePlugin;
|
|
|
|
//! struct Lowercase;
|
|
|
|
//!
|
|
|
|
//! impl PluginCommand for Lowercase {
|
|
|
|
//! type Plugin = LowercasePlugin;
|
|
|
|
//!
|
2024-03-27 11:59:57 +01:00
|
|
|
//! fn name(&self) -> &str {
|
|
|
|
//! "lowercase"
|
|
|
|
//! }
|
|
|
|
//!
|
|
|
|
//! fn usage(&self) -> &str {
|
|
|
|
//! "Convert each string in a stream to lowercase"
|
|
|
|
//! }
|
|
|
|
//!
|
|
|
|
//! fn signature(&self) -> Signature {
|
|
|
|
//! Signature::build(self.name()).input_output_type(
|
|
|
|
//! Type::List(Type::String.into()),
|
|
|
|
//! Type::List(Type::String.into()),
|
|
|
|
//! )
|
|
|
|
//! }
|
|
|
|
//!
|
|
|
|
//! fn examples(&self) -> Vec<Example> {
|
|
|
|
//! vec![Example {
|
|
|
|
//! example: r#"[Hello wORLD] | lowercase"#,
|
|
|
|
//! description: "Lowercase a list of strings",
|
|
|
|
//! result: Some(Value::test_list(vec![
|
|
|
|
//! Value::test_string("hello"),
|
|
|
|
//! Value::test_string("world"),
|
|
|
|
//! ])),
|
|
|
|
//! }]
|
2024-03-23 19:29:54 +01:00
|
|
|
//! }
|
|
|
|
//!
|
|
|
|
//! fn run(
|
|
|
|
//! &self,
|
2024-03-27 11:59:57 +01:00
|
|
|
//! _plugin: &LowercasePlugin,
|
|
|
|
//! _engine: &EngineInterface,
|
2024-03-23 19:29:54 +01:00
|
|
|
//! call: &EvaluatedCall,
|
|
|
|
//! input: PipelineData,
|
|
|
|
//! ) -> Result<PipelineData, LabeledError> {
|
|
|
|
//! let span = call.head;
|
2024-03-27 11:59:57 +01:00
|
|
|
//! Ok(input.map(
|
|
|
|
//! move |value| {
|
|
|
|
//! value
|
|
|
|
//! .as_str()
|
|
|
|
//! .map(|string| Value::string(string.to_lowercase(), span))
|
|
|
|
//! // Errors in a stream should be returned as values.
|
|
|
|
//! .unwrap_or_else(|err| Value::error(err, span))
|
|
|
|
//! },
|
2024-07-08 00:29:01 +02:00
|
|
|
//! &Signals::empty(),
|
2024-03-27 11:59:57 +01:00
|
|
|
//! )?)
|
2024-03-23 19:29:54 +01:00
|
|
|
//! }
|
|
|
|
//! }
|
|
|
|
//!
|
|
|
|
//! impl Plugin for LowercasePlugin {
|
2024-06-21 13:27:09 +02:00
|
|
|
//! fn version(&self) -> String {
|
|
|
|
//! env!("CARGO_PKG_VERSION").into()
|
|
|
|
//! }
|
|
|
|
//!
|
2024-03-23 19:29:54 +01:00
|
|
|
//! fn commands(&self) -> Vec<Box<dyn PluginCommand<Plugin=Self>>> {
|
|
|
|
//! vec![Box::new(Lowercase)]
|
|
|
|
//! }
|
|
|
|
//! }
|
|
|
|
//!
|
2024-03-27 11:59:57 +01:00
|
|
|
//! // #[test]
|
|
|
|
//! fn test_examples() -> Result<(), ShellError> {
|
|
|
|
//! PluginTest::new("lowercase", LowercasePlugin.into())?
|
|
|
|
//! .test_command_examples(&Lowercase)
|
|
|
|
//! }
|
|
|
|
//!
|
|
|
|
//! // #[test]
|
|
|
|
//! fn test_lowercase() -> Result<(), ShellError> {
|
2024-07-08 00:29:01 +02:00
|
|
|
//! let input = vec![Value::test_string("FooBar")].into_pipeline_data(Span::test_data(), Signals::empty());
|
2024-03-23 19:29:54 +01:00
|
|
|
//! let output = PluginTest::new("lowercase", LowercasePlugin.into())?
|
|
|
|
//! .eval_with("lowercase", input)?
|
Replace `ExternalStream` with new `ByteStream` type (#12774)
# Description
This PR introduces a `ByteStream` type which is a `Read`-able stream of
bytes. Internally, it has an enum over three different byte stream
sources:
```rust
pub enum ByteStreamSource {
Read(Box<dyn Read + Send + 'static>),
File(File),
Child(ChildProcess),
}
```
This is in comparison to the current `RawStream` type, which is an
`Iterator<Item = Vec<u8>>` and has to allocate for each read chunk.
Currently, `PipelineData::ExternalStream` serves a weird dual role where
it is either external command output or a wrapper around `RawStream`.
`ByteStream` makes this distinction more clear (via `ByteStreamSource`)
and replaces `PipelineData::ExternalStream` in this PR:
```rust
pub enum PipelineData {
Empty,
Value(Value, Option<PipelineMetadata>),
ListStream(ListStream, Option<PipelineMetadata>),
ByteStream(ByteStream, Option<PipelineMetadata>),
}
```
The PR is relatively large, but a decent amount of it is just repetitive
changes.
This PR fixes #7017, fixes #10763, and fixes #12369.
This PR also improves performance when piping external commands. Nushell
should, in most cases, have competitive pipeline throughput compared to,
e.g., bash.
| Command | Before (MB/s) | After (MB/s) | Bash (MB/s) |
| -------------------------------------------------- | -------------:|
------------:| -----------:|
| `throughput \| rg 'x'` | 3059 | 3744 | 3739 |
| `throughput \| nu --testbin relay o> /dev/null` | 3508 | 8087 | 8136 |
# User-Facing Changes
- This is a breaking change for the plugin communication protocol,
because the `ExternalStreamInfo` was replaced with `ByteStreamInfo`.
Plugins now only have to deal with a single input stream, as opposed to
the previous three streams: stdout, stderr, and exit code.
- The output of `describe` has been changed for external/byte streams.
- Temporary breaking change: `bytes starts-with` no longer works with
byte streams. This is to keep the PR smaller, and `bytes ends-with`
already does not work on byte streams.
- If a process core dumped, then instead of having a `Value::Error` in
the `exit_code` column of the output returned from `complete`, it now is
a `Value::Int` with the negation of the signal number.
# After Submitting
- Update docs and book as necessary
- Release notes (e.g., plugin protocol changes)
- Adapt/convert commands to work with byte streams (high priority is
`str length`, `bytes starts-with`, and maybe `bytes ends-with`).
- Refactor the `tee` code, Devyn has already done some work on this.
---------
Co-authored-by: Devyn Cairns <devyn.cairns@gmail.com>
2024-05-16 16:11:18 +02:00
|
|
|
//! .into_value(Span::test_data())?;
|
2024-03-23 19:29:54 +01:00
|
|
|
//!
|
|
|
|
//! assert_eq!(
|
|
|
|
//! Value::test_list(vec![
|
|
|
|
//! Value::test_string("foobar")
|
|
|
|
//! ]),
|
|
|
|
//! output
|
|
|
|
//! );
|
|
|
|
//! Ok(())
|
|
|
|
//! }
|
|
|
|
//! #
|
2024-03-27 11:59:57 +01:00
|
|
|
//! # test_examples().unwrap();
|
2024-03-23 19:29:54 +01:00
|
|
|
//! # test_lowercase().unwrap();
|
|
|
|
//! ```
|
|
|
|
|
2024-03-26 03:13:12 +01:00
|
|
|
mod diff;
|
2024-03-23 19:29:54 +01:00
|
|
|
mod fake_persistent_plugin;
|
|
|
|
mod fake_register;
|
|
|
|
mod plugin_test;
|
|
|
|
mod spawn_fake_plugin;
|
|
|
|
|
|
|
|
pub use plugin_test::PluginTest;
|