1use itertools::Itertools;
2use nu_cmd_base::hook::eval_hook;
3use nu_engine::{command_prelude::*, env_to_strings};
4use nu_path::{AbsolutePath, dots::expand_ndots_safe, expand_tilde};
5use nu_protocol::{
6 ByteStream, NuGlob, OutDest, Signals, UseAnsiColoring, did_you_mean,
7 process::{ChildProcess, PostWaitCallback},
8 shell_error::io::IoError,
9};
10use nu_system::{ForegroundChild, kill_by_pid};
11use nu_utils::IgnoreCaseExt;
12use pathdiff::diff_paths;
13#[cfg(windows)]
14use std::os::windows::process::CommandExt;
15use std::{
16 borrow::Cow,
17 ffi::{OsStr, OsString},
18 io::Write,
19 path::{Path, PathBuf},
20 process::Stdio,
21 sync::Arc,
22 thread,
23};
24
25#[derive(Clone)]
26pub struct External;
27
28impl Command for External {
29 fn name(&self) -> &str {
30 "run-external"
31 }
32
33 fn description(&self) -> &str {
34 "Runs external command."
35 }
36
37 fn extra_description(&self) -> &str {
38 "All externals are run with this command, whether you call it directly with `run-external external` or use `external` or `^external`.
39If you create a custom command with this name, that will be used instead."
40 }
41
42 fn signature(&self) -> nu_protocol::Signature {
43 Signature::build(self.name())
44 .input_output_types(vec![(Type::Any, Type::Any)])
45 .rest(
46 "command",
47 SyntaxShape::OneOf(vec![SyntaxShape::GlobPattern, SyntaxShape::Any]),
48 "External command to run, with arguments.",
49 )
50 .category(Category::System)
51 }
52
53 fn run(
54 &self,
55 engine_state: &EngineState,
56 stack: &mut Stack,
57 call: &Call,
58 input: PipelineData,
59 ) -> Result<PipelineData, ShellError> {
60 let cwd = engine_state.cwd(Some(stack))?;
61 let rest = call.rest::<Value>(engine_state, stack, 0)?;
62 let name_args = rest.split_first().map(|(x, y)| (x, y.to_vec()));
63
64 let Some((name, mut call_args)) = name_args else {
65 return Err(ShellError::MissingParameter {
66 param_name: "no command given".into(),
67 span: call.head,
68 });
69 };
70
71 let name_str: Cow<str> = match &name {
72 Value::Glob { val, .. } => Cow::Borrowed(val),
73 Value::String { val, .. } => Cow::Borrowed(val),
74 Value::List { vals, .. } => {
75 let Some((first, args)) = vals.split_first() else {
76 return Err(ShellError::MissingParameter {
77 param_name: "external command given as list empty".into(),
78 span: call.head,
79 });
80 };
81 call_args.splice(0..0, args.to_vec());
83 first.coerce_str()?
84 }
85 _ => Cow::Owned(name.clone().coerce_into_string()?),
86 };
87
88 let expanded_name = match &name {
89 Value::Glob { no_expand, .. } if !*no_expand => {
91 expand_ndots_safe(expand_tilde(&*name_str))
92 }
93 _ => Path::new(&*name_str).to_owned(),
94 };
95
96 let paths = nu_engine::env::path_str(engine_state, stack, call.head).unwrap_or_default();
97
98 let pathext_script_in_windows = if cfg!(windows) {
112 if let Some(executable) = which(&expanded_name, &paths, cwd.as_ref()) {
113 let ext = executable
114 .extension()
115 .unwrap_or_default()
116 .to_string_lossy()
117 .to_uppercase();
118
119 !["COM", "EXE", "BAT", "CMD", "PS1"]
120 .iter()
121 .any(|c| *c == ext)
122 } else {
123 false
124 }
125 } else {
126 false
127 };
128
129 let (potential_powershell_script, path_to_ps1_executable) = if cfg!(windows) {
131 if let Some(executable) = which(&expanded_name, &paths, cwd.as_ref()) {
132 let ext = executable
133 .extension()
134 .unwrap_or_default()
135 .to_string_lossy()
136 .to_uppercase();
137 (ext == "PS1", Some(executable))
138 } else {
139 (false, None)
140 }
141 } else {
142 (false, None)
143 };
144
145 let executable = if cfg!(windows)
149 && (is_cmd_internal_command(&name_str) || pathext_script_in_windows)
150 {
151 PathBuf::from("cmd.exe")
152 } else if cfg!(windows) && potential_powershell_script && path_to_ps1_executable.is_some() {
153 PathBuf::from("powershell.exe")
157 } else {
158 let Some(executable) = which(&expanded_name, &paths, cwd.as_ref()) else {
161 return Err(command_not_found(
162 &name_str,
163 call.head,
164 engine_state,
165 stack,
166 &cwd,
167 ));
168 };
169 executable
170 };
171
172 let mut command = std::process::Command::new(&executable);
174
175 command.current_dir(cwd);
177
178 let envs = env_to_strings(engine_state, stack)?;
180 command.env_clear();
181 command.envs(envs);
182
183 let args = eval_external_arguments(engine_state, stack, call_args)?;
185 #[cfg(windows)]
186 if is_cmd_internal_command(&name_str) || pathext_script_in_windows {
187 command.args(["/D", "/C", &expanded_name.to_string_lossy()]);
191 for arg in &args {
192 command.raw_arg(escape_cmd_argument(arg)?);
193 }
194 } else if potential_powershell_script {
195 command.args([
196 "-File",
197 &path_to_ps1_executable.unwrap_or_default().to_string_lossy(),
198 ]);
199 command.args(args.into_iter().map(|s| s.item));
200 } else {
201 command.args(args.into_iter().map(|s| s.item));
202 }
203 #[cfg(not(windows))]
204 command.args(args.into_iter().map(|s| s.item));
205
206 let stdout = stack.stdout();
209 let stderr = stack.stderr();
210 let merged_stream = if matches!(stdout, OutDest::Pipe) && matches!(stderr, OutDest::Pipe) {
211 let (reader, writer) =
212 os_pipe::pipe().map_err(|err| IoError::new(err, call.head, None))?;
213 command.stdout(
214 writer
215 .try_clone()
216 .map_err(|err| IoError::new(err, call.head, None))?,
217 );
218 command.stderr(writer);
219 Some(reader)
220 } else {
221 if engine_state.is_background_job()
222 && matches!(stdout, OutDest::Inherit | OutDest::Print)
223 {
224 command.stdout(Stdio::null());
225 } else {
226 command.stdout(
227 Stdio::try_from(stdout).map_err(|err| IoError::new(err, call.head, None))?,
228 );
229 }
230
231 if engine_state.is_background_job()
232 && matches!(stderr, OutDest::Inherit | OutDest::Print)
233 {
234 command.stderr(Stdio::null());
235 } else {
236 command.stderr(
237 Stdio::try_from(stderr).map_err(|err| IoError::new(err, call.head, None))?,
238 );
239 }
240
241 None
242 };
243
244 let data_to_copy_into_stdin = match input {
248 PipelineData::ByteStream(stream, metadata) => match stream.into_stdio() {
249 Ok(stdin) => {
250 command.stdin(stdin);
251 None
252 }
253 Err(stream) => {
254 command.stdin(Stdio::piped());
255 Some(PipelineData::byte_stream(stream, metadata))
256 }
257 },
258 PipelineData::Empty => {
259 if engine_state.is_mcp {
265 command.stdin(Stdio::null());
266 } else {
267 command.stdin(Stdio::inherit());
268 }
269 None
270 }
271 value => {
272 command.stdin(Stdio::piped());
273 Some(value)
274 }
275 };
276
277 log::trace!("run-external spawning: {command:?}");
279
280 #[cfg(windows)]
283 let child = ForegroundChild::spawn(command);
284 #[cfg(unix)]
285 let child = ForegroundChild::spawn(
286 command,
287 engine_state.is_interactive,
288 engine_state.is_background_job(),
289 &engine_state.pipeline_externals_state,
290 );
291
292 let mut child = child.map_err(|err| {
293 let context = format!("Could not spawn foreground child: {err}");
294 IoError::new_internal(err, context)
295 })?;
296
297 if let Some(thread_job) = engine_state.current_thread_job()
298 && !thread_job.try_add_pid(child.pid())
299 {
300 kill_by_pid(child.pid().into()).map_err(|err| {
301 ShellError::Io(IoError::new_internal(
302 err,
303 "Could not spawn external stdin worker",
304 ))
305 })?;
306 }
307
308 if let Some(data) = data_to_copy_into_stdin {
310 let stdin = child.as_mut().stdin.take().expect("stdin is piped");
311 let engine_state = engine_state.clone();
312 let stack = stack.clone();
313 thread::Builder::new()
314 .name("external stdin worker".into())
315 .spawn(move || {
316 let _ = write_pipeline_data(engine_state, stack, data, stdin);
317 })
318 .map_err(|err| {
319 IoError::new_with_additional_context(
320 err,
321 call.head,
322 None,
323 "Could not spawn external stdin worker",
324 )
325 })?;
326 }
327
328 let child_pid = child.pid();
329
330 let child = ChildProcess::new(
332 child,
333 merged_stream,
334 matches!(stderr, OutDest::Pipe),
335 call.head,
336 Some(PostWaitCallback::for_job_control(
337 engine_state,
338 Some(child_pid),
339 executable
340 .as_path()
341 .file_name()
342 .and_then(|it| it.to_str())
343 .map(|it| it.to_string()),
344 )),
345 )?;
346
347 Ok(PipelineData::byte_stream(
348 ByteStream::child(child, call.head),
349 None,
350 ))
351 }
352
353 fn examples(&self) -> Vec<Example<'_>> {
354 vec![
355 Example {
356 description: "Run an external command",
357 example: r#"run-external "echo" "-n" "hello""#,
358 result: None,
359 },
360 Example {
361 description: "Redirect stdout from an external command into the pipeline",
362 example: r#"run-external "echo" "-n" "hello" | split chars"#,
363 result: None,
364 },
365 Example {
366 description: "Redirect stderr from an external command into the pipeline",
367 example: r#"run-external "nu" "-c" "print -e hello" e>| split chars"#,
368 result: None,
369 },
370 ]
371 }
372}
373
374pub fn eval_external_arguments(
376 engine_state: &EngineState,
377 stack: &mut Stack,
378 call_args: Vec<Value>,
379) -> Result<Vec<Spanned<OsString>>, ShellError> {
380 let cwd = engine_state.cwd(Some(stack))?;
381 let mut args: Vec<Spanned<OsString>> = Vec::with_capacity(call_args.len());
382
383 for arg in call_args {
384 let span = arg.span();
385 match arg {
386 Value::Glob { val, no_expand, .. } if !no_expand => args.extend(
388 expand_glob(
389 &val,
390 cwd.as_std_path(),
391 span,
392 engine_state.signals().clone(),
393 )?
394 .into_iter()
395 .map(|s| s.into_spanned(span)),
396 ),
397 other => args
398 .push(OsString::from(coerce_into_string(engine_state, other)?).into_spanned(span)),
399 }
400 }
401 Ok(args)
402}
403
404fn coerce_into_string(engine_state: &EngineState, val: Value) -> Result<String, ShellError> {
407 match val {
408 Value::List { .. } => Err(ShellError::CannotPassListToExternal {
409 arg: String::from_utf8_lossy(engine_state.get_span_contents(val.span())).into_owned(),
410 span: val.span(),
411 }),
412 Value::Glob { val, .. } => Ok(val),
413 _ => val.coerce_into_string(),
414 }
415}
416
417fn expand_glob(
423 arg: &str,
424 cwd: &Path,
425 span: Span,
426 signals: Signals,
427) -> Result<Vec<OsString>, ShellError> {
428 if !nu_glob::is_glob_with_backend(arg) {
431 let path = expand_ndots_safe(expand_tilde(arg));
432 return Ok(vec![path.into()]);
433 }
434
435 let glob = NuGlob::Expand(arg.to_owned()).into_spanned(span);
438 if let Ok((prefix, matches)) = nu_engine::glob_from(&glob, cwd, span, None, signals.clone()) {
439 let mut result: Vec<OsString> = vec![];
440
441 for m in matches {
442 signals.check(&span)?;
443 if let Ok(arg) = m {
444 let arg = resolve_globbed_path_to_cwd_relative(arg, prefix.as_ref(), cwd);
445 result.push(arg.into());
446 } else {
447 result.push(arg.into());
448 }
449 }
450
451 if result.is_empty() {
454 result.push(arg.into());
455 }
456
457 Ok(result)
458 } else {
459 Ok(vec![arg.into()])
460 }
461}
462
463fn resolve_globbed_path_to_cwd_relative(
464 path: PathBuf,
465 prefix: Option<&PathBuf>,
466 cwd: &Path,
467) -> PathBuf {
468 if let Some(prefix) = prefix {
469 if let Ok(remainder) = path.strip_prefix(prefix) {
470 let new_prefix = if let Some(pfx) = diff_paths(prefix, cwd) {
471 pfx
472 } else {
473 prefix.to_path_buf()
474 };
475 new_prefix.join(remainder)
476 } else {
477 path
478 }
479 } else {
480 path
481 }
482}
483
484fn write_pipeline_data(
492 mut engine_state: EngineState,
493 mut stack: Stack,
494 data: PipelineData,
495 mut writer: impl Write,
496) -> Result<(), ShellError> {
497 if let PipelineData::ByteStream(stream, ..) = data {
498 stream.write_to(writer)?;
499 } else if let PipelineData::Value(Value::Binary { val, .. }, ..) = data {
500 writer
501 .write_all(&val)
502 .map_err(|err| IoError::new_internal(err, "Could not write pipeline data"))?;
503 } else {
504 stack.start_collect_value();
505
506 Arc::make_mut(&mut engine_state.config).use_ansi_coloring = UseAnsiColoring::False;
508
509 let output =
511 crate::Table.run(&engine_state, &mut stack, &Call::new(Span::unknown()), data)?;
512
513 for value in output {
515 let bytes = value.coerce_into_binary()?;
516 writer
517 .write_all(&bytes)
518 .map_err(|err| IoError::new_internal(err, "Could not write pipeline data"))?;
519 }
520 }
521 Ok(())
522}
523
524pub fn command_not_found(
526 name: &str,
527 span: Span,
528 engine_state: &EngineState,
529 stack: &mut Stack,
530 cwd: &AbsolutePath,
531) -> ShellError {
532 if let Some(hook) = &stack.get_config(engine_state).hooks.command_not_found {
534 let mut stack = stack.start_collect_value();
535 let canary = "ENTERED_COMMAND_NOT_FOUND";
538 if stack.has_env_var(engine_state, canary) {
539 return ShellError::ExternalCommand {
540 label: format!(
541 "Command {name} not found while running the `command_not_found` hook"
542 ),
543 help: "Make sure the `command_not_found` hook itself does not use unknown commands"
544 .into(),
545 span,
546 };
547 }
548 stack.add_env_var(canary.into(), Value::bool(true, Span::unknown()));
549
550 let output = eval_hook(
551 &mut engine_state.clone(),
552 &mut stack,
553 None,
554 vec![("cmd_name".into(), Value::string(name, span))],
555 hook,
556 "command_not_found",
557 );
558
559 stack.remove_env_var(engine_state, canary);
561
562 match output {
563 Ok(PipelineData::Value(Value::String { val, .. }, ..)) => {
564 return ShellError::ExternalCommand {
565 label: format!("Command `{name}` not found"),
566 help: val,
567 span,
568 };
569 }
570 Err(err) => {
571 return err;
572 }
573 _ => {
574 }
576 }
577 }
578
579 if let Some(replacement) = crate::removed_commands().get(&name.to_lowercase()) {
581 return ShellError::RemovedCommand {
582 removed: name.to_lowercase(),
583 replacement: replacement.clone(),
584 span,
585 };
586 }
587
588 let help = (|| {
590 if let Some(module) = engine_state.which_module_has_decl(name.as_bytes(), &[]) {
594 let module = String::from_utf8_lossy(module);
595
596 let full_name = format!("{module} {name}");
598 if engine_state.find_decl(full_name.as_bytes(), &[]).is_some() {
599 return format!("Did you mean `{full_name}`?");
600 }
601
602 return format!(
603 "A command with that name exists in module `{module}`. Try importing it with `use`"
604 );
605 }
606
607 let signatures = engine_state.get_signatures_and_declids(false);
609 if let Some((last, others)) = signatures
610 .iter()
611 .map(|(sig, _)| sig)
612 .filter(|sig| {
613 let name = name.to_folded_case(); sig.name
615 .to_folded_case()
616 .split_ascii_whitespace() .contains(name.as_str()) || sig
619 .search_terms
620 .iter()
621 .any(|term| term.to_folded_case() == name)
622 })
623 .map(|sig| format!("`{}`", sig.name))
624 .collect::<Vec<_>>()
625 .split_last()
626 {
627 let commands = if others.is_empty() {
628 last
629 } else {
630 &format!("{} or {last}", others.join(", "))
633 };
634
635 return format!("Did you mean {commands}?");
636 }
637
638 if let Some(cmd) = did_you_mean(signatures.iter().map(|(sig, _)| &sig.name), name) {
640 if cmd == name {
643 return "There is a built-in command with the same name".to_string();
644 }
645
646 return format!("Did you mean `{cmd}`?");
647 }
648
649 if cwd.join(name).is_file() {
651 return format!(
652 "`{name}` refers to a file that is not executable. Did you forget to set execute permissions?"
653 );
654 }
655
656 format!("`{name}` is neither a Nushell built-in or a known external command")
658 })();
659
660 ShellError::ExternalCommand {
661 label: format!("Command `{name}` not found"),
662 help,
663 span,
664 }
665}
666
667pub fn which(name: impl AsRef<OsStr>, paths: &str, cwd: &Path) -> Option<PathBuf> {
677 #[cfg(windows)]
678 let paths = format!("{};{}", cwd.display(), paths);
679 which::which_in(name, Some(paths), cwd).ok()
680}
681
682fn is_cmd_internal_command(name: &str) -> bool {
685 const COMMANDS: &[&str] = &[
686 "ASSOC", "CLS", "ECHO", "FTYPE", "MKLINK", "PAUSE", "START", "VER", "VOL",
687 ];
688 COMMANDS.iter().any(|cmd| cmd.eq_ignore_ascii_case(name))
689}
690
691fn has_cmd_special_character(s: impl AsRef<[u8]>) -> bool {
693 s.as_ref()
694 .iter()
695 .any(|b| matches!(b, b'<' | b'>' | b'&' | b'|' | b'^'))
696}
697
698#[cfg_attr(not(windows), allow(dead_code))]
700fn escape_cmd_argument(arg: &Spanned<OsString>) -> Result<Cow<'_, OsStr>, ShellError> {
701 let Spanned { item: arg, span } = arg;
702 let bytes = arg.as_encoded_bytes();
703 if bytes.iter().any(|b| matches!(b, b'\r' | b'\n' | b'%')) {
704 Err(ShellError::ExternalCommand {
706 label:
707 "Arguments to CMD internal commands cannot contain new lines or percent signs '%'"
708 .into(),
709 help: "some characters currently cannot be securely escaped".into(),
710 span: *span,
711 })
712 } else if bytes.contains(&b'"') {
713 if bytes.iter().filter(|b| **b == b'"').count() == 2
716 && bytes.starts_with(b"\"")
717 && bytes.ends_with(b"\"")
718 {
719 Ok(Cow::Borrowed(arg))
720 } else {
721 Err(ShellError::ExternalCommand {
722 label: "Arguments to CMD internal commands cannot contain embedded double quotes"
723 .into(),
724 help: "this case currently cannot be securely handled".into(),
725 span: *span,
726 })
727 }
728 } else if bytes.contains(&b' ') || has_cmd_special_character(bytes) {
729 let mut new_str = OsString::new();
731 new_str.push("\"");
732 new_str.push(arg);
733 new_str.push("\"");
734 Ok(Cow::Owned(new_str))
735 } else {
736 Ok(Cow::Borrowed(arg))
738 }
739}
740
741#[cfg(test)]
742mod test {
743 use super::*;
744 use nu_test_support::{fs::Stub, playground::Playground};
745
746 #[test]
747 fn test_expand_glob() {
748 Playground::setup("test_expand_glob", |dirs, play| {
749 play.with_files(&[Stub::EmptyFile("a.txt"), Stub::EmptyFile("b.txt")]);
750
751 let cwd = dirs.test().as_std_path();
752
753 let actual = expand_glob("*.txt", cwd, Span::test_data(), Signals::empty()).unwrap();
754 let expected = &["a.txt", "b.txt"];
755 assert_eq!(actual, expected);
756
757 let actual = expand_glob("./*.txt", cwd, Span::test_data(), Signals::empty()).unwrap();
758 assert_eq!(actual, expected);
759
760 let actual = expand_glob("'*.txt'", cwd, Span::test_data(), Signals::empty()).unwrap();
761 let expected = &["'*.txt'"];
762 assert_eq!(actual, expected);
763
764 let actual = expand_glob(".", cwd, Span::test_data(), Signals::empty()).unwrap();
765 let expected = &["."];
766 assert_eq!(actual, expected);
767
768 let actual = expand_glob("./a.txt", cwd, Span::test_data(), Signals::empty()).unwrap();
769 let expected = &["./a.txt"];
770 assert_eq!(actual, expected);
771
772 let actual = expand_glob("[*.txt", cwd, Span::test_data(), Signals::empty()).unwrap();
773 let expected = &["[*.txt"];
774 assert_eq!(actual, expected);
775
776 let actual =
777 expand_glob("~/foo.txt", cwd, Span::test_data(), Signals::empty()).unwrap();
778 let home = dirs::home_dir().expect("failed to get home dir");
779 let expected: Vec<OsString> = vec![home.join("foo.txt").into()];
780 assert_eq!(actual, expected);
781 })
782 }
783
784 #[test]
785 fn test_write_pipeline_data() {
786 let mut engine_state = EngineState::new();
787 let stack = Stack::new();
788 let cwd = std::env::current_dir()
789 .unwrap()
790 .into_os_string()
791 .into_string()
792 .unwrap();
793
794 engine_state.add_env_var("PWD".into(), Value::string(cwd, Span::test_data()));
796
797 let mut buf = vec![];
798 let input = PipelineData::empty();
799 write_pipeline_data(engine_state.clone(), stack.clone(), input, &mut buf).unwrap();
800 assert_eq!(buf, b"");
801
802 let mut buf = vec![];
803 let input = PipelineData::value(Value::string("foo", Span::test_data()), None);
804 write_pipeline_data(engine_state.clone(), stack.clone(), input, &mut buf).unwrap();
805 assert_eq!(buf, b"foo");
806
807 let mut buf = vec![];
808 let input = PipelineData::value(Value::binary(b"foo", Span::test_data()), None);
809 write_pipeline_data(engine_state.clone(), stack.clone(), input, &mut buf).unwrap();
810 assert_eq!(buf, b"foo");
811
812 let mut buf = vec![];
813 let input = PipelineData::byte_stream(
814 ByteStream::read(
815 b"foo".as_slice(),
816 Span::test_data(),
817 Signals::empty(),
818 ByteStreamType::Unknown,
819 ),
820 None,
821 );
822 write_pipeline_data(engine_state.clone(), stack.clone(), input, &mut buf).unwrap();
823 assert_eq!(buf, b"foo");
824 }
825}