-
Notifications
You must be signed in to change notification settings - Fork 464
/
handles.rs
106 lines (96 loc) · 3.29 KB
/
handles.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
100
101
102
103
104
105
106
use super::*;
pub fn gen(gen: &Gen, def: TypeDef) -> TokenStream {
if gen.sys {
gen_sys_handle(gen, def)
} else {
gen_win_handle(gen, def)
}
}
pub fn gen_sys_handle(gen: &Gen, def: TypeDef) -> TokenStream {
let ident = to_ident(gen.reader.type_def_name(def));
let signature = gen.type_default_name(&gen.reader.type_def_underlying_type(def));
quote! {
pub type #ident = #signature;
}
}
pub fn gen_win_handle(gen: &Gen, def: TypeDef) -> TokenStream {
let name = gen.reader.type_def_name(def);
let ident = to_ident(name);
let underlying_type = gen.reader.type_def_underlying_type(def);
let signature = gen.type_default_name(&underlying_type);
let check = if underlying_type.is_pointer() {
quote! {
impl #ident {
pub fn is_invalid(&self) -> bool {
self.0.is_null()
}
}
}
} else {
let invalid = gen.reader.type_def_invalid_values(def);
if !invalid.is_empty() {
let invalid = invalid.iter().map(|value| {
let literal = Literal::i64_unsuffixed(*value);
if *value < 0 && underlying_type.is_unsigned() {
quote! { self.0 == #literal as _ }
} else {
quote! { self.0 == #literal }
}
});
quote! {
impl #ident {
pub fn is_invalid(&self) -> bool {
#(#invalid)||*
}
}
}
} else {
quote! {}
}
};
let mut tokens = quote! {
#[repr(transparent)]
// Unfortunately, Rust requires these to be derived to allow constant patterns.
#[derive(::core::cmp::PartialEq, ::core::cmp::Eq)]
pub struct #ident(pub #signature);
#check
impl ::core::default::Default for #ident {
fn default() -> Self {
unsafe { ::core::mem::zeroed() }
}
}
impl ::core::clone::Clone for #ident {
fn clone(&self) -> Self {
*self
}
}
impl ::core::marker::Copy for #ident {}
impl ::core::fmt::Debug for #ident {
fn fmt(&self, f: &mut ::core::fmt::Formatter<'_>) -> ::core::fmt::Result {
f.debug_tuple(#name).field(&self.0).finish()
}
}
impl ::core::convert::From<::core::option::Option<#ident>> for #ident {
fn from(optional: ::core::option::Option<#ident>) -> #ident {
optional.unwrap_or_default()
}
}
unsafe impl ::windows::core::Abi for #ident {
type Abi = Self;
}
};
if let Some(dependency) = gen.reader.type_def_usable_for(def) {
let type_name = gen.reader.type_def_type_name(dependency);
let mut dependency = gen.namespace(type_name.namespace);
dependency.push_str(type_name.name);
tokens.combine("e! {
impl ::core::convert::From<#ident> for #dependency {
fn from(item: #ident) -> #dependency {
#dependency(item.0)
}
}
});
}
tokens.combine(&extensions::gen(gen.reader.type_def_type_name(def)));
tokens
}