mirror of
https://github.com/rust-lang/rust.git
synced 2025-10-30 04:24:26 +00:00
f86f7ad5f2
The end goal is to eliminate `Map` altogether. I added a `hir_` prefix to all of them, that seemed simplest. The exceptions are `module_items` which became `hir_module_free_items` because there was already a `hir_module_items`, and `items` which became `hir_free_items` for consistency with `hir_module_free_items`.
29 lines
954 B
Rust
29 lines
954 B
Rust
use rustc_data_structures::fx::FxIndexMap;
|
|
use rustc_hir as hir;
|
|
use rustc_hir::def::DefKind;
|
|
use rustc_hir::def_id::DefId;
|
|
use rustc_middle::query::LocalCrate;
|
|
use rustc_middle::ty::TyCtxt;
|
|
use rustc_session::cstore::ForeignModule;
|
|
|
|
pub(crate) fn collect(tcx: TyCtxt<'_>, LocalCrate: LocalCrate) -> FxIndexMap<DefId, ForeignModule> {
|
|
let mut modules = FxIndexMap::default();
|
|
|
|
// We need to collect all the `ForeignMod`, even if they are empty.
|
|
for id in tcx.hir_free_items() {
|
|
if !matches!(tcx.def_kind(id.owner_id), DefKind::ForeignMod) {
|
|
continue;
|
|
}
|
|
|
|
let def_id = id.owner_id.to_def_id();
|
|
let item = tcx.hir_item(id);
|
|
|
|
if let hir::ItemKind::ForeignMod { abi, items } = item.kind {
|
|
let foreign_items = items.iter().map(|it| it.id.owner_id.to_def_id()).collect();
|
|
modules.insert(def_id, ForeignModule { def_id, abi, foreign_items });
|
|
}
|
|
}
|
|
|
|
modules
|
|
}
|