statue 0.3.1

Easier way to query selectors for static HTML pages.
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

use proc_macro::{Delimiter, Group, Ident, Literal, Punct, Spacing, Span, TokenStream, TokenTree};

use crate::{
    elements::{Elements, MultipleElements, SingleElement},
    sel_queries::RetTyKind,
};

pub(crate) trait ExtendTokenStream {
    fn extend_token_stream(self, ts: &mut TokenStream);
}

impl<'a> ExtendTokenStream for SingleElement<'a> {
    fn extend_token_stream(self, ts: &mut TokenStream) {
        let Self {
            name,
            kind,
            syn,
            ret_ty,
            phantom: _phantom,
        } = self;
        ts.extend([
            TokenTree::Ident(Ident::new("let", Span::call_site())),
            TokenTree::Ident(Ident::new(name.as_str(), Span::call_site())),
            TokenTree::Punct(Punct::new(':', Spacing::Alone)),
        ]);
        RetTyKind::fmt_as_concrete_ty(ts, &ret_ty, &kind);
        ts.extend([
            TokenTree::Punct(Punct::new('=', Spacing::Alone)),
            TokenTree::Ident(Ident::new("document", Span::call_site())),
            TokenTree::Punct(Punct::new('.', Spacing::Alone)),
            TokenTree::Ident(Ident::new("query_selector", Span::call_site())),
            TokenTree::Group(Group::new(Delimiter::Parenthesis, {
                let mut ts = TokenStream::new();
                ts.extend([TokenTree::Literal(Literal::string(&syn.0))]);
                ts
            })),
            TokenTree::Punct(Punct::new('.', Spacing::Alone)),
            TokenTree::Ident(Ident::new("unwrap", Span::call_site())),
            TokenTree::Group(Group::new(Delimiter::Parenthesis, TokenStream::new())),
            TokenTree::Punct(Punct::new('.', Spacing::Alone)),
            TokenTree::Ident(Ident::new("unwrap", Span::call_site())),
            TokenTree::Group(Group::new(Delimiter::Parenthesis, TokenStream::new())),
            TokenTree::Punct(Punct::new('.', Spacing::Alone)),
            TokenTree::Ident(Ident::new("dyn_into", Span::call_site())),
            TokenTree::Punct(Punct::new(':', Spacing::Joint)),
            TokenTree::Punct(Punct::new(':', Spacing::Joint)),
            TokenTree::Punct(Punct::new('<', Spacing::Alone)),
            TokenTree::Punct(Punct::new(':', Spacing::Joint)),
            TokenTree::Punct(Punct::new(':', Spacing::Joint)),
            TokenTree::Ident(Ident::new("web_sys", Span::call_site())),
            TokenTree::Punct(Punct::new(':', Spacing::Joint)),
            TokenTree::Punct(Punct::new(':', Spacing::Joint)),
            TokenTree::Ident(Ident::new(kind.to_web_sys_name(), Span::call_site())),
            TokenTree::Punct(Punct::new('>', Spacing::Alone)),
            TokenTree::Group(Group::new(Delimiter::Parenthesis, TokenStream::new())),
            TokenTree::Punct(Punct::new('.', Spacing::Alone)),
            TokenTree::Ident(Ident::new("unwrap", Span::call_site())),
            TokenTree::Group(Group::new(Delimiter::Parenthesis, TokenStream::new())),
        ]);
        RetTyKind::extend_with_optional_into(ts, &ret_ty);
        ts.extend([TokenTree::Punct(Punct::new(';', Spacing::Alone))]);
    }
}

impl<'a> ExtendTokenStream for MultipleElements<'a> {
    fn extend_token_stream(self, ts: &mut TokenStream) {
        let Self {
            name,
            count: _count,
            ret_ty,
            common_kind: _common_kind,
            syn,
            phantom: _phantom,
        } = self;
        ts.extend([
            TokenTree::Ident(Ident::new("let", Span::call_site())),
            TokenTree::Ident(Ident::new(name.as_str(), Span::call_site())),
            TokenTree::Punct(Punct::new(':', Spacing::Alone)),
        ]);
        RetTyKind::fmt_as_inferred_ty(ts, &ret_ty);
        ts.extend([
            TokenTree::Punct(Punct::new('=', Spacing::Alone)),
            TokenTree::Ident(Ident::new("document", Span::call_site())),
            TokenTree::Punct(Punct::new('.', Spacing::Alone)),
            TokenTree::Ident(Ident::new("query_selector_all", Span::call_site())),
            TokenTree::Group(Group::new(Delimiter::Parenthesis, {
                let mut ts = TokenStream::new();
                ts.extend([TokenTree::Literal(Literal::string(&syn.0))]);
                ts
            })),
            TokenTree::Punct(Punct::new('.', Spacing::Alone)),
            TokenTree::Ident(Ident::new("unwrap", Span::call_site())),
            TokenTree::Group(Group::new(Delimiter::Parenthesis, TokenStream::new())),
        ]);
        RetTyKind::extend_with_optional_into(ts, &ret_ty);
        ts.extend([TokenTree::Punct(Punct::new(';', Spacing::Alone))]);
    }
}

impl<'a> ExtendTokenStream for Vec<SingleElement<'a>> {
    fn extend_token_stream(self, ts: &mut TokenStream) {
        for single_element in self {
            single_element.extend_token_stream(ts);
        }
    }
}

impl<'a> ExtendTokenStream for Vec<MultipleElements<'a>> {
    fn extend_token_stream(self, ts: &mut TokenStream) {
        for multiple_elements in self {
            multiple_elements.extend_token_stream(ts);
        }
    }
}

impl<'a> ExtendTokenStream for Elements<'a> {
    fn extend_token_stream(self, ts: &mut TokenStream) {
        let Self { single, multiple } = self;
        single.extend_token_stream(ts);
        multiple.extend_token_stream(ts);
    }
}