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
   8 pub fn find<'tcx>(tcx: TyCtxt<'_, 'tcx, 'tcx>) -> Option<DefId> {
   9     tcx.proc_macro_decls_static(LOCAL_CRATE)
  10 }
  11
  12 fn proc_macro_decls_static<'tcx>(
  13     tcx: TyCtxt<'_, 'tcx, 'tcx>,
  14     cnum: CrateNum,
  15 ) -> Option<DefId> {
  16     assert_eq!(cnum, LOCAL_CRATE);
  17
  18     let mut finder = Finder { decls: None };
  19     tcx.hir().krate().visit_all_item_likes(&mut finder);
  20
  21     finder.decls.map(|id| tcx.hir().local_def_id_from_hir_id(id))
  22 }
  23
  24 struct Finder {
  25     decls: Option<hir::HirId>,
  26 }
  27
  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);
  32         }
  33     }
  34
  35     fn visit_trait_item(&mut self, _trait_item: &hir::TraitItem) {
  36     }
  37
  38     fn visit_impl_item(&mut self, _impl_item: &hir::ImplItem) {
  39     }
  40 }
  41
  42 pub(crate) fn provide(providers: &mut Providers<'_>) {
  43     *providers = Providers {
  44         proc_macro_decls_static,
  45         ..*providers
  46     };
  47 }