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
use ::proc_macro::{
TokenStream,
};
use ::proc_macro2::{
Span,
TokenStream as TokenStream2,
};
use ::quote::{
format_ident,
quote,
ToTokens,
};
use ::syn::{*,
parse::{
Nothing,
Parse,
ParseStream,
},
spanned::Spanned,
Result,
visit_mut::{self, VisitMut},
};
use ::core::{
mem,
ops::Not as _,
};
use self::{
helpers::{Fields as __, LifetimeVisitor},
};
#[macro_use]
mod helpers;
include!("handle_returning_locals.rs");
mod handle_let_bindings;
mod parse;
mod wrap_statements_inside_closure_body;
enum Input {
TraitItemMethod(TraitItemMethod),
ImplItemMethod(ImplItemMethod),
ItemFn(ItemFn),
}
type Str = ::std::borrow::Cow<'static, str>;
struct Attrs {
lifetime: Str,
continuation: Option<Ident>,
}
#[proc_macro_attribute] pub
fn with (
attrs: TokenStream,
input: TokenStream,
) -> TokenStream
{
let (ref attrs, ref mut fun) = (
parse_macro_input!(attrs as Attrs),
parse_macro_input!(input as Input),
);
handle_returning_locals(&mut *fun, attrs);
if let Err(err) = handle_let_bindings::f(&mut *fun, attrs) {
return err.to_compile_error().into();
}
let ret = fun.to_token_stream();
#[cfg(feature = "verbose-expansions")] {
helpers::pretty_print_tokenstream(
&ret,
&fun.fields().sig.ident,
);
}
ret.into()
}