vh/termscp
1
0
Fork 0
mirror of https://github.com/veeso/termscp.git synced 2025-12-07 09:36:00 -08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Find command now supports also directories

Browse Source
This commit is contained in:
ChristianVisintin
2021-03-08 14:35:59 +01:00
committed by veeso
parent f3788ef61a
commit 66f9ace7bd
5 changed files with 44 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/filetransfer/ftp_transfer.rs
View File

@@ -1105,11 +1105,14 @@ mod tests {
// Pwd
assert_eq!(client.pwd().ok().unwrap(), PathBuf::from("/"));
// Search for file (let's search for pop3-*.png); there should be 2
let search_res: Vec<FsFile> = client.find("pop3-*.png").ok().unwrap();
let search_res: Vec<FsEntry> = client.find("pop3-*.png").ok().unwrap();
assert_eq!(search_res.len(), 2);
// verify names
assert_eq!(search_res[0].name.as_str(), "pop3-browser.png");
assert_eq!(search_res[1].name.as_str(), "pop3-console-client.png");
assert_eq!(search_res[0].get_name(), "pop3-browser.png");
assert_eq!(search_res[1].get_name(), "pop3-console-client.png");
// Search directory
let search_res: Vec<FsEntry> = client.find("pub").ok().unwrap();
assert_eq!(search_res.len(), 1);
// Disconnect
assert!(client.disconnect().is_ok());
// Verify err

12
src/filetransfer/mod.rs
View File

@@ -237,7 +237,7 @@ pub trait FileTransfer {
///
/// Find files from current directory (in all subdirectories) whose name matches the provided search
/// Search supports wildcards ('?', '*')
fn find(&mut self, search: &str) -> Result<Vec<FsFile>, FileTransferError> {
fn find(&mut self, search: &str) -> Result<Vec<FsEntry>, FileTransferError> {
match self.is_connected() {
true => {
// Starting from current directory, iter dir
@@ -261,8 +261,8 @@ pub trait FileTransfer {
&mut self,
dir: &Path,
filter: &WildMatch,
) -> Result<Vec<FsFile>, FileTransferError> {
let mut drained: Vec<FsFile> = Vec::new();
) -> Result<Vec<FsEntry>, FileTransferError> {
let mut drained: Vec<FsEntry> = Vec::new();
// Scan directory
match self.list_dir(dir) {
Ok(entries) => {
@@ -275,6 +275,10 @@ pub trait FileTransfer {
for entry in entries.iter() {
match entry {
FsEntry::Directory(dir) => {
// If directory name, matches wildcard, push it to drained
if filter.is_match(dir.name.as_str()) {
drained.push(FsEntry::Directory(dir.clone()));
}
match self.iter_search(dir.abs_path.as_path(), filter) {
Ok(mut filtered) => drained.append(&mut filtered),
Err(err) => return Err(err),
@@ -282,7 +286,7 @@ pub trait FileTransfer {
}
FsEntry::File(file) => {
if filter.is_match(file.name.as_str()) {
drained.push(file.clone());
drained.push(FsEntry::File(file.clone()));
}
}
}

9
src/filetransfer/scp_transfer.rs
View File

@@ -1070,11 +1070,14 @@ mod tests {
// Check session and scp
assert!(client.session.is_some());
// Search for file (let's search for pop3-*.png); there should be 2
let search_res: Vec<FsFile> = client.find("pop3-*.png").ok().unwrap();
let search_res: Vec<FsEntry> = client.find("pop3-*.png").ok().unwrap();
assert_eq!(search_res.len(), 2);
// verify names
assert_eq!(search_res[0].name.as_str(), "pop3-browser.png");
assert_eq!(search_res[1].name.as_str(), "pop3-console-client.png");
assert_eq!(search_res[0].get_name(), "pop3-browser.png");
assert_eq!(search_res[1].get_name(), "pop3-console-client.png");
// Search directory
let search_res: Vec<FsEntry> = client.find("pub").ok().unwrap();
assert_eq!(search_res.len(), 1);
// Disconnect
assert!(client.disconnect().is_ok());
// Verify err

9
src/filetransfer/sftp_transfer.rs
View File

@@ -965,11 +965,14 @@ mod tests {
// Check session and scp
assert!(client.session.is_some());
// Search for file (let's search for pop3-*.png); there should be 2
let search_res: Vec<FsFile> = client.find("pop3-*.png").ok().unwrap();
let search_res: Vec<FsEntry> = client.find("pop3-*.png").ok().unwrap();
assert_eq!(search_res.len(), 2);
// verify names
assert_eq!(search_res[0].name.as_str(), "pop3-browser.png");
assert_eq!(search_res[1].name.as_str(), "pop3-console-client.png");
assert_eq!(search_res[0].get_name(), "pop3-browser.png");
assert_eq!(search_res[1].get_name(), "pop3-console-client.png");
// Search directory
let search_res: Vec<FsEntry> = client.find("pub").ok().unwrap();
assert_eq!(search_res.len(), 1);
// Disconnect
assert!(client.disconnect().is_ok());
// Verify err