venial 0.6.1

A very small syn
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
90
91
92
93
94
95

use crate::parse::consume_item;
use crate::parse_type::consume_item_name;
use crate::parse_utils::{
    consume_ident, consume_inner_attributes, consume_stuff_until, parse_ident, parse_punct,
    TokenIter,
};
use crate::{Attribute, GroupSpan, Module, TypeExpr, UseDeclaration, VisMarker};
use proc_macro2::token_stream::IntoIter;
use proc_macro2::{Delimiter, TokenTree};
use std::iter::Peekable;

pub(crate) fn parse_mod(
    tokens: &mut Peekable<IntoIter>,
    attributes: Vec<Attribute>,
    vis_marker: Option<VisMarker>,
) -> Module {
    let tk_unsafe = consume_ident(tokens, "unsafe");
    let tk_mod = parse_ident(tokens, "mod", "module declaration");
    let module_name = consume_item_name(tokens);

    let (group, tk_semicolon) = match tokens.next().unwrap() {
        TokenTree::Group(group) if group.delimiter() == Delimiter::Brace => (Some(group), None),
        TokenTree::Punct(punct) if punct.as_char() == ';' => (None, Some(punct)),
        token => panic!(
            "cannot parse mod: expected `{{ }}` or `;`, but got token {:?}",
            token
        ),
    };

    let inner_attributes;
    let tk_braces;
    let members;
    if let Some(group) = group {
        // Parse mod block body
        let mut mod_members = vec![];
        let mut tokens = group.stream().into_iter().peekable();

        tk_braces = Some(GroupSpan::new(&group));
        inner_attributes = consume_inner_attributes(&mut tokens);
        loop {
            if tokens.peek().is_none() {
                break;
            }
            let item =
                consume_item(&mut tokens).unwrap_or_else(|e| panic!("declaration in mod: {}", e));
            mod_members.push(item);
        }
        members = mod_members;
    } else {
        tk_braces = None;
        inner_attributes = vec![];
        members = vec![];
    }

    Module {
        attributes,
        vis_marker,
        tk_unsafe,
        tk_mod,
        name: module_name,
        tk_semicolon,
        tk_braces,
        inner_attributes,
        members,
    }
}

pub(crate) fn parse_use_declaration(
    tokens: &mut TokenIter,
    attributes: Vec<Attribute>,
    vis_marker: Option<VisMarker>,
) -> UseDeclaration {
    let tk_use = parse_ident(tokens, "use", "use declaration");

    let import_tree = consume_stuff_until(
        tokens,
        |token| match token {
            TokenTree::Punct(punct) if punct.as_char() == ';' => true,
            _ => false,
        },
        true,
    );

    let tk_semicolon = parse_punct(tokens, ';', "use declaration");

    UseDeclaration {
        attributes,
        vis_marker,
        tk_use,
        import_tree: TypeExpr {
            tokens: import_tree,
        },
        tk_semicolon,
    }
}