frontend 0.4.0

rustc's frontend with no LLVM and no std: parsing through MIR, as a library
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

// `#![no_std]`: these arrive with the standard prelude and name no path, so a `std::`
// search cannot see them - and a `#[derive]` can use them without the name appearing
// in this file at all, which is why they are not trimmed by inspection.
use alloc::borrow::ToOwned;
use alloc::boxed::Box;
use alloc::format;
use alloc::string::{String, ToString};
use alloc::vec;
use alloc::vec::Vec;

use crate::rustc_data_structures::fx::FxIndexMap;
use crate::rustc_hir::attrs::{EiiDecl, EiiImpl, EiiImplResolution};
use crate::rustc_hir::def_id::DefId;
use crate::find_attr;
use crate::bug;
use crate::rustc_middle::query::LocalCrate;
use crate::rustc_middle::ty::TyCtxt;

// basically the map below but flattened out
pub(crate) type EiiMapEncodedKeyValue = (DefId, (EiiDecl, Vec<(DefId, EiiImpl)>));

pub(crate) type EiiMap = FxIndexMap<
    DefId, // the defid of the foreign item associated with the eii
    (
        // the corresponding declaration
        EiiDecl,
        // all the given implementations, indexed by defid.
        // We expect there to be only one, but collect them all to give errors if there are more
        // (or if there are none) in the final crate we build.
        FxIndexMap<DefId, EiiImpl>,
    ),
>;

pub(crate) fn collect<'tcx>(tcx: TyCtxt<'tcx>, LocalCrate: LocalCrate) -> EiiMap {
    let mut eiis = EiiMap::default();

    // Needed because Known only stores a DefId (not the full EiiDecl),
    // and we can't call the externally_implementable_items query (cycle).
    let decls_by_foreign_item: FxIndexMap<DefId, EiiDecl> = tcx
        .hir_crate_items(())
        .eiis()
        .filter_map(|id| find_attr!(tcx, id, EiiDeclaration(d) => *d))
        .map(|decl| (decl.foreign_item, decl))
        .collect();

    // iterate over all items in the current crate
    for id in tcx.hir_crate_items(()).eiis() {
        // if we find a new declaration, add it to the list without a known implementation
        if let Some(decl) = find_attr!(tcx, id, EiiDeclaration(d) => *d) {
            eiis.entry(decl.foreign_item).or_insert((decl, Default::default()));
        }

        if let Some(i) = find_attr!(tcx, id, EiiImpl(i) => i) {
            let (foreign_item, decl) = match i.resolution {
                EiiImplResolution::Macro(macro_defid) => {
                    // find the decl for this one if it wasn't in yet (maybe it's from the local crate? not very useful but not illegal)
                    let Some(decl) = find_attr!(tcx, macro_defid, EiiDeclaration(d) => *d) else {
                        // skip if it doesn't have eii_declaration (if we resolved to another macro that's not an EII)
                        tcx.dcx()
                            .span_delayed_bug(i.span, "resolved to something that's not an EII");
                        continue;
                    };
                    (decl.foreign_item, decl)
                }
                // Recover the EiiDecl from the local lookup map.
                EiiImplResolution::Known(foreign_item_did) => {
                    let decl = decls_by_foreign_item.get(&foreign_item_did).unwrap_or_else(|| {
                        bug!(
                            "EII impl has Known resolution but can't find EiiDeclaration for {:?}",
                            foreign_item_did
                        )
                    });
                    (foreign_item_did, *decl)
                }
                EiiImplResolution::Error(_eg) => continue,
            };

            eiis.entry(foreign_item)
                .or_insert_with(|| (decl, Default::default()))
                .1
                .insert(id.into(), **i);
        }
    }

    eiis
}