src/test/ui-fulldeps/auxiliary/lint-tool-test.rs

   1 #![feature(plugin_registrar)]
   2 #![feature(box_syntax, rustc_private)]
   3
   4 extern crate rustc_ast;
   5
   6 // Load rustc as a plugin to get macros
   7 extern crate rustc_driver;
   8 #[macro_use]
   9 extern crate rustc_lint;
  10 #[macro_use]
  11 extern crate rustc_session;
  12
  13 use rustc_driver::plugin::Registry;
  14 use rustc_lint::{EarlyContext, EarlyLintPass, LintArray, LintContext, LintId, LintPass};
  15 use rustc_ast as ast;
  16 declare_tool_lint!(pub clippy::TEST_LINT, Warn, "Warn about stuff");
  17 declare_tool_lint!(
  18     /// Some docs
  19     pub clippy::TEST_GROUP,
  20     Warn, "Warn about other stuff"
  21 );
  22
  23 declare_tool_lint!(
  24     /// Some docs
  25     pub rustc::TEST_RUSTC_TOOL_LINT,
  26     Deny,
  27     "Deny internal stuff"
  28 );
  29
  30 declare_lint_pass!(Pass => [TEST_LINT, TEST_GROUP, TEST_RUSTC_TOOL_LINT]);
  31
  32 impl EarlyLintPass for Pass {
  33     fn check_item(&mut self, cx: &EarlyContext, it: &ast::Item) {
  34         if it.ident.name.as_str() == "lintme" {
  35             cx.lint(TEST_LINT, |lint| {
  36                 lint.build("item is named 'lintme'").set_span(it.span).emit()
  37             });
  38         }
  39         if it.ident.name.as_str() == "lintmetoo" {
  40             cx.lint(TEST_GROUP, |lint| {
  41                 lint.build("item is named 'lintmetoo'").set_span(it.span).emit()
  42             });
  43         }
  44     }
  45 }
  46
  47 #[plugin_registrar]
  48 pub fn plugin_registrar(reg: &mut Registry) {
  49     reg.lint_store.register_lints(&[&TEST_RUSTC_TOOL_LINT, &TEST_LINT, &TEST_GROUP]);
  50     reg.lint_store.register_early_pass(|| box Pass);
  51     reg.lint_store.register_group(
  52         true,
  53         "clippy::group",
  54         Some("clippy_group"),
  55         vec![LintId::of(&TEST_LINT), LintId::of(&TEST_GROUP)],
  56     );
  57 }