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
|
use quote::quote;
use syn::visit_mut::{self, VisitMut};
use syn::{parse_quote, Attribute, Expr, ExprMatch, ItemFn, Local};
use crate::parse::Input;
pub fn check(input: &mut ItemFn) {
Checker.visit_item_fn_mut(input);
}
struct Checker;
impl VisitMut for Checker {
fn visit_expr_mut(&mut self, expr: &mut Expr) {
visit_mut::visit_expr_mut(self, expr);
let expr_match = match expr {
Expr::Match(expr) => expr,
_ => return,
};
if !take_sorted_attr(&mut expr_match.attrs) {
return;
}
let input = expr_match.clone();
check_and_insert_error(input, expr);
}
fn visit_local_mut(&mut self, local: &mut Local) {
visit_mut::visit_local_mut(self, local);
let init = match &local.init {
Some(init) => init,
None => return,
};
let expr_match = match init.expr.as_ref() {
Expr::Match(expr) => expr,
_ => return,
};
if !take_sorted_attr(&mut local.attrs) {
return;
}
let input = expr_match.clone();
let expr = local.init.as_mut().unwrap().expr.as_mut();
check_and_insert_error(input, expr);
}
}
fn take_sorted_attr(attrs: &mut Vec<Attribute>) -> bool {
for i in 0..attrs.len() {
let path = &attrs[i].path();
let path = quote!(#path).to_string();
if path == "sorted" || path == "remain :: sorted" {
attrs.remove(i);
return true;
}
}
false
}
fn check_and_insert_error(input: ExprMatch, out: &mut Expr) {
let mut input = Input::Match(input);
*out = match crate::check::sorted(&mut input) {
Ok(()) => parse_quote!(#input),
Err(err) => {
let err = err.to_compile_error();
parse_quote!({
#err
#input
})
}
};
}
|
