-
Notifications
You must be signed in to change notification settings - Fork 464
/
signatures.rs
99 lines (80 loc) · 2.61 KB
/
signatures.rs
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
use super::*;
pub fn gen_sig(sig: &Signature, gen: &Gen) -> TokenStream {
gen_sig_with_const(sig, gen, sig.is_const)
}
pub fn gen_abi_sig(sig: &Signature, gen: &Gen) -> TokenStream {
gen_abi_sig_with_const(sig, gen, sig.is_const)
}
pub fn gen_param_sig(param: &MethodParam, gen: &Gen) -> TokenStream {
gen_sig_with_const(¶m.signature, gen, !param.param.flags().output())
}
pub fn gen_abi_param_sig(param: &MethodParam, gen: &Gen) -> TokenStream {
gen_abi_sig_with_const(¶m.signature, gen, !param.param.flags().output())
}
// TODO: suspect - check for UDT
pub fn gen_return_sig(signature: &MethodSignature, gen: &Gen) -> TokenStream {
if let Some(return_sig) = &signature.return_sig {
let tokens = gen_sig(return_sig, gen);
quote! { -> #tokens }
} else {
quote! {}
}
}
pub fn gen_param_constraints(params: &[MethodParam], gen: &Gen) -> TokenStream {
let mut tokens = quote! {};
for (position, param) in params.iter().enumerate() {
if param.is_convertible() {
let name: TokenStream = format!("Param{}", position).into();
let into = gen_element_name(¶m.signature.kind, gen);
tokens.combine("e! { #name: ::windows::core::IntoParam<'a, #into>, });
}
}
if !tokens.is_empty() {
quote! { 'a, #tokens }
} else {
tokens
}
}
fn gen_abi_sig_with_const(sig: &Signature, gen: &Gen, is_const: bool) -> TokenStream {
let mut tokens = TokenStream::new();
for _ in 0..sig.pointers {
if is_const {
tokens.combine("e! { *const });
} else {
tokens.combine("e! { *mut });
}
}
tokens.combine(&gen_abi_element_name(sig, gen));
tokens
}
pub fn gen_result_sig(sig: &Signature, gen: &Gen) -> TokenStream {
let mut tokens = TokenStream::new();
for _ in 0..sig.pointers {
if sig.is_const {
tokens.combine("e! { *const });
} else {
tokens.combine("e! { *mut });
}
}
tokens.combine(&gen_element_name(&sig.kind, gen));
tokens
}
fn gen_sig_with_const(sig: &Signature, gen: &Gen, is_const: bool) -> TokenStream {
let mut tokens = TokenStream::new();
for _ in 0..sig.pointers {
if is_const {
tokens.combine("e! { *const });
} else {
tokens.combine("e! { *mut });
}
}
let kind = gen_element_name(&sig.kind, gen);
if sig.kind.is_nullable() && !gen.sys {
tokens.combine("e! {
::core::option::Option<#kind>
});
} else {
tokens.combine(&kind)
}
tokens
}