diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/index.rs | 2 | ||||
| -rw-r--r-- | src/text/mod.rs | 6 |
2 files changed, 7 insertions, 1 deletions
diff --git a/src/index.rs b/src/index.rs index d3aab9e..92a12b6 100644 --- a/src/index.rs +++ b/src/index.rs @@ -50,7 +50,7 @@ impl Index { for entry in WalkDir::new(input_path) .into_iter() .filter_map(|e| e.ok()) { - if entry.path().is_file() { + if entry.path().is_file() && text::is_supported(entry.path().to_str().unwrap()) { nof += 1; paths.push(entry.path().to_str().unwrap().to_string()); } diff --git a/src/text/mod.rs b/src/text/mod.rs index c3fd8f6..ac474bc 100644 --- a/src/text/mod.rs +++ b/src/text/mod.rs @@ -44,3 +44,9 @@ pub fn extract_text(path : &str) -> String { let extenstion = p.extension().unwrap_or_else(|| OsStr::new("")).to_str().unwrap(); EXT.get(extenstion).unwrap_or(&(empty_extractor as ExtFn))(path) } + +pub fn is_supported(path : &str) -> bool { + let p = Path::new(&path); + let extenstion = p.extension().unwrap_or_else(|| OsStr::new("")).to_str().unwrap(); + EXT.get(extenstion).is_some() +} |