1 use rustc::hir::itemlikevisit::ItemLikeVisitor;
2 use rustc::hir::def_id::{CrateNum, DefId, LOCAL_CRATE};
5 use rustc::ty::query::Providers;
8 pub fn find<'tcx>(tcx: TyCtxt<'_, 'tcx, 'tcx>) -> Option<DefId> {
9 tcx.proc_macro_decls_static(LOCAL_CRATE)
12 fn proc_macro_decls_static<'tcx>(
13 tcx: TyCtxt<'_, 'tcx, 'tcx>,
16 assert_eq!(cnum, LOCAL_CRATE);
18 let mut finder = Finder { decls: None };
19 tcx.hir().krate().visit_all_item_likes(&mut finder);
21 finder.decls.map(|id| tcx.hir().local_def_id_from_hir_id(id))
25 decls: Option<hir::HirId>,
28 impl<'v> ItemLikeVisitor<'v> for Finder {
29 fn visit_item(&mut self, item: &hir::Item) {
30 if attr::contains_name(&item.attrs, "rustc_proc_macro_decls") {
31 self.decls = Some(item.hir_id);
35 fn visit_trait_item(&mut self, _trait_item: &hir::TraitItem) {
38 fn visit_impl_item(&mut self, _impl_item: &hir::ImplItem) {
42 pub(crate) fn provide(providers: &mut Providers<'_>) {
43 *providers = Providers {
44 proc_macro_decls_static,