src/util/encode.rs


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

//! Utilities to encode HTML.

/// Encode dangerous html characters.
///
/// This ensures that certain characters which have special meaning in HTML are
/// dealt with.
/// Technically, we can skip `>` and `"` in many cases, but CM includes them.
///
/// This behavior is not explained in prose in `CommonMark` but can be inferred
/// from the input/output test cases.
///
/// ## Examples
///
/// ```rust ignore
/// use micromark::util::encode;
///
/// assert_eq!(encode("I <3 🦀"), "I &lt;3 🦀");
/// ```
///
/// ## References
///
/// *   [`micromark-util-encode` in `micromark`](https://github.com/micromark/micromark/tree/main/packages/micromark-util-encode)
pub fn encode<S: Into<String>>(value: S, encode_html: bool) -> String {
    let check = if encode_html { check_all } else { check_nil };
    let mut value = value.into();

    // It’ll grow a bit bigger for each dangerous character.
    let mut result = String::with_capacity(value.len());

    while let Some(indice) = value.find(check) {
        let after = value.split_off(indice + 1);
        let dangerous = value.pop().unwrap();
        result.push_str(&value);
        result.push_str(match dangerous {
            '\0' => "�",
            '&' => "&amp;",
            '"' => "&quot;",
            '<' => "&lt;",
            '>' => "&gt;",
            _ => unreachable!("xxx"),
        });
        value = after;
    }

    result.push_str(&value);

    result
}

fn check_all(char: char) -> bool {
    matches!(char, '\0' | '&' | '"' | '<' | '>')
}

fn check_nil(char: char) -> bool {
    matches!(char, '\0')
}