extern/nushell
1
0
Fork 1
mirror of https://github.com/nushell/nushell.git synced 2024-11-23 08:53:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

cp, mv, and rm commands need to support -i flag (#5523)

Browse Source 
* restored interactive mode to rm command

* removed unnecessary whitespace in rm file

* removed unnecessary whitespace in rm file

* fixed python-vertualenv build issue

* moved interactive logic to utils file

* restored interactive mode to cp command

* interactive mode for mv wip

* finished mv implementation

* removed unnecessary whitespace

* changed unwrap to expect
This commit is contained in:
njbull4 2022-05-18 07:53:46 -07:00 committed by GitHub
parent 9c779b071b
commit 2c58beec13
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 147 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

98
crates/nu-command/src/filesystem/cp.rs
View File

@ -6,10 +6,12 @@ use nu_path::canonicalize_with;
use nu_protocol::ast::Call;
use nu_protocol::engine::{Command, EngineState, Stack};
use nu_protocol::{
Category, Example, IntoInterruptiblePipelineData, PipelineData, ShellError, Signature, Spanned,
SyntaxShape, Value,
Category, Example, IntoInterruptiblePipelineData, PipelineData, ShellError, Signature, Span,
Spanned, SyntaxShape, Value,
};
use super::util::try_interaction;
use crate::filesystem::util::FileStructure;
const GLOB_PARAMS: nu_glob::MatchOptions = nu_glob::MatchOptions {
@ -51,7 +53,7 @@ impl Command for Cp {
)
// TODO: add back in additional features
// .switch("force", "suppress error when no file", Some('f'))
// .switch("interactive", "ask user to confirm action", Some('i'))
.switch("interactive", "ask user to confirm action", Some('i'))
.category(Category::FileSystem)
}
@ -66,6 +68,7 @@ impl Command for Cp {
let dst: Spanned<String> = call.req(engine_state, stack, 1)?;
let recursive = call.has_flag("recursive");
let verbose = call.has_flag("verbose");
let interactive = call.has_flag("interactive");
let path = current_dir(engine_state, stack)?;
let source = path.join(src.item.as_str());
@ -140,25 +143,12 @@ impl Command for Cp {
for (src, dst) in sources {
if src.is_file() {
match std::fs::copy(&src, &dst) {
Ok(_) => {
let msg =
format!("copied {:} to {:}", src.display(), dst.display());
result.push(Value::String { val: msg, span });
}
Err(e) => {
let error = Value::Error {
error: ShellError::GenericError(
e.to_string(),
e.to_string(),
Some(span),
None,
Vec::new(),
),
};
result.push(error);
}
}
let res = if interactive && dst.exists() {
interactive_copy_file(interactive, src, dst, span)
} else {
copy_file(src, dst, span)
};
result.push(res);
}
}
} else if entry.is_dir() {
@ -222,25 +212,12 @@ impl Command for Cp {
}
if s.is_file() {
match std::fs::copy(&s, &d) {
Ok(_) => {
let msg = format!("copied {:} to {:}", &s.display(), &d.display());
result.push(Value::String { val: msg, span });
}
Err(e) => {
let msg = "Can not copy source".to_string();
let error = Value::Error {
error: ShellError::GenericError(
msg,
e.to_string(),
Some(span),
None,
Vec::new(),
),
};
result.push(error);
}
}
let res = if interactive && d.exists() {
interactive_copy_file(interactive, s, d, span)
} else {
copy_file(s, d, span)
};
result.push(res);
}
}
}
@ -457,3 +434,42 @@ impl Command for Cp {
// Ok(PipelineData::new(call.head))
// }
}
fn interactive_copy_file(interactive: bool, src: PathBuf, dst: PathBuf, span: Span) -> Value {
let (interaction, confirmed) =
try_interaction(interactive, "cp: overwrite", &dst.to_string_lossy());
if let Err(e) = interaction {
Value::Error {
error: ShellError::GenericError(
e.to_string(),
e.to_string(),
Some(span),
None,
Vec::new(),
),
}
} else if !confirmed {
let msg = format!("{:} not copied to {:}", src.display(), dst.display());
Value::String { val: msg, span }
} else {
copy_file(src, dst, span)
}
}
fn copy_file(src: PathBuf, dst: PathBuf, span: Span) -> Value {
match std::fs::copy(&src, &dst) {
Ok(_) => {
let msg = format!("copied {:} to {:}", src.display(), dst.display());
Value::String { val: msg, span }
}
Err(e) => Value::Error {
error: ShellError::GenericError(
e.to_string(),
e.to_string(),
Some(span),
None,
Vec::new(),
),
},
}
}

49
crates/nu-command/src/filesystem/mv.rs
View File

@ -1,6 +1,6 @@
use std::path::{Path, PathBuf};
// use super::util::get_interactive_confirmation;
use super::util::try_interaction;
use nu_engine::env::current_dir;
use nu_engine::CallExt;
use nu_protocol::ast::Call;
@ -50,7 +50,7 @@ impl Command for Mv {
"make mv to be verbose, showing files been moved.",
Some('v'),
)
// .switch("interactive", "ask user to confirm action", Some('i'))
.switch("interactive", "ask user to confirm action", Some('i'))
// .switch("force", "suppress error when no file", Some('f'))
.category(Category::FileSystem)
}
@ -66,7 +66,7 @@ impl Command for Mv {
let spanned_source: Spanned<String> = call.req(engine_state, stack, 0)?;
let spanned_destination: Spanned<String> = call.req(engine_state, stack, 1)?;
let verbose = call.has_flag("verbose");
// let interactive = call.has_flag("interactive");
let interactive = call.has_flag("interactive");
// let force = call.has_flag("force");
let ctrlc = engine_state.ctrlc.clone();
@ -149,15 +149,24 @@ impl Command for Mv {
item: destination.clone(),
span: spanned_destination.span,
},
interactive,
);
if let Err(error) = result {
Some(Value::Error { error })
} else if verbose {
let val = format!(
"moved {:} to {:}",
entry.to_string_lossy(),
destination.to_string_lossy()
);
let val = if result.expect("Error value when unwrapping mv result") {
format!(
"moved {:} to {:}",
entry.to_string_lossy(),
destination.to_string_lossy()
)
} else {
format!(
"{:} not moved to {:}",
entry.to_string_lossy(),
destination.to_string_lossy()
)
};
Some(Value::String { val, span })
} else {
None
@ -190,7 +199,8 @@ impl Command for Mv {
fn move_file(
spanned_from: Spanned<PathBuf>,
spanned_to: Spanned<PathBuf>,
) -> Result<(), ShellError> {
interactive: bool,
) -> Result<bool, ShellError> {
let Spanned {
item: from,
span: from_span,
@ -229,7 +239,26 @@ fn move_file(
to.push(from_file_name);
}
move_item(&from, from_span, &to)
if interactive && to.exists() {
let (interaction, confirmed) =
try_interaction(interactive, "mv: overwrite", &to.to_string_lossy());
if let Err(e) = interaction {
return Err(ShellError::GenericError(
format!("Error during interaction: {:}", e),
"could not move".into(),
None,
None,
Vec::new(),
));
} else if !confirmed {
return Ok(false);
}
}
match move_item(&from, from_span, &to) {
Ok(()) => Ok(true),
Err(e) => Err(e),
}
}
fn move_item(from: &Path, from_span: Span, to: &Path) -> Result<(), ShellError> {

35
crates/nu-command/src/filesystem/rm.rs
View File

@ -9,7 +9,7 @@ use std::io::ErrorKind;
use std::os::unix::prelude::FileTypeExt;
use std::path::PathBuf;
// use super::util::get_interactive_confirmation;
use super::util::try_interaction;
use nu_engine::env::current_dir;
use nu_engine::CallExt;
@ -67,7 +67,7 @@ impl Command for Rm {
"make rm to be verbose, showing files been deleted",
Some('v'),
)
// .switch("interactive", "ask user to confirm action", Some('i'))
.switch("interactive", "ask user to confirm action", Some('i'))
.rest(
"rest",
SyntaxShape::GlobPattern,
@ -134,7 +134,7 @@ fn rm(
let recursive = call.has_flag("recursive");
let force = call.has_flag("force");
let verbose = call.has_flag("verbose");
// let interactive = call.has_flag("interactive");
let interactive = call.has_flag("interactive");
let ctrlc = engine_state.ctrlc.clone();
@ -289,6 +289,9 @@ fn rm(
|| is_fifo
|| is_empty()
{
let (interaction, confirmed) =
try_interaction(interactive, "rm: remove", &f.to_string_lossy());
let result;
#[cfg(all(
feature = "trash-support",
@ -297,9 +300,14 @@ fn rm(
))]
{
use std::io::Error;
result = if trash || (rm_always_trash && !permanent) {
result = if let Err(e) = interaction {
let e = Error::new(ErrorKind::Other, &*e.to_string());
Err(e)
} else if interactive && !confirmed {
Ok(())
} else if trash || (rm_always_trash && !permanent) {
trash::delete(&f).map_err(|e: trash::Error| {
Error::new(ErrorKind::Other, format!("{:?}", e))
Error::new(ErrorKind::Other, format!("{:?}\nTry '--trash' flag", e))
})
} else if metadata.is_file() || is_socket || is_fifo {
std::fs::remove_file(&f)
@ -313,7 +321,13 @@ fn rm(
target_os = "ios"
))]
{
result = if metadata.is_file() || is_socket || is_fifo {
use std::io::{Error, ErrorKind};
result = if let Err(e) = interaction {
let e = Error::new(ErrorKind::Other, &*e.to_string());
Err(e)
} else if interactive && !confirmed {
Ok(())
} else if metadata.is_file() || is_socket || is_fifo {
std::fs::remove_file(&f)
} else {
std::fs::remove_dir_all(&f)
@ -321,7 +335,7 @@ fn rm(
}
if let Err(e) = result {
let msg = format!("Could not delete because: {:}\nTry '--trash' flag", e);
let msg = format!("Could not delete because: {:}", e);
Value::Error {
error: ShellError::GenericError(
msg,
@ -332,7 +346,12 @@ fn rm(
),
}
} else if verbose {
let val = format!("deleted {:}", f.to_string_lossy());
let msg = if interactive && !confirmed {
"not deleted"
} else {
"deleted"
};
let val = format!("{} {:}", msg, f.to_string_lossy());
Value::String { val, span }
} else {
Value::Nothing { span }

25
crates/nu-command/src/filesystem/util.rs
View File

@ -97,8 +97,31 @@ pub struct Resource {
impl Resource {}
pub fn try_interaction(
interactive: bool,
prompt_msg: &str,
file_name: &str,
) -> (Result<Option<bool>, Box<dyn Error>>, bool) {
let interaction = if interactive {
let prompt = format!("{} '{}'? ", prompt_msg, file_name);
match get_interactive_confirmation(prompt) {
Ok(i) => Ok(Some(i)),
Err(e) => Err(e),
}
} else {
Ok(None)
};
let confirmed = match interaction {
Ok(maybe_input) => maybe_input.unwrap_or(false),
Err(_) => false,
};
(interaction, confirmed)
}
#[allow(dead_code)]
pub fn get_interactive_confirmation(prompt: String) -> Result<bool, Box<dyn Error>> {
fn get_interactive_confirmation(prompt: String) -> Result<bool, Box<dyn Error>> {
let input = Input::new()
.with_prompt(prompt)
.validate_with(|c_input: &String| -> Result<(), String> {