Crates.io | macro_magic |
lib.rs | macro_magic |
version | 0.5.1 |
source | src |
created_at | 2023-03-05 18:45:35.153703 |
updated_at | 2024-07-04 14:40:16.857594 |
description | Allows the exporting and importing of the tokens of items across module, file, and crate boundaries |
homepage | https://sam0x17.dev |
repository | https://github.com/sam0x17/macro_magic |
max_upload_size | |
id | 801658 |
size | 26,219 |
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).
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.
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]
.
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:
use
statements in the foreign module may or
may not be available in their new context without additional use statements.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.
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.
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).
#[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 our macro_rules!
declarations and remove the need for #[use_attr]
/
#[use_proc]
.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.