mirror of
https://github.com/nushell/nushell.git
synced 2024-12-12 10:10:51 +01:00
d06f457b2a
* move commands, futures.rs, script.rs, utils * move over maybe_print_errors * add nu_command crate references to nu_cli * in commands.rs open up to pub mod from pub(crate) * nu-cli, nu-command, and nu tests are now passing * cargo fmt * clean up nu-cli/src/prelude.rs * code cleanup * for some reason lex.rs was not formatted, may be causing my error * remove mod completion from lib.rs which was not being used along with quickcheck macros * add in allow unused imports * comment out one failing external test; comment out one failing internal test * revert commenting out failing tests; something else might be going on; someone with a windows machine should check and see what is going on with these failing windows tests * Update Cargo.toml Extend the optional features to nu-command Co-authored-by: Jonathan Turner <jonathandturner@users.noreply.github.com>
126 lines
2.9 KiB
Rust
126 lines
2.9 KiB
Rust
use crate::prelude::*;
|
|
use nu_engine::WholeStreamCommand;
|
|
use nu_errors::ShellError;
|
|
use nu_protocol::{Signature, SyntaxShape};
|
|
use nu_source::Tagged;
|
|
use std::process::{Command, Stdio};
|
|
|
|
pub struct Kill;
|
|
|
|
#[derive(Deserialize)]
|
|
pub struct KillArgs {
|
|
pub pid: Tagged<u64>,
|
|
pub rest: Vec<Tagged<u64>>,
|
|
pub force: Tagged<bool>,
|
|
pub quiet: Tagged<bool>,
|
|
}
|
|
|
|
#[async_trait]
|
|
impl WholeStreamCommand for Kill {
|
|
fn name(&self) -> &str {
|
|
"kill"
|
|
}
|
|
|
|
fn signature(&self) -> Signature {
|
|
Signature::build("kill")
|
|
.required(
|
|
"pid",
|
|
SyntaxShape::Int,
|
|
"process id of process that is to be killed",
|
|
)
|
|
.rest(SyntaxShape::Int, "rest of processes to kill")
|
|
.switch("force", "forcefully kill the process", Some('f'))
|
|
.switch("quiet", "won't print anything to the console", Some('q'))
|
|
}
|
|
|
|
fn usage(&self) -> &str {
|
|
"Kill a process using the process id."
|
|
}
|
|
|
|
async fn run(&self, args: CommandArgs) -> Result<OutputStream, ShellError> {
|
|
kill(args).await
|
|
}
|
|
|
|
fn examples(&self) -> Vec<Example> {
|
|
vec![
|
|
Example {
|
|
description: "Kill the pid using the most memory",
|
|
example: "ps | sort-by mem | last | kill $it.pid",
|
|
result: None,
|
|
},
|
|
Example {
|
|
description: "Force kill a given pid",
|
|
example: "kill --force 12345",
|
|
result: None,
|
|
},
|
|
]
|
|
}
|
|
}
|
|
|
|
async fn kill(args: CommandArgs) -> Result<OutputStream, ShellError> {
|
|
let (
|
|
KillArgs {
|
|
pid,
|
|
rest,
|
|
force,
|
|
quiet,
|
|
},
|
|
..,
|
|
) = args.process().await?;
|
|
let mut cmd = if cfg!(windows) {
|
|
let mut cmd = Command::new("taskkill");
|
|
|
|
if *force {
|
|
cmd.arg("/F");
|
|
}
|
|
|
|
cmd.arg("/PID");
|
|
cmd.arg(pid.item().to_string());
|
|
|
|
// each pid must written as `/PID 0` otherwise
|
|
// taskkill will act as `killall` unix command
|
|
for id in &rest {
|
|
cmd.arg("/PID");
|
|
cmd.arg(id.item().to_string());
|
|
}
|
|
|
|
cmd
|
|
} else {
|
|
let mut cmd = Command::new("kill");
|
|
|
|
if *force {
|
|
cmd.arg("-9");
|
|
}
|
|
|
|
cmd.arg(pid.item().to_string());
|
|
|
|
cmd.args(rest.iter().map(move |id| id.item().to_string()));
|
|
|
|
cmd
|
|
};
|
|
|
|
// pipe everything to null
|
|
if *quiet {
|
|
cmd.stdin(Stdio::null())
|
|
.stdout(Stdio::null())
|
|
.stderr(Stdio::null());
|
|
}
|
|
|
|
cmd.status().expect("failed to execute shell command");
|
|
|
|
Ok(OutputStream::empty())
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::Kill;
|
|
use super::ShellError;
|
|
|
|
#[test]
|
|
fn examples_work_as_expected() -> Result<(), ShellError> {
|
|
use crate::examples::test as test_examples;
|
|
|
|
Ok(test_examples(Kill {})?)
|
|
}
|
|
}
|