gaze-pii 0.6.6

Reversible PII pseudonymization runtime for agentic workflows
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

use crate::detector::PiiClass;
pub use gaze_types::Action;

#[derive(Debug, Clone, Default)]
pub struct RuleContext {
    pub field_name: Option<String>,
}

pub trait Rule: Send + Sync {
    fn action(&self, class: &PiiClass, context: &RuleContext) -> Option<Action>;
}

pub struct ClassRule {
    class: PiiClass,
    action: Action,
}

impl ClassRule {
    pub fn new(class: PiiClass, action: Action) -> Self {
        Self { class, action }
    }
}

impl Rule for ClassRule {
    fn action(&self, class: &PiiClass, _context: &RuleContext) -> Option<Action> {
        (self.class == *class).then_some(self.action)
    }
}

pub struct ColumnRule {
    field_name: String,
    action: Action,
}

impl ColumnRule {
    pub fn new(field_name: &str, action: Action) -> Self {
        Self {
            field_name: field_name.to_string(),
            action,
        }
    }
}

impl Rule for ColumnRule {
    fn action(&self, _class: &PiiClass, context: &RuleContext) -> Option<Action> {
        (context.field_name.as_deref() == Some(self.field_name.as_str())).then_some(self.action)
    }
}

pub struct DefaultRule {
    action: Action,
}

impl DefaultRule {
    pub fn new(action: Action) -> Self {
        Self { action }
    }
}

impl Rule for DefaultRule {
    fn action(&self, _class: &PiiClass, _context: &RuleContext) -> Option<Action> {
        Some(self.action)
    }
}