compiler-construction - 如何在编译器插件中修改 crate 的所有项目？

Question

我正在尝试构建一个将属性扩展为调用的语法扩展。前：

#[flame]
fn flamed() {
    ..
}

后：

fn flamed() {
    flame::start_guard("flamed");
    ..
}

这已经奏效了。但是，如果我有#[flame]箱子级别的属性（如#![flame]），我也希望它能够工作。这可能吗？如果可以，怎么做？

Answer 1

@huon 的评论

您是否尝试在 github.com/lllogiq/flamer/blob/... 中捕获 ItemKind::Mod 并迭代其内容（递归）？

恰到好处——我刚刚添加了一个提交，通过遍历它们来处理 mod 和 trait 项目。我也可能会在 walk 函数中添加代码来处理内部项目和 fns。

代码如下所示：

fn flame_item(i: &Item) -> Item {
    let base = i.clone();
    Item {
        node: match i.node {
            ItemKind::Mod(ref m) =>
                ItemKind::Mod(flame_mod(m)),
            ItemKind::Trait(unsafety, ref generic, ref bounds, ref tis) =>
                ItemKind::Trait(unsafety,
                                generic.clone(),
                                bounds.clone(),
                                flame_items(tis)),
        .. // other item types as usual: items, traitimpls, implitems

            _ => return base
        },
        ..base
    }
}

fn flame_mod(m: &Mod) -> Mod {
    Mod {
        inner: m.inner,
        items: m.items.iter().map(|i| P(flame_item(i))).collect()
    }
}

fn flame_items(items: &[TraitItem]) -> Vec<TraitItem> {
    items.iter().map(flame_trait_item).collect()
}