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
extern crate proc_macro;
use proc_macro::TokenStream;
use serde_json::Value;
fn split_to_vec(input: &str) -> Vec<&str> {
let mut start = 0;
let mut cnt = 0;
let mut ret = vec![];
for (idx, c) in input.chars().enumerate() {
match c {
'[' => {
if cnt == 0 {
start = idx;
}
cnt += 1;
},
']' => {
cnt -= 1;
if cnt == 0 {
ret.push(&input[start..=idx]);
start = idx + 1;
}
},
_ => (),
}
}
ret
}
fn turn_to_legal_name(name: &str) -> String {
let mut ret = String::new();
let mut skiped_first = false;
for c in name.chars() {
if c.is_ascii_uppercase() && skiped_first {
ret.push('_');
ret.push(c.to_ascii_lowercase());
} else {
ret.push(c);
}
if c == ',' {
skiped_first = true;
}
}
ret
}
trait ToArgs {
fn to_args(&self) -> String;
}
impl ToArgs for Value {
fn to_args(&self) -> String {
let array = self.as_array().unwrap();
if array.len() == 0 {
"".to_owned()
} else {
array.iter().map(|value| {
if value.is_string() {
format!("{}.to_owned()", value)
} else {
value.to_string()
}
}).collect::<Vec<_>>().join(", ")
}
}
}
#[proc_macro]
pub fn leetcode_test(input: TokenStream) -> TokenStream {
let input = input.to_string();
let v = split_to_vec(&input);
let (funcs, args, rets) = (v[0], v[1], v[2]);
let funcs = serde_json::from_str::<Value>(&turn_to_legal_name(funcs)).unwrap();
let args = serde_json::from_str::<Value>(&args.replace("- ", "-")).unwrap();
let rets = serde_json::from_str::<Value>(&rets.replace("- ", "-")).unwrap();
let mut code = String::new();
code.push_str(&format!("let mut obj = {}::new();\n", funcs[0].as_str().unwrap()));
for i in 1..funcs.as_array().unwrap().len() {
let mut stmt = format!("obj.{}({})", funcs[i].as_str().unwrap(), args[i].to_args());
if !rets[i].is_null() {
stmt = format!(r##"assert_eq!({}, {}, r#"{}"#)"##, stmt, rets[i].to_string(), stmt);
}
stmt.push_str(";\n");
code.push_str(&stmt);
}
code.parse().unwrap()
}
#[proc_macro]
pub fn leetcode_test_debug(input: TokenStream) -> TokenStream {
format!(r##"r#"{}"#"##, leetcode_test(input)).parse().unwrap()
}