diff options
Diffstat (limited to 'crates/ra_hir/src')
-rw-r--r-- | crates/ra_hir/src/ty.rs | 35 | ||||
-rw-r--r-- | crates/ra_hir/src/ty/tests.rs | 24 | ||||
-rw-r--r-- | crates/ra_hir/src/ty/tests/data/0008_boolean_op.txt | 31 |
3 files changed, 88 insertions, 2 deletions
diff --git a/crates/ra_hir/src/ty.rs b/crates/ra_hir/src/ty.rs index 8c320a705..b685259d7 100644 --- a/crates/ra_hir/src/ty.rs +++ b/crates/ra_hir/src/ty.rs | |||
@@ -26,7 +26,7 @@ use ena::unify::{InPlaceUnificationTable, UnifyKey, UnifyValue, NoError}; | |||
26 | 26 | ||
27 | use ra_db::{LocalSyntaxPtr, Cancelable}; | 27 | use ra_db::{LocalSyntaxPtr, Cancelable}; |
28 | use ra_syntax::{ | 28 | use ra_syntax::{ |
29 | ast::{self, AstNode, LoopBodyOwner, ArgListOwner, PrefixOp}, | 29 | ast::{self, AstNode, LoopBodyOwner, ArgListOwner, PrefixOp, BinOp}, |
30 | SyntaxNodeRef | 30 | SyntaxNodeRef |
31 | }; | 31 | }; |
32 | 32 | ||
@@ -527,6 +527,20 @@ struct InferenceContext<'a, D: HirDatabase> { | |||
527 | return_ty: Ty, | 527 | return_ty: Ty, |
528 | } | 528 | } |
529 | 529 | ||
530 | // helper function that determines whether a binary operator | ||
531 | // always returns a boolean | ||
532 | fn is_boolean_operator(op: BinOp) -> bool { | ||
533 | match op { | ||
534 | BinOp::BooleanOr | ||
535 | | BinOp::BooleanAnd | ||
536 | | BinOp::EqualityTest | ||
537 | | BinOp::LesserEqualTest | ||
538 | | BinOp::GreaterEqualTest | ||
539 | | BinOp::LesserTest | ||
540 | | BinOp::GreaterTest => true, | ||
541 | } | ||
542 | } | ||
543 | |||
530 | impl<'a, D: HirDatabase> InferenceContext<'a, D> { | 544 | impl<'a, D: HirDatabase> InferenceContext<'a, D> { |
531 | fn new( | 545 | fn new( |
532 | db: &'a D, | 546 | db: &'a D, |
@@ -899,7 +913,24 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> { | |||
899 | } | 913 | } |
900 | } | 914 | } |
901 | ast::Expr::RangeExpr(_e) => Ty::Unknown, | 915 | ast::Expr::RangeExpr(_e) => Ty::Unknown, |
902 | ast::Expr::BinExpr(_e) => Ty::Unknown, | 916 | ast::Expr::BinExpr(e) => match e.op() { |
917 | Some(op) => { | ||
918 | let subtype_expectation = match op { | ||
919 | BinOp::BooleanAnd | BinOp::BooleanOr => Expectation::has_type(Ty::Bool), | ||
920 | _ => Expectation::none(), | ||
921 | }; | ||
922 | let (lhs, rhs) = e.sub_exprs(); | ||
923 | let _lhs_ty = self.infer_expr_opt(lhs, &subtype_expectation)?; | ||
924 | let _rhs_ty = self.infer_expr_opt(rhs, &subtype_expectation)?; | ||
925 | |||
926 | if is_boolean_operator(op) { | ||
927 | Ty::Bool | ||
928 | } else { | ||
929 | Ty::Unknown | ||
930 | } | ||
931 | } | ||
932 | _ => Ty::Unknown, | ||
933 | }, | ||
903 | ast::Expr::Literal(_e) => Ty::Unknown, | 934 | ast::Expr::Literal(_e) => Ty::Unknown, |
904 | }; | 935 | }; |
905 | // use a new type variable if we got Ty::Unknown here | 936 | // use a new type variable if we got Ty::Unknown here |
diff --git a/crates/ra_hir/src/ty/tests.rs b/crates/ra_hir/src/ty/tests.rs index 515c66e85..25a354947 100644 --- a/crates/ra_hir/src/ty/tests.rs +++ b/crates/ra_hir/src/ty/tests.rs | |||
@@ -156,6 +156,30 @@ impl S { | |||
156 | ); | 156 | ); |
157 | } | 157 | } |
158 | 158 | ||
159 | #[test] | ||
160 | fn infer_boolean_op() { | ||
161 | check_inference( | ||
162 | r#" | ||
163 | fn f(x: bool) -> i32 { | ||
164 | 0i32 | ||
165 | } | ||
166 | |||
167 | fn test() { | ||
168 | let x = a && b; | ||
169 | let y = true || false; | ||
170 | let z = x == y; | ||
171 | let h = CONST_1 <= CONST_2; | ||
172 | let c = f(z || y) + 5; | ||
173 | let d = b; | ||
174 | let e = 3i32 && "hello world"; | ||
175 | |||
176 | 10 < 3 | ||
177 | } | ||
178 | "#, | ||
179 | "0008_boolean_op.txt", | ||
180 | ); | ||
181 | } | ||
182 | |||
159 | fn infer(content: &str) -> String { | 183 | fn infer(content: &str) -> String { |
160 | let (db, _, file_id) = MockDatabase::with_single_file(content); | 184 | let (db, _, file_id) = MockDatabase::with_single_file(content); |
161 | let source_file = db.source_file(file_id); | 185 | let source_file = db.source_file(file_id); |
diff --git a/crates/ra_hir/src/ty/tests/data/0008_boolean_op.txt b/crates/ra_hir/src/ty/tests/data/0008_boolean_op.txt new file mode 100644 index 000000000..ca01ad159 --- /dev/null +++ b/crates/ra_hir/src/ty/tests/data/0008_boolean_op.txt | |||
@@ -0,0 +1,31 @@ | |||
1 | [28; 32) '0i32': i32 | ||
2 | [22; 34) '{ 0i32 }': i32 | ||
3 | [6; 7) 'x': [unknown] | ||
4 | [127; 134) 'CONST_1': [unknown] | ||
5 | [201; 205) '3i32': bool | ||
6 | [76; 77) 'y': bool | ||
7 | [65; 66) 'b': bool | ||
8 | [60; 66) 'a && b': bool | ||
9 | [127; 145) 'CONST_...ONST_2': bool | ||
10 | [182; 183) 'd': [unknown] | ||
11 | [229; 231) '10': [unknown] | ||
12 | [209; 222) '"hello world"': bool | ||
13 | [229; 235) '10 < 3': bool | ||
14 | [186; 187) 'b': [unknown] | ||
15 | [159; 172) 'f(z || y) + 5': [unknown] | ||
16 | [56; 57) 'x': bool | ||
17 | [112; 113) 'y': bool | ||
18 | [201; 222) '3i32 &...world"': bool | ||
19 | [234; 235) '3': [unknown] | ||
20 | [138; 145) 'CONST_2': [unknown] | ||
21 | [80; 93) 'true || false': bool | ||
22 | [46; 237) '{ ... < 3 }': bool | ||
23 | [197; 198) 'e': bool | ||
24 | [107; 113) 'x == y': bool | ||
25 | [88; 93) 'false': bool | ||
26 | [80; 84) 'true': bool | ||
27 | [123; 124) 'h': bool | ||
28 | [155; 156) 'c': [unknown] | ||
29 | [103; 104) 'z': bool | ||
30 | [60; 61) 'a': bool | ||
31 | [107; 108) 'x': bool | ||