Skip to content

Commit 49a028d

Browse files
bors[bot]jonas-schievink
bors[bot]
and
jonas-schievink
authored
Merge #10474
10474: feat: Support `let...else` r=jonas-schievink a=jonas-schievink bors r+ Co-authored-by: Jonas Schievink <[email protected]>
2 parents 94fa49c + 28f945a commit 49a028d

File tree

12 files changed

+133
-8
lines changed

12 files changed

+133
-8
lines changed

‎Cargo.lock

Lines changed: 2 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎crates/hir_def/src/body/lower.rs

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -639,7 +639,13 @@ impl ExprCollector<'_> {
639639
let type_ref =
640640
stmt.ty().map(|it| Interned::new(TypeRef::from_ast(&self.ctx(), it)));
641641
let initializer = stmt.initializer().map(|e| self.collect_expr(e));
642-
self.statements_in_scope.push(Statement::Let { pat, type_ref, initializer });
642+
let else_branch = stmt.else_branch().map(|block| self.collect_block(block));
643+
self.statements_in_scope.push(Statement::Let {
644+
pat,
645+
type_ref,
646+
initializer,
647+
else_branch,
648+
});
643649
}
644650
ast::Stmt::ExprStmt(stmt) => {
645651
if let Some(expr) = stmt.expr() {

‎crates/hir_def/src/body/scope.rs

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,11 +149,15 @@ fn compute_block_scopes(
149149
) {
150150
for stmt in statements {
151151
match stmt {
152-
Statement::Let { pat, initializer, .. } => {
152+
Statement::Let { pat, initializer, else_branch, .. } => {
153153
if let Some(expr) = initializer {
154154
scopes.set_scope(*expr, scope);
155155
compute_expr_scopes(*expr, body, scopes, scope);
156156
}
157+
if let Some(expr) = else_branch {
158+
scopes.set_scope(*expr, scope);
159+
compute_expr_scopes(*expr, body, scopes, scope);
160+
}
157161
scope = scopes.new_scope(scope);
158162
scopes.add_bindings(body, scope, *pat);
159163
}

‎crates/hir_def/src/expr.rs

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -208,8 +208,16 @@ pub struct RecordLitField {
208208

209209
#[derive(Debug, Clone, Eq, PartialEq)]
210210
pub enum Statement {
211-
Let { pat: PatId, type_ref: Option<Interned<TypeRef>>, initializer: Option<ExprId> },
212-
Expr { expr: ExprId, has_semi: bool },
211+
Let {
212+
pat: PatId,
213+
type_ref: Option<Interned<TypeRef>>,
214+
initializer: Option<ExprId>,
215+
else_branch: Option<ExprId>,
216+
},
217+
Expr {
218+
expr: ExprId,
219+
has_semi: bool,
220+
},
213221
}
214222

215223
impl Expr {

‎crates/hir_ty/src/infer/expr.rs

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -914,7 +914,7 @@ impl<'a> InferenceContext<'a> {
914914
) -> Ty {
915915
for stmt in statements {
916916
match stmt {
917-
Statement::Let { pat, type_ref, initializer } => {
917+
Statement::Let { pat, type_ref, initializer, else_branch } => {
918918
let decl_ty = type_ref
919919
.as_ref()
920920
.map(|tr| self.make_ty(tr))
@@ -931,6 +931,13 @@ impl<'a> InferenceContext<'a> {
931931
}
932932
}
933933

934+
if let Some(expr) = else_branch {
935+
self.infer_expr_coerce(
936+
*expr,
937+
&Expectation::has_type(Ty::new(&Interner, TyKind::Never)),
938+
);
939+
}
940+
934941
self.infer_pat(*pat, &ty, BindingMode::default());
935942
}
936943
Statement::Expr { expr, .. } => {

‎crates/hir_ty/src/tests/never_type.rs

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -407,3 +407,39 @@ fn diverging_expression_3_break() {
407407
"]],
408408
);
409409
}
410+
411+
#[test]
412+
fn let_else_must_diverge() {
413+
check_infer_with_mismatches(
414+
r#"
415+
fn f() {
416+
let 1 = 2 else {
417+
return;
418+
};
419+
}
420+
"#,
421+
expect![[r#"
422+
7..54 '{ ... }; }': ()
423+
17..18 '1': i32
424+
17..18 '1': i32
425+
21..22 '2': i32
426+
28..51 '{ ... }': !
427+
38..44 'return': !
428+
"#]],
429+
);
430+
check_infer_with_mismatches(
431+
r#"
432+
fn f() {
433+
let 1 = 2 else {};
434+
}
435+
"#,
436+
expect![[r#"
437+
7..33 '{ ... {}; }': ()
438+
17..18 '1': i32
439+
17..18 '1': i32
440+
21..22 '2': i32
441+
28..30 '{}': ()
442+
28..30: expected !, got ()
443+
"#]],
444+
);
445+
}

‎crates/parser/src/grammar/expressions.rs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,12 @@ pub(super) fn stmt(p: &mut Parser, with_semi: StmtWithSemi, prefer_expr: bool) {
102102
expressions::expr(p);
103103
}
104104

105+
if p.eat(T![else]) {
106+
// test let_else
107+
// fn f() { let Some(x) = opt else { return }; }
108+
block_expr(p);
109+
}
110+
105111
match with_semi {
106112
StmtWithSemi::No => (),
107113
StmtWithSemi::Optional => {

‎crates/syntax/Cargo.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ rayon = "1"
2929
expect-test = "1.1"
3030
proc-macro2 = "1.0.8"
3131
quote = "1.0.2"
32-
ungrammar = "=1.14.6"
32+
ungrammar = "=1.14.7"
3333

3434
test_utils = { path = "../test_utils" }
3535
sourcegen = { path = "../sourcegen" }

‎crates/syntax/src/ast/generated/nodes.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -722,6 +722,7 @@ impl LetStmt {
722722
pub fn ty(&self) -> Option<Type> { support::child(&self.syntax) }
723723
pub fn eq_token(&self) -> Option<SyntaxToken> { support::token(&self.syntax, T![=]) }
724724
pub fn initializer(&self) -> Option<Expr> { support::child(&self.syntax) }
725+
pub fn else_token(&self) -> Option<SyntaxToken> { support::token(&self.syntax, T![else]) }
725726
pub fn semicolon_token(&self) -> Option<SyntaxToken> { support::token(&self.syntax, T![;]) }
726727
}
727728

‎crates/syntax/src/ast/node_ext.rs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -754,6 +754,12 @@ impl ast::GenericParamList {
754754
}
755755
}
756756

757+
impl ast::LetStmt {
758+
pub fn else_branch(&self) -> Option<ast::BlockExpr> {
759+
support::children(self.syntax()).next()
760+
}
761+
}
762+
757763
impl ast::HasDocComments for ast::SourceFile {}
758764
impl ast::HasDocComments for ast::Fn {}
759765
impl ast::HasDocComments for ast::Struct {}

‎crates/syntax/test_data/parser/inline/ok/0194_let_else.rast

Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,50 @@
1+
[email protected]
2+
[email protected]
3+
[email protected] "fn"
4+
[email protected] " "
5+
[email protected]
6+
[email protected] "f"
7+
[email protected]
8+
[email protected] "("
9+
[email protected] ")"
10+
[email protected] " "
11+
[email protected]
12+
[email protected]
13+
[email protected] "{"
14+
[email protected] " "
15+
[email protected]
16+
[email protected] "let"
17+
[email protected] " "
18+
[email protected]
19+
[email protected]
20+
[email protected]
21+
[email protected]
22+
[email protected] "Some"
23+
[email protected] "("
24+
[email protected]
25+
[email protected]
26+
[email protected] "x"
27+
[email protected] ")"
28+
[email protected] " "
29+
[email protected] "="
30+
[email protected] " "
31+
[email protected]
32+
[email protected]
33+
[email protected]
34+
[email protected]
35+
[email protected] "opt"
36+
[email protected] " "
37+
[email protected] "else"
38+
[email protected] " "
39+
[email protected]
40+
[email protected]
41+
[email protected] "{"
42+
[email protected] " "
43+
[email protected]
44+
[email protected] "return"
45+
[email protected] " "
46+
[email protected] "}"
47+
[email protected] ";"
48+
[email protected] " "
49+
[email protected] "}"
50+
[email protected] "\n"

‎crates/syntax/test_data/parser/inline/ok/0194_let_else.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
fn f() { let Some(x) = opt else { return }; }

0 commit comments

Comments
 (0)