| David Tolnay | 17e137f | 2020-05-08 15:55:28 -0700 | [diff] [blame] | 1 | use crate::syntax::discriminant::DiscriminantSet; |
| David Tolnay | c598a27 | 2020-08-29 15:10:18 -0700 | [diff] [blame] | 2 | use crate::syntax::file::{Item, ItemForeignMod}; |
| David Tolnay | 5275978 | 2020-05-03 23:59:40 -0700 | [diff] [blame] | 3 | use crate::syntax::report::Errors; |
| David Tolnay | eebe9b7 | 2020-04-14 16:32:18 -0700 | [diff] [blame] | 4 | use crate::syntax::Atom::*; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 5 | use crate::syntax::{ |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 6 | attrs, error, Api, Array, Derive, Doc, Enum, ExternFn, ExternType, Impl, Include, IncludeKind, |
| David Tolnay | 75ea17c | 2020-12-06 21:08:34 -0800 | [diff] [blame] | 7 | Lang, Namespace, Pair, Receiver, Ref, RustName, Signature, SliceRef, Struct, Ty1, Type, |
| David Tolnay | e0dca7b | 2020-11-25 17:18:57 -0800 | [diff] [blame] | 8 | TypeAlias, Var, Variant, |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 9 | }; |
| David Tolnay | 9a69d8a | 2020-11-16 09:16:41 -0800 | [diff] [blame] | 10 | use proc_macro2::{Delimiter, Group, Span, TokenStream, TokenTree}; |
| David Tolnay | e2e303f | 2020-05-10 20:52:00 -0700 | [diff] [blame] | 11 | use quote::{format_ident, quote, quote_spanned}; |
| David Tolnay | e2f9ec4 | 2020-05-07 16:19:33 -0700 | [diff] [blame] | 12 | use syn::parse::{ParseStream, Parser}; |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 13 | use syn::punctuated::Punctuated; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 14 | use syn::{ |
| Xiangpeng Hao | 7876235 | 2020-11-12 10:24:18 +0800 | [diff] [blame] | 15 | Abi, Attribute, Error, Expr, Fields, FnArg, ForeignItem, ForeignItemFn, ForeignItemType, |
| 16 | GenericArgument, GenericParam, Generics, Ident, ItemEnum, ItemImpl, ItemStruct, Lit, LitStr, |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 17 | Pat, PathArguments, Result, ReturnType, Token, TraitBound, TraitBoundModifier, |
| 18 | Type as RustType, TypeArray, TypeBareFn, TypeParamBound, TypePath, TypeReference, |
| David Tolnay | 2680923 | 2020-12-21 16:06:47 -0800 | [diff] [blame] | 19 | Variant as RustVariant, |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 20 | }; |
| 21 | |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 22 | pub mod kw { |
| David Tolnay | b27f787 | 2020-11-15 15:07:04 -0800 | [diff] [blame] | 23 | syn::custom_keyword!(Pin); |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 24 | syn::custom_keyword!(Result); |
| 25 | } |
| 26 | |
| David Tolnay | d7a3a18 | 2020-11-01 20:45:14 -0800 | [diff] [blame] | 27 | pub fn parse_items( |
| 28 | cx: &mut Errors, |
| 29 | items: Vec<Item>, |
| 30 | trusted: bool, |
| 31 | namespace: &Namespace, |
| 32 | ) -> Vec<Api> { |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 33 | let mut apis = Vec::new(); |
| 34 | for item in items { |
| 35 | match item { |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 36 | Item::Struct(item) => match parse_struct(cx, item, namespace) { |
| David Tolnay | 5275978 | 2020-05-03 23:59:40 -0700 | [diff] [blame] | 37 | Ok(strct) => apis.push(strct), |
| 38 | Err(err) => cx.push(err), |
| 39 | }, |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 40 | Item::Enum(item) => match parse_enum(cx, item, namespace) { |
| David Tolnay | 5275978 | 2020-05-03 23:59:40 -0700 | [diff] [blame] | 41 | Ok(enm) => apis.push(enm), |
| 42 | Err(err) => cx.push(err), |
| 43 | }, |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 44 | Item::ForeignMod(foreign_mod) => { |
| David Tolnay | d7a3a18 | 2020-11-01 20:45:14 -0800 | [diff] [blame] | 45 | parse_foreign_mod(cx, foreign_mod, &mut apis, trusted, namespace) |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 46 | } |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 47 | Item::Impl(item) => match parse_impl(item) { |
| David Tolnay | 7e69f89 | 2020-10-03 22:20:22 -0700 | [diff] [blame] | 48 | Ok(imp) => apis.push(imp), |
| 49 | Err(err) => cx.push(err), |
| 50 | }, |
| David Tolnay | 5275978 | 2020-05-03 23:59:40 -0700 | [diff] [blame] | 51 | Item::Use(item) => cx.error(item, error::USE_NOT_ALLOWED), |
| David Tolnay | 00a8385 | 2020-08-29 15:06:16 -0700 | [diff] [blame] | 52 | Item::Other(item) => cx.error(item, "unsupported item"), |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 53 | } |
| 54 | } |
| David Tolnay | 5275978 | 2020-05-03 23:59:40 -0700 | [diff] [blame] | 55 | apis |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 56 | } |
| 57 | |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 58 | fn parse_struct(cx: &mut Errors, item: ItemStruct, namespace: &Namespace) -> Result<Api> { |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 59 | let mut doc = Doc::new(); |
| 60 | let mut derives = Vec::new(); |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 61 | let mut namespace = namespace.clone(); |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 62 | let mut cxx_name = None; |
| 63 | let mut rust_name = None; |
| David Tolnay | b129ea7 | 2020-05-10 14:29:30 -0700 | [diff] [blame] | 64 | attrs::parse( |
| David Tolnay | 3e62888 | 2020-05-10 15:30:14 -0700 | [diff] [blame] | 65 | cx, |
| David Tolnay | b129ea7 | 2020-05-10 14:29:30 -0700 | [diff] [blame] | 66 | &item.attrs, |
| 67 | attrs::Parser { |
| 68 | doc: Some(&mut doc), |
| 69 | derives: Some(&mut derives), |
| David Tolnay | d7a3a18 | 2020-11-01 20:45:14 -0800 | [diff] [blame] | 70 | namespace: Some(&mut namespace), |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 71 | cxx_name: Some(&mut cxx_name), |
| 72 | rust_name: Some(&mut rust_name), |
| David Tolnay | ddf69e2 | 2020-05-10 22:08:20 -0700 | [diff] [blame] | 73 | ..Default::default() |
| David Tolnay | b129ea7 | 2020-05-10 14:29:30 -0700 | [diff] [blame] | 74 | }, |
| David Tolnay | 3e62888 | 2020-05-10 15:30:14 -0700 | [diff] [blame] | 75 | ); |
| David Tolnay | 09462ac | 2020-03-20 14:58:41 -0700 | [diff] [blame] | 76 | |
| David Tolnay | bc954d9 | 2020-12-21 15:17:10 -0800 | [diff] [blame] | 77 | let generics = &item.generics; |
| 78 | if !generics.params.is_empty() || generics.where_clause.is_some() { |
| 79 | let struct_token = item.struct_token; |
| 80 | let ident = &item.ident; |
| 81 | let where_clause = &generics.where_clause; |
| 82 | let span = quote!(#struct_token #ident #generics #where_clause); |
| 83 | return Err(Error::new_spanned( |
| 84 | span, |
| 85 | "struct with generic parameters is not supported yet", |
| 86 | )); |
| 87 | } |
| 88 | |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 89 | let named_fields = match item.fields { |
| David Tolnay | 09462ac | 2020-03-20 14:58:41 -0700 | [diff] [blame] | 90 | Fields::Named(fields) => fields, |
| 91 | Fields::Unit => return Err(Error::new_spanned(item, "unit structs are not supported")), |
| 92 | Fields::Unnamed(_) => { |
| David Tolnay | 2c4b35f | 2020-10-04 21:22:30 -0700 | [diff] [blame] | 93 | return Err(Error::new_spanned(item, "tuple structs are not supported")); |
| David Tolnay | 09462ac | 2020-03-20 14:58:41 -0700 | [diff] [blame] | 94 | } |
| 95 | }; |
| 96 | |
| David Tolnay | 38c1121 | 2020-11-25 15:50:19 -0800 | [diff] [blame] | 97 | let mut fields = Vec::new(); |
| 98 | for field in named_fields.named { |
| 99 | let ident = field.ident.unwrap(); |
| 100 | let ty = match parse_type(&field.ty) { |
| 101 | Ok(ty) => ty, |
| 102 | Err(err) => { |
| 103 | cx.push(err); |
| 104 | continue; |
| 105 | } |
| 106 | }; |
| 107 | fields.push(Var { ident, ty }); |
| 108 | } |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 109 | |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 110 | let struct_token = item.struct_token; |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 111 | let name = pair(namespace, &item.ident, cxx_name, rust_name); |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 112 | let brace_token = named_fields.brace_token; |
| 113 | |
| David Tolnay | 09462ac | 2020-03-20 14:58:41 -0700 | [diff] [blame] | 114 | Ok(Api::Struct(Struct { |
| 115 | doc, |
| 116 | derives, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 117 | struct_token, |
| 118 | name, |
| 119 | brace_token, |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 120 | fields, |
| David Tolnay | 09462ac | 2020-03-20 14:58:41 -0700 | [diff] [blame] | 121 | })) |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 122 | } |
| 123 | |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 124 | fn parse_enum(cx: &mut Errors, item: ItemEnum, namespace: &Namespace) -> Result<Api> { |
| David Tolnay | ddf69e2 | 2020-05-10 22:08:20 -0700 | [diff] [blame] | 125 | let mut doc = Doc::new(); |
| David Tolnay | fbc4669 | 2020-11-27 11:33:29 -0800 | [diff] [blame] | 126 | let mut derives = Vec::new(); |
| David Tolnay | ddf69e2 | 2020-05-10 22:08:20 -0700 | [diff] [blame] | 127 | let mut repr = None; |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 128 | let mut namespace = namespace.clone(); |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 129 | let mut cxx_name = None; |
| 130 | let mut rust_name = None; |
| David Tolnay | ddf69e2 | 2020-05-10 22:08:20 -0700 | [diff] [blame] | 131 | attrs::parse( |
| 132 | cx, |
| 133 | &item.attrs, |
| 134 | attrs::Parser { |
| 135 | doc: Some(&mut doc), |
| David Tolnay | fbc4669 | 2020-11-27 11:33:29 -0800 | [diff] [blame] | 136 | derives: Some(&mut derives), |
| David Tolnay | ddf69e2 | 2020-05-10 22:08:20 -0700 | [diff] [blame] | 137 | repr: Some(&mut repr), |
| David Tolnay | d7a3a18 | 2020-11-01 20:45:14 -0800 | [diff] [blame] | 138 | namespace: Some(&mut namespace), |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 139 | cxx_name: Some(&mut cxx_name), |
| 140 | rust_name: Some(&mut rust_name), |
| David Tolnay | ddf69e2 | 2020-05-10 22:08:20 -0700 | [diff] [blame] | 141 | ..Default::default() |
| 142 | }, |
| 143 | ); |
| Joel Galenson | c03402a | 2020-04-23 17:31:09 -0700 | [diff] [blame] | 144 | |
| David Tolnay | bc954d9 | 2020-12-21 15:17:10 -0800 | [diff] [blame] | 145 | let generics = &item.generics; |
| 146 | if !generics.params.is_empty() || generics.where_clause.is_some() { |
| 147 | let enum_token = item.enum_token; |
| 148 | let ident = &item.ident; |
| 149 | let where_clause = &generics.where_clause; |
| 150 | let span = quote!(#enum_token #ident #generics #where_clause); |
| 151 | return Err(Error::new_spanned( |
| 152 | span, |
| 153 | "enums with generic parameters are not allowed", |
| 154 | )); |
| 155 | } |
| 156 | |
| Joel Galenson | 8854773 | 2020-05-05 08:23:42 -0700 | [diff] [blame] | 157 | let mut variants = Vec::new(); |
| David Tolnay | ddf69e2 | 2020-05-10 22:08:20 -0700 | [diff] [blame] | 158 | let mut discriminants = DiscriminantSet::new(repr); |
| Joel Galenson | 8854773 | 2020-05-05 08:23:42 -0700 | [diff] [blame] | 159 | for variant in item.variants { |
| David Tolnay | 2680923 | 2020-12-21 16:06:47 -0800 | [diff] [blame] | 160 | match parse_variant(variant, &mut discriminants) { |
| 161 | Ok(variant) => variants.push(variant), |
| 162 | Err(err) => cx.push(err), |
| Joel Galenson | c03402a | 2020-04-23 17:31:09 -0700 | [diff] [blame] | 163 | } |
| 164 | } |
| 165 | |
| David Tolnay | e2e303f | 2020-05-10 20:52:00 -0700 | [diff] [blame] | 166 | let enum_token = item.enum_token; |
| 167 | let brace_token = item.brace_token; |
| 168 | |
| David Tolnay | 58eee39 | 2020-11-20 20:37:58 -0800 | [diff] [blame] | 169 | let explicit_repr = repr.is_some(); |
| David Tolnay | 0435a81 | 2020-05-10 21:58:15 -0700 | [diff] [blame] | 170 | let mut repr = U8; |
| 171 | match discriminants.inferred_repr() { |
| 172 | Ok(inferred) => repr = inferred, |
| David Tolnay | e2e303f | 2020-05-10 20:52:00 -0700 | [diff] [blame] | 173 | Err(err) => { |
| 174 | let span = quote_spanned!(brace_token.span=> #enum_token {}); |
| David Tolnay | 0435a81 | 2020-05-10 21:58:15 -0700 | [diff] [blame] | 175 | cx.error(span, err); |
| 176 | variants.clear(); |
| David Tolnay | e2e303f | 2020-05-10 20:52:00 -0700 | [diff] [blame] | 177 | } |
| David Tolnay | 0435a81 | 2020-05-10 21:58:15 -0700 | [diff] [blame] | 178 | } |
| David Tolnay | e2e303f | 2020-05-10 20:52:00 -0700 | [diff] [blame] | 179 | |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 180 | let name = pair(namespace, &item.ident, cxx_name, rust_name); |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 181 | let repr_ident = Ident::new(repr.as_ref(), Span::call_site()); |
| 182 | let repr_type = Type::Ident(RustName::new(repr_ident)); |
| David Tolnay | 9a69d8a | 2020-11-16 09:16:41 -0800 | [diff] [blame] | 183 | |
| Joel Galenson | c03402a | 2020-04-23 17:31:09 -0700 | [diff] [blame] | 184 | Ok(Api::Enum(Enum { |
| 185 | doc, |
| David Tolnay | fbc4669 | 2020-11-27 11:33:29 -0800 | [diff] [blame] | 186 | derives, |
| David Tolnay | e2e303f | 2020-05-10 20:52:00 -0700 | [diff] [blame] | 187 | enum_token, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 188 | name, |
| David Tolnay | e2e303f | 2020-05-10 20:52:00 -0700 | [diff] [blame] | 189 | brace_token, |
| David Tolnay | 7ae018f | 2020-05-05 10:11:30 -0700 | [diff] [blame] | 190 | variants, |
| David Tolnay | e2e303f | 2020-05-10 20:52:00 -0700 | [diff] [blame] | 191 | repr, |
| David Tolnay | 9a69d8a | 2020-11-16 09:16:41 -0800 | [diff] [blame] | 192 | repr_type, |
| David Tolnay | 58eee39 | 2020-11-20 20:37:58 -0800 | [diff] [blame] | 193 | explicit_repr, |
| Joel Galenson | c03402a | 2020-04-23 17:31:09 -0700 | [diff] [blame] | 194 | })) |
| 195 | } |
| 196 | |
| David Tolnay | 2680923 | 2020-12-21 16:06:47 -0800 | [diff] [blame] | 197 | fn parse_variant(variant: RustVariant, discriminants: &mut DiscriminantSet) -> Result<Variant> { |
| 198 | match variant.fields { |
| 199 | Fields::Unit => {} |
| 200 | _ => { |
| 201 | let msg = "enums with data are not supported yet"; |
| 202 | return Err(Error::new_spanned(variant, msg)); |
| 203 | } |
| 204 | } |
| 205 | |
| 206 | let expr = variant.discriminant.as_ref().map(|(_, expr)| expr); |
| 207 | let try_discriminant = match &expr { |
| 208 | Some(lit) => discriminants.insert(lit), |
| 209 | None => discriminants.insert_next(), |
| 210 | }; |
| 211 | let discriminant = match try_discriminant { |
| 212 | Ok(discriminant) => discriminant, |
| 213 | Err(err) => return Err(Error::new_spanned(variant, err)), |
| 214 | }; |
| 215 | let expr = variant.discriminant.map(|(_, expr)| expr); |
| 216 | |
| 217 | Ok(Variant { |
| 218 | ident: variant.ident, |
| 219 | discriminant, |
| 220 | expr, |
| 221 | }) |
| 222 | } |
| 223 | |
| David Tolnay | 805dca3 | 2020-08-29 19:09:55 -0700 | [diff] [blame] | 224 | fn parse_foreign_mod( |
| 225 | cx: &mut Errors, |
| 226 | foreign_mod: ItemForeignMod, |
| 227 | out: &mut Vec<Api>, |
| 228 | trusted: bool, |
| David Tolnay | d7a3a18 | 2020-11-01 20:45:14 -0800 | [diff] [blame] | 229 | namespace: &Namespace, |
| David Tolnay | 805dca3 | 2020-08-29 19:09:55 -0700 | [diff] [blame] | 230 | ) { |
| David Tolnay | 3a45173 | 2020-08-29 18:54:55 -0700 | [diff] [blame] | 231 | let lang = match parse_lang(&foreign_mod.abi) { |
| David Tolnay | 5275978 | 2020-05-03 23:59:40 -0700 | [diff] [blame] | 232 | Ok(lang) => lang, |
| 233 | Err(err) => return cx.push(err), |
| 234 | }; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 235 | |
| David Tolnay | 3a45173 | 2020-08-29 18:54:55 -0700 | [diff] [blame] | 236 | match lang { |
| 237 | Lang::Rust => { |
| David Tolnay | 805dca3 | 2020-08-29 19:09:55 -0700 | [diff] [blame] | 238 | if foreign_mod.unsafety.is_some() { |
| David Tolnay | 3a45173 | 2020-08-29 18:54:55 -0700 | [diff] [blame] | 239 | let unsafety = foreign_mod.unsafety; |
| David Tolnay | 7be5b1f | 2020-11-11 10:48:32 -0800 | [diff] [blame] | 240 | let abi = &foreign_mod.abi; |
| David Tolnay | 3a45173 | 2020-08-29 18:54:55 -0700 | [diff] [blame] | 241 | let span = quote!(#unsafety #abi); |
| 242 | cx.error(span, "extern \"Rust\" block does not need to be unsafe"); |
| 243 | } |
| 244 | } |
| 245 | Lang::Cxx => {} |
| 246 | } |
| 247 | |
| David Tolnay | 805dca3 | 2020-08-29 19:09:55 -0700 | [diff] [blame] | 248 | let trusted = trusted || foreign_mod.unsafety.is_some(); |
| 249 | |
| David Tolnay | 7feb264 | 2020-11-10 18:49:04 -0800 | [diff] [blame] | 250 | let mut namespace = namespace.clone(); |
| 251 | attrs::parse( |
| 252 | cx, |
| 253 | &foreign_mod.attrs, |
| 254 | attrs::Parser { |
| 255 | namespace: Some(&mut namespace), |
| 256 | ..Default::default() |
| 257 | }, |
| 258 | ); |
| 259 | |
| Joel Galenson | e1e969d | 2020-04-21 12:50:20 -0700 | [diff] [blame] | 260 | let mut items = Vec::new(); |
| 261 | for foreign in &foreign_mod.items { |
| 262 | match foreign { |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 263 | ForeignItem::Type(foreign) => { |
| David Tolnay | f5ead10 | 2020-11-24 17:08:05 -0800 | [diff] [blame] | 264 | let ety = parse_extern_type(cx, foreign, lang, trusted, &namespace); |
| 265 | items.push(ety); |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 266 | } |
| David Tolnay | 556738d | 2020-11-15 13:58:44 -0800 | [diff] [blame] | 267 | ForeignItem::Fn(foreign) => { |
| 268 | match parse_extern_fn(cx, foreign, lang, trusted, &namespace) { |
| 269 | Ok(efn) => items.push(efn), |
| 270 | Err(err) => cx.push(err), |
| 271 | } |
| 272 | } |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 273 | ForeignItem::Macro(foreign) if foreign.mac.path.is_ident("include") => { |
| David Tolnay | 91e87fa | 2020-05-11 19:10:23 -0700 | [diff] [blame] | 274 | match foreign.mac.parse_body_with(parse_include) { |
| David Tolnay | 5275978 | 2020-05-03 23:59:40 -0700 | [diff] [blame] | 275 | Ok(include) => items.push(Api::Include(include)), |
| 276 | Err(err) => cx.push(err), |
| 277 | } |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 278 | } |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 279 | ForeignItem::Verbatim(tokens) => { |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 280 | match parse_extern_verbatim(cx, tokens, lang, trusted, &namespace) { |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 281 | Ok(api) => items.push(api), |
| 282 | Err(err) => cx.push(err), |
| 283 | } |
| 284 | } |
| David Tolnay | 5275978 | 2020-05-03 23:59:40 -0700 | [diff] [blame] | 285 | _ => cx.error(foreign, "unsupported foreign item"), |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 286 | } |
| 287 | } |
| David Tolnay | a1f29c4 | 2020-04-22 18:01:38 -0700 | [diff] [blame] | 288 | |
| David Tolnay | 7be5b1f | 2020-11-11 10:48:32 -0800 | [diff] [blame] | 289 | if !trusted |
| 290 | && items.iter().any(|api| match api { |
| 291 | Api::CxxFunction(efn) => efn.unsafety.is_none(), |
| 292 | _ => false, |
| 293 | }) |
| 294 | { |
| 295 | cx.error( |
| 296 | foreign_mod.abi, |
| 297 | "block must be declared `unsafe extern \"C++\"` if it contains any safe-to-call C++ functions", |
| 298 | ); |
| 299 | } |
| 300 | |
| David Tolnay | a1f29c4 | 2020-04-22 18:01:38 -0700 | [diff] [blame] | 301 | let mut types = items.iter().filter_map(|item| match item { |
| David Tolnay | 17a934c | 2020-11-02 00:40:04 -0800 | [diff] [blame] | 302 | Api::CxxType(ety) | Api::RustType(ety) => Some(&ety.name), |
| 303 | Api::TypeAlias(alias) => Some(&alias.name), |
| David Tolnay | a1f29c4 | 2020-04-22 18:01:38 -0700 | [diff] [blame] | 304 | _ => None, |
| 305 | }); |
| 306 | if let (Some(single_type), None) = (types.next(), types.next()) { |
| David Tolnay | e5a015a | 2020-05-07 21:54:26 -0700 | [diff] [blame] | 307 | let single_type = single_type.clone(); |
| David Tolnay | a1f29c4 | 2020-04-22 18:01:38 -0700 | [diff] [blame] | 308 | for item in &mut items { |
| 309 | if let Api::CxxFunction(efn) | Api::RustFunction(efn) = item { |
| 310 | if let Some(receiver) = &mut efn.receiver { |
| David Tolnay | 2d99b00 | 2020-12-21 14:34:35 -0800 | [diff] [blame] | 311 | if receiver.ty.rust == "Self" { |
| 312 | receiver.ty.rust = single_type.rust.clone(); |
| David Tolnay | a1f29c4 | 2020-04-22 18:01:38 -0700 | [diff] [blame] | 313 | } |
| 314 | } |
| 315 | } |
| 316 | } |
| 317 | } |
| 318 | |
| David Tolnay | 5275978 | 2020-05-03 23:59:40 -0700 | [diff] [blame] | 319 | out.extend(items); |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 320 | } |
| 321 | |
| David Tolnay | 3a45173 | 2020-08-29 18:54:55 -0700 | [diff] [blame] | 322 | fn parse_lang(abi: &Abi) -> Result<Lang> { |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 323 | let name = match &abi.name { |
| 324 | Some(name) => name, |
| 325 | None => { |
| 326 | return Err(Error::new_spanned( |
| 327 | abi, |
| David Tolnay | c72a9f6 | 2020-11-11 10:56:26 -0800 | [diff] [blame] | 328 | "ABI name is required, extern \"C++\" or extern \"Rust\"", |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 329 | )); |
| 330 | } |
| 331 | }; |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 332 | |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 333 | match name.value().as_str() { |
| David Tolnay | ea25ac8 | 2020-11-11 10:35:23 -0800 | [diff] [blame] | 334 | "C++" => Ok(Lang::Cxx), |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 335 | "Rust" => Ok(Lang::Rust), |
| David Tolnay | 2757ca8 | 2020-11-14 15:12:06 -0800 | [diff] [blame] | 336 | _ => Err(Error::new_spanned( |
| 337 | abi, |
| 338 | "unrecognized ABI, requires either \"C++\" or \"Rust\"", |
| 339 | )), |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 340 | } |
| 341 | } |
| 342 | |
| David Tolnay | 00f236a | 2020-08-29 19:07:18 -0700 | [diff] [blame] | 343 | fn parse_extern_type( |
| 344 | cx: &mut Errors, |
| 345 | foreign_type: &ForeignItemType, |
| 346 | lang: Lang, |
| 347 | trusted: bool, |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 348 | namespace: &Namespace, |
| David Tolnay | f5ead10 | 2020-11-24 17:08:05 -0800 | [diff] [blame] | 349 | ) -> Api { |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 350 | let mut doc = Doc::new(); |
| David Tolnay | 16e2620 | 2020-11-27 19:28:37 -0800 | [diff] [blame] | 351 | let mut derives = Vec::new(); |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 352 | let mut namespace = namespace.clone(); |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 353 | let mut cxx_name = None; |
| 354 | let mut rust_name = None; |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 355 | attrs::parse( |
| 356 | cx, |
| 357 | &foreign_type.attrs, |
| 358 | attrs::Parser { |
| 359 | doc: Some(&mut doc), |
| David Tolnay | 16e2620 | 2020-11-27 19:28:37 -0800 | [diff] [blame] | 360 | derives: Some(&mut derives), |
| David Tolnay | d7a3a18 | 2020-11-01 20:45:14 -0800 | [diff] [blame] | 361 | namespace: Some(&mut namespace), |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 362 | cxx_name: Some(&mut cxx_name), |
| 363 | rust_name: Some(&mut rust_name), |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 364 | ..Default::default() |
| 365 | }, |
| 366 | ); |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 367 | |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 368 | let type_token = foreign_type.type_token; |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 369 | let name = pair(namespace, &foreign_type.ident, cxx_name, rust_name); |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 370 | let colon_token = None; |
| 371 | let bounds = Vec::new(); |
| David Tolnay | c836102 | 2020-08-25 21:57:53 -0700 | [diff] [blame] | 372 | let semi_token = foreign_type.semi_token; |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 373 | |
| 374 | (match lang { |
| David Tolnay | 295ef6b | 2020-05-07 16:10:22 -0700 | [diff] [blame] | 375 | Lang::Cxx => Api::CxxType, |
| 376 | Lang::Rust => Api::RustType, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 377 | })(ExternType { |
| David Tolnay | 16e2620 | 2020-11-27 19:28:37 -0800 | [diff] [blame] | 378 | lang, |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 379 | doc, |
| David Tolnay | 16e2620 | 2020-11-27 19:28:37 -0800 | [diff] [blame] | 380 | derives, |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 381 | type_token, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 382 | name, |
| 383 | colon_token, |
| 384 | bounds, |
| David Tolnay | c836102 | 2020-08-25 21:57:53 -0700 | [diff] [blame] | 385 | semi_token, |
| David Tolnay | 00f236a | 2020-08-29 19:07:18 -0700 | [diff] [blame] | 386 | trusted, |
| David Tolnay | f5ead10 | 2020-11-24 17:08:05 -0800 | [diff] [blame] | 387 | }) |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 388 | } |
| 389 | |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 390 | fn parse_extern_fn( |
| 391 | cx: &mut Errors, |
| 392 | foreign_fn: &ForeignItemFn, |
| 393 | lang: Lang, |
| David Tolnay | 556738d | 2020-11-15 13:58:44 -0800 | [diff] [blame] | 394 | trusted: bool, |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 395 | namespace: &Namespace, |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 396 | ) -> Result<Api> { |
| David Tolnay | bc954d9 | 2020-12-21 15:17:10 -0800 | [diff] [blame] | 397 | let mut doc = Doc::new(); |
| David Tolnay | a2dfcbf | 2020-12-21 15:46:24 -0800 | [diff] [blame] | 398 | let mut namespace = namespace.clone(); |
| David Tolnay | bc954d9 | 2020-12-21 15:17:10 -0800 | [diff] [blame] | 399 | let mut cxx_name = None; |
| 400 | let mut rust_name = None; |
| David Tolnay | bc954d9 | 2020-12-21 15:17:10 -0800 | [diff] [blame] | 401 | attrs::parse( |
| 402 | cx, |
| 403 | &foreign_fn.attrs, |
| 404 | attrs::Parser { |
| 405 | doc: Some(&mut doc), |
| David Tolnay | a2dfcbf | 2020-12-21 15:46:24 -0800 | [diff] [blame] | 406 | namespace: Some(&mut namespace), |
| David Tolnay | bc954d9 | 2020-12-21 15:17:10 -0800 | [diff] [blame] | 407 | cxx_name: Some(&mut cxx_name), |
| 408 | rust_name: Some(&mut rust_name), |
| David Tolnay | bc954d9 | 2020-12-21 15:17:10 -0800 | [diff] [blame] | 409 | ..Default::default() |
| 410 | }, |
| 411 | ); |
| 412 | |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 413 | let generics = &foreign_fn.sig.generics; |
| David Tolnay | 9938b64 | 2020-11-15 18:11:40 -0800 | [diff] [blame] | 414 | if generics.where_clause.is_some() |
| 415 | || generics.params.iter().any(|param| match param { |
| 416 | GenericParam::Lifetime(lifetime) => !lifetime.bounds.is_empty(), |
| 417 | GenericParam::Type(_) | GenericParam::Const(_) => true, |
| 418 | }) |
| 419 | { |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 420 | return Err(Error::new_spanned( |
| 421 | foreign_fn, |
| 422 | "extern function with generic parameters is not supported yet", |
| 423 | )); |
| 424 | } |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 425 | |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 426 | if let Some(variadic) = &foreign_fn.sig.variadic { |
| 427 | return Err(Error::new_spanned( |
| 428 | variadic, |
| 429 | "variadic function is not supported yet", |
| 430 | )); |
| 431 | } |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 432 | |
| David Tolnay | 8de461f | 2020-11-19 13:46:14 -0800 | [diff] [blame] | 433 | if foreign_fn.sig.asyncness.is_some() { |
| 434 | return Err(Error::new_spanned( |
| 435 | foreign_fn, |
| 436 | "async function is not directly supported yet, but see https://cxx.rs/async.html for a working approach", |
| 437 | )); |
| 438 | } |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 439 | |
| David Tolnay | 80ef068 | 2020-11-19 13:56:07 -0800 | [diff] [blame] | 440 | if foreign_fn.sig.constness.is_some() { |
| 441 | return Err(Error::new_spanned( |
| 442 | foreign_fn, |
| 443 | "const extern function is not supported", |
| 444 | )); |
| 445 | } |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 446 | |
| David Tolnay | 80ef068 | 2020-11-19 13:56:07 -0800 | [diff] [blame] | 447 | if let Some(abi) = &foreign_fn.sig.abi { |
| 448 | return Err(Error::new_spanned( |
| 449 | abi, |
| 450 | "explicit ABI on extern function is not supported", |
| 451 | )); |
| 452 | } |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 453 | |
| 454 | let mut receiver = None; |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 455 | let mut args = Punctuated::new(); |
| 456 | for arg in foreign_fn.sig.inputs.pairs() { |
| 457 | let (arg, comma) = arg.into_tuple(); |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 458 | match arg { |
| David Tolnay | 1dd11a1 | 2020-04-22 12:31:48 -0700 | [diff] [blame] | 459 | FnArg::Receiver(arg) => { |
| David Tolnay | a1f29c4 | 2020-04-22 18:01:38 -0700 | [diff] [blame] | 460 | if let Some((ampersand, lifetime)) = &arg.reference { |
| 461 | receiver = Some(Receiver { |
| David Tolnay | 9c76c39 | 2020-11-15 16:07:08 -0800 | [diff] [blame] | 462 | pinned: false, |
| David Tolnay | a1f29c4 | 2020-04-22 18:01:38 -0700 | [diff] [blame] | 463 | ampersand: *ampersand, |
| 464 | lifetime: lifetime.clone(), |
| David Tolnay | 9c4ac2e | 2020-11-15 21:14:03 -0800 | [diff] [blame] | 465 | mutable: arg.mutability.is_some(), |
| David Tolnay | a1f29c4 | 2020-04-22 18:01:38 -0700 | [diff] [blame] | 466 | var: arg.self_token, |
| David Tolnay | 2d99b00 | 2020-12-21 14:34:35 -0800 | [diff] [blame] | 467 | ty: RustName::new(Ident::new("Self", arg.self_token.span)), |
| David Tolnay | a1f29c4 | 2020-04-22 18:01:38 -0700 | [diff] [blame] | 468 | shorthand: true, |
| David Tolnay | 9c76c39 | 2020-11-15 16:07:08 -0800 | [diff] [blame] | 469 | pin_tokens: None, |
| David Tolnay | 9c4ac2e | 2020-11-15 21:14:03 -0800 | [diff] [blame] | 470 | mutability: arg.mutability, |
| David Tolnay | a1f29c4 | 2020-04-22 18:01:38 -0700 | [diff] [blame] | 471 | }); |
| 472 | continue; |
| Joel Galenson | e1e969d | 2020-04-21 12:50:20 -0700 | [diff] [blame] | 473 | } |
| David Tolnay | 1dd11a1 | 2020-04-22 12:31:48 -0700 | [diff] [blame] | 474 | return Err(Error::new_spanned(arg, "unsupported signature")); |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 475 | } |
| 476 | FnArg::Typed(arg) => { |
| 477 | let ident = match arg.pat.as_ref() { |
| 478 | Pat::Ident(pat) => pat.ident.clone(), |
| Joel Galenson | ba67607 | 2020-04-27 15:55:45 -0700 | [diff] [blame] | 479 | Pat::Wild(pat) => { |
| 480 | Ident::new(&format!("_{}", args.len()), pat.underscore_token.span) |
| 481 | } |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 482 | _ => return Err(Error::new_spanned(arg, "unsupported signature")), |
| 483 | }; |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 484 | let ty = parse_type(&arg.ty)?; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 485 | if ident != "self" { |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 486 | args.push_value(Var { ident, ty }); |
| 487 | if let Some(comma) = comma { |
| 488 | args.push_punct(*comma); |
| 489 | } |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 490 | continue; |
| 491 | } |
| 492 | if let Type::Ref(reference) = ty { |
| 493 | if let Type::Ident(ident) = reference.inner { |
| 494 | receiver = Some(Receiver { |
| David Tolnay | 9c76c39 | 2020-11-15 16:07:08 -0800 | [diff] [blame] | 495 | pinned: reference.pinned, |
| David Tolnay | fb6e386 | 2020-04-20 01:33:23 -0700 | [diff] [blame] | 496 | ampersand: reference.ampersand, |
| David Tolnay | 0bd50fa | 2020-04-22 15:31:33 -0700 | [diff] [blame] | 497 | lifetime: reference.lifetime, |
| David Tolnay | 9c4ac2e | 2020-11-15 21:14:03 -0800 | [diff] [blame] | 498 | mutable: reference.mutable, |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 499 | var: Token), |
| David Tolnay | 05e11cc | 2020-04-20 02:13:56 -0700 | [diff] [blame] | 500 | ty: ident, |
| David Tolnay | 62d360c | 2020-04-22 16:26:21 -0700 | [diff] [blame] | 501 | shorthand: false, |
| David Tolnay | 9c76c39 | 2020-11-15 16:07:08 -0800 | [diff] [blame] | 502 | pin_tokens: reference.pin_tokens, |
| David Tolnay | 9c4ac2e | 2020-11-15 21:14:03 -0800 | [diff] [blame] | 503 | mutability: reference.mutability, |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 504 | }); |
| 505 | continue; |
| 506 | } |
| 507 | } |
| 508 | return Err(Error::new_spanned(arg, "unsupported method receiver")); |
| 509 | } |
| 510 | } |
| 511 | } |
| David Tolnay | 59b7ede | 2020-03-16 00:30:23 -0700 | [diff] [blame] | 512 | |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 513 | let mut throws_tokens = None; |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 514 | let ret = parse_return_type(&foreign_fn.sig.output, &mut throws_tokens)?; |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 515 | let throws = throws_tokens.is_some(); |
| David Tolnay | bdb576c | 2020-09-06 23:45:55 -0700 | [diff] [blame] | 516 | let unsafety = foreign_fn.sig.unsafety; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 517 | let fn_token = foreign_fn.sig.fn_token; |
| David Tolnay | 6a990f8 | 2020-12-21 15:42:31 -0800 | [diff] [blame] | 518 | let name = pair(namespace, &foreign_fn.sig.ident, cxx_name, rust_name); |
| David Tolnay | 9938b64 | 2020-11-15 18:11:40 -0800 | [diff] [blame] | 519 | let generics = generics.clone(); |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 520 | let paren_token = foreign_fn.sig.paren_token; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 521 | let semi_token = foreign_fn.semi_token; |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 522 | |
| 523 | Ok(match lang { |
| David Tolnay | 295ef6b | 2020-05-07 16:10:22 -0700 | [diff] [blame] | 524 | Lang::Cxx => Api::CxxFunction, |
| 525 | Lang::Rust => Api::RustFunction, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 526 | }(ExternFn { |
| David Tolnay | 6cde49f | 2020-03-16 12:25:45 -0700 | [diff] [blame] | 527 | lang, |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 528 | doc, |
| David Tolnay | 17a934c | 2020-11-02 00:40:04 -0800 | [diff] [blame] | 529 | name, |
| David Tolnay | 1644873 | 2020-03-18 12:39:36 -0700 | [diff] [blame] | 530 | sig: Signature { |
| David Tolnay | bdb576c | 2020-09-06 23:45:55 -0700 | [diff] [blame] | 531 | unsafety, |
| David Tolnay | 1644873 | 2020-03-18 12:39:36 -0700 | [diff] [blame] | 532 | fn_token, |
| David Tolnay | 9938b64 | 2020-11-15 18:11:40 -0800 | [diff] [blame] | 533 | generics, |
| David Tolnay | 1644873 | 2020-03-18 12:39:36 -0700 | [diff] [blame] | 534 | receiver, |
| 535 | args, |
| 536 | ret, |
| 537 | throws, |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 538 | paren_token, |
| 539 | throws_tokens, |
| David Tolnay | 1644873 | 2020-03-18 12:39:36 -0700 | [diff] [blame] | 540 | }, |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 541 | semi_token, |
| David Tolnay | 556738d | 2020-11-15 13:58:44 -0800 | [diff] [blame] | 542 | trusted, |
| David Tolnay | 295ef6b | 2020-05-07 16:10:22 -0700 | [diff] [blame] | 543 | })) |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 544 | } |
| 545 | |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 546 | fn parse_extern_verbatim( |
| 547 | cx: &mut Errors, |
| 548 | tokens: &TokenStream, |
| 549 | lang: Lang, |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 550 | trusted: bool, |
| David Tolnay | 302b175 | 2020-11-10 18:43:24 -0800 | [diff] [blame] | 551 | namespace: &Namespace, |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 552 | ) -> Result<Api> { |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 553 | |input: ParseStream| -> Result<Api> { |
| David Tolnay | e2f9ec4 | 2020-05-07 16:19:33 -0700 | [diff] [blame] | 554 | let attrs = input.call(Attribute::parse_outer)?; |
| 555 | let type_token: Token![type] = match input.parse()? { |
| 556 | Some(type_token) => type_token, |
| 557 | None => { |
| 558 | let span = input.cursor().token_stream(); |
| 559 | return Err(Error::new_spanned(span, "unsupported foreign item")); |
| 560 | } |
| 561 | }; |
| 562 | let ident: Ident = input.parse()?; |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 563 | let lookahead = input.lookahead1(); |
| 564 | if lookahead.peek(Token![=]) { |
| 565 | // type Alias = crate::path::to::Type; |
| 566 | parse_type_alias(cx, attrs, type_token, ident, input, lang, namespace) |
| 567 | } else if lookahead.peek(Token![:]) { |
| 568 | // type Opaque: Bound2 + Bound2; |
| 569 | parse_extern_type_bounded( |
| 570 | cx, attrs, type_token, ident, input, lang, trusted, namespace, |
| 571 | ) |
| 572 | } else { |
| 573 | Err(lookahead.error()) |
| David Tolnay | e2f9ec4 | 2020-05-07 16:19:33 -0700 | [diff] [blame] | 574 | } |
| 575 | } |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 576 | .parse2(tokens.clone()) |
| 577 | } |
| 578 | |
| 579 | fn parse_type_alias( |
| 580 | cx: &mut Errors, |
| 581 | attrs: Vec<Attribute>, |
| 582 | type_token: Token![type], |
| 583 | ident: Ident, |
| 584 | input: ParseStream, |
| 585 | lang: Lang, |
| 586 | namespace: &Namespace, |
| 587 | ) -> Result<Api> { |
| 588 | let eq_token: Token![=] = input.parse()?; |
| 589 | let ty: RustType = input.parse()?; |
| 590 | let semi_token: Token![;] = input.parse()?; |
| 591 | |
| 592 | let mut doc = Doc::new(); |
| 593 | let mut derives = Vec::new(); |
| 594 | let mut namespace = namespace.clone(); |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 595 | let mut cxx_name = None; |
| 596 | let mut rust_name = None; |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 597 | attrs::parse( |
| 598 | cx, |
| 599 | &attrs, |
| 600 | attrs::Parser { |
| 601 | doc: Some(&mut doc), |
| 602 | derives: Some(&mut derives), |
| 603 | namespace: Some(&mut namespace), |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 604 | cxx_name: Some(&mut cxx_name), |
| 605 | rust_name: Some(&mut rust_name), |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 606 | ..Default::default() |
| 607 | }, |
| 608 | ); |
| 609 | |
| 610 | if lang == Lang::Rust { |
| 611 | let span = quote!(#type_token #semi_token); |
| 612 | let msg = "type alias in extern \"Rust\" block is not supported"; |
| 613 | return Err(Error::new_spanned(span, msg)); |
| 614 | } |
| 615 | |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 616 | let name = pair(namespace, &ident, cxx_name, rust_name); |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 617 | |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 618 | Ok(Api::TypeAlias(TypeAlias { |
| 619 | doc, |
| 620 | derives, |
| 621 | type_token, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 622 | name, |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 623 | eq_token, |
| 624 | ty, |
| 625 | semi_token, |
| 626 | })) |
| 627 | } |
| 628 | |
| 629 | fn parse_extern_type_bounded( |
| 630 | cx: &mut Errors, |
| 631 | attrs: Vec<Attribute>, |
| 632 | type_token: Token![type], |
| 633 | ident: Ident, |
| 634 | input: ParseStream, |
| 635 | lang: Lang, |
| 636 | trusted: bool, |
| 637 | namespace: &Namespace, |
| 638 | ) -> Result<Api> { |
| 639 | let colon_token: Token![:] = input.parse()?; |
| 640 | let mut bounds = Vec::new(); |
| 641 | loop { |
| 642 | match input.parse()? { |
| 643 | TypeParamBound::Trait(TraitBound { |
| 644 | paren_token: None, |
| 645 | modifier: TraitBoundModifier::None, |
| 646 | lifetimes: None, |
| 647 | path, |
| 648 | }) if if let Some(derive) = path.get_ident().and_then(Derive::from) { |
| 649 | bounds.push(derive); |
| 650 | true |
| 651 | } else { |
| 652 | false |
| 653 | } => {} |
| 654 | bound @ TypeParamBound::Trait(_) | bound @ TypeParamBound::Lifetime(_) => { |
| 655 | cx.error(bound, "unsupported trait"); |
| 656 | } |
| 657 | } |
| 658 | |
| 659 | let lookahead = input.lookahead1(); |
| 660 | if lookahead.peek(Token![+]) { |
| 661 | input.parse::<Token![+]>()?; |
| 662 | } else if lookahead.peek(Token![;]) { |
| 663 | break; |
| 664 | } else { |
| 665 | return Err(lookahead.error()); |
| 666 | } |
| 667 | } |
| 668 | let semi_token: Token![;] = input.parse()?; |
| 669 | |
| 670 | let mut doc = Doc::new(); |
| 671 | let mut derives = Vec::new(); |
| 672 | let mut namespace = namespace.clone(); |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 673 | let mut cxx_name = None; |
| 674 | let mut rust_name = None; |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 675 | attrs::parse( |
| 676 | cx, |
| 677 | &attrs, |
| 678 | attrs::Parser { |
| 679 | doc: Some(&mut doc), |
| 680 | derives: Some(&mut derives), |
| 681 | namespace: Some(&mut namespace), |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 682 | cxx_name: Some(&mut cxx_name), |
| 683 | rust_name: Some(&mut rust_name), |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 684 | ..Default::default() |
| 685 | }, |
| 686 | ); |
| 687 | |
| David Tolnay | 885be13 | 2020-12-21 15:52:47 -0800 | [diff] [blame] | 688 | let name = pair(namespace, &ident, cxx_name, rust_name); |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 689 | let colon_token = Some(colon_token); |
| 690 | |
| 691 | Ok(match lang { |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 692 | Lang::Cxx => Api::CxxType, |
| 693 | Lang::Rust => Api::RustType, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 694 | }(ExternType { |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 695 | lang, |
| 696 | doc, |
| 697 | derives, |
| 698 | type_token, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 699 | name, |
| 700 | colon_token, |
| David Tolnay | b7eb65e | 2020-11-27 20:00:53 -0800 | [diff] [blame] | 701 | bounds, |
| 702 | semi_token, |
| 703 | trusted, |
| 704 | })) |
| David Tolnay | e2f9ec4 | 2020-05-07 16:19:33 -0700 | [diff] [blame] | 705 | } |
| 706 | |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 707 | fn parse_impl(imp: ItemImpl) -> Result<Api> { |
| David Tolnay | 7e69f89 | 2020-10-03 22:20:22 -0700 | [diff] [blame] | 708 | if !imp.items.is_empty() { |
| 709 | let mut span = Group::new(Delimiter::Brace, TokenStream::new()); |
| 710 | span.set_span(imp.brace_token.span); |
| 711 | return Err(Error::new_spanned(span, "expected an empty impl block")); |
| 712 | } |
| 713 | |
| David Tolnay | 7e69f89 | 2020-10-03 22:20:22 -0700 | [diff] [blame] | 714 | if let Some((bang, path, for_token)) = &imp.trait_ { |
| David Tolnay | 028d3d2 | 2020-11-29 18:09:52 -0800 | [diff] [blame] | 715 | let self_ty = &imp.self_ty; |
| David Tolnay | 7e69f89 | 2020-10-03 22:20:22 -0700 | [diff] [blame] | 716 | let span = quote!(#bang #path #for_token #self_ty); |
| 717 | return Err(Error::new_spanned( |
| 718 | span, |
| 719 | "unexpected impl, expected something like `impl UniquePtr<T> {}`", |
| 720 | )); |
| 721 | } |
| 722 | |
| 723 | let generics = &imp.generics; |
| 724 | if !generics.params.is_empty() || generics.where_clause.is_some() { |
| 725 | return Err(Error::new_spanned( |
| 726 | imp, |
| 727 | "generic parameters on an impl is not supported", |
| 728 | )); |
| 729 | } |
| 730 | |
| David Tolnay | 028d3d2 | 2020-11-29 18:09:52 -0800 | [diff] [blame] | 731 | let mut negative_token = None; |
| 732 | let mut self_ty = *imp.self_ty; |
| 733 | if let RustType::Verbatim(ty) = &self_ty { |
| 734 | let mut iter = ty.clone().into_iter(); |
| 735 | if let Some(TokenTree::Punct(punct)) = iter.next() { |
| 736 | if punct.as_char() == '!' { |
| 737 | let ty = iter.collect::<TokenStream>(); |
| 738 | if !ty.is_empty() { |
| 739 | negative_token = Some(Token)); |
| 740 | self_ty = syn::parse2(ty)?; |
| 741 | } |
| 742 | } |
| 743 | } |
| 744 | } |
| 745 | |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 746 | let impl_token = imp.impl_token; |
| 747 | let negative = negative_token.is_some(); |
| 748 | let ty = parse_type(&self_ty)?; |
| 749 | let brace_token = imp.brace_token; |
| 750 | |
| David Tolnay | 7e69f89 | 2020-10-03 22:20:22 -0700 | [diff] [blame] | 751 | Ok(Api::Impl(Impl { |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 752 | impl_token, |
| 753 | negative, |
| 754 | ty, |
| 755 | brace_token, |
| David Tolnay | 028d3d2 | 2020-11-29 18:09:52 -0800 | [diff] [blame] | 756 | negative_token, |
| David Tolnay | 7e69f89 | 2020-10-03 22:20:22 -0700 | [diff] [blame] | 757 | })) |
| 758 | } |
| 759 | |
| David Tolnay | b0cd327 | 2020-10-27 21:32:54 -0700 | [diff] [blame] | 760 | fn parse_include(input: ParseStream) -> Result<Include> { |
| David Tolnay | 91e87fa | 2020-05-11 19:10:23 -0700 | [diff] [blame] | 761 | if input.peek(LitStr) { |
| David Tolnay | b0cd327 | 2020-10-27 21:32:54 -0700 | [diff] [blame] | 762 | let lit: LitStr = input.parse()?; |
| 763 | let span = lit.span(); |
| 764 | return Ok(Include { |
| 765 | path: lit.value(), |
| 766 | kind: IncludeKind::Quoted, |
| 767 | begin_span: span, |
| 768 | end_span: span, |
| 769 | }); |
| David Tolnay | 91e87fa | 2020-05-11 19:10:23 -0700 | [diff] [blame] | 770 | } |
| 771 | |
| 772 | if input.peek(Token![<]) { |
| 773 | let mut path = String::new(); |
| David Tolnay | b0cd327 | 2020-10-27 21:32:54 -0700 | [diff] [blame] | 774 | |
| David Tolnay | 2cc2e3a | 2020-10-28 13:12:35 -0700 | [diff] [blame] | 775 | let langle: Token![<] = input.parse()?; |
| David Tolnay | 91e87fa | 2020-05-11 19:10:23 -0700 | [diff] [blame] | 776 | while !input.is_empty() && !input.peek(Token![>]) { |
| 777 | let token: TokenTree = input.parse()?; |
| 778 | match token { |
| 779 | TokenTree::Ident(token) => path += &token.to_string(), |
| 780 | TokenTree::Literal(token) |
| 781 | if token |
| 782 | .to_string() |
| 783 | .starts_with(|ch: char| ch.is_ascii_digit()) => |
| 784 | { |
| 785 | path += &token.to_string(); |
| 786 | } |
| 787 | TokenTree::Punct(token) => path.push(token.as_char()), |
| 788 | _ => return Err(Error::new(token.span(), "unexpected token in include path")), |
| 789 | } |
| 790 | } |
| David Tolnay | b0cd327 | 2020-10-27 21:32:54 -0700 | [diff] [blame] | 791 | let rangle: Token![>] = input.parse()?; |
| David Tolnay | b0cd327 | 2020-10-27 21:32:54 -0700 | [diff] [blame] | 792 | |
| 793 | return Ok(Include { |
| 794 | path, |
| 795 | kind: IncludeKind::Bracketed, |
| David Tolnay | 2cc2e3a | 2020-10-28 13:12:35 -0700 | [diff] [blame] | 796 | begin_span: langle.span, |
| 797 | end_span: rangle.span, |
| David Tolnay | b0cd327 | 2020-10-27 21:32:54 -0700 | [diff] [blame] | 798 | }); |
| David Tolnay | 91e87fa | 2020-05-11 19:10:23 -0700 | [diff] [blame] | 799 | } |
| 800 | |
| 801 | Err(input.error("expected \"quoted/path/to\" or <bracketed/path/to>")) |
| 802 | } |
| 803 | |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 804 | fn parse_type(ty: &RustType) -> Result<Type> { |
| David Tolnay | 59b7ede | 2020-03-16 00:30:23 -0700 | [diff] [blame] | 805 | match ty { |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 806 | RustType::Reference(ty) => parse_type_reference(ty), |
| 807 | RustType::Path(ty) => parse_type_path(ty), |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 808 | RustType::Array(ty) => parse_type_array(ty), |
| 809 | RustType::BareFn(ty) => parse_type_fn(ty), |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 810 | RustType::Tuple(ty) if ty.elems.is_empty() => Ok(Type::Void(ty.paren_token.span)), |
| 811 | _ => Err(Error::new_spanned(ty, "unsupported type")), |
| 812 | } |
| 813 | } |
| 814 | |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 815 | fn parse_type_reference(ty: &TypeReference) -> Result<Type> { |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 816 | let ampersand = ty.and_token; |
| 817 | let lifetime = ty.lifetime.clone(); |
| 818 | let mutable = ty.mutability.is_some(); |
| 819 | let mutability = ty.mutability; |
| 820 | |
| David Tolnay | e0dca7b | 2020-11-25 17:18:57 -0800 | [diff] [blame] | 821 | if let RustType::Slice(slice) = ty.elem.as_ref() { |
| 822 | let inner = parse_type(&slice.elem)?; |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 823 | let bracket = slice.bracket_token; |
| David Tolnay | 69db814 | 2020-11-25 17:49:25 -0800 | [diff] [blame] | 824 | return Ok(Type::SliceRef(Box::new(SliceRef { |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 825 | ampersand, |
| 826 | lifetime, |
| 827 | mutable, |
| 828 | bracket, |
| David Tolnay | 69db814 | 2020-11-25 17:49:25 -0800 | [diff] [blame] | 829 | inner, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 830 | mutability, |
| David Tolnay | 69db814 | 2020-11-25 17:49:25 -0800 | [diff] [blame] | 831 | }))); |
| David Tolnay | e0dca7b | 2020-11-25 17:18:57 -0800 | [diff] [blame] | 832 | } |
| 833 | |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 834 | let inner = parse_type(&ty.elem)?; |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 835 | let pinned = false; |
| 836 | let pin_tokens = None; |
| 837 | |
| 838 | Ok(match &inner { |
| Adrian Taylor | c871343 | 2020-10-21 18:20:55 -0700 | [diff] [blame] | 839 | Type::Ident(ident) if ident.rust == "str" => { |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 840 | if ty.mutability.is_some() { |
| 841 | return Err(Error::new_spanned(ty, "unsupported type")); |
| 842 | } else { |
| 843 | Type::Str |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 844 | } |
| 845 | } |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 846 | _ => Type::Ref, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 847 | }(Box::new(Ref { |
| 848 | pinned, |
| 849 | ampersand, |
| 850 | lifetime, |
| 851 | mutable, |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 852 | inner, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 853 | pin_tokens, |
| 854 | mutability, |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 855 | }))) |
| 856 | } |
| 857 | |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 858 | fn parse_type_path(ty: &TypePath) -> Result<Type> { |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 859 | let path = &ty.path; |
| 860 | if ty.qself.is_none() && path.leading_colon.is_none() && path.segments.len() == 1 { |
| 861 | let segment = &path.segments[0]; |
| 862 | let ident = segment.ident.clone(); |
| 863 | match &segment.arguments { |
| David Tolnay | 75ea17c | 2020-12-06 21:08:34 -0800 | [diff] [blame] | 864 | PathArguments::None => return Ok(Type::Ident(RustName::new(ident))), |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 865 | PathArguments::AngleBracketed(generic) => { |
| 866 | if ident == "UniquePtr" && generic.args.len() == 1 { |
| 867 | if let GenericArgument::Type(arg) = &generic.args[0] { |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 868 | let inner = parse_type(arg)?; |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 869 | return Ok(Type::UniquePtr(Box::new(Ty1 { |
| David Tolnay | da71ef6 | 2020-11-02 00:36:00 -0800 | [diff] [blame] | 870 | name: ident, |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 871 | langle: generic.lt_token, |
| 872 | inner, |
| 873 | rangle: generic.gt_token, |
| 874 | }))); |
| 875 | } |
| David Tolnay | b3b24a1 | 2020-12-01 15:27:43 -0800 | [diff] [blame] | 876 | } else if ident == "SharedPtr" && generic.args.len() == 1 { |
| 877 | if let GenericArgument::Type(arg) = &generic.args[0] { |
| 878 | let inner = parse_type(arg)?; |
| 879 | return Ok(Type::SharedPtr(Box::new(Ty1 { |
| 880 | name: ident, |
| 881 | langle: generic.lt_token, |
| 882 | inner, |
| 883 | rangle: generic.gt_token, |
| 884 | }))); |
| 885 | } |
| David Tolnay | e1dcdf7 | 2020-04-24 17:40:55 -0700 | [diff] [blame] | 886 | } else if ident == "CxxVector" && generic.args.len() == 1 { |
| Myron Ahn | eba35cf | 2020-02-05 19:41:51 +0700 | [diff] [blame] | 887 | if let GenericArgument::Type(arg) = &generic.args[0] { |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 888 | let inner = parse_type(arg)?; |
| David Tolnay | 4377a9e | 2020-04-24 15:20:26 -0700 | [diff] [blame] | 889 | return Ok(Type::CxxVector(Box::new(Ty1 { |
| David Tolnay | da71ef6 | 2020-11-02 00:36:00 -0800 | [diff] [blame] | 890 | name: ident, |
| Myron Ahn | eba35cf | 2020-02-05 19:41:51 +0700 | [diff] [blame] | 891 | langle: generic.lt_token, |
| 892 | inner, |
| 893 | rangle: generic.gt_token, |
| 894 | }))); |
| 895 | } |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 896 | } else if ident == "Box" && generic.args.len() == 1 { |
| 897 | if let GenericArgument::Type(arg) = &generic.args[0] { |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 898 | let inner = parse_type(arg)?; |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 899 | return Ok(Type::RustBox(Box::new(Ty1 { |
| David Tolnay | da71ef6 | 2020-11-02 00:36:00 -0800 | [diff] [blame] | 900 | name: ident, |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 901 | langle: generic.lt_token, |
| 902 | inner, |
| 903 | rangle: generic.gt_token, |
| 904 | }))); |
| 905 | } |
| Myron Ahn | eba35cf | 2020-02-05 19:41:51 +0700 | [diff] [blame] | 906 | } else if ident == "Vec" && generic.args.len() == 1 { |
| 907 | if let GenericArgument::Type(arg) = &generic.args[0] { |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 908 | let inner = parse_type(arg)?; |
| Myron Ahn | eba35cf | 2020-02-05 19:41:51 +0700 | [diff] [blame] | 909 | return Ok(Type::RustVec(Box::new(Ty1 { |
| David Tolnay | da71ef6 | 2020-11-02 00:36:00 -0800 | [diff] [blame] | 910 | name: ident, |
| Myron Ahn | eba35cf | 2020-02-05 19:41:51 +0700 | [diff] [blame] | 911 | langle: generic.lt_token, |
| 912 | inner, |
| 913 | rangle: generic.gt_token, |
| 914 | }))); |
| 915 | } |
| David Tolnay | cbc2a4b | 2020-11-15 15:08:44 -0800 | [diff] [blame] | 916 | } else if ident == "Pin" && generic.args.len() == 1 { |
| 917 | if let GenericArgument::Type(arg) = &generic.args[0] { |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 918 | let inner = parse_type(arg)?; |
| David Tolnay | c94035a | 2020-11-15 16:49:18 -0800 | [diff] [blame] | 919 | let pin_token = kw::Pin(ident.span()); |
| David Tolnay | cbc2a4b | 2020-11-15 15:08:44 -0800 | [diff] [blame] | 920 | if let Type::Ref(mut inner) = inner { |
| David Tolnay | cbc2a4b | 2020-11-15 15:08:44 -0800 | [diff] [blame] | 921 | inner.pinned = true; |
| 922 | inner.pin_tokens = |
| 923 | Some((pin_token, generic.lt_token, generic.gt_token)); |
| 924 | return Ok(Type::Ref(inner)); |
| 925 | } |
| 926 | } |
| David Tolnay | b40b9db | 2020-03-18 13:50:31 -0700 | [diff] [blame] | 927 | } |
| 928 | } |
| 929 | PathArguments::Parenthesized(_) => {} |
| David Tolnay | fb134ed | 2020-03-15 23:17:48 -0700 | [diff] [blame] | 930 | } |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 931 | } |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 932 | |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 933 | Err(Error::new_spanned(ty, "unsupported type")) |
| 934 | } |
| 935 | |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 936 | fn parse_type_array(ty: &TypeArray) -> Result<Type> { |
| 937 | let inner = parse_type(&ty.elem)?; |
| David Tolnay | d9f9784 | 2020-11-24 21:00:05 -0800 | [diff] [blame] | 938 | |
| 939 | let len_expr = if let Expr::Lit(lit) = &ty.len { |
| 940 | lit |
| 941 | } else { |
| 942 | let msg = "unsupported expression, array length must be an integer literal"; |
| 943 | return Err(Error::new_spanned(&ty.len, msg)); |
| 944 | }; |
| 945 | |
| 946 | let len_token = if let Lit::Int(int) = &len_expr.lit { |
| 947 | int.clone() |
| 948 | } else { |
| 949 | let msg = "array length must be an integer literal"; |
| 950 | return Err(Error::new_spanned(len_expr, msg)); |
| 951 | }; |
| 952 | |
| David Tolnay | 581f72d | 2020-11-25 15:48:59 -0800 | [diff] [blame] | 953 | let len = len_token.base10_parse::<usize>()?; |
| 954 | if len == 0 { |
| 955 | let msg = "array with zero size is not supported"; |
| 956 | return Err(Error::new_spanned(ty, msg)); |
| 957 | } |
| 958 | |
| David Tolnay | d9f9784 | 2020-11-24 21:00:05 -0800 | [diff] [blame] | 959 | let bracket = ty.bracket_token; |
| 960 | let semi_token = ty.semi_token; |
| David Tolnay | d9f9784 | 2020-11-24 21:00:05 -0800 | [diff] [blame] | 961 | |
| 962 | Ok(Type::Array(Box::new(Array { |
| 963 | bracket, |
| 964 | inner, |
| 965 | semi_token, |
| 966 | len, |
| 967 | len_token, |
| 968 | }))) |
| David Tolnay | 2d0df23 | 2020-11-24 20:50:32 -0800 | [diff] [blame] | 969 | } |
| 970 | |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 971 | fn parse_type_fn(ty: &TypeBareFn) -> Result<Type> { |
| David Tolnay | c071b89 | 2020-03-18 16:59:53 -0700 | [diff] [blame] | 972 | if ty.lifetimes.is_some() { |
| 973 | return Err(Error::new_spanned( |
| 974 | ty, |
| 975 | "function pointer with lifetime parameters is not supported yet", |
| 976 | )); |
| 977 | } |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 978 | |
| David Tolnay | c071b89 | 2020-03-18 16:59:53 -0700 | [diff] [blame] | 979 | if ty.variadic.is_some() { |
| 980 | return Err(Error::new_spanned( |
| 981 | ty, |
| 982 | "variadic function pointer is not supported yet", |
| 983 | )); |
| 984 | } |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 985 | |
| David Tolnay | c071b89 | 2020-03-18 16:59:53 -0700 | [diff] [blame] | 986 | let args = ty |
| 987 | .inputs |
| 988 | .iter() |
| 989 | .enumerate() |
| 990 | .map(|(i, arg)| { |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 991 | let ty = parse_type(&arg.ty)?; |
| David Tolnay | c071b89 | 2020-03-18 16:59:53 -0700 | [diff] [blame] | 992 | let ident = match &arg.name { |
| 993 | Some(ident) => ident.0.clone(), |
| 994 | None => format_ident!("_{}", i), |
| 995 | }; |
| 996 | Ok(Var { ident, ty }) |
| 997 | }) |
| 998 | .collect::<Result<_>>()?; |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 999 | |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 1000 | let mut throws_tokens = None; |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 1001 | let ret = parse_return_type(&ty.output, &mut throws_tokens)?; |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 1002 | let throws = throws_tokens.is_some(); |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 1003 | |
| 1004 | let unsafety = ty.unsafety; |
| 1005 | let fn_token = ty.fn_token; |
| 1006 | let generics = Generics::default(); |
| 1007 | let receiver = None; |
| 1008 | let paren_token = ty.paren_token; |
| 1009 | |
| David Tolnay | c071b89 | 2020-03-18 16:59:53 -0700 | [diff] [blame] | 1010 | Ok(Type::Fn(Box::new(Signature { |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 1011 | unsafety, |
| 1012 | fn_token, |
| 1013 | generics, |
| 1014 | receiver, |
| David Tolnay | c071b89 | 2020-03-18 16:59:53 -0700 | [diff] [blame] | 1015 | args, |
| 1016 | ret, |
| 1017 | throws, |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 1018 | paren_token, |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 1019 | throws_tokens, |
| David Tolnay | c071b89 | 2020-03-18 16:59:53 -0700 | [diff] [blame] | 1020 | }))) |
| 1021 | } |
| 1022 | |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 1023 | fn parse_return_type( |
| 1024 | ty: &ReturnType, |
| 1025 | throws_tokens: &mut Option<(kw::Result, Token![<], Token![>])>, |
| 1026 | ) -> Result<Option<Type>> { |
| David Tolnay | c071b89 | 2020-03-18 16:59:53 -0700 | [diff] [blame] | 1027 | let mut ret = match ty { |
| 1028 | ReturnType::Default => return Ok(None), |
| 1029 | ReturnType::Type(_, ret) => ret.as_ref(), |
| 1030 | }; |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 1031 | |
| David Tolnay | c071b89 | 2020-03-18 16:59:53 -0700 | [diff] [blame] | 1032 | if let RustType::Path(ty) = ret { |
| 1033 | let path = &ty.path; |
| 1034 | if ty.qself.is_none() && path.leading_colon.is_none() && path.segments.len() == 1 { |
| 1035 | let segment = &path.segments[0]; |
| 1036 | let ident = segment.ident.clone(); |
| 1037 | if let PathArguments::AngleBracketed(generic) = &segment.arguments { |
| 1038 | if ident == "Result" && generic.args.len() == 1 { |
| 1039 | if let GenericArgument::Type(arg) = &generic.args[0] { |
| 1040 | ret = arg; |
| David Tolnay | e3a4815 | 2020-04-08 19:38:05 -0700 | [diff] [blame] | 1041 | *throws_tokens = |
| 1042 | Some((kw::Result(ident.span()), generic.lt_token, generic.gt_token)); |
| David Tolnay | c071b89 | 2020-03-18 16:59:53 -0700 | [diff] [blame] | 1043 | } |
| 1044 | } |
| 1045 | } |
| 1046 | } |
| 1047 | } |
| David Tolnay | ddbcb52 | 2020-12-21 15:24:45 -0800 | [diff] [blame] | 1048 | |
| David Tolnay | 313cc0a | 2020-11-24 21:47:01 -0800 | [diff] [blame] | 1049 | match parse_type(ret)? { |
| David Tolnay | c071b89 | 2020-03-18 16:59:53 -0700 | [diff] [blame] | 1050 | Type::Void(_) => Ok(None), |
| 1051 | ty => Ok(Some(ty)), |
| 1052 | } |
| 1053 | } |
| David Tolnay | 6a990f8 | 2020-12-21 15:42:31 -0800 | [diff] [blame] | 1054 | |
| 1055 | fn pair(namespace: Namespace, default: &Ident, cxx: Option<Ident>, rust: Option<Ident>) -> Pair { |
| 1056 | let default = || default.clone(); |
| David Tolnay | c2a4c78 | 2020-12-21 15:55:31 -0800 | [diff] [blame] | 1057 | Pair { |
| David Tolnay | 6a990f8 | 2020-12-21 15:42:31 -0800 | [diff] [blame] | 1058 | namespace, |
| David Tolnay | c2a4c78 | 2020-12-21 15:55:31 -0800 | [diff] [blame] | 1059 | cxx: cxx.unwrap_or_else(default), |
| 1060 | rust: rust.unwrap_or_else(default), |
| 1061 | } |
| David Tolnay | 6a990f8 | 2020-12-21 15:42:31 -0800 | [diff] [blame] | 1062 | } |