ripgrep-all/src/adapters.rs

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!(".*\\.{}", &regex::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;
    });
}