Strip unconfigured nodes from decorator-generated AST

This commit is contained in:
Jeffrey Seyfried
2016-06-16 06:44:53 +00:00
parent 83d283b67b
commit c41cf30e9d
3 changed files with 25 additions and 4 deletions

View File

@@ -95,6 +95,16 @@ impl Annotatable {
_ => panic!("expected Item")
}
}
pub fn fold_with<F: Folder>(self, folder: &mut F) -> SmallVector<Self> {
match self {
Annotatable::Item(item) => folder.fold_item(item).map(Annotatable::Item),
Annotatable::ImplItem(item) =>
folder.fold_impl_item(item.unwrap()).map(|item| Annotatable::ImplItem(P(item))),
Annotatable::TraitItem(item) =>
folder.fold_trait_item(item.unwrap()).map(|item| Annotatable::TraitItem(P(item))),
}
}
}
// A more flexible ItemDecorator.