You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.5 KiB
Rust
77 lines
2.5 KiB
Rust
pub mod ffmpeg;
|
|
pub mod pandoc;
|
|
pub mod poppler;
|
|
pub mod spawning;
|
|
|
|
//pub use ffmpeg::FffmpegAdapter;
|
|
|
|
use regex::{Regex, RegexSet};
|
|
use std::collections::HashMap;
|
|
use std::ffi::OsString;
|
|
use std::io::BufRead;
|
|
use std::io::Write;
|
|
use std::rc::Rc;
|
|
|
|
pub enum Matcher {
|
|
MimeType(Regex), // todo: generic pattern?
|
|
FileName(Regex),
|
|
}
|
|
|
|
pub struct AdapterMeta {
|
|
pub name: String,
|
|
pub version: i32,
|
|
pub matchers: Vec<Matcher>,
|
|
}
|
|
|
|
pub struct FileMeta {
|
|
// filename is not actually a utf8 string, but since we can't do regex on OsStr and can't get a &[u8] from OsStr either,
|
|
// and since we probably only want to do matching on ascii stuff anyways, this is the filename as a string with non-valid bytes removed
|
|
pub lossy_filename: String,
|
|
pub mimetype: String,
|
|
}
|
|
|
|
pub trait GetMetadata {
|
|
fn metadata<'a>(&'a self) -> &'a AdapterMeta;
|
|
}
|
|
pub trait FileAdapter: GetMetadata {
|
|
fn adapt(&self, inp_fname: &str, oup: &mut dyn Write) -> std::io::Result<()>;
|
|
}
|
|
|
|
pub fn ExtensionMatcher(extension: &str) -> Matcher {
|
|
let regex = Regex::new(&format!(".*\\.{}", ®ex::escape(extension)))
|
|
.expect("we know this regex compiles");
|
|
Matcher::FileName(regex)
|
|
}
|
|
|
|
pub fn init_adapters() -> Result<impl Fn(FileMeta) -> Option<Rc<dyn FileAdapter>>, regex::Error> {
|
|
let adapters: Vec<Rc<dyn FileAdapter>> = vec![
|
|
Rc::new(crate::adapters::ffmpeg::FFmpegAdapter::new()),
|
|
Rc::new(crate::adapters::pandoc::PandocAdapter::new()),
|
|
Rc::new(crate::adapters::poppler::PopplerAdapter::new()),
|
|
];
|
|
|
|
let mut fname_regexes = vec![];
|
|
let mut mime_regexes = vec![];
|
|
for adapter in adapters.into_iter() {
|
|
let metadata = adapter.metadata();
|
|
for matcher in &metadata.matchers {
|
|
match matcher {
|
|
Matcher::MimeType(re) => mime_regexes.push((re.clone(), adapter.clone())),
|
|
Matcher::FileName(re) => fname_regexes.push((re.clone(), adapter.clone())),
|
|
};
|
|
}
|
|
}
|
|
let fname_regex_set = RegexSet::new(fname_regexes.iter().map(|p| p.0.as_str()))?;
|
|
let mime_regex_set = RegexSet::new(mime_regexes.iter().map(|p| p.0.as_str()))?;
|
|
return Ok(move |meta: FileMeta| {
|
|
// todo: handle multiple matches
|
|
for m in fname_regex_set.matches(&meta.lossy_filename) {
|
|
return Some(fname_regexes[m].1.clone());
|
|
}
|
|
for m in mime_regex_set.matches(&meta.mimetype) {
|
|
return Some(mime_regexes[m].1.clone());
|
|
}
|
|
return None;
|
|
});
|
|
}
|