diff --git a/src/data/files.rs b/src/data/files.rs index a0601c8b5..6543b86b4 100644 --- a/src/data/files.rs +++ b/src/data/files.rs @@ -8,9 +8,25 @@ pub(crate) fn dir_entry_dict( tag: impl Into, full: bool, with_symlink_targets: bool, + name_only: bool, ) -> Result { - let mut dict = TaggedDictBuilder::new(tag); - dict.insert_untagged("name", UntaggedValue::string(filename.to_string_lossy())); + let tag = tag.into(); + let mut dict = TaggedDictBuilder::new(&tag); + + let name = if name_only { + filename.file_name().and_then(|s| s.to_str()) + } else { + filename.to_str() + } + .ok_or_else(|| { + ShellError::labeled_error( + format!("Invalid file name: {:}", filename.to_string_lossy()), + "invalid file name", + tag, + ) + })?; + + dict.insert_untagged("name", UntaggedValue::string(name)); if metadata.is_dir() { dict.insert_untagged("type", UntaggedValue::string("Dir")); diff --git a/src/shell/filesystem_shell.rs b/src/shell/filesystem_shell.rs index d0139a431..92932f4d3 100644 --- a/src/shell/filesystem_shell.rs +++ b/src/shell/filesystem_shell.rs @@ -14,7 +14,7 @@ use nu_parser::ExpandContext; use nu_protocol::{Primitive, ReturnSuccess, UntaggedValue}; use rustyline::completion::FilenameCompleter; use rustyline::hint::{Hinter, HistoryHinter}; -use std::path::{Path, PathBuf}; +use std::path::PathBuf; use std::sync::atomic::Ordering; use trash as SendToTrash; @@ -96,132 +96,60 @@ impl Shell for FilesystemShell { }: LsArgs, context: &RunnablePerItemContext, ) -> Result { - let cwd = self.path(); - let mut full_path = PathBuf::from(self.path()); - - if let Some(value) = &path { - full_path.push((*value).as_ref()) - } - let ctrl_c = context.ctrl_c.clone(); let name_tag = context.name.clone(); - //If it's not a glob, try to display the contents of the entry if it's a directory - let lossy_path = full_path.to_string_lossy(); - if !lossy_path.contains('*') && !lossy_path.contains('?') { - let entry = Path::new(&full_path); - if entry.is_dir() { - let entries = std::fs::read_dir(&entry); - let entries = match entries { - Err(e) => { - if let Some(s) = path { - return Err(ShellError::labeled_error( - e.to_string(), - e.to_string(), - s.tag(), - )); - } else { - return Err(ShellError::labeled_error( - e.to_string(), - e.to_string(), - name_tag, - )); - } + let (path, p_tag) = match path { + Some(p) => { + let p_tag = p.tag; + let mut p = p.item; + if p.is_dir() { + if is_dir_empty(&p) { + return Ok(OutputStream::empty()); } - Ok(o) => o, - }; - let mut entries = entries.collect::>>(); - entries.sort_by(|x, y| match (x, y) { - (Ok(entry1), Ok(entry2)) => entry1 - .path() - .to_string_lossy() - .to_lowercase() - .cmp(&entry2.path().to_string_lossy().to_lowercase()), - (Err(_), Ok(_)) => std::cmp::Ordering::Greater, - (Ok(_), Err(_)) => std::cmp::Ordering::Greater, - _ => std::cmp::Ordering::Equal, - }); - let stream = async_stream! { - for entry in entries { - if ctrl_c.load(Ordering::SeqCst) { - break; - } - if let Ok(entry) = entry { - let filepath = entry.path(); - if let Ok(metadata) = std::fs::symlink_metadata(&filepath) { - let mut filename = if let Ok(fname) = filepath.strip_prefix(&cwd) { - fname - } else { - Path::new(&filepath) - }; - - if short_names { - filename = if let Some(fname) = filename.file_name() { - Path::new(fname) - } else { - let fname = filename - .file_name() - .or_else(|| Path::new(filename).file_name()) - .unwrap_or(filename.as_os_str()); - Path::new(fname) - } - } - - let value = dir_entry_dict(filename, &metadata, &name_tag, full, with_symlink_targets)?; - yield ReturnSuccess::value(value); - } - } - } - }; - return Ok(stream.to_output_stream()); + p.push("*"); + } + (p, p_tag) } - } - - let entries = match glob::glob(&full_path.to_string_lossy()) { - Ok(files) => files, - Err(_) => { - if let Some(source) = path { - return Err(ShellError::labeled_error( - "Invalid pattern", - "Invalid pattern", - source.tag(), - )); + None => { + if is_dir_empty(&self.path().into()) { + return Ok(OutputStream::empty()); } else { - return Err(ShellError::untagged_runtime_error("Invalid pattern.")); + (PathBuf::from("./*"), context.name.clone()) } } }; - // Enumerate the entries from the glob and add each + let mut paths = match glob::glob(&path.to_string_lossy()) { + Ok(g) => Ok(g), + Err(e) => Err(ShellError::labeled_error("Glob error", e.msg, &p_tag)), + }? + .peekable(); + + if paths.peek().is_none() { + return Err(ShellError::labeled_error( + "Invalid File or Pattern", + "Invalid File or Pattern", + &p_tag, + )); + } + let stream = async_stream! { - for entry in entries { + for path in paths { if ctrl_c.load(Ordering::SeqCst) { break; } - if let Ok(entry) = entry { - if let Ok(metadata) = std::fs::symlink_metadata(&entry) { - let mut filename = if let Ok(fname) = entry.strip_prefix(&cwd) { - fname - } else { - Path::new(&entry) - }; - - if short_names { - filename = if let Some(fname) = filename.file_name() { - Path::new(fname) - } else { - let fname = filename - .file_name() - .or_else(|| Path::new(filename).file_name()) - .unwrap_or(filename.as_os_str()); - Path::new(fname) + match path { + Ok(p) => match std::fs::symlink_metadata(&p) { + Ok(m) => { + match dir_entry_dict(&p, &m, name_tag.clone(), full, short_names, with_symlink_targets) { + Ok(d) => yield ReturnSuccess::value(d), + Err(e) => yield Err(e) } } - - if let Ok(value) = dir_entry_dict(filename, &metadata, &name_tag, full, with_symlink_targets) { - yield ReturnSuccess::value(value); - } + Err(e) => yield Err(ShellError::from(e)) } + Err(e) => yield Err(e.into_error().into()), } } }; @@ -1196,3 +1124,10 @@ impl Shell for FilesystemShell { self.hinter.hint(line, pos, ctx) } } + +fn is_dir_empty(d: &PathBuf) -> bool { + match d.read_dir() { + Err(_e) => true, + Ok(mut s) => s.next().is_none(), + } +}