charon_lib/transform/
utils.rs

1use crate::ast::*;
2use crate::formatter::{AstFormatter, FmtCtx};
3use crate::pretty::FmtWithCtx;
4use derive_generic_visitor::*;
5use macros::EnumIsA;
6use std::fmt::{self, Debug};
7
8/// Each `GenericArgs` is meant for a corresponding `GenericParams`; this describes which one.
9#[derive(Debug, Clone, Eq, PartialEq, Hash, EnumIsA, Drive, DriveMut)]
10pub enum GenericsSource {
11    /// A top-level item.
12    Item(ItemId),
13    /// A trait method.
14    Method(TraitDeclId, TraitMethodId),
15    /// A trait associated type.
16    TraitType(TraitDeclId, AssocTypeId),
17    /// A builtin item like `Box`.
18    Builtin,
19    /// Some other use of generics outside the main Charon ast.
20    Other,
21}
22
23impl GenericsSource {
24    pub fn item<I: Into<ItemId>>(id: I) -> Self {
25        Self::Item(id.into())
26    }
27
28    /// Return a path that represents the target item.
29    pub fn item_name(&self, translated: &TranslatedCrate, fmt_ctx: &FmtCtx) -> String {
30        match self {
31            GenericsSource::Item(id) => translated.item_name(*id).to_string_with_ctx(fmt_ctx),
32            GenericsSource::Method(trait_id, method_name) => format!(
33                "{}::{method_name}",
34                translated.item_name(*trait_id).to_string_with_ctx(fmt_ctx),
35            ),
36            GenericsSource::TraitType(trait_id, type_id) => {
37                let type_name =
38                    fmt::from_fn(|f| fmt_ctx.format_assoc_type_name(f, *trait_id, *type_id));
39                format!(
40                    "{}::{type_name}",
41                    translated.item_name(*trait_id).to_string_with_ctx(fmt_ctx),
42                )
43            }
44            GenericsSource::Builtin => "<built-in>".to_string(),
45            GenericsSource::Other => "<unknown>".to_string(),
46        }
47    }
48}
49
50impl TypeDeclId {
51    pub fn generics_target(&self) -> GenericsSource {
52        GenericsSource::item(*self)
53    }
54}
55impl FnPtrKind {
56    pub fn generics_target(&self) -> GenericsSource {
57        match self {
58            FnPtrKind::Fun(fun_id) => GenericsSource::item(*fun_id),
59            FnPtrKind::Trait(trait_ref, name) => {
60                GenericsSource::Method(trait_ref.trait_decl_ref.skip_binder.id, *name)
61            }
62        }
63    }
64}
charon_lib/transform/utils.rs

charon_lib/transform/
utils.rs
