crates/hir-ty/src/mapping.rs

   1 //! This module contains the implementations of the `ToChalk` trait, which
   2 //! handles conversion between our data types and their corresponding types in
   3 //! Chalk (in both directions); plus some helper functions for more specialized
   4 //! conversions.
   5
   6 use chalk_solve::rust_ir;
   7
   8 use base_db::salsa::{self, InternKey};
   9 use hir_def::{LifetimeParamId, TraitId, TypeAliasId, TypeOrConstParamId};
  10
  11 use crate::{
  12     chalk_db, db::HirDatabase, AssocTypeId, CallableDefId, ChalkTraitId, FnDefId, ForeignDefId,
  13     Interner, OpaqueTyId, PlaceholderIndex,
  14 };
  15
  16 pub(crate) trait ToChalk {
  17     type Chalk;
  18     fn to_chalk(self, db: &dyn HirDatabase) -> Self::Chalk;
  19     fn from_chalk(db: &dyn HirDatabase, chalk: Self::Chalk) -> Self;
  20 }
  21
  22 pub(crate) fn from_chalk<T, ChalkT>(db: &dyn HirDatabase, chalk: ChalkT) -> T
  23 where
  24     T: ToChalk<Chalk = ChalkT>,
  25 {
  26     T::from_chalk(db, chalk)
  27 }
  28
  29 impl ToChalk for hir_def::ImplId {
  30     type Chalk = chalk_db::ImplId;
  31
  32     fn to_chalk(self, _db: &dyn HirDatabase) -> chalk_db::ImplId {
  33         chalk_ir::ImplId(self.as_intern_id())
  34     }
  35
  36     fn from_chalk(_db: &dyn HirDatabase, impl_id: chalk_db::ImplId) -> hir_def::ImplId {
  37         InternKey::from_intern_id(impl_id.0)
  38     }
  39 }
  40
  41 impl ToChalk for CallableDefId {
  42     type Chalk = FnDefId;
  43
  44     fn to_chalk(self, db: &dyn HirDatabase) -> FnDefId {
  45         db.intern_callable_def(self).into()
  46     }
  47
  48     fn from_chalk(db: &dyn HirDatabase, fn_def_id: FnDefId) -> CallableDefId {
  49         db.lookup_intern_callable_def(fn_def_id.into())
  50     }
  51 }
  52
  53 pub(crate) struct TypeAliasAsValue(pub(crate) TypeAliasId);
  54
  55 impl ToChalk for TypeAliasAsValue {
  56     type Chalk = chalk_db::AssociatedTyValueId;
  57
  58     fn to_chalk(self, _db: &dyn HirDatabase) -> chalk_db::AssociatedTyValueId {
  59         rust_ir::AssociatedTyValueId(self.0.as_intern_id())
  60     }
  61
  62     fn from_chalk(
  63         _db: &dyn HirDatabase,
  64         assoc_ty_value_id: chalk_db::AssociatedTyValueId,
  65     ) -> TypeAliasAsValue {
  66         TypeAliasAsValue(TypeAliasId::from_intern_id(assoc_ty_value_id.0))
  67     }
  68 }
  69
  70 impl From<FnDefId> for crate::db::InternedCallableDefId {
  71     fn from(fn_def_id: FnDefId) -> Self {
  72         InternKey::from_intern_id(fn_def_id.0)
  73     }
  74 }
  75
  76 impl From<crate::db::InternedCallableDefId> for FnDefId {
  77     fn from(callable_def_id: crate::db::InternedCallableDefId) -> Self {
  78         chalk_ir::FnDefId(callable_def_id.as_intern_id())
  79     }
  80 }
  81
  82 impl From<OpaqueTyId> for crate::db::InternedOpaqueTyId {
  83     fn from(id: OpaqueTyId) -> Self {
  84         InternKey::from_intern_id(id.0)
  85     }
  86 }
  87
  88 impl From<crate::db::InternedOpaqueTyId> for OpaqueTyId {
  89     fn from(id: crate::db::InternedOpaqueTyId) -> Self {
  90         chalk_ir::OpaqueTyId(id.as_intern_id())
  91     }
  92 }
  93
  94 impl From<chalk_ir::ClosureId<Interner>> for crate::db::InternedClosureId {
  95     fn from(id: chalk_ir::ClosureId<Interner>) -> Self {
  96         Self::from_intern_id(id.0)
  97     }
  98 }
  99
 100 impl From<crate::db::InternedClosureId> for chalk_ir::ClosureId<Interner> {
 101     fn from(id: crate::db::InternedClosureId) -> Self {
 102         chalk_ir::ClosureId(id.as_intern_id())
 103     }
 104 }
 105
 106 pub fn to_foreign_def_id(id: TypeAliasId) -> ForeignDefId {
 107     chalk_ir::ForeignDefId(salsa::InternKey::as_intern_id(&id))
 108 }
 109
 110 pub fn from_foreign_def_id(id: ForeignDefId) -> TypeAliasId {
 111     salsa::InternKey::from_intern_id(id.0)
 112 }
 113
 114 pub fn to_assoc_type_id(id: TypeAliasId) -> AssocTypeId {
 115     chalk_ir::AssocTypeId(salsa::InternKey::as_intern_id(&id))
 116 }
 117
 118 pub fn from_assoc_type_id(id: AssocTypeId) -> TypeAliasId {
 119     salsa::InternKey::from_intern_id(id.0)
 120 }
 121
 122 pub fn from_placeholder_idx(db: &dyn HirDatabase, idx: PlaceholderIndex) -> TypeOrConstParamId {
 123     assert_eq!(idx.ui, chalk_ir::UniverseIndex::ROOT);
 124     let interned_id = salsa::InternKey::from_intern_id(salsa::InternId::from(idx.idx));
 125     db.lookup_intern_type_or_const_param_id(interned_id)
 126 }
 127
 128 pub fn to_placeholder_idx(db: &dyn HirDatabase, id: TypeOrConstParamId) -> PlaceholderIndex {
 129     let interned_id = db.intern_type_or_const_param_id(id);
 130     PlaceholderIndex {
 131         ui: chalk_ir::UniverseIndex::ROOT,
 132         idx: salsa::InternKey::as_intern_id(&interned_id).as_usize(),
 133     }
 134 }
 135
 136 pub fn lt_from_placeholder_idx(db: &dyn HirDatabase, idx: PlaceholderIndex) -> LifetimeParamId {
 137     assert_eq!(idx.ui, chalk_ir::UniverseIndex::ROOT);
 138     let interned_id = salsa::InternKey::from_intern_id(salsa::InternId::from(idx.idx));
 139     db.lookup_intern_lifetime_param_id(interned_id)
 140 }
 141
 142 pub fn to_chalk_trait_id(id: TraitId) -> ChalkTraitId {
 143     chalk_ir::TraitId(salsa::InternKey::as_intern_id(&id))
 144 }
 145
 146 pub fn from_chalk_trait_id(id: ChalkTraitId) -> TraitId {
 147     salsa::InternKey::from_intern_id(id.0)
 148 }