indoc/
expr.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
58
59
60
61
use crate::error::{Error, Result};
use proc_macro::token_stream::IntoIter as TokenIter;
use proc_macro::{Spacing, Span, TokenStream, TokenTree};
use std::iter::{self, Peekable};

pub fn parse(input: &mut Peekable<TokenIter>, require_comma: bool) -> Result<TokenStream> {
    #[derive(PartialEq)]
    enum Lookbehind {
        JointColon,
        DoubleColon,
        JointHyphen,
        Other,
    }

    let mut expr = TokenStream::new();
    let mut lookbehind = Lookbehind::Other;
    let mut angle_bracket_depth = 0;

    loop {
        if angle_bracket_depth == 0 {
            match input.peek() {
                Some(TokenTree::Punct(punct)) if punct.as_char() == ',' => {
                    return Ok(expr);
                }
                _ => {}
            }
        }
        match input.next() {
            Some(TokenTree::Punct(punct)) => {
                let ch = punct.as_char();
                let spacing = punct.spacing();
                expr.extend(iter::once(TokenTree::Punct(punct)));
                lookbehind = match ch {
                    ':' if lookbehind == Lookbehind::JointColon => Lookbehind::DoubleColon,
                    ':' if spacing == Spacing::Joint => Lookbehind::JointColon,
                    '<' if lookbehind == Lookbehind::DoubleColon => {
                        angle_bracket_depth += 1;
                        Lookbehind::Other
                    }
                    '>' if angle_bracket_depth > 0 && lookbehind != Lookbehind::JointHyphen => {
                        angle_bracket_depth -= 1;
                        Lookbehind::Other
                    }
                    '-' if spacing == Spacing::Joint => Lookbehind::JointHyphen,
                    _ => Lookbehind::Other,
                };
            }
            Some(token) => expr.extend(iter::once(token)),
            None => {
                return if require_comma {
                    Err(Error::new(
                        Span::call_site(),
                        "unexpected end of macro input",
                    ))
                } else {
                    Ok(expr)
                };
            }
        }
    }
}