Compiler construction 在编写语法扩展时,是否可以查找有关注释类型以外的类型的信息?
我想写一个语法扩展,它在生成新函数时结合了相关类型的信息。作为一个无意义的例子,假设我有以下代码:Compiler construction 在编写语法扩展时,是否可以查找有关注释类型以外的类型的信息?,compiler-construction,rust,rust-compiler-plugin,Compiler Construction,Rust,Rust Compiler Plugin,我想写一个语法扩展,它在生成新函数时结合了相关类型的信息。作为一个无意义的例子,假设我有以下代码: struct Monster { health: u8, } impl Monster { fn health(&self) { self.health } } #[attack(Monster)] struct Player { has_weapon: true, } 我希望将攻击属性扩展为了解怪物方法的函数。一个简单的例子是 impl Player {
struct Monster {
health: u8,
}
impl Monster {
fn health(&self) { self.health }
}
#[attack(Monster)]
struct Player {
has_weapon: true,
}
我希望将攻击
属性扩展为了解怪物
方法的函数。一个简单的例子是
impl Player {
fn attack_monster(m: &Monster) {
println!("{}", m.health());
}
}
具体来说,我希望能够获得类型的固有方法的函数签名。我也可以确定一个特征的功能特征。重要的区别在于,我的扩展名不提前知道要查找哪种类型或特征-它将由用户作为注释的参数提供
我目前有一个修饰类型的语法扩展,因为我想添加方法。为此,我实现了一个多项目装饰器
。查看expand
函数的参数后,我无法找到任何查找类型或特征的方法,只能找到生成全新类型或特征的方法:
trait MultiItemDecorator {
fn expand(&self,
ctx: &mut ExtCtxt,
sp: Span,
meta_item: &MetaItem,
item: &Annotatable,
push: &mut FnMut(Annotatable));
}
类型信息不适用于语法扩展。它们可用于lint插件 但是,您可以为您的
impl Monster
编写另一个decorator来自己获取类型
例如:
#![feature(plugin_registrar, rustc_private)]
extern crate rustc;
extern crate syntax;
use rustc::plugin::Registry;
use syntax::ast::MetaItem;
use syntax::ast::Item_::ItemImpl;
use syntax::ast::MetaItem_::{MetaList, MetaWord};
use syntax::codemap::Span;
use syntax::ext::base::{Annotatable, ExtCtxt};
use syntax::ext::base::Annotatable::Item;
use syntax::ext::base::SyntaxExtension::MultiDecorator;
use syntax::parse::token::intern;
use std::collections::hash_map::HashMap;
use std::collections::hash_set::HashSet;
use std::mem;
type Structs = HashMap<String, HashSet<String>>;
fn singleton() -> &'static mut Structs {
static mut hash_set: *mut Structs = 0 as *mut Structs;
let set: Structs = HashMap::new();
unsafe {
if hash_set == 0 as *mut Structs {
hash_set = mem::transmute(Box::new(set));
}
&mut *hash_set
}
}
fn expand_attack(cx: &mut ExtCtxt, sp: Span, meta_item: &MetaItem, _: &Annotatable, _: &mut FnMut(Annotatable)) {
let structs = singleton();
if let MetaList(_, ref items) = meta_item.node {
if let MetaWord(ref word) = items[0].node {
let struct_name = word.to_string();
if let Some(ref methods) = structs.get(&struct_name) {
if let Some(method_name) = methods.iter().next() {
cx.span_warn(sp, &format!("{}.{}()", struct_name, method_name));
// TODO: generate the impl.
}
}
}
}
}
fn expand_register(_: &mut ExtCtxt, _: Span, _: &MetaItem, item: &Annotatable, _: &mut FnMut(Annotatable)) {
let mut structs = singleton();
if let &Annotatable::Item(ref item) = item {
let name = item.ident.to_string();
if let ItemImpl(_, _, _, _, _, ref items) = item.node {
let mut methods = HashSet::new();
for item in items {
methods.insert(item.ident.to_string());
}
structs.insert(name, methods);
}
}
}
#[plugin_registrar]
pub fn plugin_register(reg: &mut Registry) {
reg.register_syntax_extension(intern("attack"), MultiDecorator(Box::new(expand_attack)));
reg.register_syntax_extension(intern("register"), MultiDecorator(Box::new(expand_register)));
}
这与我在本文中所做的类似,我仍在寻找更好的方式来共享这个全局可变状态(
singleton
)。谢谢!不幸的是,我希望能够将其用于无法修改以添加注释的类型,例如其他板条箱或标准库中的类型。@Shepmaster:我不会等待。你就是不能按你的要求去做。通常,当我编写宏来生成必须与其他类型交互的代码时,我要么使用宏的许多变体来处理不同的类型,要么直接将必要的信息传递给宏。
#[register]
impl Monster {
fn health(&self) -> u8 { self.health }
}