holm/nushell
1
1
Fork 0
forked from extern/nushell
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make more visible the strategies for figuring out where to copy files.

Browse Source
This commit is contained in:
Andrés N. Robalino 2019-08-07 13:40:38 -05:00
parent e0bacaaf37
commit 50393bdf42
4 changed files with 39 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -129,7 +129,7 @@ Nu adheres closely to a set of goals that make up its design philosophy. As feat
| cd path | Change to a new path |
| cp source path | Copy files |
| ls (path) | View the contents of the current or given path |
| mkdir path | Make directories, (to create intermediary directories append '--p') |
| mkdir path | Make directories, (to create intermediary directories append '--create-all') |
| date (--utc) | Get the current datetime |
| ps | View current processes |
| sys | View information about the current system |

61
src/commands/cp.rs
View File

@ -56,7 +56,7 @@ impl FileStructure {
self.root = path.to_path_buf();
}
pub fn translate<F>(&mut self, to: F) -> Vec<(PathBuf, PathBuf)>
pub fn paths_applying_with<F>(&mut self, to: F) -> Vec<(PathBuf, PathBuf)>
where
F: Fn((PathBuf, usize)) -> (PathBuf, PathBuf),
{
@ -140,8 +140,8 @@ pub fn cp(args: CommandArgs) -> Result<OutputStream, ShellError> {
let sources: Vec<_> = sources.unwrap().collect();
if sources.len() == 1 {
if let Ok(val) = &sources[0] {
if val.is_dir() && !args.has("recursive") {
if let Ok(entry) = &sources[0] {
if entry.is_dir() && !args.has("recursive") {
return Err(ShellError::labeled_error(
"is a directory (not copied). Try using \"--recursive\".",
"is a directory (not copied). Try using \"--recursive\".",
@ -151,18 +151,21 @@ pub fn cp(args: CommandArgs) -> Result<OutputStream, ShellError> {
let mut sources: FileStructure = FileStructure::new();
sources.walk_decorate(&val);
sources.walk_decorate(&entry);
if val.is_file() {
for (ref src, ref dst) in sources.translate(|(src, _)| {
if entry.is_file() {
let strategy = |(source_file, _depth_level)| {
if destination.exists() {
let mut dst = dunce::canonicalize(destination.clone()).unwrap();
dst.push(val.file_name().unwrap());
(src, dst)
let mut new_dst = dunce::canonicalize(destination.clone()).unwrap();
new_dst.push(entry.file_name().unwrap());
(source_file, new_dst)
} else {
(src, destination.clone())
(source_file, destination.clone())
}
}) {
};
for (ref src, ref dst) in sources.paths_applying_with(strategy) {
if src.is_file() {
match std::fs::copy(src, dst) {
Err(e) => {
@ -178,7 +181,7 @@ pub fn cp(args: CommandArgs) -> Result<OutputStream, ShellError> {
}
}
if val.is_dir() {
if entry.is_dir() {
if !destination.exists() {
match std::fs::create_dir_all(&destination) {
Err(e) => {
@ -191,25 +194,27 @@ pub fn cp(args: CommandArgs) -> Result<OutputStream, ShellError> {
Ok(o) => o,
};
for (ref src, ref dst) in sources.translate(|(src, loc)| {
let mut final_path = destination.clone();
let path = dunce::canonicalize(&src).unwrap();
let strategy = |(source_file, depth_level)| {
let mut new_dst = destination.clone();
let path = dunce::canonicalize(&source_file).unwrap();
let mut comps: Vec<_> = path
.components()
.map(|fragment| fragment.as_os_str())
.rev()
.take(1 + loc)
.take(1 + depth_level)
.collect();
comps.reverse();
for fragment in comps.iter() {
final_path.push(fragment);
new_dst.push(fragment);
}
(PathBuf::from(&src), PathBuf::from(final_path))
}) {
(PathBuf::from(&source_file), PathBuf::from(new_dst))
};
for (ref src, ref dst) in sources.paths_applying_with(strategy) {
if src.is_dir() {
if !dst.exists() {
match std::fs::create_dir_all(dst) {
@ -239,7 +244,7 @@ pub fn cp(args: CommandArgs) -> Result<OutputStream, ShellError> {
}
}
} else {
destination.push(val.file_name().unwrap());
destination.push(entry.file_name().unwrap());
match std::fs::create_dir_all(&destination) {
Err(e) => {
@ -252,25 +257,27 @@ pub fn cp(args: CommandArgs) -> Result<OutputStream, ShellError> {
Ok(o) => o,
};
for (ref src, ref dst) in sources.translate(|(src, loc)| {
let mut final_path = dunce::canonicalize(&destination).unwrap();
let path = dunce::canonicalize(&src).unwrap();
let strategy = |(source_file, depth_level)| {
let mut new_dst = dunce::canonicalize(&destination).unwrap();
let path = dunce::canonicalize(&source_file).unwrap();
let mut comps: Vec<_> = path
.components()
.map(|fragment| fragment.as_os_str())
.rev()
.take(1 + loc)
.take(1 + depth_level)
.collect();
comps.reverse();
for fragment in comps.iter() {
final_path.push(fragment);
new_dst.push(fragment);
}
(PathBuf::from(&src), PathBuf::from(final_path))
}) {
(PathBuf::from(&source_file), PathBuf::from(new_dst))
};
for (ref src, ref dst) in sources.paths_applying_with(strategy) {
if src.is_dir() {
if !dst.exists() {
match std::fs::create_dir_all(dst) {

8
src/commands/mkdir.rs
View File

@ -18,7 +18,7 @@ impl Command for Mkdir {
fn config(&self) -> CommandConfig {
let mut named: IndexMap<String, NamedType> = IndexMap::new();
named.insert("p".to_string(), NamedType::Switch);
named.insert("create-all".to_string(), NamedType::Switch);
CommandConfig {
name: self.name().to_string(),
@ -32,16 +32,14 @@ impl Command for Mkdir {
}
pub fn mkdir(args: CommandArgs) -> Result<OutputStream, ShellError> {
let env = args.env.lock().unwrap();
let path = env.path.to_path_buf();
let mut full_path = PathBuf::from(path);
let mut full_path = PathBuf::from(args.shell_manager.path());
match &args.nth(0) {
Some(Tagged { item: value, .. }) => full_path.push(Path::new(&value.as_string()?)),
_ => {}
}
if !args.has("p") {
if !args.has("create-all") {
match std::fs::create_dir(full_path) {
Err(_) => Err(ShellError::labeled_error(
"No such file or directory",

2
tests/command_mkdir_tests.rs
View File

@ -43,7 +43,7 @@ fn creates_intermediary_directories_with_p_flag() {
nu!(
_output,
cwd(&full_path),
"mkdir some_folder/another/deeper_one --p"
"mkdir some_folder/another/deeper_one --create-all"
);
let mut expected = PathBuf::from(full_path);