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::{ConstParamId, LifetimeParamId, TraitId, TypeAliasId, TypeParamId};
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;
to_chalk(self, db: &dyn HirDatabase) -> Self::Chalk18 fn to_chalk(self, db: &dyn HirDatabase) -> Self::Chalk;
from_chalk(db: &dyn HirDatabase, chalk: Self::Chalk) -> Self19 fn from_chalk(db: &dyn HirDatabase, chalk: Self::Chalk) -> Self;
20 }
21
from_chalk<T, ChalkT>(db: &dyn HirDatabase, chalk: ChalkT) -> T where T: ToChalk<Chalk = ChalkT>,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
to_chalk(self, _db: &dyn HirDatabase) -> chalk_db::ImplId32 fn to_chalk(self, _db: &dyn HirDatabase) -> chalk_db::ImplId {
33 chalk_ir::ImplId(self.as_intern_id())
34 }
35
from_chalk(_db: &dyn HirDatabase, impl_id: chalk_db::ImplId) -> hir_def::ImplId36 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
to_chalk(self, db: &dyn HirDatabase) -> FnDefId44 fn to_chalk(self, db: &dyn HirDatabase) -> FnDefId {
45 db.intern_callable_def(self).into()
46 }
47
from_chalk(db: &dyn HirDatabase, fn_def_id: FnDefId) -> CallableDefId48 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
to_chalk(self, _db: &dyn HirDatabase) -> chalk_db::AssociatedTyValueId58 fn to_chalk(self, _db: &dyn HirDatabase) -> chalk_db::AssociatedTyValueId {
59 rust_ir::AssociatedTyValueId(self.0.as_intern_id())
60 }
61
from_chalk( _db: &dyn HirDatabase, assoc_ty_value_id: chalk_db::AssociatedTyValueId, ) -> TypeAliasAsValue62 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 {
from(fn_def_id: FnDefId) -> Self71 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 {
from(callable_def_id: crate::db::InternedCallableDefId) -> Self77 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 {
from(id: OpaqueTyId) -> Self83 fn from(id: OpaqueTyId) -> Self {
84 InternKey::from_intern_id(id.0)
85 }
86 }
87
88 impl From<crate::db::InternedOpaqueTyId> for OpaqueTyId {
from(id: crate::db::InternedOpaqueTyId) -> Self89 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 {
from(id: chalk_ir::ClosureId<Interner>) -> Self95 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> {
from(id: crate::db::InternedClosureId) -> Self101 fn from(id: crate::db::InternedClosureId) -> Self {
102 chalk_ir::ClosureId(id.as_intern_id())
103 }
104 }
105
to_foreign_def_id(id: TypeAliasId) -> ForeignDefId106 pub fn to_foreign_def_id(id: TypeAliasId) -> ForeignDefId {
107 chalk_ir::ForeignDefId(salsa::InternKey::as_intern_id(&id))
108 }
109
from_foreign_def_id(id: ForeignDefId) -> TypeAliasId110 pub fn from_foreign_def_id(id: ForeignDefId) -> TypeAliasId {
111 salsa::InternKey::from_intern_id(id.0)
112 }
113
to_assoc_type_id(id: TypeAliasId) -> AssocTypeId114 pub fn to_assoc_type_id(id: TypeAliasId) -> AssocTypeId {
115 chalk_ir::AssocTypeId(salsa::InternKey::as_intern_id(&id))
116 }
117
from_assoc_type_id(id: AssocTypeId) -> TypeAliasId118 pub fn from_assoc_type_id(id: AssocTypeId) -> TypeAliasId {
119 salsa::InternKey::from_intern_id(id.0)
120 }
121
from_placeholder_idx(db: &dyn HirDatabase, idx: PlaceholderIndex) -> TypeParamId122 pub fn from_placeholder_idx(db: &dyn HirDatabase, idx: PlaceholderIndex) -> TypeParamId {
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_param_id(interned_id)
126 }
127
to_placeholder_idx(db: &dyn HirDatabase, id: TypeParamId) -> PlaceholderIndex128 pub fn to_placeholder_idx(db: &dyn HirDatabase, id: TypeParamId) -> PlaceholderIndex {
129 let interned_id = db.intern_type_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
lt_from_placeholder_idx(db: &dyn HirDatabase, idx: PlaceholderIndex) -> LifetimeParamId136 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
const_from_placeholder_idx(db: &dyn HirDatabase, idx: PlaceholderIndex) -> ConstParamId142 pub fn const_from_placeholder_idx(db: &dyn HirDatabase, idx: PlaceholderIndex) -> ConstParamId {
143 assert_eq!(idx.ui, chalk_ir::UniverseIndex::ROOT);
144 let interned_id = salsa::InternKey::from_intern_id(salsa::InternId::from(idx.idx));
145 db.lookup_intern_const_param_id(interned_id)
146 }
147
to_chalk_trait_id(id: TraitId) -> ChalkTraitId148 pub fn to_chalk_trait_id(id: TraitId) -> ChalkTraitId {
149 chalk_ir::TraitId(salsa::InternKey::as_intern_id(&id))
150 }
151
from_chalk_trait_id(id: ChalkTraitId) -> TraitId152 pub fn from_chalk_trait_id(id: ChalkTraitId) -> TraitId {
153 salsa::InternKey::from_intern_id(id.0)
154 }
155