compiler/rustc_interface/src/proc_macro_decls.rs

   1 use rustc_hir as hir;
   2 use rustc_hir::def_id::LocalDefId;
   3 use rustc_hir::itemlikevisit::ItemLikeVisitor;
   4 use rustc_middle::ty::query::Providers;
   5 use rustc_middle::ty::TyCtxt;
   6 use rustc_span::symbol::sym;
   7
   8 fn proc_macro_decls_static(tcx: TyCtxt<'_>, (): ()) -> Option<LocalDefId> {
   9     let mut finder = Finder { tcx, decls: None };
  10     tcx.hir().visit_all_item_likes(&mut finder);
  11
  12     finder.decls.map(|id| tcx.hir().local_def_id(id))
  13 }
  14
  15 struct Finder<'tcx> {
  16     tcx: TyCtxt<'tcx>,
  17     decls: Option<hir::HirId>,
  18 }
  19
  20 impl<'v> ItemLikeVisitor<'v> for Finder<'_> {
  21     fn visit_item(&mut self, item: &hir::Item<'_>) {
  22         let attrs = self.tcx.hir().attrs(item.hir_id());
  23         if self.tcx.sess.contains_name(attrs, sym::rustc_proc_macro_decls) {
  24             self.decls = Some(item.hir_id());
  25         }
  26     }
  27
  28     fn visit_trait_item(&mut self, _trait_item: &hir::TraitItem<'_>) {}
  29
  30     fn visit_impl_item(&mut self, _impl_item: &hir::ImplItem<'_>) {}
  31
  32     fn visit_foreign_item(&mut self, _foreign_item: &hir::ForeignItem<'_>) {}
  33 }
  34
  35 pub(crate) fn provide(providers: &mut Providers) {
  36     *providers = Providers { proc_macro_decls_static, ..*providers };
  37 }