src/test/auxiliary/lint_group_plugin_test.rs

   1 // Copyright 2014 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 // force-host
  12
  13 #![feature(plugin_registrar)]
  14 #![feature(box_syntax, rustc_private)]
  15
  16 extern crate rustc_front;
  17
  18 // Load rustc as a plugin to get macros
  19 #[macro_use]
  20 extern crate rustc;
  21
  22 use rustc_front::hir;
  23 use rustc::lint::{LateContext, LintContext, LintPass, LateLintPass, LateLintPassObject, LintArray};
  24 use rustc::plugin::Registry;
  25
  26 declare_lint!(TEST_LINT, Warn, "Warn about items named 'lintme'");
  27
  28 declare_lint!(PLEASE_LINT, Warn, "Warn about items named 'pleaselintme'");
  29
  30 struct Pass;
  31
  32 impl LintPass for Pass {
  33     fn get_lints(&self) -> LintArray {
  34         lint_array!(TEST_LINT, PLEASE_LINT)
  35     }
  36 }
  37
  38 impl LateLintPass for Pass {
  39     fn check_item(&mut self, cx: &LateContext, it: &hir::Item) {
  40         match &*it.name.as_str() {
  41             "lintme" => cx.span_lint(TEST_LINT, it.span, "item is named 'lintme'"),
  42             "pleaselintme" => cx.span_lint(PLEASE_LINT, it.span, "item is named 'pleaselintme'"),
  43             _ => {}
  44         }
  45     }
  46 }
  47
  48 #[plugin_registrar]
  49 pub fn plugin_registrar(reg: &mut Registry) {
  50     reg.register_late_lint_pass(box Pass as LateLintPassObject);
  51     reg.register_lint_group("lint_me", vec![TEST_LINT, PLEASE_LINT]);
  52 }