volumen_parser_core 0.1.0

Volumen parser core crate
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

const PROMPT_LENGTH: usize = "@prompt".len();

/// Parses comment text to determine if it contains a valid @prompt annotation.
pub fn parse_comment(comment: &str) -> Option<bool> {
    let comment = comment.to_lowercase();
    if let Some(pos) = comment.find("@prompt") {
        let before_char = if pos == 0 {
            None
        } else {
            comment.chars().nth(pos - 1)
        };

        let after_pos = pos + PROMPT_LENGTH;
        let after_char = comment.chars().nth(after_pos);

        let valid_before = before_char.map_or(true, |c| !c.is_alphanumeric() && c != '_');
        let valid_after = after_char.map_or(true, |c| !c.is_alphanumeric() && c != '_');

        Some(valid_before && valid_after)
    } else {
        None
    }
}

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

    #[test]
    fn basic() {
        assert_eq!(parse_comment("@prompt"), Some(true));
        assert_eq!(parse_comment(" @prompt "), Some(true));
        assert_eq!(parse_comment("@prompt greeting"), Some(true));
        assert_eq!(parse_comment("greeting @prompt"), Some(true));
    }

    #[test]
    fn extra() {
        assert_eq!(parse_comment("@prompt for user greeting"), Some(true));
        assert_eq!(parse_comment("This is a @prompt comment"), Some(true));
        assert_eq!(parse_comment("* @prompt"), Some(true));
        assert_eq!(parse_comment("*@prompt"), Some(true));
        assert_eq!(parse_comment("* @prompt greeting"), Some(true));
        assert_eq!(parse_comment("  * @prompt  "), Some(true));
    }

    #[test]
    fn case() {
        assert_eq!(parse_comment("@PROMPT"), Some(true));
        assert_eq!(parse_comment("@Prompt"), Some(true));
        assert_eq!(parse_comment("@PrOmPt"), Some(true));
        assert_eq!(parse_comment("* @PROMPT"), Some(true));
        assert_eq!(parse_comment("@PROMPT for testing"), Some(true));
        assert_eq!(parse_comment("@Prompt with mixed case"), Some(true));
    }

    #[test]
    fn not_exact() {
        assert_eq!(parse_comment("@prompting"), Some(false));
        assert_eq!(parse_comment("my@prompt"), Some(false));
        assert_eq!(parse_comment("@prompter"), Some(false));
        assert_eq!(parse_comment("@prompt_var"), Some(false));
        assert_eq!(parse_comment("* @prompting"), Some(false));
        assert_eq!(parse_comment("* my@prompt"), Some(false));
    }

    #[test]
    fn unrelated_text() {
        assert_eq!(parse_comment("regular comment"), None);
        assert_eq!(parse_comment("* This is documentation"), None);
        assert_eq!(parse_comment("TODO: fix this"), None);
        assert_eq!(parse_comment(""), None);
        assert_eq!(parse_comment("   "), None);
    }

    #[test]
    fn punctuation() {
        assert_eq!(parse_comment("@prompt!"), Some(true));
        assert_eq!(parse_comment("@prompt."), Some(true));
        assert_eq!(parse_comment("@prompt,"), Some(true));
        assert_eq!(parse_comment("(@prompt)"), Some(true));
    }

    #[test]
    fn repeating() {
        assert_eq!(parse_comment("@prompt for @prompt usage"), Some(true));
    }
}