| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 1 | #![allow( |
| David Tolnay | 30d214c | 2020-03-15 23:54:34 -0700 | [diff] [blame] | 2 | clippy::inherent_to_string, |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 3 | clippy::large_enum_variant, |
| 4 | clippy::new_without_default, |
| David Tolnay | ef665fd | 2020-11-15 17:25:31 -0800 | [diff] [blame] | 5 | clippy::nonminimal_bool, |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 6 | clippy::or_fun_call, |
| David Tolnay | 982158b | 2020-11-27 15:58:49 -0800 | [diff] [blame^] | 7 | clippy::single_match, |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 8 | clippy::toplevel_ref_arg, |
| 9 | clippy::useless_let_if_seq |
| 10 | )] |
| 11 | |
| 12 | extern crate proc_macro; |
| 13 | |
| David Tolnay | d8ad970 | 2020-11-27 12:43:59 -0800 | [diff] [blame] | 14 | mod derive; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 15 | mod expand; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 16 | mod syntax; |
| David Tolnay | d09e012 | 2020-05-04 02:34:23 -0700 | [diff] [blame] | 17 | mod type_id; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 18 | |
| David Tolnay | 05ef6ff | 2020-08-29 11:27:05 -0700 | [diff] [blame] | 19 | use crate::syntax::file::Module; |
| David Tolnay | 0841930 | 2020-04-19 20:38:20 -0700 | [diff] [blame] | 20 | use crate::syntax::namespace::Namespace; |
| David Tolnay | a8d94a1 | 2020-09-06 23:28:18 -0700 | [diff] [blame] | 21 | use crate::syntax::qualified::QualifiedName; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 22 | use proc_macro::TokenStream; |
| Adrian Taylor | 3e5cff4 | 2020-10-29 19:48:45 -0700 | [diff] [blame] | 23 | use syn::parse::{Parse, ParseStream, Parser, Result}; |
| David Tolnay | a8d94a1 | 2020-09-06 23:28:18 -0700 | [diff] [blame] | 24 | use syn::parse_macro_input; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 25 | |
| 26 | /// `#[cxx::bridge] mod ffi { ... }` |
| 27 | /// |
| 28 | /// Refer to the crate-level documentation for the explanation of how this macro |
| 29 | /// is intended to be used. |
| 30 | /// |
| 31 | /// The only additional thing to note here is namespace support — if the |
| David Tolnay | c72a9f6 | 2020-11-11 10:56:26 -0800 | [diff] [blame] | 32 | /// types and functions on the `extern "C++"` side of our bridge are in a |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 33 | /// namespace, specify that namespace as an argument of the cxx::bridge |
| 34 | /// attribute macro. |
| 35 | /// |
| 36 | /// ``` |
| David Tolnay | 7907966 | 2020-11-01 09:11:55 -0800 | [diff] [blame] | 37 | /// #[cxx::bridge(namespace = "mycompany::rust")] |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 38 | /// # mod ffi {} |
| 39 | /// ``` |
| 40 | /// |
| 41 | /// The types and functions from the `extern "Rust"` side of the bridge will be |
| 42 | /// placed into that same namespace in the generated C++ code. |
| 43 | #[proc_macro_attribute] |
| 44 | pub fn bridge(args: TokenStream, input: TokenStream) -> TokenStream { |
| 45 | let _ = syntax::error::ERRORS; |
| 46 | |
| Adrian Taylor | 3e5cff4 | 2020-10-29 19:48:45 -0700 | [diff] [blame] | 47 | let namespace = match Namespace::parse_bridge_attr_namespace.parse(args) { |
| David Tolnay | d7a3a18 | 2020-11-01 20:45:14 -0800 | [diff] [blame] | 48 | Ok(namespace) => namespace, |
| Adrian Taylor | 3e5cff4 | 2020-10-29 19:48:45 -0700 | [diff] [blame] | 49 | Err(err) => return err.to_compile_error().into(), |
| 50 | }; |
| David Tolnay | 3c64a4e | 2020-08-29 14:07:38 -0700 | [diff] [blame] | 51 | let mut ffi = parse_macro_input!(input as Module); |
| 52 | ffi.namespace = namespace; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 53 | |
| David Tolnay | 3c64a4e | 2020-08-29 14:07:38 -0700 | [diff] [blame] | 54 | expand::bridge(ffi) |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 55 | .unwrap_or_else(|err| err.to_compile_error()) |
| 56 | .into() |
| 57 | } |
| David Tolnay | d09e012 | 2020-05-04 02:34:23 -0700 | [diff] [blame] | 58 | |
| 59 | #[proc_macro] |
| 60 | pub fn type_id(input: TokenStream) -> TokenStream { |
| David Tolnay | a8d94a1 | 2020-09-06 23:28:18 -0700 | [diff] [blame] | 61 | struct TypeId(QualifiedName); |
| 62 | |
| 63 | impl Parse for TypeId { |
| 64 | fn parse(input: ParseStream) -> Result<Self> { |
| 65 | QualifiedName::parse_quoted_or_unquoted(input).map(TypeId) |
| 66 | } |
| 67 | } |
| 68 | |
| 69 | let arg = parse_macro_input!(input as TypeId); |
| 70 | type_id::expand(arg.0).into() |
| David Tolnay | d09e012 | 2020-05-04 02:34:23 -0700 | [diff] [blame] | 71 | } |