diff options
Diffstat (limited to 'src/text')
| -rw-r--r-- | src/text/mod.rs | 31 | ||||
| -rw-r--r-- | src/text/txt.rs | 5 |
2 files changed, 36 insertions, 0 deletions
diff --git a/src/text/mod.rs b/src/text/mod.rs new file mode 100644 index 0000000..4e1b9a4 --- /dev/null +++ b/src/text/mod.rs @@ -0,0 +1,31 @@ +use lazy_static::lazy_static; +use std::ffi::OsStr; +use std::path::Path; +use std::collections::HashMap; + +mod txt; + +fn empty_extractor(_ : &str) -> String { + "".to_string() +} + +macro_rules! ext { + ($f:ident) => {(stringify!($f), $f::get_text as ExtFn)} +} + +type ExtFn = fn(&str) -> String; + +lazy_static! { + static ref EXT: HashMap<&'static str, ExtFn> = { + HashMap::from([ + ext!(txt), + ]) + }; +} + + +pub fn extract_text(path : &str) -> String { + let p = Path::new(&path); + let extenstion = p.extension().unwrap_or_else(|| OsStr::new("")).to_str().unwrap(); + EXT.get(extenstion).unwrap_or(&(empty_extractor as ExtFn))(path) +} diff --git a/src/text/txt.rs b/src/text/txt.rs new file mode 100644 index 0000000..14e7422 --- /dev/null +++ b/src/text/txt.rs @@ -0,0 +1,5 @@ +use std::fs; + +pub fn get_text(path : &str) -> String { + fs::read_to_string(path).unwrap_or_default() +} |