src/test/ui-fulldeps/auxiliary/issue-40001-plugin.rs

   1 #![feature(box_syntax, plugin, plugin_registrar, rustc_private)]
   2 #![crate_type = "dylib"]
   3
   4 extern crate rustc_ast_pretty;
   5 extern crate rustc_driver;
   6 extern crate rustc_hir;
   7 #[macro_use]
   8 extern crate rustc_lint;
   9 #[macro_use]
  10 extern crate rustc_session;
  11 extern crate rustc_span;
  12 extern crate rustc_ast;
  13
  14 use rustc_ast_pretty::pprust;
  15 use rustc_driver::plugin::Registry;
  16 use rustc_hir as hir;
  17 use rustc_hir::intravisit;
  18 use rustc_hir::Node;
  19 use rustc_lint::{LateContext, LateLintPass, LintArray, LintContext, LintPass};
  20 use rustc_span::source_map;
  21
  22 #[plugin_registrar]
  23 pub fn plugin_registrar(reg: &mut Registry) {
  24     reg.lint_store.register_lints(&[&MISSING_ALLOWED_ATTR]);
  25     reg.lint_store.register_late_pass(|| box MissingAllowedAttrPass);
  26 }
  27
  28 declare_lint! {
  29     MISSING_ALLOWED_ATTR,
  30     Deny,
  31     "Checks for missing `allowed_attr` attribute"
  32 }
  33
  34 declare_lint_pass!(MissingAllowedAttrPass => [MISSING_ALLOWED_ATTR]);
  35
  36 impl<'tcx> LateLintPass<'tcx> for MissingAllowedAttrPass {
  37     fn check_fn(
  38         &mut self,
  39         cx: &LateContext<'tcx>,
  40         _: intravisit::FnKind<'tcx>,
  41         _: &'tcx hir::FnDecl,
  42         _: &'tcx hir::Body,
  43         span: source_map::Span,
  44         id: hir::HirId,
  45     ) {
  46         let item = match cx.tcx.hir().get(id) {
  47             Node::Item(item) => item,
  48             _ => cx.tcx.hir().expect_item(cx.tcx.hir().get_parent_item(id)),
  49         };
  50
  51         let allowed = |attr| pprust::attribute_to_string(attr).contains("allowed_attr");
  52         if !cx.tcx.hir().attrs(item.hir_id()).iter().any(allowed) {
  53             cx.lint(MISSING_ALLOWED_ATTR, |lint| {
  54                 lint.build("Missing 'allowed_attr' attribute").set_span(span).emit()
  55             });
  56         }
  57     }
  58 }