miasma 0.2.0

Trap malicious AI web scrapers in an endless pit of poisoned training data.
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use crate::config::{LinkPrefix, MiasmaConfig};

/// Controls link generation for a given response.
#[cfg_attr(test, derive(Debug, PartialEq))]
pub enum LinkSettings<'a> {
    NoLinks,
    Links(LinkSettingsInner<'a>),
}

#[cfg_attr(test, derive(Debug, PartialEq))]
pub struct LinkSettingsInner<'a> {
    pub count: u8,
    pub prefix: &'a LinkPrefix,
    pub next_depth: Option<u32>,
}

impl<'a> LinkSettings<'a> {
    /// Determine link generation settings based on config and current depth.
    pub fn next(config: &'a MiasmaConfig, current_depth: u32) -> Self {
        let at_max_depth = config.max_depth.0.is_some_and(|max| current_depth >= max);
        if at_max_depth {
            return Self::NoLinks;
        }

        let next_depth = if config.max_depth.0.is_some() {
            Some(current_depth + 1)
        } else {
            None
        };

        Self::Links(LinkSettingsInner {
            count: config.link_count,
            prefix: &config.link_prefix,
            next_depth,
        })
    }
}

#[cfg(test)]
mod test {
    use super::*;
    use crate::config::MaxDepth;

    #[test]
    fn links_if_no_max_depth() {
        let config = MiasmaConfig {
            max_depth: MaxDepth(None),
            ..MiasmaConfig::default()
        };

        let link_settings = LinkSettings::next(&config, 0);
        let expected = LinkSettings::Links(LinkSettingsInner {
            count: config.link_count,
            prefix: &config.link_prefix,
            next_depth: None,
        });
        assert_eq!(link_settings, expected);
    }

    #[test]
    fn links_if_not_at_max_depth() {
        let config = MiasmaConfig {
            max_depth: MaxDepth(Some(5)),
            ..MiasmaConfig::default()
        };
        for current_depth in [0, 1, 2, 3, 4] {
            let link_settings = LinkSettings::next(&config, current_depth);
            let expected = LinkSettings::Links(LinkSettingsInner {
                count: config.link_count,
                prefix: &config.link_prefix,
                next_depth: Some(current_depth + 1),
            });
            assert_eq!(link_settings, expected);
        }
    }

    #[test]
    fn no_links_if_at_max_depth() {
        let config = MiasmaConfig {
            max_depth: MaxDepth(Some(5)),
            ..MiasmaConfig::default()
        };
        for current_depth in [5, 6, 10] {
            let link_settings = LinkSettings::next(&config, current_depth);
            assert!(matches!(link_settings, LinkSettings::NoLinks))
        }
    }
}