1 use rustc::hir::itemlikevisit::ItemLikeVisitor;
2 use rustc::hir::def_id::{CrateNum, DefId, LOCAL_CRATE};
5 use rustc::ty::query::Providers;
7 use syntax::symbol::sym;
9 pub fn find(tcx: TyCtxt<'_>) -> Option<DefId> {
10 tcx.proc_macro_decls_static(LOCAL_CRATE)
13 fn proc_macro_decls_static(tcx: TyCtxt<'_>, cnum: CrateNum) -> Option<DefId> {
14 assert_eq!(cnum, LOCAL_CRATE);
16 let mut finder = Finder { decls: None };
17 tcx.hir().krate().visit_all_item_likes(&mut finder);
19 finder.decls.map(|id| tcx.hir().local_def_id(id))
23 decls: Option<hir::HirId>,
26 impl<'v> ItemLikeVisitor<'v> for Finder {
27 fn visit_item(&mut self, item: &hir::Item<'_>) {
28 if attr::contains_name(&item.attrs, sym::rustc_proc_macro_decls) {
29 self.decls = Some(item.hir_id);
33 fn visit_trait_item(&mut self, _trait_item: &hir::TraitItem<'_>) {
36 fn visit_impl_item(&mut self, _impl_item: &hir::ImplItem<'_>) {
40 pub(crate) fn provide(providers: &mut Providers<'_>) {
41 *providers = Providers {
42 proc_macro_decls_static,