compiler/rustc_interface/src/proc_macro_decls.rs

   1 use rustc_hir as hir;
   2 use rustc_hir::def_id::LocalDefId;
   3 use rustc_middle::ty::query::Providers;
   4 use rustc_middle::ty::TyCtxt;
   5 use rustc_span::symbol::sym;
   6
   7 fn proc_macro_decls_static(tcx: TyCtxt<'_>, (): ()) -> Option<LocalDefId> {
   8     let mut finder = Finder { tcx, decls: None };
   9
  10     for id in tcx.hir().items() {
  11         let attrs = finder.tcx.hir().attrs(id.hir_id());
  12         if finder.tcx.sess.contains_name(attrs, sym::rustc_proc_macro_decls) {
  13             finder.decls = Some(id.def_id);
  14         }
  15     }
  16
  17     finder.decls
  18 }
  19
  20 struct Finder<'tcx> {
  21     tcx: TyCtxt<'tcx>,
  22     decls: Option<hir::def_id::LocalDefId>,
  23 }
  24
  25 pub(crate) fn provide(providers: &mut Providers) {
  26     *providers = Providers { proc_macro_decls_static, ..*providers };
  27 }