src/librustc_interface/proc_macro_decls.rs

   1 use rustc::hir::itemlikevisit::ItemLikeVisitor;
   2 use rustc::hir::def_id::{CrateNum, DefId, LOCAL_CRATE};
   3 use rustc::hir;
   4 use rustc::ty::TyCtxt;
   5 use rustc::ty::query::Providers;
   6 use syntax::attr;
   7 use syntax::symbol::sym;
   8
   9 pub fn find<'tcx>(tcx: TyCtxt<'tcx, 'tcx>) -> Option<DefId> {
  10     tcx.proc_macro_decls_static(LOCAL_CRATE)
  11 }
  12
  13 fn proc_macro_decls_static<'tcx>(
  14     tcx: TyCtxt<'tcx, 'tcx>,
  15     cnum: CrateNum,
  16 ) -> Option<DefId> {
  17     assert_eq!(cnum, LOCAL_CRATE);
  18
  19     let mut finder = Finder { decls: None };
  20     tcx.hir().krate().visit_all_item_likes(&mut finder);
  21
  22     finder.decls.map(|id| tcx.hir().local_def_id_from_hir_id(id))
  23 }
  24
  25 struct Finder {
  26     decls: Option<hir::HirId>,
  27 }
  28
  29 impl<'v> ItemLikeVisitor<'v> for Finder {
  30     fn visit_item(&mut self, item: &hir::Item) {
  31         if attr::contains_name(&item.attrs, sym::rustc_proc_macro_decls) {
  32             self.decls = Some(item.hir_id);
  33         }
  34     }
  35
  36     fn visit_trait_item(&mut self, _trait_item: &hir::TraitItem) {
  37     }
  38
  39     fn visit_impl_item(&mut self, _impl_item: &hir::ImplItem) {
  40     }
  41 }
  42
  43 pub(crate) fn provide(providers: &mut Providers<'_>) {
  44     *providers = Providers {
  45         proc_macro_decls_static,
  46         ..*providers
  47     };
  48 }