Switch stdin redirect to manual. Add test (#1367)

This commit is contained in:
Jonathan Turner 2020-02-09 22:55:07 -08:00 committed by GitHub
parent a2668e3327
commit dc50e61f26
5 changed files with 113 additions and 42 deletions

105
Cargo.lock generated
View File

@ -587,7 +587,23 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5750773d74a7dc612eac2ded3f55e9cdeeaa072210cd17c0192aedb48adb3618" checksum = "5750773d74a7dc612eac2ded3f55e9cdeeaa072210cd17c0192aedb48adb3618"
dependencies = [ dependencies = [
"bitflags", "bitflags",
"crossterm_winapi", "crossterm_winapi 0.5.1",
"lazy_static 1.4.0",
"libc",
"mio",
"parking_lot",
"signal-hook",
"winapi 0.3.8",
]
[[package]]
name = "crossterm"
version = "0.16.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b8a3223215bc00c666d6be730e88aef245ad4a4f837e87a16c347e8acf701643"
dependencies = [
"bitflags",
"crossterm_winapi 0.6.1",
"lazy_static 1.4.0", "lazy_static 1.4.0",
"libc", "libc",
"mio", "mio",
@ -605,6 +621,15 @@ dependencies = [
"winapi 0.3.8", "winapi 0.3.8",
] ]
[[package]]
name = "crossterm_winapi"
version = "0.6.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "057b7146d02fb50175fd7dbe5158f6097f33d02831f43b4ee8ae4ddf67b68f5c"
dependencies = [
"winapi 0.3.8",
]
[[package]] [[package]]
name = "csv" name = "csv"
version = "1.1.3" version = "1.1.3"
@ -863,6 +888,19 @@ dependencies = [
"termcolor", "termcolor",
] ]
[[package]]
name = "env_logger"
version = "0.7.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "44533bbbb3bb3c1fa17d9f2e4e38bbbaf8396ba82193c4cb1b6445d711445d36"
dependencies = [
"atty",
"humantime",
"log",
"regex",
"termcolor",
]
[[package]] [[package]]
name = "erased-serde" name = "erased-serde"
version = "0.3.10" version = "0.3.10"
@ -956,9 +994,9 @@ dependencies = [
[[package]] [[package]]
name = "futures-channel" name = "futures-channel"
version = "0.3.3" version = "0.3.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7264eb65b194d2fa6ec31b898ead7c332854bfa42521659226e72a585fca5b85" checksum = "f0c77d04ce8edd9cb903932b608268b3fffec4163dc053b3b402bf47eac1f1a8"
dependencies = [ dependencies = [
"futures-core", "futures-core",
"futures-sink", "futures-sink",
@ -976,9 +1014,9 @@ dependencies = [
[[package]] [[package]]
name = "futures-core" name = "futures-core"
version = "0.3.3" version = "0.3.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b597b16aa1a19ce2dfde5128a7c656d75346b35601a640be2d9efd4e9c83609d" checksum = "f25592f769825e89b92358db00d26f965761e094951ac44d3663ef25b7ac464a"
[[package]] [[package]]
name = "futures-core-preview" name = "futures-core-preview"
@ -1010,9 +1048,9 @@ dependencies = [
[[package]] [[package]]
name = "futures-io" name = "futures-io"
version = "0.3.3" version = "0.3.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3d429f824b5e5dbd45fc8e54e1005a37e1f8c6d570cd64d0b59b24d3a80b8b8e" checksum = "a638959aa96152c7a4cddf50fcb1e3fede0583b27157c26e67d6f99904090dc6"
[[package]] [[package]]
name = "futures-io-preview" name = "futures-io-preview"
@ -1022,9 +1060,9 @@ checksum = "f4914ae450db1921a56c91bde97a27846287d062087d4a652efc09bb3a01ebda"
[[package]] [[package]]
name = "futures-macro" name = "futures-macro"
version = "0.3.3" version = "0.3.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a1d75b72904b78044e0091355fc49d29f48bff07a68a719a41cf059711e071b4" checksum = "9a5081aa3de1f7542a794a397cde100ed903b0630152d0973479018fd85423a7"
dependencies = [ dependencies = [
"proc-macro-hack", "proc-macro-hack",
"proc-macro2", "proc-macro2",
@ -1048,9 +1086,9 @@ dependencies = [
[[package]] [[package]]
name = "futures-sink" name = "futures-sink"
version = "0.3.3" version = "0.3.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "04299e123547ea7c56f3e1b376703142f5fc0b6700433eed549e9d0b8a75a66c" checksum = "3466821b4bc114d95b087b850a724c6f83115e929bc88f1fa98a3304a944c8a6"
[[package]] [[package]]
name = "futures-sink-preview" name = "futures-sink-preview"
@ -1060,9 +1098,9 @@ checksum = "86f148ef6b69f75bb610d4f9a2336d4fc88c4b5b67129d1a340dd0fd362efeec"
[[package]] [[package]]
name = "futures-task" name = "futures-task"
version = "0.3.3" version = "0.3.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "86f9ceab4bce46555ee608b1ec7c414d6b2e76e196ef46fa5a8d4815a8571398" checksum = "7b0a34e53cf6cdcd0178aa573aed466b646eb3db769570841fda0c7ede375a27"
[[package]] [[package]]
name = "futures-timer" name = "futures-timer"
@ -1078,9 +1116,9 @@ checksum = "3de1a2b2a2a33d9e60e17980b60ee061eeaae96a5abe9121db0fdb9af167a1c5"
[[package]] [[package]]
name = "futures-util" name = "futures-util"
version = "0.3.3" version = "0.3.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7d2f1296f7644d2cd908ebb2fa74645608e39f117c72bac251d40418c6d74c4f" checksum = "22766cf25d64306bedf0384da004d05c9974ab104fcc4528f1236181c18004c5"
dependencies = [ dependencies = [
"futures 0.1.29", "futures 0.1.29",
"futures-channel", "futures-channel",
@ -2147,7 +2185,6 @@ dependencies = [
"ansi_term 0.12.1", "ansi_term 0.12.1",
"app_dirs", "app_dirs",
"async-stream", "async-stream",
"atty",
"base64 0.11.0", "base64 0.11.0",
"bigdecimal", "bigdecimal",
"bson", "bson",
@ -2158,7 +2195,7 @@ dependencies = [
"chrono", "chrono",
"clap", "clap",
"clipboard", "clipboard",
"crossterm", "crossterm 0.16.0",
"csv", "csv",
"ctrlc", "ctrlc",
"derive-new", "derive-new",
@ -2209,7 +2246,7 @@ dependencies = [
"pin-utils", "pin-utils",
"pretty-hex", "pretty-hex",
"pretty_assertions", "pretty_assertions",
"pretty_env_logger", "pretty_env_logger 0.4.0",
"prettytable-rs", "prettytable-rs",
"ptree", "ptree",
"query_interface", "query_interface",
@ -2306,7 +2343,7 @@ dependencies = [
"num-traits 0.2.11", "num-traits 0.2.11",
"pretty", "pretty",
"pretty_assertions", "pretty_assertions",
"pretty_env_logger", "pretty_env_logger 0.3.1",
"ptree", "ptree",
"serde 1.0.104", "serde 1.0.104",
"shellexpand", "shellexpand",
@ -2420,7 +2457,7 @@ name = "nu_plugin_binaryview"
version = "0.9.0" version = "0.9.0"
dependencies = [ dependencies = [
"ansi_term 0.12.1", "ansi_term 0.12.1",
"crossterm", "crossterm 0.14.2",
"image", "image",
"neso", "neso",
"nu-build", "nu-build",
@ -2549,7 +2586,7 @@ name = "nu_plugin_textview"
version = "0.9.0" version = "0.9.0"
dependencies = [ dependencies = [
"ansi_term 0.12.1", "ansi_term 0.12.1",
"crossterm", "crossterm 0.14.2",
"nu-build", "nu-build",
"nu-errors", "nu-errors",
"nu-plugin", "nu-plugin",
@ -2946,7 +2983,17 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "717ee476b1690853d222af4634056d830b5197ffd747726a9a1eee6da9f49074" checksum = "717ee476b1690853d222af4634056d830b5197ffd747726a9a1eee6da9f49074"
dependencies = [ dependencies = [
"chrono", "chrono",
"env_logger", "env_logger 0.6.2",
"log",
]
[[package]]
name = "pretty_env_logger"
version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "926d36b9553851b8b0005f1275891b392ee4d2d833852c417ed025477350fb9d"
dependencies = [
"env_logger 0.7.1",
"log", "log",
] ]
@ -3255,9 +3302,9 @@ dependencies = [
[[package]] [[package]]
name = "roxmltree" name = "roxmltree"
version = "0.9.0" version = "0.9.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "708ee8345e5b70c86aca5ab03bb7d961349b5b2fa7939d3b131af487101de2f3" checksum = "99d696b20b92d3e02e08fd8456f0ab03007c99e6b111a6205b9cb6fc044d0957"
dependencies = [ dependencies = [
"xmlparser", "xmlparser",
] ]
@ -3526,9 +3573,9 @@ dependencies = [
[[package]] [[package]]
name = "serde_json" name = "serde_json"
version = "1.0.46" version = "1.0.47"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "21b01d7f0288608a01dca632cf1df859df6fd6ffa885300fc275ce2ba6221953" checksum = "15913895b61e0be854afd32fd4163fcd2a3df34142cf2cb961b310ce694cbf90"
dependencies = [ dependencies = [
"indexmap", "indexmap",
"itoa", "itoa",
@ -3658,9 +3705,9 @@ checksum = "6e63cff320ae2c57904679ba7cb63280a3dc4613885beafb148ee7bf9aa9042d"
[[package]] [[package]]
name = "starship" name = "starship"
version = "0.33.1" version = "0.35.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "91efa8a7ad3da400fa610a1f49f9aca455de4082913a48e2f3d6cad64813459d" checksum = "5e9fd041a10d378370baa43669ccf11a8385a495d492ce57e5d144f83007a8ba"
dependencies = [ dependencies = [
"ansi_term 0.12.1", "ansi_term 0.12.1",
"battery", "battery",
@ -3676,7 +3723,7 @@ dependencies = [
"open", "open",
"os_info", "os_info",
"path-slash", "path-slash",
"pretty_env_logger", "pretty_env_logger 0.4.0",
"rayon", "rayon",
"reqwest", "reqwest",
"serde_json", "serde_json",

View File

@ -70,7 +70,7 @@ itertools = "0.8.2"
ansi_term = "0.12.1" ansi_term = "0.12.1"
nom = "5.0.1" nom = "5.0.1"
dunce = "1.0.0" dunce = "1.0.0"
indexmap = { version = "1.3.1", features = ["serde-1"] } indexmap = { version = "1.3.2", features = ["serde-1"] }
byte-unit = "3.0.3" byte-unit = "3.0.3"
base64 = "0.11" base64 = "0.11"
futures = { version = "0.3", features = ["compat", "io-compat"] } futures = { version = "0.3", features = ["compat", "io-compat"] }
@ -80,10 +80,10 @@ num-traits = "0.2.11"
term = "0.5.2" term = "0.5.2"
bytes = "0.4.12" bytes = "0.4.12"
log = "0.4.8" log = "0.4.8"
pretty_env_logger = "0.3.1" pretty_env_logger = "0.4.0"
serde = { version = "1.0.104", features = ["derive"] } serde = { version = "1.0.104", features = ["derive"] }
bson = { version = "0.14.0", features = ["decimal128"] } bson = { version = "0.14.0", features = ["decimal128"] }
serde_json = "1.0.44" serde_json = "1.0.47"
serde-hjson = "0.9.1" serde-hjson = "0.9.1"
serde_yaml = "0.8" serde_yaml = "0.8"
serde_bytes = "0.11.3" serde_bytes = "0.11.3"
@ -97,7 +97,7 @@ git2 = { version = "0.11.0", default_features = false }
dirs = "2.0.2" dirs = "2.0.2"
glob = "0.3.0" glob = "0.3.0"
ctrlc = "3.1.3" ctrlc = "3.1.3"
roxmltree = "0.9.0" roxmltree = "0.9.1"
nom_locate = "1.0.0" nom_locate = "1.0.0"
nom-tracable = "0.4.1" nom-tracable = "0.4.1"
unicode-xid = "0.2.0" unicode-xid = "0.2.0"
@ -110,7 +110,7 @@ ichwh = "0.3"
textwrap = {version = "0.11.0", features = ["term_size"]} textwrap = {version = "0.11.0", features = ["term_size"]}
shellexpand = "1.1.1" shellexpand = "1.1.1"
pin-utils = "0.1.0-alpha.4" pin-utils = "0.1.0-alpha.4"
num-bigint = { version = "0.2.5", features = ["serde"] } num-bigint = { version = "0.2.6", features = ["serde"] }
bigdecimal = { version = "0.1.0", features = ["serde"] } bigdecimal = { version = "0.1.0", features = ["serde"] }
serde_urlencoded = "0.6.1" serde_urlencoded = "0.6.1"
trash = "1.0.0" trash = "1.0.0"
@ -119,19 +119,18 @@ cfg-if = "0.1"
strip-ansi-escapes = "0.1.0" strip-ansi-escapes = "0.1.0"
calamine = "0.16" calamine = "0.16"
umask = "0.1" umask = "0.1"
futures-util = "0.3.1" futures-util = "0.3.4"
termcolor = "1.1.0" termcolor = "1.1.0"
natural = "0.3.0" natural = "0.3.0"
parking_lot = "0.10.0" parking_lot = "0.10.0"
meval = "0.2" meval = "0.2"
atty = "0.2"
clipboard = {version = "0.5", optional = true } clipboard = {version = "0.5", optional = true }
ptree = {version = "0.2" } ptree = {version = "0.2" }
starship = { version = "0.33.1", optional = true} starship = { version = "0.35.1", optional = true}
syntect = {version = "3.2.0", optional = true } syntect = {version = "3.2.0", optional = true }
onig_sys = {version = "=69.1.0", optional = true } onig_sys = {version = "=69.1.0", optional = true }
crossterm = {version = "0.14.2", optional = true} crossterm = {version = "0.16.0", optional = true}
url = {version = "2.1.1", optional = true} url = {version = "2.1.1", optional = true}
semver = {version = "0.9.0", optional = true} semver = {version = "0.9.0", optional = true}

View File

@ -373,7 +373,10 @@ pub fn create_default_context(
Ok(context) Ok(context)
} }
pub async fn run_pipeline_standalone(pipeline: String) -> Result<(), Box<dyn Error>> { pub async fn run_pipeline_standalone(
pipeline: String,
redirect_stdin: bool,
) -> Result<(), Box<dyn Error>> {
let mut syncer = crate::env::environment_syncer::EnvironmentSyncer::new(); let mut syncer = crate::env::environment_syncer::EnvironmentSyncer::new();
let mut context = create_default_context(&mut syncer)?; let mut context = create_default_context(&mut syncer)?;
@ -390,7 +393,7 @@ pub async fn run_pipeline_standalone(pipeline: String) -> Result<(), Box<dyn Err
context.ctrl_c.store(false, Ordering::SeqCst); context.ctrl_c.store(false, Ordering::SeqCst);
} }
let line = process_line(Ok(pipeline), &mut context, true).await; let line = process_line(Ok(pipeline), &mut context, redirect_stdin).await;
match line { match line {
LineResult::Success(line) => { LineResult::Success(line) => {
@ -644,7 +647,7 @@ async fn process_line(
return LineResult::Error(line.to_string(), failure.into()); return LineResult::Error(line.to_string(), failure.into());
} }
let input_stream = if redirect_stdin && !atty::is(atty::Stream::Stdin) { let input_stream = if redirect_stdin {
let file = futures::io::AllowStdIo::new(std::io::stdin()); let file = futures::io::AllowStdIo::new(std::io::stdin());
let stream = FramedRead::new(file, LinesCodec).map(|line| { let stream = FramedRead::new(file, LinesCodec).map(|line| {
if let Ok(line) = line { if let Ok(line) = line {

View File

@ -32,6 +32,12 @@ fn main() -> Result<(), Box<dyn Error>> {
.multiple(true) .multiple(true)
.takes_value(true), .takes_value(true),
) )
.arg(
Arg::with_name("stdin")
.long("stdin")
.multiple(false)
.takes_value(false),
)
.get_matches(); .get_matches();
let loglevel = match matches.value_of("loglevel") { let loglevel = match matches.value_of("loglevel") {
@ -74,7 +80,10 @@ fn main() -> Result<(), Box<dyn Error>> {
None => {} None => {}
Some(values) => { Some(values) => {
for item in values { for item in values {
futures::executor::block_on(nu::run_pipeline_standalone(item.into()))?; futures::executor::block_on(nu::run_pipeline_standalone(
item.into(),
matches.is_present("stdin"),
))?;
} }
return Ok(()); return Ok(());
} }

View File

@ -91,6 +91,19 @@ mod external_words {
} }
} }
mod nu_commands {
use super::nu;
#[test]
fn echo_internally_externally() {
let actual = nu!(cwd: ".", r#"
nu -c "echo 'foo'"
"#);
assert_eq!(actual, "foo");
}
}
mod tilde_expansion { mod tilde_expansion {
use super::nu; use super::nu;