-i flag finished, lacking tests

This commit is contained in:
Gabriel B Gutierrez 2021-10-14 14:54:51 -03:00
parent 9ea7cdfc33
commit 8c2ae1eed1
8 changed files with 145 additions and 15 deletions

40
Cargo.lock generated
View file

@ -154,6 +154,21 @@ dependencies = [
"chrono", "chrono",
] ]
[[package]]
name = "console"
version = "0.14.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3993e6445baa160675931ec041a5e03ca84b9c6e32a056150d3aa2bdda0a1f45"
dependencies = [
"encode_unicode",
"lazy_static",
"libc",
"regex",
"terminal_size",
"unicode-width",
"winapi",
]
[[package]] [[package]]
name = "core-foundation-sys" name = "core-foundation-sys"
version = "0.8.2" version = "0.8.2"
@ -240,6 +255,18 @@ dependencies = [
"syn", "syn",
] ]
[[package]]
name = "dialoguer"
version = "0.8.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c9dd058f8b65922819fabb4a41e7d1964e56344042c26efbccd465202c23fa0c"
dependencies = [
"console",
"lazy_static",
"tempfile",
"zeroize",
]
[[package]] [[package]]
name = "diff" name = "diff"
version = "0.1.12" version = "0.1.12"
@ -291,6 +318,12 @@ version = "1.6.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e78d4f1cc4ae33bbfc157ed5d5a5ef3bc29227303d595861deb238fcec4e9457" checksum = "e78d4f1cc4ae33bbfc157ed5d5a5ef3bc29227303d595861deb238fcec4e9457"
[[package]]
name = "encode_unicode"
version = "0.3.6"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a357d28ed41a50f9c765dbfe56cbc04a64e53e5fc58ba79fbc34c10ef3df831f"
[[package]] [[package]]
name = "engine-q" name = "engine-q"
version = "0.1.0" version = "0.1.0"
@ -541,6 +574,7 @@ version = "0.1.0"
dependencies = [ dependencies = [
"bytesize", "bytesize",
"chrono", "chrono",
"dialoguer",
"glob", "glob",
"lscolors", "lscolors",
"nu-engine", "nu-engine",
@ -1258,3 +1292,9 @@ name = "winapi-x86_64-pc-windows-gnu"
version = "0.4.0" version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f" checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f"
[[package]]
name = "zeroize"
version = "1.4.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bf68b08513768deaa790264a7fac27a58cbf2705cfcdc9448362229217d7e970"

View file

@ -25,6 +25,7 @@ chrono = { version = "0.4.19", features = ["serde"] }
terminal_size = "0.1.17" terminal_size = "0.1.17"
lscolors = { version = "0.8.0", features = ["crossterm"] } lscolors = { version = "0.8.0", features = ["crossterm"] }
bytesize = "1.1.0" bytesize = "1.1.0"
dialoguer = "0.8.0"
[features] [features]
trash-support = ["trash"] trash-support = ["trash"]

View file

@ -1,6 +1,7 @@
use std::env::current_dir; use std::env::current_dir;
use std::path::PathBuf; use std::path::PathBuf;
use super::interactive_helper::get_confirmation;
use nu_engine::CallExt; use nu_engine::CallExt;
use nu_path::canonicalize_with; use nu_path::canonicalize_with;
use nu_protocol::ast::Call; use nu_protocol::ast::Call;
@ -29,6 +30,7 @@ impl Command for Cp {
"copy recursively through subdirectories", "copy recursively through subdirectories",
Some('r'), Some('r'),
) )
.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'))
} }
@ -41,19 +43,13 @@ impl Command for Cp {
let source: String = call.req(context, 0)?; let source: String = call.req(context, 0)?;
let destination: String = call.req(context, 1)?; let destination: String = call.req(context, 1)?;
let interactive = call.has_flag("interactive"); let interactive = call.has_flag("interactive");
let force = call.has_flag("force");
if interactive {
println!(
"Are you shure that you want to move {} to {}?",
source, destination
);
}
let path: PathBuf = current_dir().unwrap(); let path: PathBuf = current_dir().unwrap();
let source = path.join(source.as_str()); let source = path.join(source.as_str());
let destination = path.join(destination.as_str()); let destination = path.join(destination.as_str());
let sources = let mut sources =
glob::glob(&source.to_string_lossy()).map_or_else(|_| Vec::new(), Iterator::collect); glob::glob(&source.to_string_lossy()).map_or_else(|_| Vec::new(), Iterator::collect);
if sources.is_empty() { if sources.is_empty() {
return Err(ShellError::FileNotFound(call.positional[0].span)); return Err(ShellError::FileNotFound(call.positional[0].span));
@ -77,6 +73,35 @@ impl Command for Cp {
)); ));
} }
if interactive && !force {
let mut remove_index: Vec<usize> = vec![];
for (index, file) in sources.iter().enumerate() {
let prompt = format!(
"Are you shure that you want to copy {} to {}?",
file.as_ref()
.unwrap()
.file_name()
.unwrap()
.to_str()
.unwrap(),
destination.file_name().unwrap().to_str().unwrap()
);
let input = get_confirmation(prompt)?;
if !input {
remove_index.push(index);
}
}
for index in remove_index {
sources.remove(index);
}
if sources.len() == 0 {
return Err(ShellError::NoFileToBeCopied());
}
}
for entry in sources.into_iter().flatten() { for entry in sources.into_iter().flatten() {
let mut sources = FileStructure::new(); let mut sources = FileStructure::new();
sources.walk_decorate(&entry)?; sources.walk_decorate(&entry)?;

View file

@ -0,0 +1,26 @@
use dialoguer::Input;
use std::error::Error;
pub fn get_confirmation(prompt: String) -> Result<bool, Box<dyn Error>> {
let input = Input::new()
.with_prompt(prompt)
.validate_with(|c_input: &String| -> Result<(), String> {
if c_input.len() == 1
&& (c_input == "y" || c_input == "Y" || c_input == "n" || c_input == "N")
{
Ok(())
} else if c_input.len() > 1 {
Err("Enter only one letter (Y/N)".to_string())
} else {
Err("Input not valid".to_string())
}
})
.default("Y/N".into())
.interact_text()?;
if input == "y" || input == "Y" {
Ok(true)
} else {
Ok(false)
}
}

View file

@ -1,5 +1,6 @@
mod cd; mod cd;
mod cp; mod cp;
mod interactive_helper;
mod ls; mod ls;
mod mkdir; mod mkdir;
mod mv; mod mv;

View file

@ -1,6 +1,7 @@
use std::env::current_dir; use std::env::current_dir;
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
use super::interactive_helper::get_confirmation;
use nu_engine::CallExt; use nu_engine::CallExt;
use nu_protocol::ast::Call; use nu_protocol::ast::Call;
use nu_protocol::engine::{Command, EvaluationContext}; use nu_protocol::engine::{Command, EvaluationContext};
@ -59,8 +60,9 @@ impl Command for Mv {
} }
if interactive && !force { if interactive && !force {
for file in &sources { let mut remove_index: Vec<usize> = vec![];
println!( for (index, file) in sources.iter().enumerate() {
let prompt = format!(
"Are you shure that you want to move {} to {}?", "Are you shure that you want to move {} to {}?",
file.as_ref() file.as_ref()
.unwrap() .unwrap()
@ -70,6 +72,19 @@ impl Command for Mv {
.unwrap(), .unwrap(),
destination.file_name().unwrap().to_str().unwrap() destination.file_name().unwrap().to_str().unwrap()
); );
let input = get_confirmation(prompt)?;
if !input {
remove_index.push(index);
}
}
for index in remove_index {
sources.remove(index);
}
if sources.len() == 0 {
return Err(ShellError::NoFileToBeMoved());
} }
} }

View file

@ -3,6 +3,8 @@ use std::env::current_dir;
use std::os::unix::prelude::FileTypeExt; use std::os::unix::prelude::FileTypeExt;
use std::path::PathBuf; use std::path::PathBuf;
use super::interactive_helper::get_confirmation;
use nu_engine::CallExt; use nu_engine::CallExt;
use nu_protocol::ast::Call; use nu_protocol::ast::Call;
use nu_protocol::engine::{Command, EvaluationContext}; use nu_protocol::engine::{Command, EvaluationContext};
@ -19,7 +21,6 @@ struct RmArgs {
trash: bool, trash: bool,
permanent: bool, permanent: bool,
force: bool, force: bool,
interactive: bool,
} }
impl Command for Rm { impl Command for Rm {
@ -126,11 +127,26 @@ fn rm(context: &EvaluationContext, call: &Call) -> Result<Value, ShellError> {
let force = call.has_flag("force"); let force = call.has_flag("force");
if interactive && !force { if interactive && !force {
for file in &targets { let mut remove_index: Vec<usize> = vec![];
println!( for (index, file) in targets.iter().enumerate() {
"Are you shure that you want to delete {}?", let prompt: String = format!(
"Are you sure that you what to delete {}?",
file.1.file_name().unwrap().to_str().unwrap() file.1.file_name().unwrap().to_str().unwrap()
); );
let input = get_confirmation(prompt)?;
if !input {
remove_index.push(index);
}
}
for index in remove_index {
targets.remove(index);
}
if targets.len() == 0 {
return Err(ShellError::NoFileToBeRemoved());
} }
} }
@ -140,7 +156,6 @@ fn rm(context: &EvaluationContext, call: &Call) -> Result<Value, ShellError> {
trash, trash,
permanent, permanent,
force, force,
interactive,
}; };
let response = rm_helper(call, args); let response = rm_helper(call, args);

View file

@ -158,6 +158,13 @@ pub enum ShellError {
#[error("Remove not possible")] #[error("Remove not possible")]
#[diagnostic(code(nu::shell::remove_not_possible), url(docsrs))] #[diagnostic(code(nu::shell::remove_not_possible), url(docsrs))]
RemoveNotPossible(String, #[label("{0}")] Span), RemoveNotPossible(String, #[label("{0}")] Span),
#[error("No file to be removed")]
NoFileToBeRemoved(),
#[error("No file to be moved")]
NoFileToBeMoved(),
#[error("No file to be copied")]
NoFileToBeCopied(),
} }
impl From<std::io::Error> for ShellError { impl From<std::io::Error> for ShellError {