rustc_metadata/
foreign_modules.rs

1use rustc_data_structures::fx::FxIndexMap;
2use rustc_hir as hir;
3use rustc_hir::def::DefKind;
4use rustc_hir::def_id::DefId;
5use rustc_middle::query::LocalCrate;
6use rustc_middle::ty::TyCtxt;
7use rustc_session::cstore::ForeignModule;
8
9pub(crate) fn collect(tcx: TyCtxt<'_>, LocalCrate: LocalCrate) -> FxIndexMap<DefId, ForeignModule> {
10    let mut modules = FxIndexMap::default();
11
12    // We need to collect all the `ForeignMod`, even if they are empty.
13    for id in tcx.hir_free_items() {
14        if !matches!(tcx.def_kind(id.owner_id), DefKind::ForeignMod) {
15            continue;
16        }
17
18        let def_id = id.owner_id.to_def_id();
19        let item = tcx.hir_item(id);
20
21        if let hir::ItemKind::ForeignMod { abi, items } = item.kind {
22            let foreign_items = items.iter().map(|it| it.owner_id.to_def_id()).collect();
23            modules.insert(def_id, ForeignModule { def_id, abi, foreign_items });
24        }
25    }
26
27    modules
28}