proc_macro/
escape.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
57
#[derive(Copy, Clone)]
pub(crate) struct EscapeOptions {
    /// Produce \'.
    pub escape_single_quote: bool,
    /// Produce \".
    pub escape_double_quote: bool,
    /// Produce \x escapes for non-ASCII, and use \x rather than \u for ASCII
    /// control characters.
    pub escape_nonascii: bool,
}

pub(crate) fn escape_bytes(bytes: &[u8], opt: EscapeOptions) -> String {
    let mut repr = String::new();

    if opt.escape_nonascii {
        for &byte in bytes {
            escape_single_byte(byte, opt, &mut repr);
        }
    } else {
        let mut chunks = bytes.utf8_chunks();
        while let Some(chunk) = chunks.next() {
            for ch in chunk.valid().chars() {
                escape_single_char(ch, opt, &mut repr);
            }
            for &byte in chunk.invalid() {
                escape_single_byte(byte, opt, &mut repr);
            }
        }
    }

    repr
}

fn escape_single_byte(byte: u8, opt: EscapeOptions, repr: &mut String) {
    if byte == b'\0' {
        repr.push_str("\\0");
    } else if (byte == b'\'' && !opt.escape_single_quote)
        || (byte == b'"' && !opt.escape_double_quote)
    {
        repr.push(byte as char);
    } else {
        // Escapes \t, \r, \n, \\, \', \", and uses \x## for non-ASCII and
        // for ASCII control characters.
        repr.extend(byte.escape_ascii().map(char::from));
    }
}

fn escape_single_char(ch: char, opt: EscapeOptions, repr: &mut String) {
    if (ch == '\'' && !opt.escape_single_quote) || (ch == '"' && !opt.escape_double_quote) {
        repr.push(ch);
    } else {
        // Escapes \0, \t, \r, \n, \\, \', \", and uses \u{...} for
        // non-printable characters and for Grapheme_Extend characters, which
        // includes things like U+0300 "Combining Grave Accent".
        repr.extend(ch.escape_debug());
    }
}