bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

Add method to get all attributes on a definition

Browse source
This commit is contained in:
Adwin White 2024-06-28 13:22:15 +08:00
parent 84071e2662
commit 9387b0bad9
5 changed files with 85 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

25
compiler/rustc_smir/src/rustc_smir/context.rs
View file

@ -232,7 +232,7 @@ impl<'tcx> Context for TablesWrapper<'tcx> {
&self,
def_id: stable_mir::DefId,
attr: &[stable_mir::Symbol],
) -> Vec<stable_mir::ty::Attribute> {
) -> Vec<stable_mir::crate_def::Attribute> {
let mut tables = self.0.borrow_mut();
let tcx = tables.tcx;
let did = tables[def_id];
@ -242,7 +242,28 @@ impl<'tcx> Context for TablesWrapper<'tcx> {
.map(|attribute| {
let attr_str = rustc_ast_pretty::pprust::attribute_to_string(attribute);
let span = attribute.span;
stable_mir::ty::Attribute::new(attr_str, span.stable(&mut *tables))
stable_mir::crate_def::Attribute::new(attr_str, span.stable(&mut *tables))
})
.collect()
}
fn get_all_attrs(&self, def_id: stable_mir::DefId) -> Vec<stable_mir::crate_def::Attribute> {
let mut tables = self.0.borrow_mut();
let tcx = tables.tcx;
let did = tables[def_id];
let filter_fn = move |a: &&rustc_ast::ast::Attribute| {
matches!(a.kind, rustc_ast::ast::AttrKind::Normal(_))
};
let attrs_iter = if let Some(did) = did.as_local() {
tcx.hir().attrs(tcx.local_def_id_to_hir_id(did)).iter().filter(filter_fn)
} else {
tcx.item_attrs(did).iter().filter(filter_fn)
};
attrs_iter
.map(|attribute| {
let attr_str = rustc_ast_pretty::pprust::attribute_to_string(attribute);
let span = attribute.span;
stable_mir::crate_def::Attribute::new(attr_str, span.stable(&mut *tables))
})
.collect()
}