mirror of
https://github.com/rust-lang/rust.git
synced 2025-05-14 02:49:40 +00:00
Merge #11664
11664: fix: Properly handle proc-macro crate types for nameres r=Veykril a=Veykril bors r+ Co-authored-by: Lukas Wirth <lukastw97@gmail.com>
This commit is contained in:
bors[bot]
GitHub
commit
5b51cb835a
@ -160,6 +160,7 @@ impl ChangeFixture {
|
||||
meta.cfg,
|
||||
meta.env,
|
||||
Default::default(),
|
||||
false,
|
||||
origin,
|
||||
);
|
||||
let prev = crates.insert(crate_name.clone(), crate_id);
|
||||
@ -194,6 +195,7 @@ impl ChangeFixture {
|
||||
default_cfg,
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
} else {
|
||||
@ -230,6 +232,7 @@ impl ChangeFixture {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
Vec::new(),
|
||||
false,
|
||||
CrateOrigin::Lang,
|
||||
);
|
||||
|
||||
@ -266,6 +269,7 @@ impl ChangeFixture {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
proc_macro,
|
||||
true,
|
||||
CrateOrigin::Lang,
|
||||
);
|
||||
|
||||
|
||||
@ -124,7 +124,7 @@ pub enum CrateOrigin {
|
||||
/// Crates that are provided by the language, like std, core, proc-macro, ...
|
||||
Lang,
|
||||
/// Crates that we don't know their origin.
|
||||
// Idealy this enum should cover all cases, and then we remove this variant.
|
||||
// Ideally this enum should cover all cases, and then we remove this variant.
|
||||
Unknown,
|
||||
}
|
||||
|
||||
@ -228,6 +228,7 @@ pub struct CrateData {
|
||||
pub dependencies: Vec<Dependency>,
|
||||
pub proc_macro: Vec<ProcMacro>,
|
||||
pub origin: CrateOrigin,
|
||||
pub is_proc_macro: bool,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
|
||||
@ -279,6 +280,7 @@ impl CrateGraph {
|
||||
potential_cfg_options: CfgOptions,
|
||||
env: Env,
|
||||
proc_macro: Vec<ProcMacro>,
|
||||
is_proc_macro: bool,
|
||||
origin: CrateOrigin,
|
||||
) -> CrateId {
|
||||
let data = CrateData {
|
||||
@ -292,6 +294,7 @@ impl CrateGraph {
|
||||
proc_macro,
|
||||
dependencies: Vec::new(),
|
||||
origin,
|
||||
is_proc_macro,
|
||||
};
|
||||
let crate_id = CrateId(self.arena.len() as u32);
|
||||
let prev = self.arena.insert(crate_id, data);
|
||||
@ -596,6 +599,7 @@ mod tests {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
let crate2 = graph.add_crate_root(
|
||||
@ -607,6 +611,7 @@ mod tests {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
let crate3 = graph.add_crate_root(
|
||||
@ -618,6 +623,7 @@ mod tests {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
assert!(graph
|
||||
@ -643,6 +649,7 @@ mod tests {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
let crate2 = graph.add_crate_root(
|
||||
@ -654,6 +661,7 @@ mod tests {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
assert!(graph
|
||||
@ -676,6 +684,7 @@ mod tests {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
let crate2 = graph.add_crate_root(
|
||||
@ -687,6 +696,7 @@ mod tests {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
let crate3 = graph.add_crate_root(
|
||||
@ -698,6 +708,7 @@ mod tests {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
assert!(graph
|
||||
@ -720,6 +731,7 @@ mod tests {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
let crate2 = graph.add_crate_root(
|
||||
@ -731,6 +743,7 @@ mod tests {
|
||||
CfgOptions::default(),
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
assert!(graph
|
||||
|
||||
@ -154,18 +154,21 @@ impl RawAttrs {
|
||||
return smallvec![attr.clone()];
|
||||
}
|
||||
|
||||
let subtree = match attr.input.as_deref() {
|
||||
Some(AttrInput::TokenTree(it, _)) => it,
|
||||
let subtree = match attr.token_tree_value() {
|
||||
Some(it) => it,
|
||||
_ => return smallvec![attr.clone()],
|
||||
};
|
||||
|
||||
// Input subtree is: `(cfg, $(attr),+)`
|
||||
// Split it up into a `cfg` subtree and the `attr` subtrees.
|
||||
// FIXME: There should be a common API for this.
|
||||
let mut parts = subtree.token_trees.split(
|
||||
|tt| matches!(tt, tt::TokenTree::Leaf(tt::Leaf::Punct(p)) if p.char == ','),
|
||||
);
|
||||
let cfg = parts.next().unwrap();
|
||||
let mut parts = subtree.token_trees.split(|tt| {
|
||||
matches!(tt, tt::TokenTree::Leaf(tt::Leaf::Punct(Punct { char: ',', .. })))
|
||||
});
|
||||
let cfg = match parts.next() {
|
||||
Some(it) => it,
|
||||
None => return smallvec![],
|
||||
};
|
||||
let cfg = Subtree { delimiter: subtree.delimiter, token_trees: cfg.to_vec() };
|
||||
let cfg = CfgExpr::parse(&cfg);
|
||||
let index = attr.id;
|
||||
@ -259,17 +262,8 @@ impl Attrs {
|
||||
}
|
||||
|
||||
pub fn docs(&self) -> Option<Documentation> {
|
||||
let docs = self.by_key("doc").attrs().flat_map(|attr| match attr.input.as_deref()? {
|
||||
AttrInput::Literal(s) => Some(s),
|
||||
AttrInput::TokenTree(..) => None,
|
||||
});
|
||||
let indent = docs
|
||||
.clone()
|
||||
.flat_map(|s| s.lines())
|
||||
.filter(|line| !line.chars().all(|c| c.is_whitespace()))
|
||||
.map(|line| line.chars().take_while(|c| c.is_whitespace()).count())
|
||||
.min()
|
||||
.unwrap_or(0);
|
||||
let docs = self.by_key("doc").attrs().filter_map(|attr| attr.string_value());
|
||||
let indent = doc_indent(self);
|
||||
let mut buf = String::new();
|
||||
for doc in docs {
|
||||
// str::lines doesn't yield anything for the empty string
|
||||
@ -507,18 +501,9 @@ impl AttrsWithOwner {
|
||||
&self,
|
||||
db: &dyn DefDatabase,
|
||||
) -> Option<(Documentation, DocsRangeMap)> {
|
||||
// FIXME: code duplication in `docs` above
|
||||
let docs = self.by_key("doc").attrs().flat_map(|attr| match attr.input.as_deref()? {
|
||||
AttrInput::Literal(s) => Some((s, attr.id)),
|
||||
AttrInput::TokenTree(..) => None,
|
||||
});
|
||||
let indent = docs
|
||||
.clone()
|
||||
.flat_map(|(s, _)| s.lines())
|
||||
.filter(|line| !line.chars().all(|c| c.is_whitespace()))
|
||||
.map(|line| line.chars().take_while(|c| c.is_whitespace()).count())
|
||||
.min()
|
||||
.unwrap_or(0);
|
||||
let docs =
|
||||
self.by_key("doc").attrs().filter_map(|attr| attr.string_value().map(|s| (s, attr.id)));
|
||||
let indent = doc_indent(self);
|
||||
let mut buf = String::new();
|
||||
let mut mapping = Vec::new();
|
||||
for (doc, idx) in docs {
|
||||
@ -557,6 +542,18 @@ impl AttrsWithOwner {
|
||||
}
|
||||
}
|
||||
|
||||
fn doc_indent(attrs: &Attrs) -> usize {
|
||||
attrs
|
||||
.by_key("doc")
|
||||
.attrs()
|
||||
.filter_map(|attr| attr.string_value())
|
||||
.flat_map(|s| s.lines())
|
||||
.filter(|line| !line.chars().all(|c| c.is_whitespace()))
|
||||
.map(|line| line.chars().take_while(|c| c.is_whitespace()).count())
|
||||
.min()
|
||||
.unwrap_or(0)
|
||||
}
|
||||
|
||||
fn inner_attributes(
|
||||
syntax: &SyntaxNode,
|
||||
) -> Option<impl Iterator<Item = Either<ast::Attr, ast::Comment>>> {
|
||||
@ -773,45 +770,58 @@ impl Attr {
|
||||
Self::from_src(db, ast, hygiene, id)
|
||||
}
|
||||
|
||||
pub fn path(&self) -> &ModPath {
|
||||
&self.path
|
||||
}
|
||||
}
|
||||
|
||||
impl Attr {
|
||||
/// #[path = "string"]
|
||||
pub fn string_value(&self) -> Option<&SmolStr> {
|
||||
match self.input.as_deref()? {
|
||||
AttrInput::Literal(it) => Some(it),
|
||||
_ => None,
|
||||
}
|
||||
}
|
||||
|
||||
/// #[path(ident)]
|
||||
pub fn single_ident_value(&self) -> Option<&tt::Ident> {
|
||||
match self.input.as_deref()? {
|
||||
AttrInput::TokenTree(subtree, _) => match &*subtree.token_trees {
|
||||
[tt::TokenTree::Leaf(tt::Leaf::Ident(ident))] => Some(ident),
|
||||
_ => None,
|
||||
},
|
||||
_ => None,
|
||||
}
|
||||
}
|
||||
|
||||
/// #[path TokenTree]
|
||||
pub fn token_tree_value(&self) -> Option<&Subtree> {
|
||||
match self.input.as_deref()? {
|
||||
AttrInput::TokenTree(subtree, _) => Some(subtree),
|
||||
_ => None,
|
||||
}
|
||||
}
|
||||
|
||||
/// Parses this attribute as a token tree consisting of comma separated paths.
|
||||
pub fn parse_path_comma_token_tree(&self) -> Option<impl Iterator<Item = ModPath> + '_> {
|
||||
let args = match self.input.as_deref() {
|
||||
Some(AttrInput::TokenTree(args, _)) => args,
|
||||
_ => return None,
|
||||
};
|
||||
let args = self.token_tree_value()?;
|
||||
|
||||
if args.delimiter_kind() != Some(DelimiterKind::Parenthesis) {
|
||||
return None;
|
||||
}
|
||||
let paths = args
|
||||
.token_trees
|
||||
.iter()
|
||||
.group_by(|tt| {
|
||||
matches!(tt, tt::TokenTree::Leaf(tt::Leaf::Punct(Punct { char: ',', .. })))
|
||||
})
|
||||
.into_iter()
|
||||
.filter(|(comma, _)| !*comma)
|
||||
.map(|(_, tts)| {
|
||||
let segments = tts.filter_map(|tt| match tt {
|
||||
.split(|tt| matches!(tt, tt::TokenTree::Leaf(tt::Leaf::Punct(Punct { char: ',', .. }))))
|
||||
.map(|tts| {
|
||||
let segments = tts.iter().filter_map(|tt| match tt {
|
||||
tt::TokenTree::Leaf(tt::Leaf::Ident(id)) => Some(id.as_name()),
|
||||
_ => None,
|
||||
});
|
||||
ModPath::from_segments(PathKind::Plain, segments)
|
||||
})
|
||||
.collect::<Vec<_>>();
|
||||
});
|
||||
|
||||
Some(paths.into_iter())
|
||||
}
|
||||
|
||||
pub fn path(&self) -> &ModPath {
|
||||
&self.path
|
||||
}
|
||||
|
||||
pub fn string_value(&self) -> Option<&SmolStr> {
|
||||
match self.input.as_deref()? {
|
||||
AttrInput::Literal(it) => Some(it),
|
||||
_ => None,
|
||||
}
|
||||
Some(paths)
|
||||
}
|
||||
}
|
||||
|
||||
@ -823,17 +833,11 @@ pub struct AttrQuery<'attr> {
|
||||
|
||||
impl<'attr> AttrQuery<'attr> {
|
||||
pub fn tt_values(self) -> impl Iterator<Item = &'attr Subtree> {
|
||||
self.attrs().filter_map(|attr| match attr.input.as_deref()? {
|
||||
AttrInput::TokenTree(it, _) => Some(it),
|
||||
_ => None,
|
||||
})
|
||||
self.attrs().filter_map(|attr| attr.token_tree_value())
|
||||
}
|
||||
|
||||
pub fn string_value(self) -> Option<&'attr SmolStr> {
|
||||
self.attrs().find_map(|attr| match attr.input.as_deref()? {
|
||||
AttrInput::Literal(it) => Some(it),
|
||||
_ => None,
|
||||
})
|
||||
self.attrs().find_map(|attr| attr.string_value())
|
||||
}
|
||||
|
||||
pub fn exists(self) -> bool {
|
||||
|
||||
@ -22,10 +22,10 @@ use itertools::Itertools;
|
||||
use la_arena::Idx;
|
||||
use limit::Limit;
|
||||
use rustc_hash::{FxHashMap, FxHashSet};
|
||||
use syntax::ast;
|
||||
use syntax::{ast, SmolStr};
|
||||
|
||||
use crate::{
|
||||
attr::{Attr, AttrId, AttrInput, Attrs},
|
||||
attr::{Attr, AttrId, Attrs},
|
||||
attr_macro_as_call_id,
|
||||
db::DefDatabase,
|
||||
derive_macro_as_call_id,
|
||||
@ -61,7 +61,8 @@ pub(super) fn collect_defs(db: &dyn DefDatabase, mut def_map: DefMap, tree_id: T
|
||||
|
||||
let mut deps = FxHashMap::default();
|
||||
// populate external prelude and dependency list
|
||||
for dep in &crate_graph[def_map.krate].dependencies {
|
||||
let krate = &crate_graph[def_map.krate];
|
||||
for dep in &krate.dependencies {
|
||||
tracing::debug!("crate dep {:?} -> {:?}", dep.name, dep.crate_id);
|
||||
let dep_def_map = db.crate_def_map(dep.crate_id);
|
||||
let dep_root = dep_def_map.module_id(dep_def_map.root);
|
||||
@ -73,9 +74,9 @@ pub(super) fn collect_defs(db: &dyn DefDatabase, mut def_map: DefMap, tree_id: T
|
||||
}
|
||||
}
|
||||
|
||||
let cfg_options = &crate_graph[def_map.krate].cfg_options;
|
||||
let proc_macros = &crate_graph[def_map.krate].proc_macro;
|
||||
let proc_macros = proc_macros
|
||||
let cfg_options = &krate.cfg_options;
|
||||
let proc_macros = krate
|
||||
.proc_macro
|
||||
.iter()
|
||||
.enumerate()
|
||||
.map(|(idx, it)| {
|
||||
@ -87,6 +88,7 @@ pub(super) fn collect_defs(db: &dyn DefDatabase, mut def_map: DefMap, tree_id: T
|
||||
)
|
||||
})
|
||||
.collect();
|
||||
let is_proc_macro = krate.is_proc_macro;
|
||||
|
||||
let mut collector = DefCollector {
|
||||
db,
|
||||
@ -103,6 +105,7 @@ pub(super) fn collect_defs(db: &dyn DefDatabase, mut def_map: DefMap, tree_id: T
|
||||
from_glob_import: Default::default(),
|
||||
skip_attrs: Default::default(),
|
||||
derive_helpers_in_scope: Default::default(),
|
||||
is_proc_macro,
|
||||
};
|
||||
if tree_id.is_block() {
|
||||
collector.seed_with_inner(tree_id);
|
||||
@ -243,6 +246,7 @@ struct DefCollector<'a> {
|
||||
/// empty when proc. macro support is disabled (in which case we still do name resolution for
|
||||
/// them).
|
||||
proc_macros: Vec<(Name, ProcMacroExpander)>,
|
||||
is_proc_macro: bool,
|
||||
exports_proc_macros: bool,
|
||||
from_glob_import: PerNsGlobImports,
|
||||
/// If we fail to resolve an attribute on a `ModItem`, we fall back to ignoring the attribute.
|
||||
@ -277,27 +281,29 @@ impl DefCollector<'_> {
|
||||
};
|
||||
|
||||
if *attr_name == hir_expand::name![recursion_limit] {
|
||||
if let Some(input) = &attr.input {
|
||||
if let AttrInput::Literal(limit) = &**input {
|
||||
if let Ok(limit) = limit.parse() {
|
||||
self.def_map.recursion_limit = Some(limit);
|
||||
}
|
||||
if let Some(limit) = attr.string_value() {
|
||||
if let Ok(limit) = limit.parse() {
|
||||
self.def_map.recursion_limit = Some(limit);
|
||||
}
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
if *attr_name == hir_expand::name![crate_type] {
|
||||
if let Some("proc-macro") = attr.string_value().map(SmolStr::as_str) {
|
||||
self.is_proc_macro = true;
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
let attr_is_register_like = *attr_name == hir_expand::name![register_attr]
|
||||
|| *attr_name == hir_expand::name![register_tool];
|
||||
if !attr_is_register_like {
|
||||
continue;
|
||||
}
|
||||
|
||||
let registered_name = match attr.input.as_deref() {
|
||||
Some(AttrInput::TokenTree(subtree, _)) => match &*subtree.token_trees {
|
||||
[tt::TokenTree::Leaf(tt::Leaf::Ident(name))] => name.as_name(),
|
||||
_ => continue,
|
||||
},
|
||||
let registered_name = match attr.single_ident_value() {
|
||||
Some(ident) => ident.as_name(),
|
||||
_ => continue,
|
||||
};
|
||||
|
||||
@ -404,8 +410,7 @@ impl DefCollector<'_> {
|
||||
}
|
||||
self.unresolved_imports = unresolved_imports;
|
||||
|
||||
// FIXME: This condition should instead check if this is a `proc-macro` type crate.
|
||||
if self.exports_proc_macros {
|
||||
if self.is_proc_macro {
|
||||
// A crate exporting procedural macros is not allowed to export anything else.
|
||||
//
|
||||
// Additionally, while the proc macro entry points must be `pub`, they are not publicly
|
||||
@ -1555,22 +1560,21 @@ impl ModCollector<'_, '_> {
|
||||
let fn_id =
|
||||
FunctionLoc { container, id: ItemTreeId::new(self.tree_id, id) }.intern(db);
|
||||
|
||||
let is_proc_macro = attrs.parse_proc_macro_decl(&it.name);
|
||||
let vis = match is_proc_macro {
|
||||
Some(proc_macro) => {
|
||||
// FIXME: this should only be done in the root module of `proc-macro` crates, not everywhere
|
||||
let module_id = def_map.module_id(def_map.root());
|
||||
|
||||
self.def_collector.export_proc_macro(
|
||||
proc_macro,
|
||||
ItemTreeId::new(self.tree_id, id),
|
||||
fn_id,
|
||||
module_id,
|
||||
);
|
||||
Visibility::Module(module_id)
|
||||
let vis = resolve_vis(def_map, &self.item_tree[it.visibility]);
|
||||
if self.def_collector.is_proc_macro {
|
||||
if self.module_id == def_map.root {
|
||||
if let Some(proc_macro) = attrs.parse_proc_macro_decl(&it.name) {
|
||||
let crate_root = def_map.module_id(def_map.root);
|
||||
self.def_collector.export_proc_macro(
|
||||
proc_macro,
|
||||
ItemTreeId::new(self.tree_id, id),
|
||||
fn_id,
|
||||
crate_root,
|
||||
);
|
||||
}
|
||||
}
|
||||
None => resolve_vis(def_map, &self.item_tree[it.visibility]),
|
||||
};
|
||||
}
|
||||
|
||||
update_def(self.def_collector, fn_id.into(), &it.name, vis, false);
|
||||
}
|
||||
ModItem::Struct(id) => {
|
||||
@ -2099,6 +2103,7 @@ mod tests {
|
||||
from_glob_import: Default::default(),
|
||||
skip_attrs: Default::default(),
|
||||
derive_helpers_in_scope: Default::default(),
|
||||
is_proc_macro: false,
|
||||
};
|
||||
collector.seed_with_top_level();
|
||||
collector.collect();
|
||||
|
||||
@ -809,6 +809,7 @@ pub macro derive($item:item) {}
|
||||
struct S;
|
||||
|
||||
//- /proc.rs crate:proc
|
||||
#![crate_type="proc-macro"]
|
||||
#[proc_macro_derive(Derive, attributes(helper))]
|
||||
fn derive() {}
|
||||
"#,
|
||||
@ -886,29 +887,30 @@ indirect_macro!();
|
||||
#[test]
|
||||
fn resolves_proc_macros() {
|
||||
check(
|
||||
r"
|
||||
struct TokenStream;
|
||||
r#"
|
||||
#![crate_type="proc-macro"]
|
||||
struct TokenStream;
|
||||
|
||||
#[proc_macro]
|
||||
pub fn function_like_macro(args: TokenStream) -> TokenStream {
|
||||
args
|
||||
}
|
||||
#[proc_macro]
|
||||
pub fn function_like_macro(args: TokenStream) -> TokenStream {
|
||||
args
|
||||
}
|
||||
|
||||
#[proc_macro_attribute]
|
||||
pub fn attribute_macro(_args: TokenStream, item: TokenStream) -> TokenStream {
|
||||
item
|
||||
}
|
||||
#[proc_macro_attribute]
|
||||
pub fn attribute_macro(_args: TokenStream, item: TokenStream) -> TokenStream {
|
||||
item
|
||||
}
|
||||
|
||||
#[proc_macro_derive(DummyTrait)]
|
||||
pub fn derive_macro(_item: TokenStream) -> TokenStream {
|
||||
TokenStream
|
||||
}
|
||||
#[proc_macro_derive(DummyTrait)]
|
||||
pub fn derive_macro(_item: TokenStream) -> TokenStream {
|
||||
TokenStream
|
||||
}
|
||||
|
||||
#[proc_macro_derive(AnotherTrait, attributes(helper_attr))]
|
||||
pub fn derive_macro_2(_item: TokenStream) -> TokenStream {
|
||||
TokenStream
|
||||
}
|
||||
",
|
||||
#[proc_macro_derive(AnotherTrait, attributes(helper_attr))]
|
||||
pub fn derive_macro_2(_item: TokenStream) -> TokenStream {
|
||||
TokenStream
|
||||
}
|
||||
"#,
|
||||
expect![[r#"
|
||||
crate
|
||||
AnotherTrait: m
|
||||
@ -927,33 +929,34 @@ fn proc_macro_censoring() {
|
||||
// Make sure that only proc macros are publicly exported from proc-macro crates.
|
||||
|
||||
check(
|
||||
r"
|
||||
//- /main.rs crate:main deps:macros
|
||||
pub use macros::*;
|
||||
r#"
|
||||
//- /main.rs crate:main deps:macros
|
||||
pub use macros::*;
|
||||
|
||||
//- /macros.rs crate:macros
|
||||
pub struct TokenStream;
|
||||
//- /macros.rs crate:macros
|
||||
#![crate_type="proc-macro"]
|
||||
pub struct TokenStream;
|
||||
|
||||
#[proc_macro]
|
||||
pub fn function_like_macro(args: TokenStream) -> TokenStream {
|
||||
args
|
||||
}
|
||||
#[proc_macro]
|
||||
pub fn function_like_macro(args: TokenStream) -> TokenStream {
|
||||
args
|
||||
}
|
||||
|
||||
#[proc_macro_attribute]
|
||||
pub fn attribute_macro(_args: TokenStream, item: TokenStream) -> TokenStream {
|
||||
item
|
||||
}
|
||||
#[proc_macro_attribute]
|
||||
pub fn attribute_macro(_args: TokenStream, item: TokenStream) -> TokenStream {
|
||||
item
|
||||
}
|
||||
|
||||
#[proc_macro_derive(DummyTrait)]
|
||||
pub fn derive_macro(_item: TokenStream) -> TokenStream {
|
||||
TokenStream
|
||||
}
|
||||
#[proc_macro_derive(DummyTrait)]
|
||||
pub fn derive_macro(_item: TokenStream) -> TokenStream {
|
||||
TokenStream
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! mbe {
|
||||
() => {};
|
||||
}
|
||||
",
|
||||
#[macro_export]
|
||||
macro_rules! mbe {
|
||||
() => {};
|
||||
}
|
||||
"#,
|
||||
expect![[r#"
|
||||
crate
|
||||
DummyTrait: m
|
||||
@ -966,14 +969,15 @@ fn proc_macro_censoring() {
|
||||
#[test]
|
||||
fn collects_derive_helpers() {
|
||||
let def_map = compute_crate_def_map(
|
||||
r"
|
||||
struct TokenStream;
|
||||
r#"
|
||||
#![crate_type="proc-macro"]
|
||||
struct TokenStream;
|
||||
|
||||
#[proc_macro_derive(AnotherTrait, attributes(helper_attr))]
|
||||
pub fn derive_macro_2(_item: TokenStream) -> TokenStream {
|
||||
TokenStream
|
||||
}
|
||||
",
|
||||
#[proc_macro_derive(AnotherTrait, attributes(helper_attr))]
|
||||
pub fn derive_macro_2(_item: TokenStream) -> TokenStream {
|
||||
TokenStream
|
||||
}
|
||||
"#,
|
||||
);
|
||||
|
||||
assert_eq!(def_map.exported_derives.len(), 1);
|
||||
|
||||
@ -268,6 +268,7 @@ pub mod known {
|
||||
bench,
|
||||
cfg_accessible,
|
||||
cfg_eval,
|
||||
crate_type,
|
||||
derive,
|
||||
global_allocator,
|
||||
test,
|
||||
|
||||
@ -1244,6 +1244,7 @@ use mac::fn_macro;
|
||||
fn_macro$0!();
|
||||
|
||||
//- /mac.rs crate:mac
|
||||
#![crate_type="proc-macro"]
|
||||
#[proc_macro]
|
||||
fn fn_macro() {}
|
||||
//^^^^^^^^
|
||||
|
||||
@ -231,6 +231,7 @@ impl Analysis {
|
||||
cfg_options,
|
||||
Env::default(),
|
||||
Default::default(),
|
||||
false,
|
||||
Default::default(),
|
||||
);
|
||||
change.change_file(file_id, Some(Arc::new(text)));
|
||||
|
||||
@ -1501,16 +1501,15 @@ fn f() {
|
||||
check(
|
||||
r#"
|
||||
//- proc_macros: identity
|
||||
|
||||
#[proc_macros::identity]
|
||||
fn func$0() {
|
||||
func();
|
||||
}
|
||||
"#,
|
||||
expect![[r#"
|
||||
func Function FileId(0) 26..51 29..33
|
||||
func Function FileId(0) 25..50 28..32
|
||||
|
||||
FileId(0) 42..46
|
||||
FileId(0) 41..45
|
||||
"#]],
|
||||
)
|
||||
}
|
||||
@ -1555,11 +1554,12 @@ fn func() {}
|
||||
);
|
||||
check(
|
||||
r#"
|
||||
#![crate_type="proc-macro"]
|
||||
#[proc_macro_attribute]
|
||||
fn func$0() {}
|
||||
"#,
|
||||
expect![[r#"
|
||||
func Attribute FileId(0) 0..36 27..31
|
||||
func Attribute FileId(0) 28..64 55..59
|
||||
|
||||
(no references)
|
||||
"#]],
|
||||
@ -1604,11 +1604,12 @@ struct Foo;
|
||||
);
|
||||
check(
|
||||
r#"
|
||||
#![crate_type="proc-macro"]
|
||||
#[proc_macro_derive(Derive, attributes(x))]
|
||||
pub fn deri$0ve(_stream: TokenStream) -> TokenStream {}
|
||||
"#,
|
||||
expect![[r#"
|
||||
derive Derive FileId(0) 0..97 51..57
|
||||
derive Derive FileId(0) 28..125 79..85
|
||||
|
||||
(no references)
|
||||
"#]],
|
||||
|
||||
@ -34,6 +34,7 @@ pub(crate) fn shuffle_crate_graph(db: &mut RootDatabase) {
|
||||
data.potential_cfg_options.clone(),
|
||||
data.env.clone(),
|
||||
data.proc_macro.clone(),
|
||||
data.is_proc_macro,
|
||||
data.origin.clone(),
|
||||
);
|
||||
map.insert(old_id, new_id);
|
||||
|
||||
@ -176,6 +176,7 @@ fn cargo_hello_world_project_model_with_wildcard_overrides() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
2,
|
||||
@ -248,6 +249,7 @@ fn cargo_hello_world_project_model_with_wildcard_overrides() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
4,
|
||||
@ -312,6 +314,7 @@ fn cargo_hello_world_project_model_with_wildcard_overrides() {
|
||||
"https://github.com/rust-lang/libc",
|
||||
),
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
1,
|
||||
@ -384,6 +387,7 @@ fn cargo_hello_world_project_model_with_wildcard_overrides() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
3,
|
||||
@ -456,6 +460,7 @@ fn cargo_hello_world_project_model_with_wildcard_overrides() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
},
|
||||
}"#]],
|
||||
@ -543,6 +548,7 @@ fn cargo_hello_world_project_model_with_selective_overrides() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
2,
|
||||
@ -617,6 +623,7 @@ fn cargo_hello_world_project_model_with_selective_overrides() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
4,
|
||||
@ -681,6 +688,7 @@ fn cargo_hello_world_project_model_with_selective_overrides() {
|
||||
"https://github.com/rust-lang/libc",
|
||||
),
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
1,
|
||||
@ -755,6 +763,7 @@ fn cargo_hello_world_project_model_with_selective_overrides() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
3,
|
||||
@ -829,6 +838,7 @@ fn cargo_hello_world_project_model_with_selective_overrides() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
},
|
||||
}"#]],
|
||||
@ -907,6 +917,7 @@ fn cargo_hello_world_project_model() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
2,
|
||||
@ -981,6 +992,7 @@ fn cargo_hello_world_project_model() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
4,
|
||||
@ -1047,6 +1059,7 @@ fn cargo_hello_world_project_model() {
|
||||
"https://github.com/rust-lang/libc",
|
||||
),
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
1,
|
||||
@ -1121,6 +1134,7 @@ fn cargo_hello_world_project_model() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
3,
|
||||
@ -1195,6 +1209,7 @@ fn cargo_hello_world_project_model() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
},
|
||||
}"#]],
|
||||
@ -1247,6 +1262,7 @@ fn rust_project_hello_world_project_model() {
|
||||
],
|
||||
proc_macro: [],
|
||||
origin: Lang,
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
10,
|
||||
@ -1276,6 +1292,7 @@ fn rust_project_hello_world_project_model() {
|
||||
dependencies: [],
|
||||
proc_macro: [],
|
||||
origin: Lang,
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
7,
|
||||
@ -1305,6 +1322,7 @@ fn rust_project_hello_world_project_model() {
|
||||
dependencies: [],
|
||||
proc_macro: [],
|
||||
origin: Lang,
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
4,
|
||||
@ -1344,6 +1362,7 @@ fn rust_project_hello_world_project_model() {
|
||||
],
|
||||
proc_macro: [],
|
||||
origin: Lang,
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
1,
|
||||
@ -1373,6 +1392,7 @@ fn rust_project_hello_world_project_model() {
|
||||
dependencies: [],
|
||||
proc_macro: [],
|
||||
origin: Lang,
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
11,
|
||||
@ -1441,6 +1461,7 @@ fn rust_project_hello_world_project_model() {
|
||||
origin: CratesIo {
|
||||
repo: None,
|
||||
},
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
8,
|
||||
@ -1470,6 +1491,7 @@ fn rust_project_hello_world_project_model() {
|
||||
dependencies: [],
|
||||
proc_macro: [],
|
||||
origin: Lang,
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
5,
|
||||
@ -1499,6 +1521,7 @@ fn rust_project_hello_world_project_model() {
|
||||
dependencies: [],
|
||||
proc_macro: [],
|
||||
origin: Lang,
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
2,
|
||||
@ -1528,6 +1551,7 @@ fn rust_project_hello_world_project_model() {
|
||||
dependencies: [],
|
||||
proc_macro: [],
|
||||
origin: Lang,
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
9,
|
||||
@ -1557,6 +1581,7 @@ fn rust_project_hello_world_project_model() {
|
||||
dependencies: [],
|
||||
proc_macro: [],
|
||||
origin: Lang,
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
6,
|
||||
@ -1668,6 +1693,7 @@ fn rust_project_hello_world_project_model() {
|
||||
],
|
||||
proc_macro: [],
|
||||
origin: Lang,
|
||||
is_proc_macro: false,
|
||||
},
|
||||
CrateId(
|
||||
3,
|
||||
@ -1697,6 +1723,7 @@ fn rust_project_hello_world_project_model() {
|
||||
dependencies: [],
|
||||
proc_macro: [],
|
||||
origin: Lang,
|
||||
is_proc_macro: false,
|
||||
},
|
||||
},
|
||||
}"#]],
|
||||
|
||||
@ -483,6 +483,7 @@ fn project_json_to_crate_graph(
|
||||
cfg_options,
|
||||
env,
|
||||
proc_macro.unwrap_or_default(),
|
||||
krate.is_proc_macro,
|
||||
if krate.display_name.is_some() {
|
||||
CrateOrigin::CratesIo { repo: krate.repository.clone() }
|
||||
} else {
|
||||
@ -592,6 +593,7 @@ fn cargo_to_crate_graph(
|
||||
&mut |path| load_proc_macro(&cargo[tgt].name, path),
|
||||
file_id,
|
||||
&cargo[tgt].name,
|
||||
cargo[tgt].is_proc_macro,
|
||||
);
|
||||
if cargo[tgt].kind == TargetKind::Lib {
|
||||
lib_tgt = Some((crate_id, cargo[tgt].name.clone()));
|
||||
@ -707,6 +709,7 @@ fn detached_files_to_crate_graph(
|
||||
cfg_options.clone(),
|
||||
Env::default(),
|
||||
Vec::new(),
|
||||
false,
|
||||
CrateOrigin::Unknown,
|
||||
);
|
||||
|
||||
@ -759,6 +762,7 @@ fn handle_rustc_crates(
|
||||
&mut |path| load_proc_macro(&rustc_workspace[tgt].name, path),
|
||||
file_id,
|
||||
&rustc_workspace[tgt].name,
|
||||
rustc_workspace[tgt].is_proc_macro,
|
||||
);
|
||||
pkg_to_lib_crate.insert(pkg, crate_id);
|
||||
// Add dependencies on core / std / alloc for this crate
|
||||
@ -813,6 +817,7 @@ fn add_target_crate_root(
|
||||
load_proc_macro: &mut dyn FnMut(&AbsPath) -> Vec<ProcMacro>,
|
||||
file_id: FileId,
|
||||
cargo_name: &str,
|
||||
is_proc_macro: bool,
|
||||
) -> CrateId {
|
||||
let edition = pkg.edition;
|
||||
let cfg_options = {
|
||||
@ -857,6 +862,7 @@ fn add_target_crate_root(
|
||||
potential_cfg_options,
|
||||
env,
|
||||
proc_macro,
|
||||
is_proc_macro,
|
||||
CrateOrigin::CratesIo { repo: pkg.repository.clone() },
|
||||
)
|
||||
}
|
||||
@ -901,6 +907,7 @@ fn sysroot_to_crate_graph(
|
||||
cfg_options.clone(),
|
||||
env,
|
||||
proc_macro,
|
||||
false,
|
||||
CrateOrigin::Lang,
|
||||
);
|
||||
Some((krate, crate_id))
|
||||
|
||||
Loading…
Reference in New Issue
Block a user