31 releases
0.5.1 | Jul 4, 2024 |
---|---|
0.5.0 | Oct 10, 2023 |
0.4.3 |
|
0.4.2 | Jul 24, 2023 |
0.2.11 | Mar 31, 2023 |
#13 in Procedural macros
211,995 downloads per month
Used in 359 crates
(13 directly)
65KB
961 lines
Macro Magic 🪄
This crate provides the #[export_tokens]
macro and a number of companion macros, including
the #[import_tokens_proc]
and #[import_tokens_attr]
macros. When used in tandem with
#[export_tokens]
, these macros allow you to create regular and attribute proc macros in which
you can import and make use of the tokens of external/foreign items marked with
#[export_tokens]
in other modules, files, and even in other crates merely by referring to
them by name/path.
Among other things, the patterns introduced by macro_magic
can be used to implement safe and
efficient exportation and importation of item tokens within the same file, and even across file
and crate boundaries. The only requirement is that you have control over (i.e. can add an attribute
macro to) the source code of both locations.
macro_magic
is designed to work with stable Rust, and is fully no_std
compatible (in fact,
there is a unit test to ensure everything is no_std
safe).
General Syntax
You can use macro_magic
to build regular and attribute proc macros that look like this:
#[my_attribute(path::to::MyItem)]
trait SomeTrait {
// ..
}
this:
do_something!(path::to::MyItem);
or even this:
let foreign_tokens = my_macro!(path::to::MyItem);
assert_eq!(foreign_tokens.to_string(), "struct MyItem {...}");
where path::to::MyItem
is the path to an item that has been marked with #[export_tokens]
.
All of this behavior is accomplished under the hood using proc macros that create
macro_rules
-based callbacks, but as a programmer this complexity is completely hidden from
you via simple attribute macros you can apply to your proc macros to imbue them with the power
of importing the tokens for external items based on their path.
Attribute Example
You could write an attribute macro to "inject" the fields of one struct into another as follows:
#[import_tokens_attr]
#[proc_macro_attribute]
pub fn combine_structs(attr: TokenStream, tokens: TokenStream) -> TokenStream {
let foreign_struct = parse_macro_input!(attr as ItemStruct);
let local_struct = parse_macro_input!(tokens as ItemStruct);
let Fields::Named(local_fields) = local_struct.fields else {
return Error::new(
local_struct.fields.span(),
"unnamed fields are not supported"
).to_compile_error().into()
};
let Fields::Named(foreign_fields) = foreign_struct.fields else {
return Error::new(
foreign_struct.fields.span(),
"unnamed fields are not supported"
).to_compile_error().into()
};
let local_fields = local_fields.named.iter();
let foreign_fields = foreign_fields.named.iter();
let attrs = local_struct.attrs;
let generics = local_struct.generics;
let ident = local_struct.ident;
let vis = local_struct.vis;
quote! {
#(#attrs)
*
#vis struct #ident<#generics> {
#(#local_fields),
*
,
#(#foreign_fields),
*
}
}
.into()
}
And then you could use the #[combine_structs]
attribute as follows:
#[export_tokens]
struct ExternalStruct {
foo: u32,
bar: u64,
fizz: i64,
}
#[combine_structs(ExternalStruct)]
struct LocalStruct {
biz: bool,
baz: i32,
}
Which would result in the following expanded output for LocalStruct
:
struct LocalStruct {
foo: u32,
bar: u64,
fizz: i64,
biz: bool,
baz: i32,
}
Note that the attr
variable on the combine_structs
proc macro, thanks to the powers of
#[import_tokens_attr]
, will receive the actual tokens for the ExternalStruct
item, rather
than merely receiving the tokens for the path ExternalStruct
.
This gives you the ability to write attribute macros that receive tokens for two items, one
specified by path via the first argument attr
, as well as the tokens for the item the
attribute is attached to via the 2nd argument tokens
. The only requirement is that the item
specified by attr
has been marked with #[export_tokens]
.
Proc Macro Example
You could write a PHP/ruby/crystal-style verbatim import / require
macro which blindly
imports the tokens of the specified external module into the current context (with all the good
and bad implications that would imply), like this:
#[import_tokens_proc]
#[proc_macro]
pub fn require(tokens: TokenStream) -> TokenStream {
let external_mod = parse_macro_input!(tokens as ItemMod);
let Some((_, stmts)) = external_mod.content else {
return Error::new(
external_mod.span(),
"cannot import tokens from a file-based module since custom file-level \
attributes are not yet supported by Rust"
).to_compile_error().into()
};
quote! {
#(#stmts)
*
}
.into()
}
You could then use the require!
macro like this:
// in some external crate
#[export_tokens]
mod an_external_module {
fn my_cool_function() -> u32 {
567
}
fn my_other_function() -> u32 {
116
}
}
// in another crate where we will use the `require!` macro
mod my_module {
use my_macros::require;
fn existing_stuff() {
println!("foo!");
}
require!(external_crate::an_external_module);
}
which would result in this expansion of require!
within my_module
:
mod my_module {
use my_macros::require;
fn existing_stuff() {
println!("foo!");
}
fn my_cool_function() -> u32 {
567
}
fn my_other_function() -> u32 {
116
}
}
Notice that this hypothetical require!
macro is dangerous for two reasons:
- Any types you may have brought into scope with
use
statements in the foreign module may or may not be available in their new context without additional use statements. - If existing items in the module or context where you use the
require!
macro conflict with something you are importing, you will get a compiler error (this is good, though).
These are just some of the capabilities of macro_magic
🪄.
See the docs
for more information.
Features
proc_support
The proc_support
feature must be enabled in proc macro crates that make use of any import
tokens functionality, including #[import_tokens_attr]
, #[import_tokens_proc]
and
import_tokens!
. Otherwise these macros will not function correctly and will issue compiler
errors complaining about items not existing under macro_magic::mm_core
. The
#[export_tokens]
macro does not require this feature to function correctly, so you can safely
use it without enabling this feature.
The reason for this feature gating is that things like syn
, quote
, proc_macro2
, etc., are
not 100% no_std
compatible and should only be enabled in proc macro crates. For this reason,
you should not enable this feature in crates where you are merely using #[export_tokens]
and nothing else within that crate.
Limitations
One thing that macro_magic
doesn't provide is the ability to build up state information
across multiple macro invocations, however this problem can be tackled effectively using the
outer macro pattern or in some cases using
static atomics and mutexes in your proc macro crate (which we actually do in this crate to keep
track of unique identifiers).
Breaking Changes
- 0.4x removed
#[use_attr]
and#[use_proc]
(they are no longer needed with the new self-calling macro style that has been adopted in 0.4x) and also removed the ability to access#[export_tokens]
invocations in inaccessible locations like inside of functions and across module permission boundaries like in an inaccessible private module. This feature may be re-added in the future if there is interest, however removing it allowed us to consolidate naming of ourmacro_rules!
declarations and remove the need for#[use_attr]
/#[use_proc]
. - 0.2x removed and/or re-wrote a number of features that relied on a non-future-proof
behavior of writing/reading files in the
OUT_DIR
. Versions >= 0.2.0 are completely safe and no longer contain this behavior, however features that provided the ability to enumerate all the#[export_tokens]
calls in a namespace have been removed. The proper way to do this is with the outer macro pattern or with global state mutexes/atomics in your proc macro crate, as mentioned above.
More detailed historical change information can be found in releases.
Dependencies
~0.4–1MB
~22K SLoC