rustc
可以加载编译器插件,它是由用户提供的库用来扩充编译器的行为,例如新的语法扩展,lint检查等。
一个插件是带有设计好的用来在rustc
中注册扩展的注册(registrar)函数的一个动态库包装箱。其它包装箱可以使用#![plugin(...)]
属性来装载这个扩展。查看rustc::plugin文档来获取更多关于定义和装载插件的机制。
如果属性存在的话,#![plugin(foo(... args ...))]
传递的参数并不由rustc
自身解释。它们被传递给插件的Registry
args方法。
在绝大多数情况中,一个插件应该只通过#![plugin]
而不通过extern crate
来使用。链接一个插件会将libsyntax
和librustc
加入到你的包装箱的依赖中。基本上你不会希望如此除非你在构建另一个插件。plugin_as_library
lint会检查这些原则。
通常的做法是将插件放到它们自己的包装箱中,与任何那些会被库的调用者使用的macro_rules!
宏或 Rust 代码分开。
插件可以有多种方法来扩展 Rust 的语法。一种语法扩展是宏过程。它们与[普通宏](5.35.Macros 宏.md)的调用方法一样,不过扩展是通过执行任意Rust代码在编译时操作语法树进行的。
让我们写一个实现了罗马数字的插件roman_numerals.rs。
#![crate_type="dylib"]
#![feature(plugin_registrar, rustc_private)]
extern crate syntax;
extern crate rustc;
extern crate rustc_plugin;
use syntax::codemap::Span;
use syntax::parse::token;
use syntax::ast::TokenTree;
use syntax::ext::base::{ExtCtxt, MacResult, DummyResult, MacEager};
use syntax::ext::build::AstBuilder; // trait for expr_usize
use rustc_plugin::Registry;
fn expand_rn(cx: &mut ExtCtxt, sp: Span, args: &[TokenTree])
-> Box<MacResult + 'static> {
static NUMERALS: &'static [(&'static str, usize)] = &[
("M", 1000), ("CM", 900), ("D", 500), ("CD", 400),
("C", 100), ("XC", 90), ("L", 50), ("XL", 40),
("X", 10), ("IX", 9), ("V", 5), ("IV", 4),
("I", 1)];
if args.len() != 1 {
cx.span_err(
sp,
&format!("argument should be a single identifier, but got {} arguments", args.len()));
return DummyResult::any(sp);
}
let text = match args[0] {
TokenTree::Token(_, token::Ident(s, _)) => s.to_string(),
_ => {
cx.span_err(sp, "argument should be a single identifier");
return DummyResult::any(sp);
}
};
let mut text = &*text;
let mut total = 0;
while !text.is_empty() {
match NUMERALS.iter().find(|&&(rn, _)| text.starts_with(rn)) {
Some(&(rn, val)) => {
total += val;
text = &text[rn.len()..];
}
None => {
cx.span_err(sp, "invalid Roman numeral");
return DummyResult::any(sp);
}
}
}
MacEager::expr(cx.expr_usize(sp, total))
}
#[plugin_registrar]
pub fn plugin_registrar(reg: &mut Registry) {
reg.register_macro("rn", expand_rn);
}
我们可以像其它宏那样使用rn!()
:
#![feature(plugin)]
#![plugin(roman_numerals)]
fn main() {
assert_eq!(rn!(MMXV), 2015);
}
与一个简单的fn(&str) -> u32
函数相比的优势有:
除了宏过程,你可以定义新的类derive属性和其它类型的扩展。查看Registry::register_syntax_extension和SyntaxExtension enum。对于更复杂的宏例子,查看regex_macros。
这里提供一些[宏调试的提示](5.35.Macros 宏.md#debugging-macro-code)。
你可以使用syntax::parse来将记号树转换为像表达式这样的更高级的语法元素:
fn expand_foo(cx: &mut ExtCtxt, sp: Span, args: &[TokenTree])
-> Box<MacResult+'static> {
let mut parser = cx.new_parser_from_tts(args);
let expr: P<Expr> = parser.parse_expr();
看完libsyntax解析器代码会给你一个解析基础设施如何工作的感觉。
保留你解析所有的Span,以便更好的报告错误。你可以用Spanned包围你的自定数据结构。
调用ExtCtxt::span_fatal将会立即终止编译。相反最好调用ExtCtxt::span_err并返回DummyResult,这样编译器可以继续并找到更多错误。
为了打印用于调试的语法段,你可以同时使用span_note和syntax::print::pprust::*_to_string。
上面的例子使用AstBuilder::expr_usize产生了一个普通整数。作为一个AstBuilder
特性的额外选择,libsyntax
提供了一个准引用宏的集合。它们并没有文档并且非常边缘化。然而,这些将会是实现一个作为一个普通插件库的改进准引用的好的出发点。
插件可以扩展Rust Lint基础设施来添加额外的代码风格,安全检查等。你可以查看src/test/auxiliary/lint_plugin_test.rs来了解一个完整的例子,我们在这里重现它的核心部分:
#![feature(plugin_registrar)]
#![feature(box_syntax, rustc_private)]
extern crate syntax;
// Load rustc as a plugin to get macros
#[macro_use]
extern crate rustc;
extern crate rustc_plugin;
use rustc::lint::{EarlyContext, LintContext, LintPass, EarlyLintPass,
EarlyLintPassObject, LintArray};
use rustc_plugin::Registry;
use syntax::ast;
declare_lint!(TEST_LINT, Warn, "Warn about items named 'lintme'");
struct Pass;
impl LintPass for Pass {
fn get_lints(&self) -> LintArray {
lint_array!(TEST_LINT)
}
}
impl EarlyLintPass for Pass {
fn check_item(&mut self, cx: &EarlyContext, it: &ast::Item) {
if it.ident.name.as_str() == "lintme" {
cx.span_lint(TEST_LINT, it.span, "item is named 'lintme'");
}
}
}
#[plugin_registrar]
pub fn plugin_registrar(reg: &mut Registry) {
reg.register_early_lint_pass(box Pass as EarlyLintPassObject);
}
那么像这样的代码:
#![plugin(lint_plugin_test)]
fn lintme() { }
将产生一个编译警告:
foo.rs:4:1: 4:16 warning: item is named 'lintme', #[warn(test_lint)] on by default
foo.rs:4 fn lintme() { }
^~~~~~~~~~~~~~~
Lint插件的组件有:
declare_lint!
调用,它定义了Lint结构LintPass
可能会对多个不同的Lint
调用span_lint
,不过它们都需要用get_lints
方法进行注册。Lint过程是语法遍历,不过它们运行在编译的晚期,这时类型信息时可用的。rustc
的内建lint与lint插件使用相同的基础构架,并提供了如何访问类型信息的例子。
由插件定义的语法通常通过属性和插件标识控制,例如,[#[allow(test_lint)]]
,-A test-lint
。这些标识符来自于declare_lint!
的第一个参数,经过合适的大小写和标点转换。
你可以运行rustc -W help foo.rs
来见检查lint列表是否为rustc
所知,包括由foo.rs
加载的插件。