veil-engine 0.2.1

Detection, placeholders and restoration. No I/O.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use std::sync::Arc;

use regex::{Regex, RegexBuilder};

use super::{Candidate, Detector, Source, P_CUSTOM};
use crate::kind::EntityKind;
use crate::rules::{kind, RegexRule, RulesError};

/// A user-defined regex rule.
pub(super) struct Custom {
    regex: Regex,
    kind: EntityKind,
    name: Arc<str>,
}

impl Custom {
    pub fn new(rule: &RegexRule) -> Result<Self, RulesError> {
        let regex = RegexBuilder::new(&rule.pattern)
            .size_limit(1 << 20)
            .build()
            .map_err(|e| RulesError::Regex { name: rule.name.clone(), error: e.to_string() })?;
        if regex.is_match("") {
            return Err(RulesError::Regex { name: rule.name.clone(), error: "the pattern matches empty text".into() });
        }
        Ok(Custom { regex, kind: kind(&rule.kind)?, name: rule.name.as_str().into() })
    }
}

impl Detector for Custom {
    fn find(&self, text: &str, out: &mut Vec<Candidate>) {
        for m in self.regex.find_iter(text) {
            out.push(Candidate {
                start: m.start(),
                end: m.end(),
                kind: self.kind.clone(),
                source: Source::Regex(self.name.clone()),
                priority: P_CUSTOM,
            });
        }
    }
}