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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186

#![allow(dead_code)] // TODO: remove

// This is inspired from `synstructure`, but `synstructure` is not adapted in severals ways
// including:
//     * `&mut` everywhere
//     * not generic, we use our own `ast`, `synstructure` only knows about `syn`
//     * missing information (what arm are we in?, what attributes? etc.)

use proc_macro2;
use quote::ToTokens;
use syn;

use ast;
use attr;
use quote;

/// The type of binding to use when generating a pattern.
#[derive(Debug, Copy, Clone)]
pub enum BindingStyle {
    /// `x`
    Move,
    /// `mut x`
    MoveMut,
    /// `ref x`
    Ref,
    /// `ref mut x`
    RefMut,
}

impl quote::ToTokens for BindingStyle {
    fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {
        match *self {
            BindingStyle::Move => (),
            BindingStyle::MoveMut => tokens.extend(quote!(mut)),
            BindingStyle::Ref => tokens.extend(quote!(ref)),
            BindingStyle::RefMut => {
                tokens.extend(quote!(ref mut));
            }
        }
    }
}

#[derive(Debug)]
pub struct BindingInfo<'a> {
    pub ident: syn::Ident,
    pub field: &'a ast::Field<'a>,
}

pub struct Matcher {
    binding_name: String,
    binding_style: BindingStyle,
}

impl Matcher {
    pub fn new(style: BindingStyle) -> Self {
        Matcher {
            binding_name: "__arg".into(),
            binding_style: style,
        }
    }

    pub fn with_name(self, name: String) -> Self {
        Matcher {
            binding_name: name,
            ..self
        }
    }

    pub fn build_arms<F>(self, input: &ast::Input, f: F) -> proc_macro2::TokenStream
    where
        F: Fn(
            syn::Path,
            &syn::Ident,
            ast::Style,
            &attr::Input,
            Vec<BindingInfo>,
        ) -> proc_macro2::TokenStream,
    {
        let ident = &input.ident;
        // Generate patterns for matching against all of the variants
        let variants = match input.body {
            ast::Body::Enum(ref variants) => variants
                .iter()
                .map(|variant| {
                    let variant_ident = &variant.ident;
                    let variant_path = parse_quote!(#ident::#variant_ident);

                    let pat =
                        self.build_match_pattern(&variant_path, variant.style, &variant.fields);

                    (
                        variant_path,
                        variant_ident,
                        variant.style,
                        &variant.attrs,
                        pat,
                    )
                })
                .collect(),
            ast::Body::Struct(style, ref vd) => {
                let path = parse_quote!(#ident);
                vec![(
                    path,
                    ident,
                    style,
                    &input.attrs,
                    self.build_match_pattern(ident, style, vd),
                )]
            }
        };

        // Now that we have the patterns, generate the actual branches of the match
        // expression
        let mut t = proc_macro2::TokenStream::new();
        for (path, name, style, attrs, (pat, bindings)) in variants {
            let body = f(path, name, style, attrs, bindings);
            quote!(#pat => { #body }).to_tokens(&mut t);
        }

        t
    }

    pub fn build_match_pattern<'a, N>(
        &self,
        name: &N,
        style: ast::Style,
        fields: &'a [ast::Field<'a>],
    ) -> (proc_macro2::TokenStream, Vec<BindingInfo<'a>>)
    where
        N: quote::ToTokens,
    {
        let binding = self.binding_style;
        let (stream, matches) = match style {
            ast::Style::Unit => (proc_macro2::TokenStream::new(), Vec::new()),
            ast::Style::Tuple => {
                let (stream, matches) = fields.iter().enumerate().fold(
                    (proc_macro2::TokenStream::new(), Vec::new()),
                    |(mut stream, mut matches), (i, field)| {
                        let ident: syn::Ident = syn::Ident::new(
                            &format!("{}_{}", self.binding_name, i),
                            proc_macro2::Span::call_site(),
                        );
                        quote!(#binding #ident ,).to_tokens(&mut stream);
                        matches.push(BindingInfo {
                            ident: ident,
                            field: field,
                        });

                        (stream, matches)
                    },
                );

                (quote! { ( #stream ) }, matches)
            }
            ast::Style::Struct => {
                let (stream, matches) = fields.iter().enumerate().fold(
                    (proc_macro2::TokenStream::new(), Vec::new()),
                    |(mut stream, mut matches), (i, field)| {
                        let ident: syn::Ident = syn::Ident::new(
                            &format!("{}_{}", self.binding_name, i),
                            proc_macro2::Span::call_site(),
                        );
                        {
                            let field_name = field.ident.as_ref().unwrap();
                            quote!(#field_name : #binding #ident ,).to_tokens(&mut stream);
                        }
                        matches.push(BindingInfo {
                            ident: ident,
                            field: field,
                        });

                        (stream, matches)
                    },
                );

                (quote! { { #stream } }, matches)
            }
        };

        let mut all_tokens = proc_macro2::TokenStream::new();
        name.to_tokens(&mut all_tokens);
        all_tokens.extend(stream);

        (all_tokens, matches)
    }
}