miasma 0.2.8

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
89

use std::sync::Arc;

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

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

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

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

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

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

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

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

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

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

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