diff options
Diffstat (limited to 'crates/ra_hir/src/ty')
-rw-r--r-- | crates/ra_hir/src/ty/infer.rs | 20 | ||||
-rw-r--r-- | crates/ra_hir/src/ty/lower.rs | 22 | ||||
-rw-r--r-- | crates/ra_hir/src/ty/snapshots/tests__infer_associated_const.snap | 14 | ||||
-rw-r--r-- | crates/ra_hir/src/ty/snapshots/tests__infer_const.snap | 14 | ||||
-rw-r--r-- | crates/ra_hir/src/ty/snapshots/tests__infer_static.snap | 16 | ||||
-rw-r--r-- | crates/ra_hir/src/ty/tests.rs | 37 |
6 files changed, 104 insertions, 19 deletions
diff --git a/crates/ra_hir/src/ty/infer.rs b/crates/ra_hir/src/ty/infer.rs index 921130b71..f5f85308c 100644 --- a/crates/ra_hir/src/ty/infer.rs +++ b/crates/ra_hir/src/ty/infer.rs | |||
@@ -393,17 +393,22 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> { | |||
393 | // Attempt to find an impl_item for the type which has a name matching | 393 | // Attempt to find an impl_item for the type which has a name matching |
394 | // the current segment | 394 | // the current segment |
395 | log::debug!("looking for path segment: {:?}", segment); | 395 | log::debug!("looking for path segment: {:?}", segment); |
396 | let item = ty.iterate_impl_items(self.db, |item| match item { | 396 | let item: crate::ModuleDef = ty.iterate_impl_items(self.db, |item| match item { |
397 | crate::ImplItem::Method(func) => { | 397 | crate::ImplItem::Method(func) => { |
398 | let sig = func.signature(self.db); | 398 | let sig = func.signature(self.db); |
399 | if segment.name == *sig.name() { | 399 | if segment.name == *sig.name() { |
400 | return Some(func); | 400 | return Some(func.into()); |
401 | } | 401 | } |
402 | None | 402 | None |
403 | } | 403 | } |
404 | 404 | ||
405 | // TODO: Resolve associated const | 405 | crate::ImplItem::Const(konst) => { |
406 | crate::ImplItem::Const(_) => None, | 406 | let sig = konst.signature(self.db); |
407 | if segment.name == *sig.name() { | ||
408 | return Some(konst.into()); | ||
409 | } | ||
410 | None | ||
411 | } | ||
407 | 412 | ||
408 | // TODO: Resolve associated types | 413 | // TODO: Resolve associated types |
409 | crate::ImplItem::TypeAlias(_) => None, | 414 | crate::ImplItem::TypeAlias(_) => None, |
@@ -477,9 +482,10 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> { | |||
477 | let ty = self.insert_type_vars(ty.apply_substs(substs)); | 482 | let ty = self.insert_type_vars(ty.apply_substs(substs)); |
478 | (ty, Some(var.into())) | 483 | (ty, Some(var.into())) |
479 | } | 484 | } |
480 | TypableDef::TypeAlias(_) | TypableDef::Function(_) | TypableDef::Enum(_) => { | 485 | TypableDef::TypeAlias(_) |
481 | (Ty::Unknown, None) | 486 | | TypableDef::Function(_) |
482 | } | 487 | | TypableDef::Enum(_) |
488 | | TypableDef::Const(_) => (Ty::Unknown, None), | ||
483 | } | 489 | } |
484 | } | 490 | } |
485 | 491 | ||
diff --git a/crates/ra_hir/src/ty/lower.rs b/crates/ra_hir/src/ty/lower.rs index b66b8e4a5..4a9b725d1 100644 --- a/crates/ra_hir/src/ty/lower.rs +++ b/crates/ra_hir/src/ty/lower.rs | |||
@@ -11,6 +11,7 @@ use std::sync::Arc; | |||
11 | use crate::{ | 11 | use crate::{ |
12 | Function, Struct, StructField, Enum, EnumVariant, Path, Name, | 12 | Function, Struct, StructField, Enum, EnumVariant, Path, Name, |
13 | ModuleDef, TypeAlias, | 13 | ModuleDef, TypeAlias, |
14 | Const, | ||
14 | HirDatabase, | 15 | HirDatabase, |
15 | type_ref::TypeRef, | 16 | type_ref::TypeRef, |
16 | name::KnownName, | 17 | name::KnownName, |
@@ -125,6 +126,7 @@ impl Ty { | |||
125 | TypableDef::Enum(e) => e.generic_params(db), | 126 | TypableDef::Enum(e) => e.generic_params(db), |
126 | TypableDef::EnumVariant(var) => var.parent_enum(db).generic_params(db), | 127 | TypableDef::EnumVariant(var) => var.parent_enum(db).generic_params(db), |
127 | TypableDef::TypeAlias(t) => t.generic_params(db), | 128 | TypableDef::TypeAlias(t) => t.generic_params(db), |
129 | TypableDef::Const(_) => GenericParams::default().into(), | ||
128 | }; | 130 | }; |
129 | let parent_param_count = def_generics.count_parent_params(); | 131 | let parent_param_count = def_generics.count_parent_params(); |
130 | substs.extend((0..parent_param_count).map(|_| Ty::Unknown)); | 132 | substs.extend((0..parent_param_count).map(|_| Ty::Unknown)); |
@@ -163,6 +165,7 @@ impl Ty { | |||
163 | TypableDef::Function(_) | 165 | TypableDef::Function(_) |
164 | | TypableDef::Struct(_) | 166 | | TypableDef::Struct(_) |
165 | | TypableDef::Enum(_) | 167 | | TypableDef::Enum(_) |
168 | | TypableDef::Const(_) | ||
166 | | TypableDef::TypeAlias(_) => last, | 169 | | TypableDef::TypeAlias(_) => last, |
167 | TypableDef::EnumVariant(_) => { | 170 | TypableDef::EnumVariant(_) => { |
168 | // the generic args for an enum variant may be either specified | 171 | // the generic args for an enum variant may be either specified |
@@ -197,12 +200,14 @@ pub(crate) fn type_for_def(db: &impl HirDatabase, def: TypableDef, ns: Namespace | |||
197 | (TypableDef::Enum(e), Namespace::Types) => type_for_enum(db, e), | 200 | (TypableDef::Enum(e), Namespace::Types) => type_for_enum(db, e), |
198 | (TypableDef::EnumVariant(v), Namespace::Values) => type_for_enum_variant_constructor(db, v), | 201 | (TypableDef::EnumVariant(v), Namespace::Values) => type_for_enum_variant_constructor(db, v), |
199 | (TypableDef::TypeAlias(t), Namespace::Types) => type_for_type_alias(db, t), | 202 | (TypableDef::TypeAlias(t), Namespace::Types) => type_for_type_alias(db, t), |
203 | (TypableDef::Const(c), Namespace::Values) => type_for_const(db, c), | ||
200 | 204 | ||
201 | // 'error' cases: | 205 | // 'error' cases: |
202 | (TypableDef::Function(_), Namespace::Types) => Ty::Unknown, | 206 | (TypableDef::Function(_), Namespace::Types) => Ty::Unknown, |
203 | (TypableDef::Enum(_), Namespace::Values) => Ty::Unknown, | 207 | (TypableDef::Enum(_), Namespace::Values) => Ty::Unknown, |
204 | (TypableDef::EnumVariant(_), Namespace::Types) => Ty::Unknown, | 208 | (TypableDef::EnumVariant(_), Namespace::Types) => Ty::Unknown, |
205 | (TypableDef::TypeAlias(_), Namespace::Values) => Ty::Unknown, | 209 | (TypableDef::TypeAlias(_), Namespace::Values) => Ty::Unknown, |
210 | (TypableDef::Const(_), Namespace::Types) => Ty::Unknown, | ||
206 | } | 211 | } |
207 | } | 212 | } |
208 | 213 | ||
@@ -233,6 +238,14 @@ fn type_for_fn(db: &impl HirDatabase, def: Function) -> Ty { | |||
233 | Ty::FnDef { def: def.into(), sig, name, substs } | 238 | Ty::FnDef { def: def.into(), sig, name, substs } |
234 | } | 239 | } |
235 | 240 | ||
241 | /// Build the declared type of a const. | ||
242 | fn type_for_const(db: &impl HirDatabase, def: Const) -> Ty { | ||
243 | let signature = def.signature(db); | ||
244 | let resolver = def.resolver(db); | ||
245 | |||
246 | Ty::from_hir(db, &resolver, signature.type_ref()) | ||
247 | } | ||
248 | |||
236 | /// Build the type of a tuple struct constructor. | 249 | /// Build the type of a tuple struct constructor. |
237 | fn type_for_struct_constructor(db: &impl HirDatabase, def: Struct) -> Ty { | 250 | fn type_for_struct_constructor(db: &impl HirDatabase, def: Struct) -> Ty { |
238 | let var_data = def.variant_data(db); | 251 | let var_data = def.variant_data(db); |
@@ -318,8 +331,9 @@ pub enum TypableDef { | |||
318 | Enum(Enum), | 331 | Enum(Enum), |
319 | EnumVariant(EnumVariant), | 332 | EnumVariant(EnumVariant), |
320 | TypeAlias(TypeAlias), | 333 | TypeAlias(TypeAlias), |
334 | Const(Const), | ||
321 | } | 335 | } |
322 | impl_froms!(TypableDef: Function, Struct, Enum, EnumVariant, TypeAlias); | 336 | impl_froms!(TypableDef: Function, Struct, Enum, EnumVariant, TypeAlias, Const); |
323 | 337 | ||
324 | impl From<ModuleDef> for Option<TypableDef> { | 338 | impl From<ModuleDef> for Option<TypableDef> { |
325 | fn from(def: ModuleDef) -> Option<TypableDef> { | 339 | fn from(def: ModuleDef) -> Option<TypableDef> { |
@@ -329,10 +343,8 @@ impl From<ModuleDef> for Option<TypableDef> { | |||
329 | ModuleDef::Enum(e) => e.into(), | 343 | ModuleDef::Enum(e) => e.into(), |
330 | ModuleDef::EnumVariant(v) => v.into(), | 344 | ModuleDef::EnumVariant(v) => v.into(), |
331 | ModuleDef::TypeAlias(t) => t.into(), | 345 | ModuleDef::TypeAlias(t) => t.into(), |
332 | ModuleDef::Const(_) | 346 | ModuleDef::Const(v) => v.into(), |
333 | | ModuleDef::Static(_) | 347 | ModuleDef::Static(_) | ModuleDef::Module(_) | ModuleDef::Trait(_) => return None, |
334 | | ModuleDef::Module(_) | ||
335 | | ModuleDef::Trait(_) => return None, | ||
336 | }; | 348 | }; |
337 | Some(res) | 349 | Some(res) |
338 | } | 350 | } |
diff --git a/crates/ra_hir/src/ty/snapshots/tests__infer_associated_const.snap b/crates/ra_hir/src/ty/snapshots/tests__infer_associated_const.snap index 131d1fa16..51f3fd4c0 100644 --- a/crates/ra_hir/src/ty/snapshots/tests__infer_associated_const.snap +++ b/crates/ra_hir/src/ty/snapshots/tests__infer_associated_const.snap | |||
@@ -1,14 +1,14 @@ | |||
1 | --- | 1 | --- |
2 | created: "2019-02-21T21:51:46.497925200Z" | 2 | created: "2019-02-25T08:36:33.885804400Z" |
3 | creator: [email protected] | 3 | creator: [email protected] |
4 | source: crates/ra_hir/src/ty/tests.rs | 4 | source: crates/ra_hir/src/ty/tests.rs |
5 | expression: "&result" | 5 | expression: "&result" |
6 | --- | 6 | --- |
7 | [227; 305) '{ ...:ID; }': () | 7 | [227; 305) '{ ...:ID; }': () |
8 | [237; 238) 'x': [unknown] | 8 | [237; 238) 'x': u32 |
9 | [241; 252) 'Struct::FOO': [unknown] | 9 | [241; 252) 'Struct::FOO': u32 |
10 | [262; 263) 'y': [unknown] | 10 | [262; 263) 'y': u32 |
11 | [266; 275) 'Enum::BAR': [unknown] | 11 | [266; 275) 'Enum::BAR': u32 |
12 | [285; 286) 'z': [unknown] | 12 | [285; 286) 'z': u32 |
13 | [289; 302) 'TraitTest::ID': [unknown] | 13 | [289; 302) 'TraitTest::ID': u32 |
14 | 14 | ||
diff --git a/crates/ra_hir/src/ty/snapshots/tests__infer_const.snap b/crates/ra_hir/src/ty/snapshots/tests__infer_const.snap new file mode 100644 index 000000000..a5eba3980 --- /dev/null +++ b/crates/ra_hir/src/ty/snapshots/tests__infer_const.snap | |||
@@ -0,0 +1,14 @@ | |||
1 | --- | ||
2 | created: "2019-02-25T07:26:34.115351100Z" | ||
3 | creator: [email protected] | ||
4 | source: crates/ra_hir/src/ty/tests.rs | ||
5 | expression: "&result" | ||
6 | --- | ||
7 | [95; 213) '{ ...NST; }': () | ||
8 | [138; 139) 'x': [unknown] | ||
9 | [142; 153) 'LOCAL_CONST': [unknown] | ||
10 | [163; 164) 'z': u32 | ||
11 | [167; 179) 'GLOBAL_CONST': u32 | ||
12 | [189; 191) 'id': u32 | ||
13 | [194; 210) 'Foo::A..._CONST': u32 | ||
14 | |||
diff --git a/crates/ra_hir/src/ty/snapshots/tests__infer_static.snap b/crates/ra_hir/src/ty/snapshots/tests__infer_static.snap new file mode 100644 index 000000000..ffe9ca66d --- /dev/null +++ b/crates/ra_hir/src/ty/snapshots/tests__infer_static.snap | |||
@@ -0,0 +1,16 @@ | |||
1 | --- | ||
2 | created: "2019-02-25T07:26:41.480764900Z" | ||
3 | creator: [email protected] | ||
4 | source: crates/ra_hir/src/ty/tests.rs | ||
5 | expression: "&result" | ||
6 | --- | ||
7 | [85; 280) '{ ...MUT; }': () | ||
8 | [173; 174) 'x': [unknown] | ||
9 | [177; 189) 'LOCAL_STATIC': [unknown] | ||
10 | [199; 200) 'y': [unknown] | ||
11 | [203; 219) 'LOCAL_...IC_MUT': [unknown] | ||
12 | [229; 230) 'z': [unknown] | ||
13 | [233; 246) 'GLOBAL_STATIC': [unknown] | ||
14 | [256; 257) 'w': [unknown] | ||
15 | [260; 277) 'GLOBAL...IC_MUT': [unknown] | ||
16 | |||
diff --git a/crates/ra_hir/src/ty/tests.rs b/crates/ra_hir/src/ty/tests.rs index 642259225..8de46a29e 100644 --- a/crates/ra_hir/src/ty/tests.rs +++ b/crates/ra_hir/src/ty/tests.rs | |||
@@ -1006,6 +1006,43 @@ mod foo { | |||
1006 | assert_eq!("i128", type_at_pos(&db, pos)); | 1006 | assert_eq!("i128", type_at_pos(&db, pos)); |
1007 | } | 1007 | } |
1008 | 1008 | ||
1009 | #[test] | ||
1010 | fn infer_const() { | ||
1011 | check_inference( | ||
1012 | "infer_const", | ||
1013 | r#" | ||
1014 | struct Foo; | ||
1015 | impl Foo { const ASSOC_CONST: u32 = 0; } | ||
1016 | const GLOBAL_CONST: u32 = 101; | ||
1017 | fn test() { | ||
1018 | const LOCAL_CONST: u32 = 99; | ||
1019 | let x = LOCAL_CONST; | ||
1020 | let z = GLOBAL_CONST; | ||
1021 | let id = Foo::ASSOC_CONST; | ||
1022 | } | ||
1023 | "#, | ||
1024 | ); | ||
1025 | } | ||
1026 | |||
1027 | #[test] | ||
1028 | fn infer_static() { | ||
1029 | check_inference( | ||
1030 | "infer_static", | ||
1031 | r#" | ||
1032 | static GLOBAL_STATIC: u32 = 101; | ||
1033 | static mut GLOBAL_STATIC_MUT: u32 = 101; | ||
1034 | fn test() { | ||
1035 | static LOCAL_STATIC: u32 = 99; | ||
1036 | static mut LOCAL_STATIC_MUT: u32 = 99; | ||
1037 | let x = LOCAL_STATIC; | ||
1038 | let y = LOCAL_STATIC_MUT; | ||
1039 | let z = GLOBAL_STATIC; | ||
1040 | let w = GLOBAL_STATIC_MUT; | ||
1041 | } | ||
1042 | "#, | ||
1043 | ); | ||
1044 | } | ||
1045 | |||
1009 | fn type_at_pos(db: &MockDatabase, pos: FilePosition) -> String { | 1046 | fn type_at_pos(db: &MockDatabase, pos: FilePosition) -> String { |
1010 | let func = source_binder::function_from_position(db, pos).unwrap(); | 1047 | let func = source_binder::function_from_position(db, pos).unwrap(); |
1011 | let body_syntax_mapping = func.body_syntax_mapping(db); | 1048 | let body_syntax_mapping = func.body_syntax_mapping(db); |