blob: ec89cbb737cffbb1ecb7a03e54d48c13ba853b00 [file] [log] [blame]
David Tolnay17e137f2020-05-08 15:55:28 -07001use crate::syntax::discriminant::DiscriminantSet;
David Tolnayc598a272020-08-29 15:10:18 -07002use crate::syntax::file::{Item, ItemForeignMod};
David Tolnay52759782020-05-03 23:59:40 -07003use crate::syntax::report::Errors;
David Tolnayeebe9b72020-04-14 16:32:18 -07004use crate::syntax::Atom::*;
David Tolnay7db73692019-10-20 14:51:12 -04005use crate::syntax::{
Adrian Taylor0f8ab222020-10-29 21:31:56 -07006 attrs, error, Api, Doc, Enum, ExternFn, ExternType, Impl, Include, IncludeKind, Lang,
David Tolnayd60c07b2020-10-29 13:18:49 -07007 Namespace, Pair, Receiver, Ref, ResolvableName, Signature, Slice, Struct, Ty1, Type, TypeAlias,
8 Var, Variant,
David Tolnay7db73692019-10-20 14:51:12 -04009};
David Tolnay9a69d8a2020-11-16 09:16:41 -080010use proc_macro2::{Delimiter, Group, Span, TokenStream, TokenTree};
David Tolnaye2e303f2020-05-10 20:52:00 -070011use quote::{format_ident, quote, quote_spanned};
David Tolnaye2f9ec42020-05-07 16:19:33 -070012use syn::parse::{ParseStream, Parser};
David Tolnaye3a48152020-04-08 19:38:05 -070013use syn::punctuated::Punctuated;
David Tolnay7db73692019-10-20 14:51:12 -040014use syn::{
David Tolnay17e137f2020-05-08 15:55:28 -070015 Abi, Attribute, Error, Fields, FnArg, ForeignItem, ForeignItemFn, ForeignItemType,
David Tolnay9938b642020-11-15 18:11:40 -080016 GenericArgument, GenericParam, Generics, Ident, ItemEnum, ItemImpl, ItemStruct, LitStr, Pat,
17 PathArguments, Result, ReturnType, Token, Type as RustType, TypeBareFn, TypePath,
18 TypeReference, TypeSlice,
David Tolnay7db73692019-10-20 14:51:12 -040019};
20
David Tolnaye3a48152020-04-08 19:38:05 -070021pub mod kw {
David Tolnayb27f7872020-11-15 15:07:04 -080022 syn::custom_keyword!(Pin);
David Tolnaye3a48152020-04-08 19:38:05 -070023 syn::custom_keyword!(Result);
24}
25
David Tolnayd7a3a182020-11-01 20:45:14 -080026pub fn parse_items(
27 cx: &mut Errors,
28 items: Vec<Item>,
29 trusted: bool,
30 namespace: &Namespace,
31) -> Vec<Api> {
David Tolnay7db73692019-10-20 14:51:12 -040032 let mut apis = Vec::new();
33 for item in items {
34 match item {
David Tolnay302b1752020-11-10 18:43:24 -080035 Item::Struct(item) => match parse_struct(cx, item, namespace) {
David Tolnay52759782020-05-03 23:59:40 -070036 Ok(strct) => apis.push(strct),
37 Err(err) => cx.push(err),
38 },
David Tolnay302b1752020-11-10 18:43:24 -080039 Item::Enum(item) => match parse_enum(cx, item, namespace) {
David Tolnay52759782020-05-03 23:59:40 -070040 Ok(enm) => apis.push(enm),
41 Err(err) => cx.push(err),
42 },
Adrian Taylorc8713432020-10-21 18:20:55 -070043 Item::ForeignMod(foreign_mod) => {
David Tolnayd7a3a182020-11-01 20:45:14 -080044 parse_foreign_mod(cx, foreign_mod, &mut apis, trusted, namespace)
Adrian Taylorc8713432020-10-21 18:20:55 -070045 }
David Tolnayd7a3a182020-11-01 20:45:14 -080046 Item::Impl(item) => match parse_impl(item, namespace) {
David Tolnay7e69f892020-10-03 22:20:22 -070047 Ok(imp) => apis.push(imp),
48 Err(err) => cx.push(err),
49 },
David Tolnay52759782020-05-03 23:59:40 -070050 Item::Use(item) => cx.error(item, error::USE_NOT_ALLOWED),
David Tolnay00a83852020-08-29 15:06:16 -070051 Item::Other(item) => cx.error(item, "unsupported item"),
David Tolnay7db73692019-10-20 14:51:12 -040052 }
53 }
David Tolnay52759782020-05-03 23:59:40 -070054 apis
David Tolnay7db73692019-10-20 14:51:12 -040055}
56
David Tolnay302b1752020-11-10 18:43:24 -080057fn parse_struct(cx: &mut Errors, item: ItemStruct, namespace: &Namespace) -> Result<Api> {
David Tolnay7db73692019-10-20 14:51:12 -040058 let generics = &item.generics;
59 if !generics.params.is_empty() || generics.where_clause.is_some() {
60 let struct_token = item.struct_token;
61 let ident = &item.ident;
62 let where_clause = &generics.where_clause;
63 let span = quote!(#struct_token #ident #generics #where_clause);
64 return Err(Error::new_spanned(
65 span,
66 "struct with generic parameters is not supported yet",
67 ));
68 }
69
70 let mut doc = Doc::new();
71 let mut derives = Vec::new();
David Tolnay302b1752020-11-10 18:43:24 -080072 let mut namespace = namespace.clone();
David Tolnayb129ea72020-05-10 14:29:30 -070073 attrs::parse(
David Tolnay3e628882020-05-10 15:30:14 -070074 cx,
David Tolnayb129ea72020-05-10 14:29:30 -070075 &item.attrs,
76 attrs::Parser {
77 doc: Some(&mut doc),
78 derives: Some(&mut derives),
David Tolnayd7a3a182020-11-01 20:45:14 -080079 namespace: Some(&mut namespace),
David Tolnayddf69e22020-05-10 22:08:20 -070080 ..Default::default()
David Tolnayb129ea72020-05-10 14:29:30 -070081 },
David Tolnay3e628882020-05-10 15:30:14 -070082 );
David Tolnay09462ac2020-03-20 14:58:41 -070083
David Tolnay302b1752020-11-10 18:43:24 -080084 let named_fields = match item.fields {
David Tolnay09462ac2020-03-20 14:58:41 -070085 Fields::Named(fields) => fields,
86 Fields::Unit => return Err(Error::new_spanned(item, "unit structs are not supported")),
87 Fields::Unnamed(_) => {
David Tolnay2c4b35f2020-10-04 21:22:30 -070088 return Err(Error::new_spanned(item, "tuple structs are not supported"));
David Tolnay09462ac2020-03-20 14:58:41 -070089 }
90 };
91
David Tolnay302b1752020-11-10 18:43:24 -080092 let fields = named_fields
93 .named
94 .into_iter()
95 .map(|field| {
96 Ok(Var {
97 ident: field.ident.unwrap(),
98 ty: parse_type(&field.ty, &namespace)?,
99 })
100 })
101 .collect::<Result<_>>()?;
102
David Tolnay09462ac2020-03-20 14:58:41 -0700103 Ok(Api::Struct(Struct {
104 doc,
105 derives,
106 struct_token: item.struct_token,
David Tolnay302b1752020-11-10 18:43:24 -0800107 name: Pair::new(namespace, item.ident),
108 brace_token: named_fields.brace_token,
109 fields,
David Tolnay09462ac2020-03-20 14:58:41 -0700110 }))
David Tolnay7db73692019-10-20 14:51:12 -0400111}
112
David Tolnay302b1752020-11-10 18:43:24 -0800113fn parse_enum(cx: &mut Errors, item: ItemEnum, namespace: &Namespace) -> Result<Api> {
Joel Galensonc03402a2020-04-23 17:31:09 -0700114 let generics = &item.generics;
115 if !generics.params.is_empty() || generics.where_clause.is_some() {
116 let enum_token = item.enum_token;
117 let ident = &item.ident;
118 let where_clause = &generics.where_clause;
119 let span = quote!(#enum_token #ident #generics #where_clause);
120 return Err(Error::new_spanned(
121 span,
122 "enums with generic parameters are not allowed",
123 ));
124 }
125
David Tolnayddf69e22020-05-10 22:08:20 -0700126 let mut doc = Doc::new();
127 let mut repr = None;
David Tolnay302b1752020-11-10 18:43:24 -0800128 let mut namespace = namespace.clone();
David Tolnayddf69e22020-05-10 22:08:20 -0700129 attrs::parse(
130 cx,
131 &item.attrs,
132 attrs::Parser {
133 doc: Some(&mut doc),
134 repr: Some(&mut repr),
David Tolnayd7a3a182020-11-01 20:45:14 -0800135 namespace: Some(&mut namespace),
David Tolnayddf69e22020-05-10 22:08:20 -0700136 ..Default::default()
137 },
138 );
Joel Galensonc03402a2020-04-23 17:31:09 -0700139
Joel Galenson88547732020-05-05 08:23:42 -0700140 let mut variants = Vec::new();
David Tolnayddf69e22020-05-10 22:08:20 -0700141 let mut discriminants = DiscriminantSet::new(repr);
Joel Galenson88547732020-05-05 08:23:42 -0700142 for variant in item.variants {
143 match variant.fields {
Joel Galensonc03402a2020-04-23 17:31:09 -0700144 Fields::Unit => {}
145 _ => {
David Tolnay0435a812020-05-10 21:58:15 -0700146 cx.error(variant, "enums with data are not supported yet");
147 break;
Joel Galensonc03402a2020-04-23 17:31:09 -0700148 }
149 }
David Tolnay17e137f2020-05-08 15:55:28 -0700150 let expr = variant.discriminant.as_ref().map(|(_, expr)| expr);
151 let try_discriminant = match &expr {
152 Some(lit) => discriminants.insert(lit),
153 None => discriminants.insert_next(),
154 };
155 let discriminant = match try_discriminant {
156 Ok(discriminant) => discriminant,
David Tolnay0435a812020-05-10 21:58:15 -0700157 Err(err) => {
158 cx.error(variant, err);
159 break;
160 }
David Tolnay17e137f2020-05-08 15:55:28 -0700161 };
David Tolnay2b8bf6d2020-05-10 17:37:16 -0700162 let expr = variant.discriminant.map(|(_, expr)| expr);
Joel Galenson88547732020-05-05 08:23:42 -0700163 variants.push(Variant {
164 ident: variant.ident,
David Tolnay7ae018f2020-05-05 10:11:30 -0700165 discriminant,
David Tolnay2b8bf6d2020-05-10 17:37:16 -0700166 expr,
Joel Galenson88547732020-05-05 08:23:42 -0700167 });
Joel Galensonc03402a2020-04-23 17:31:09 -0700168 }
169
David Tolnaye2e303f2020-05-10 20:52:00 -0700170 let enum_token = item.enum_token;
171 let brace_token = item.brace_token;
172
David Tolnay0435a812020-05-10 21:58:15 -0700173 let mut repr = U8;
174 match discriminants.inferred_repr() {
175 Ok(inferred) => repr = inferred,
David Tolnaye2e303f2020-05-10 20:52:00 -0700176 Err(err) => {
177 let span = quote_spanned!(brace_token.span=> #enum_token {});
David Tolnay0435a812020-05-10 21:58:15 -0700178 cx.error(span, err);
179 variants.clear();
David Tolnaye2e303f2020-05-10 20:52:00 -0700180 }
David Tolnay0435a812020-05-10 21:58:15 -0700181 }
David Tolnaye2e303f2020-05-10 20:52:00 -0700182
David Tolnay9a69d8a2020-11-16 09:16:41 -0800183 let ident = Ident::new(repr.as_ref(), Span::call_site());
184 let repr_type = Type::Ident(ResolvableName::new(ident));
185
Joel Galensonc03402a2020-04-23 17:31:09 -0700186 Ok(Api::Enum(Enum {
187 doc,
David Tolnaye2e303f2020-05-10 20:52:00 -0700188 enum_token,
David Tolnay17a934c2020-11-02 00:40:04 -0800189 name: Pair::new(namespace, item.ident),
David Tolnaye2e303f2020-05-10 20:52:00 -0700190 brace_token,
David Tolnay7ae018f2020-05-05 10:11:30 -0700191 variants,
David Tolnaye2e303f2020-05-10 20:52:00 -0700192 repr,
David Tolnay9a69d8a2020-11-16 09:16:41 -0800193 repr_type,
Joel Galensonc03402a2020-04-23 17:31:09 -0700194 }))
195}
196
David Tolnay805dca32020-08-29 19:09:55 -0700197fn parse_foreign_mod(
198 cx: &mut Errors,
199 foreign_mod: ItemForeignMod,
200 out: &mut Vec<Api>,
201 trusted: bool,
David Tolnayd7a3a182020-11-01 20:45:14 -0800202 namespace: &Namespace,
David Tolnay805dca32020-08-29 19:09:55 -0700203) {
David Tolnay3a451732020-08-29 18:54:55 -0700204 let lang = match parse_lang(&foreign_mod.abi) {
David Tolnay52759782020-05-03 23:59:40 -0700205 Ok(lang) => lang,
206 Err(err) => return cx.push(err),
207 };
David Tolnay7db73692019-10-20 14:51:12 -0400208
David Tolnay3a451732020-08-29 18:54:55 -0700209 match lang {
210 Lang::Rust => {
David Tolnay805dca32020-08-29 19:09:55 -0700211 if foreign_mod.unsafety.is_some() {
David Tolnay3a451732020-08-29 18:54:55 -0700212 let unsafety = foreign_mod.unsafety;
David Tolnay7be5b1f2020-11-11 10:48:32 -0800213 let abi = &foreign_mod.abi;
David Tolnay3a451732020-08-29 18:54:55 -0700214 let span = quote!(#unsafety #abi);
215 cx.error(span, "extern \"Rust\" block does not need to be unsafe");
216 }
217 }
218 Lang::Cxx => {}
219 }
220
David Tolnay805dca32020-08-29 19:09:55 -0700221 let trusted = trusted || foreign_mod.unsafety.is_some();
222
David Tolnay7feb2642020-11-10 18:49:04 -0800223 let mut namespace = namespace.clone();
224 attrs::parse(
225 cx,
226 &foreign_mod.attrs,
227 attrs::Parser {
228 namespace: Some(&mut namespace),
229 ..Default::default()
230 },
231 );
232
Joel Galensone1e969d2020-04-21 12:50:20 -0700233 let mut items = Vec::new();
234 for foreign in &foreign_mod.items {
235 match foreign {
Adrian Taylorc8713432020-10-21 18:20:55 -0700236 ForeignItem::Type(foreign) => {
David Tolnay7feb2642020-11-10 18:49:04 -0800237 match parse_extern_type(cx, foreign, lang, trusted, &namespace) {
Adrian Taylorc8713432020-10-21 18:20:55 -0700238 Ok(ety) => items.push(ety),
239 Err(err) => cx.push(err),
240 }
241 }
David Tolnay556738d2020-11-15 13:58:44 -0800242 ForeignItem::Fn(foreign) => {
243 match parse_extern_fn(cx, foreign, lang, trusted, &namespace) {
244 Ok(efn) => items.push(efn),
245 Err(err) => cx.push(err),
246 }
247 }
David Tolnay7db73692019-10-20 14:51:12 -0400248 ForeignItem::Macro(foreign) if foreign.mac.path.is_ident("include") => {
David Tolnay91e87fa2020-05-11 19:10:23 -0700249 match foreign.mac.parse_body_with(parse_include) {
David Tolnay52759782020-05-03 23:59:40 -0700250 Ok(include) => items.push(Api::Include(include)),
251 Err(err) => cx.push(err),
252 }
David Tolnay7db73692019-10-20 14:51:12 -0400253 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700254 ForeignItem::Verbatim(tokens) => {
David Tolnay7feb2642020-11-10 18:49:04 -0800255 match parse_extern_verbatim(cx, tokens, lang, &namespace) {
Adrian Taylorc8713432020-10-21 18:20:55 -0700256 Ok(api) => items.push(api),
257 Err(err) => cx.push(err),
258 }
259 }
David Tolnay52759782020-05-03 23:59:40 -0700260 _ => cx.error(foreign, "unsupported foreign item"),
David Tolnay7db73692019-10-20 14:51:12 -0400261 }
262 }
David Tolnaya1f29c42020-04-22 18:01:38 -0700263
David Tolnay7be5b1f2020-11-11 10:48:32 -0800264 if !trusted
265 && items.iter().any(|api| match api {
266 Api::CxxFunction(efn) => efn.unsafety.is_none(),
267 _ => false,
268 })
269 {
270 cx.error(
271 foreign_mod.abi,
272 "block must be declared `unsafe extern \"C++\"` if it contains any safe-to-call C++ functions",
273 );
274 }
275
David Tolnaya1f29c42020-04-22 18:01:38 -0700276 let mut types = items.iter().filter_map(|item| match item {
David Tolnay17a934c2020-11-02 00:40:04 -0800277 Api::CxxType(ety) | Api::RustType(ety) => Some(&ety.name),
278 Api::TypeAlias(alias) => Some(&alias.name),
David Tolnaya1f29c42020-04-22 18:01:38 -0700279 _ => None,
280 });
281 if let (Some(single_type), None) = (types.next(), types.next()) {
David Tolnaye5a015a2020-05-07 21:54:26 -0700282 let single_type = single_type.clone();
David Tolnaya1f29c42020-04-22 18:01:38 -0700283 for item in &mut items {
284 if let Api::CxxFunction(efn) | Api::RustFunction(efn) = item {
285 if let Some(receiver) = &mut efn.receiver {
Adrian Taylorc8713432020-10-21 18:20:55 -0700286 if receiver.ty.is_self() {
Adrian Taylor565ddf02020-10-29 21:12:36 -0700287 receiver.ty = ResolvableName::new(single_type.rust.clone());
David Tolnaya1f29c42020-04-22 18:01:38 -0700288 }
289 }
290 }
291 }
292 }
293
David Tolnay52759782020-05-03 23:59:40 -0700294 out.extend(items);
David Tolnay7db73692019-10-20 14:51:12 -0400295}
296
David Tolnay3a451732020-08-29 18:54:55 -0700297fn parse_lang(abi: &Abi) -> Result<Lang> {
David Tolnay7db73692019-10-20 14:51:12 -0400298 let name = match &abi.name {
299 Some(name) => name,
300 None => {
301 return Err(Error::new_spanned(
302 abi,
David Tolnayc72a9f62020-11-11 10:56:26 -0800303 "ABI name is required, extern \"C++\" or extern \"Rust\"",
David Tolnay7db73692019-10-20 14:51:12 -0400304 ));
305 }
306 };
307 match name.value().as_str() {
David Tolnayea25ac82020-11-11 10:35:23 -0800308 "C++" => Ok(Lang::Cxx),
David Tolnay7db73692019-10-20 14:51:12 -0400309 "Rust" => Ok(Lang::Rust),
David Tolnay2757ca82020-11-14 15:12:06 -0800310 _ => Err(Error::new_spanned(
311 abi,
312 "unrecognized ABI, requires either \"C++\" or \"Rust\"",
313 )),
David Tolnay7db73692019-10-20 14:51:12 -0400314 }
315}
316
David Tolnay00f236a2020-08-29 19:07:18 -0700317fn parse_extern_type(
318 cx: &mut Errors,
319 foreign_type: &ForeignItemType,
320 lang: Lang,
321 trusted: bool,
David Tolnay302b1752020-11-10 18:43:24 -0800322 namespace: &Namespace,
David Tolnay00f236a2020-08-29 19:07:18 -0700323) -> Result<Api> {
Adrian Taylorc8713432020-10-21 18:20:55 -0700324 let mut doc = Doc::new();
David Tolnay302b1752020-11-10 18:43:24 -0800325 let mut namespace = namespace.clone();
Adrian Taylorc8713432020-10-21 18:20:55 -0700326 attrs::parse(
327 cx,
328 &foreign_type.attrs,
329 attrs::Parser {
330 doc: Some(&mut doc),
David Tolnayd7a3a182020-11-01 20:45:14 -0800331 namespace: Some(&mut namespace),
Adrian Taylorc8713432020-10-21 18:20:55 -0700332 ..Default::default()
333 },
334 );
David Tolnay7db73692019-10-20 14:51:12 -0400335 let type_token = foreign_type.type_token;
336 let ident = foreign_type.ident.clone();
David Tolnayc8361022020-08-25 21:57:53 -0700337 let semi_token = foreign_type.semi_token;
David Tolnay295ef6b2020-05-07 16:10:22 -0700338 let api_type = match lang {
339 Lang::Cxx => Api::CxxType,
340 Lang::Rust => Api::RustType,
341 };
342 Ok(api_type(ExternType {
David Tolnay7db73692019-10-20 14:51:12 -0400343 doc,
344 type_token,
David Tolnay17a934c2020-11-02 00:40:04 -0800345 name: Pair::new(namespace, ident),
David Tolnayc8361022020-08-25 21:57:53 -0700346 semi_token,
David Tolnay00f236a2020-08-29 19:07:18 -0700347 trusted,
David Tolnay295ef6b2020-05-07 16:10:22 -0700348 }))
David Tolnay7db73692019-10-20 14:51:12 -0400349}
350
Adrian Taylorc8713432020-10-21 18:20:55 -0700351fn parse_extern_fn(
352 cx: &mut Errors,
353 foreign_fn: &ForeignItemFn,
354 lang: Lang,
David Tolnay556738d2020-11-15 13:58:44 -0800355 trusted: bool,
David Tolnay302b1752020-11-10 18:43:24 -0800356 namespace: &Namespace,
Adrian Taylorc8713432020-10-21 18:20:55 -0700357) -> Result<Api> {
David Tolnay7db73692019-10-20 14:51:12 -0400358 let generics = &foreign_fn.sig.generics;
David Tolnay9938b642020-11-15 18:11:40 -0800359 if generics.where_clause.is_some()
360 || generics.params.iter().any(|param| match param {
361 GenericParam::Lifetime(lifetime) => !lifetime.bounds.is_empty(),
362 GenericParam::Type(_) | GenericParam::Const(_) => true,
363 })
364 {
David Tolnay7db73692019-10-20 14:51:12 -0400365 return Err(Error::new_spanned(
366 foreign_fn,
367 "extern function with generic parameters is not supported yet",
368 ));
369 }
370 if let Some(variadic) = &foreign_fn.sig.variadic {
371 return Err(Error::new_spanned(
372 variadic,
373 "variadic function is not supported yet",
374 ));
375 }
376
David Tolnay938ca852020-10-09 19:25:08 -0700377 let mut doc = Doc::new();
378 let mut cxx_name = None;
379 let mut rust_name = None;
David Tolnay302b1752020-11-10 18:43:24 -0800380 let mut namespace = namespace.clone();
David Tolnay938ca852020-10-09 19:25:08 -0700381 attrs::parse(
382 cx,
383 &foreign_fn.attrs,
384 attrs::Parser {
385 doc: Some(&mut doc),
386 cxx_name: Some(&mut cxx_name),
387 rust_name: Some(&mut rust_name),
David Tolnayd7a3a182020-11-01 20:45:14 -0800388 namespace: Some(&mut namespace),
David Tolnay938ca852020-10-09 19:25:08 -0700389 ..Default::default()
390 },
391 );
392
David Tolnay7db73692019-10-20 14:51:12 -0400393 let mut receiver = None;
David Tolnaye3a48152020-04-08 19:38:05 -0700394 let mut args = Punctuated::new();
395 for arg in foreign_fn.sig.inputs.pairs() {
396 let (arg, comma) = arg.into_tuple();
David Tolnay7db73692019-10-20 14:51:12 -0400397 match arg {
David Tolnay1dd11a12020-04-22 12:31:48 -0700398 FnArg::Receiver(arg) => {
David Tolnaya1f29c42020-04-22 18:01:38 -0700399 if let Some((ampersand, lifetime)) = &arg.reference {
400 receiver = Some(Receiver {
David Tolnay9c76c392020-11-15 16:07:08 -0800401 pinned: false,
David Tolnaya1f29c42020-04-22 18:01:38 -0700402 ampersand: *ampersand,
403 lifetime: lifetime.clone(),
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800404 mutable: arg.mutability.is_some(),
David Tolnaya1f29c42020-04-22 18:01:38 -0700405 var: arg.self_token,
Adrian Taylorc8713432020-10-21 18:20:55 -0700406 ty: ResolvableName::make_self(arg.self_token.span),
David Tolnaya1f29c42020-04-22 18:01:38 -0700407 shorthand: true,
David Tolnay9c76c392020-11-15 16:07:08 -0800408 pin_tokens: None,
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800409 mutability: arg.mutability,
David Tolnaya1f29c42020-04-22 18:01:38 -0700410 });
411 continue;
Joel Galensone1e969d2020-04-21 12:50:20 -0700412 }
David Tolnay1dd11a12020-04-22 12:31:48 -0700413 return Err(Error::new_spanned(arg, "unsupported signature"));
David Tolnay7db73692019-10-20 14:51:12 -0400414 }
415 FnArg::Typed(arg) => {
416 let ident = match arg.pat.as_ref() {
417 Pat::Ident(pat) => pat.ident.clone(),
Joel Galensonba676072020-04-27 15:55:45 -0700418 Pat::Wild(pat) => {
419 Ident::new(&format!("_{}", args.len()), pat.underscore_token.span)
420 }
David Tolnay7db73692019-10-20 14:51:12 -0400421 _ => return Err(Error::new_spanned(arg, "unsupported signature")),
422 };
David Tolnayd7a3a182020-11-01 20:45:14 -0800423 let ty = parse_type(&arg.ty, &namespace)?;
David Tolnay7db73692019-10-20 14:51:12 -0400424 if ident != "self" {
David Tolnaye3a48152020-04-08 19:38:05 -0700425 args.push_value(Var { ident, ty });
426 if let Some(comma) = comma {
427 args.push_punct(*comma);
428 }
David Tolnay7db73692019-10-20 14:51:12 -0400429 continue;
430 }
431 if let Type::Ref(reference) = ty {
432 if let Type::Ident(ident) = reference.inner {
433 receiver = Some(Receiver {
David Tolnay9c76c392020-11-15 16:07:08 -0800434 pinned: reference.pinned,
David Tolnayfb6e3862020-04-20 01:33:23 -0700435 ampersand: reference.ampersand,
David Tolnay0bd50fa2020-04-22 15:31:33 -0700436 lifetime: reference.lifetime,
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800437 mutable: reference.mutable,
Adrian Taylorc8713432020-10-21 18:20:55 -0700438 var: Token![self](ident.rust.span()),
David Tolnay05e11cc2020-04-20 02:13:56 -0700439 ty: ident,
David Tolnay62d360c2020-04-22 16:26:21 -0700440 shorthand: false,
David Tolnay9c76c392020-11-15 16:07:08 -0800441 pin_tokens: reference.pin_tokens,
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800442 mutability: reference.mutability,
David Tolnay7db73692019-10-20 14:51:12 -0400443 });
444 continue;
445 }
446 }
447 return Err(Error::new_spanned(arg, "unsupported method receiver"));
448 }
449 }
450 }
David Tolnay59b7ede2020-03-16 00:30:23 -0700451
David Tolnaye3a48152020-04-08 19:38:05 -0700452 let mut throws_tokens = None;
David Tolnayd7a3a182020-11-01 20:45:14 -0800453 let ret = parse_return_type(&foreign_fn.sig.output, &mut throws_tokens, &namespace)?;
David Tolnaye3a48152020-04-08 19:38:05 -0700454 let throws = throws_tokens.is_some();
David Tolnaybdb576c2020-09-06 23:45:55 -0700455 let unsafety = foreign_fn.sig.unsafety;
David Tolnay7db73692019-10-20 14:51:12 -0400456 let fn_token = foreign_fn.sig.fn_token;
David Tolnay17a934c2020-11-02 00:40:04 -0800457 let name = Pair::new_from_differing_names(
David Tolnayd7a3a182020-11-01 20:45:14 -0800458 namespace,
Adrian Taylor0f8ab222020-10-29 21:31:56 -0700459 cxx_name.unwrap_or(foreign_fn.sig.ident.clone()),
460 rust_name.unwrap_or(foreign_fn.sig.ident.clone()),
461 );
David Tolnay9938b642020-11-15 18:11:40 -0800462 let generics = generics.clone();
David Tolnaye3a48152020-04-08 19:38:05 -0700463 let paren_token = foreign_fn.sig.paren_token;
David Tolnay7db73692019-10-20 14:51:12 -0400464 let semi_token = foreign_fn.semi_token;
David Tolnay295ef6b2020-05-07 16:10:22 -0700465 let api_function = match lang {
466 Lang::Cxx => Api::CxxFunction,
467 Lang::Rust => Api::RustFunction,
468 };
David Tolnayd95b1192020-03-18 20:07:46 -0700469
David Tolnay295ef6b2020-05-07 16:10:22 -0700470 Ok(api_function(ExternFn {
David Tolnay6cde49f2020-03-16 12:25:45 -0700471 lang,
David Tolnay7db73692019-10-20 14:51:12 -0400472 doc,
David Tolnay17a934c2020-11-02 00:40:04 -0800473 name,
David Tolnay16448732020-03-18 12:39:36 -0700474 sig: Signature {
David Tolnaybdb576c2020-09-06 23:45:55 -0700475 unsafety,
David Tolnay16448732020-03-18 12:39:36 -0700476 fn_token,
David Tolnay9938b642020-11-15 18:11:40 -0800477 generics,
David Tolnay16448732020-03-18 12:39:36 -0700478 receiver,
479 args,
480 ret,
481 throws,
David Tolnaye3a48152020-04-08 19:38:05 -0700482 paren_token,
483 throws_tokens,
David Tolnay16448732020-03-18 12:39:36 -0700484 },
David Tolnay7db73692019-10-20 14:51:12 -0400485 semi_token,
David Tolnay556738d2020-11-15 13:58:44 -0800486 trusted,
David Tolnay295ef6b2020-05-07 16:10:22 -0700487 }))
David Tolnay7db73692019-10-20 14:51:12 -0400488}
489
Adrian Taylorc8713432020-10-21 18:20:55 -0700490fn parse_extern_verbatim(
491 cx: &mut Errors,
492 tokens: &TokenStream,
493 lang: Lang,
David Tolnay302b1752020-11-10 18:43:24 -0800494 namespace: &Namespace,
Adrian Taylorc8713432020-10-21 18:20:55 -0700495) -> Result<Api> {
David Tolnaye2f9ec42020-05-07 16:19:33 -0700496 // type Alias = crate::path::to::Type;
David Tolnay3e628882020-05-10 15:30:14 -0700497 let parse = |input: ParseStream| -> Result<TypeAlias> {
David Tolnaye2f9ec42020-05-07 16:19:33 -0700498 let attrs = input.call(Attribute::parse_outer)?;
499 let type_token: Token![type] = match input.parse()? {
500 Some(type_token) => type_token,
501 None => {
502 let span = input.cursor().token_stream();
503 return Err(Error::new_spanned(span, "unsupported foreign item"));
504 }
505 };
506 let ident: Ident = input.parse()?;
507 let eq_token: Token![=] = input.parse()?;
508 let ty: RustType = input.parse()?;
509 let semi_token: Token![;] = input.parse()?;
Adrian Taylorc8713432020-10-21 18:20:55 -0700510 let mut doc = Doc::new();
David Tolnay302b1752020-11-10 18:43:24 -0800511 let mut namespace = namespace.clone();
Adrian Taylorc8713432020-10-21 18:20:55 -0700512 attrs::parse(
513 cx,
514 &attrs,
515 attrs::Parser {
516 doc: Some(&mut doc),
David Tolnayd7a3a182020-11-01 20:45:14 -0800517 namespace: Some(&mut namespace),
Adrian Taylorc8713432020-10-21 18:20:55 -0700518 ..Default::default()
519 },
520 );
David Tolnaye2f9ec42020-05-07 16:19:33 -0700521
522 Ok(TypeAlias {
Bryan Henry890083d2020-09-13 10:34:31 -0700523 doc,
David Tolnaye2f9ec42020-05-07 16:19:33 -0700524 type_token,
David Tolnay17a934c2020-11-02 00:40:04 -0800525 name: Pair::new(namespace, ident),
David Tolnaye2f9ec42020-05-07 16:19:33 -0700526 eq_token,
527 ty,
528 semi_token,
529 })
David Tolnay3e628882020-05-10 15:30:14 -0700530 };
David Tolnaye2f9ec42020-05-07 16:19:33 -0700531
532 let type_alias = parse.parse2(tokens.clone())?;
533 match lang {
534 Lang::Cxx => Ok(Api::TypeAlias(type_alias)),
535 Lang::Rust => {
536 let (type_token, semi_token) = (type_alias.type_token, type_alias.semi_token);
537 let span = quote!(#type_token #semi_token);
538 let msg = "type alias in extern \"Rust\" block is not supported";
539 Err(Error::new_spanned(span, msg))
540 }
541 }
542}
543
David Tolnayd7a3a182020-11-01 20:45:14 -0800544fn parse_impl(imp: ItemImpl, namespace: &Namespace) -> Result<Api> {
David Tolnay7e69f892020-10-03 22:20:22 -0700545 if !imp.items.is_empty() {
546 let mut span = Group::new(Delimiter::Brace, TokenStream::new());
547 span.set_span(imp.brace_token.span);
548 return Err(Error::new_spanned(span, "expected an empty impl block"));
549 }
550
551 let self_ty = &imp.self_ty;
552 if let Some((bang, path, for_token)) = &imp.trait_ {
553 let span = quote!(#bang #path #for_token #self_ty);
554 return Err(Error::new_spanned(
555 span,
556 "unexpected impl, expected something like `impl UniquePtr<T> {}`",
557 ));
558 }
559
560 let generics = &imp.generics;
561 if !generics.params.is_empty() || generics.where_clause.is_some() {
562 return Err(Error::new_spanned(
563 imp,
564 "generic parameters on an impl is not supported",
565 ));
566 }
567
568 Ok(Api::Impl(Impl {
569 impl_token: imp.impl_token,
David Tolnayd7a3a182020-11-01 20:45:14 -0800570 ty: parse_type(&self_ty, namespace)?,
David Tolnay7e69f892020-10-03 22:20:22 -0700571 brace_token: imp.brace_token,
572 }))
573}
574
David Tolnayb0cd3272020-10-27 21:32:54 -0700575fn parse_include(input: ParseStream) -> Result<Include> {
David Tolnay91e87fa2020-05-11 19:10:23 -0700576 if input.peek(LitStr) {
David Tolnayb0cd3272020-10-27 21:32:54 -0700577 let lit: LitStr = input.parse()?;
578 let span = lit.span();
579 return Ok(Include {
580 path: lit.value(),
581 kind: IncludeKind::Quoted,
582 begin_span: span,
583 end_span: span,
584 });
David Tolnay91e87fa2020-05-11 19:10:23 -0700585 }
586
587 if input.peek(Token![<]) {
588 let mut path = String::new();
David Tolnayb0cd3272020-10-27 21:32:54 -0700589
David Tolnay2cc2e3a2020-10-28 13:12:35 -0700590 let langle: Token![<] = input.parse()?;
David Tolnay91e87fa2020-05-11 19:10:23 -0700591 while !input.is_empty() && !input.peek(Token![>]) {
592 let token: TokenTree = input.parse()?;
593 match token {
594 TokenTree::Ident(token) => path += &token.to_string(),
595 TokenTree::Literal(token)
596 if token
597 .to_string()
598 .starts_with(|ch: char| ch.is_ascii_digit()) =>
599 {
600 path += &token.to_string();
601 }
602 TokenTree::Punct(token) => path.push(token.as_char()),
603 _ => return Err(Error::new(token.span(), "unexpected token in include path")),
604 }
605 }
David Tolnayb0cd3272020-10-27 21:32:54 -0700606 let rangle: Token![>] = input.parse()?;
David Tolnayb0cd3272020-10-27 21:32:54 -0700607
608 return Ok(Include {
609 path,
610 kind: IncludeKind::Bracketed,
David Tolnay2cc2e3a2020-10-28 13:12:35 -0700611 begin_span: langle.span,
612 end_span: rangle.span,
David Tolnayb0cd3272020-10-27 21:32:54 -0700613 });
David Tolnay91e87fa2020-05-11 19:10:23 -0700614 }
615
616 Err(input.error("expected \"quoted/path/to\" or <bracketed/path/to>"))
617}
618
David Tolnayd7a3a182020-11-01 20:45:14 -0800619fn parse_type(ty: &RustType, namespace: &Namespace) -> Result<Type> {
David Tolnay59b7ede2020-03-16 00:30:23 -0700620 match ty {
David Tolnayd7a3a182020-11-01 20:45:14 -0800621 RustType::Reference(ty) => parse_type_reference(ty, namespace),
622 RustType::Path(ty) => parse_type_path(ty, namespace),
623 RustType::Slice(ty) => parse_type_slice(ty, namespace),
624 RustType::BareFn(ty) => parse_type_fn(ty, namespace),
David Tolnayb40b9db2020-03-18 13:50:31 -0700625 RustType::Tuple(ty) if ty.elems.is_empty() => Ok(Type::Void(ty.paren_token.span)),
626 _ => Err(Error::new_spanned(ty, "unsupported type")),
627 }
628}
629
David Tolnayd7a3a182020-11-01 20:45:14 -0800630fn parse_type_reference(ty: &TypeReference, namespace: &Namespace) -> Result<Type> {
631 let inner = parse_type(&ty.elem, namespace)?;
David Tolnayb40b9db2020-03-18 13:50:31 -0700632 let which = match &inner {
Adrian Taylorc8713432020-10-21 18:20:55 -0700633 Type::Ident(ident) if ident.rust == "str" => {
David Tolnayb40b9db2020-03-18 13:50:31 -0700634 if ty.mutability.is_some() {
635 return Err(Error::new_spanned(ty, "unsupported type"));
636 } else {
637 Type::Str
David Tolnay7db73692019-10-20 14:51:12 -0400638 }
639 }
David Tolnayeebe9b72020-04-14 16:32:18 -0700640 Type::Slice(slice) => match &slice.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -0700641 Type::Ident(ident) if ident.rust == U8 && ty.mutability.is_none() => Type::SliceRefU8,
David Tolnayeebe9b72020-04-14 16:32:18 -0700642 _ => Type::Ref,
David Tolnayeb952ba2020-04-14 15:02:24 -0700643 },
David Tolnayb40b9db2020-03-18 13:50:31 -0700644 _ => Type::Ref,
645 };
646 Ok(which(Box::new(Ref {
David Tolnayb27f7872020-11-15 15:07:04 -0800647 pinned: false,
David Tolnayb40b9db2020-03-18 13:50:31 -0700648 ampersand: ty.and_token,
David Tolnay0bd50fa2020-04-22 15:31:33 -0700649 lifetime: ty.lifetime.clone(),
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800650 mutable: ty.mutability.is_some(),
David Tolnayb40b9db2020-03-18 13:50:31 -0700651 inner,
David Tolnayb27f7872020-11-15 15:07:04 -0800652 pin_tokens: None,
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800653 mutability: ty.mutability,
David Tolnayb40b9db2020-03-18 13:50:31 -0700654 })))
655}
656
David Tolnayd7a3a182020-11-01 20:45:14 -0800657fn parse_type_path(ty: &TypePath, namespace: &Namespace) -> Result<Type> {
David Tolnayb40b9db2020-03-18 13:50:31 -0700658 let path = &ty.path;
659 if ty.qself.is_none() && path.leading_colon.is_none() && path.segments.len() == 1 {
660 let segment = &path.segments[0];
661 let ident = segment.ident.clone();
662 match &segment.arguments {
David Tolnayda71ef62020-11-02 00:36:00 -0800663 PathArguments::None => return Ok(Type::Ident(ResolvableName::new(ident))),
David Tolnayb40b9db2020-03-18 13:50:31 -0700664 PathArguments::AngleBracketed(generic) => {
665 if ident == "UniquePtr" && generic.args.len() == 1 {
666 if let GenericArgument::Type(arg) = &generic.args[0] {
David Tolnayd7a3a182020-11-01 20:45:14 -0800667 let inner = parse_type(arg, namespace)?;
David Tolnayb40b9db2020-03-18 13:50:31 -0700668 return Ok(Type::UniquePtr(Box::new(Ty1 {
David Tolnayda71ef62020-11-02 00:36:00 -0800669 name: ident,
David Tolnayb40b9db2020-03-18 13:50:31 -0700670 langle: generic.lt_token,
671 inner,
672 rangle: generic.gt_token,
673 })));
674 }
David Tolnaye1dcdf72020-04-24 17:40:55 -0700675 } else if ident == "CxxVector" && generic.args.len() == 1 {
Myron Ahneba35cf2020-02-05 19:41:51 +0700676 if let GenericArgument::Type(arg) = &generic.args[0] {
David Tolnayd7a3a182020-11-01 20:45:14 -0800677 let inner = parse_type(arg, namespace)?;
David Tolnay4377a9e2020-04-24 15:20:26 -0700678 return Ok(Type::CxxVector(Box::new(Ty1 {
David Tolnayda71ef62020-11-02 00:36:00 -0800679 name: ident,
Myron Ahneba35cf2020-02-05 19:41:51 +0700680 langle: generic.lt_token,
681 inner,
682 rangle: generic.gt_token,
683 })));
684 }
David Tolnayb40b9db2020-03-18 13:50:31 -0700685 } else if ident == "Box" && generic.args.len() == 1 {
686 if let GenericArgument::Type(arg) = &generic.args[0] {
David Tolnayd7a3a182020-11-01 20:45:14 -0800687 let inner = parse_type(arg, namespace)?;
David Tolnayb40b9db2020-03-18 13:50:31 -0700688 return Ok(Type::RustBox(Box::new(Ty1 {
David Tolnayda71ef62020-11-02 00:36:00 -0800689 name: ident,
David Tolnayb40b9db2020-03-18 13:50:31 -0700690 langle: generic.lt_token,
691 inner,
692 rangle: generic.gt_token,
693 })));
694 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700695 } else if ident == "Vec" && generic.args.len() == 1 {
696 if let GenericArgument::Type(arg) = &generic.args[0] {
David Tolnayd7a3a182020-11-01 20:45:14 -0800697 let inner = parse_type(arg, namespace)?;
Myron Ahneba35cf2020-02-05 19:41:51 +0700698 return Ok(Type::RustVec(Box::new(Ty1 {
David Tolnayda71ef62020-11-02 00:36:00 -0800699 name: ident,
Myron Ahneba35cf2020-02-05 19:41:51 +0700700 langle: generic.lt_token,
701 inner,
702 rangle: generic.gt_token,
703 })));
704 }
David Tolnaycbc2a4b2020-11-15 15:08:44 -0800705 } else if ident == "Pin" && generic.args.len() == 1 {
706 if let GenericArgument::Type(arg) = &generic.args[0] {
707 let inner = parse_type(arg, namespace)?;
David Tolnayc94035a2020-11-15 16:49:18 -0800708 let pin_token = kw::Pin(ident.span());
David Tolnaycbc2a4b2020-11-15 15:08:44 -0800709 if let Type::Ref(mut inner) = inner {
David Tolnaycbc2a4b2020-11-15 15:08:44 -0800710 inner.pinned = true;
711 inner.pin_tokens =
712 Some((pin_token, generic.lt_token, generic.gt_token));
713 return Ok(Type::Ref(inner));
714 }
715 }
David Tolnayb40b9db2020-03-18 13:50:31 -0700716 }
717 }
718 PathArguments::Parenthesized(_) => {}
David Tolnayfb134ed2020-03-15 23:17:48 -0700719 }
David Tolnay7db73692019-10-20 14:51:12 -0400720 }
721 Err(Error::new_spanned(ty, "unsupported type"))
722}
723
David Tolnayd7a3a182020-11-01 20:45:14 -0800724fn parse_type_slice(ty: &TypeSlice, namespace: &Namespace) -> Result<Type> {
725 let inner = parse_type(&ty.elem, namespace)?;
David Tolnayeebe9b72020-04-14 16:32:18 -0700726 Ok(Type::Slice(Box::new(Slice {
727 bracket: ty.bracket_token,
728 inner,
729 })))
730}
731
David Tolnayd7a3a182020-11-01 20:45:14 -0800732fn parse_type_fn(ty: &TypeBareFn, namespace: &Namespace) -> Result<Type> {
David Tolnayc071b892020-03-18 16:59:53 -0700733 if ty.lifetimes.is_some() {
734 return Err(Error::new_spanned(
735 ty,
736 "function pointer with lifetime parameters is not supported yet",
737 ));
738 }
739 if ty.variadic.is_some() {
740 return Err(Error::new_spanned(
741 ty,
742 "variadic function pointer is not supported yet",
743 ));
744 }
745 let args = ty
746 .inputs
747 .iter()
748 .enumerate()
749 .map(|(i, arg)| {
David Tolnayd7a3a182020-11-01 20:45:14 -0800750 let ty = parse_type(&arg.ty, namespace)?;
David Tolnayc071b892020-03-18 16:59:53 -0700751 let ident = match &arg.name {
752 Some(ident) => ident.0.clone(),
753 None => format_ident!("_{}", i),
754 };
755 Ok(Var { ident, ty })
756 })
757 .collect::<Result<_>>()?;
David Tolnaye3a48152020-04-08 19:38:05 -0700758 let mut throws_tokens = None;
David Tolnayd7a3a182020-11-01 20:45:14 -0800759 let ret = parse_return_type(&ty.output, &mut throws_tokens, namespace)?;
David Tolnaye3a48152020-04-08 19:38:05 -0700760 let throws = throws_tokens.is_some();
David Tolnayc071b892020-03-18 16:59:53 -0700761 Ok(Type::Fn(Box::new(Signature {
David Tolnaybdb576c2020-09-06 23:45:55 -0700762 unsafety: ty.unsafety,
David Tolnayc071b892020-03-18 16:59:53 -0700763 fn_token: ty.fn_token,
David Tolnay9938b642020-11-15 18:11:40 -0800764 generics: Generics::default(),
David Tolnayc071b892020-03-18 16:59:53 -0700765 receiver: None,
766 args,
767 ret,
768 throws,
David Tolnaye3a48152020-04-08 19:38:05 -0700769 paren_token: ty.paren_token,
770 throws_tokens,
David Tolnayc071b892020-03-18 16:59:53 -0700771 })))
772}
773
David Tolnaye3a48152020-04-08 19:38:05 -0700774fn parse_return_type(
775 ty: &ReturnType,
776 throws_tokens: &mut Option<(kw::Result, Token![<], Token![>])>,
David Tolnayd7a3a182020-11-01 20:45:14 -0800777 namespace: &Namespace,
David Tolnaye3a48152020-04-08 19:38:05 -0700778) -> Result<Option<Type>> {
David Tolnayc071b892020-03-18 16:59:53 -0700779 let mut ret = match ty {
780 ReturnType::Default => return Ok(None),
781 ReturnType::Type(_, ret) => ret.as_ref(),
782 };
783 if let RustType::Path(ty) = ret {
784 let path = &ty.path;
785 if ty.qself.is_none() && path.leading_colon.is_none() && path.segments.len() == 1 {
786 let segment = &path.segments[0];
787 let ident = segment.ident.clone();
788 if let PathArguments::AngleBracketed(generic) = &segment.arguments {
789 if ident == "Result" && generic.args.len() == 1 {
790 if let GenericArgument::Type(arg) = &generic.args[0] {
791 ret = arg;
David Tolnaye3a48152020-04-08 19:38:05 -0700792 *throws_tokens =
793 Some((kw::Result(ident.span()), generic.lt_token, generic.gt_token));
David Tolnayc071b892020-03-18 16:59:53 -0700794 }
795 }
796 }
797 }
798 }
David Tolnayd7a3a182020-11-01 20:45:14 -0800799 match parse_type(ret, namespace)? {
David Tolnayc071b892020-03-18 16:59:53 -0700800 Type::Void(_) => Ok(None),
801 ty => Ok(Some(ty)),
802 }
803}