language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-22 19:42:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

Support bound variables in multiple patterns

Browse source
This commit is contained in:
Ayaz Hafiz 2022-07-21 11:40:09 -04:00
parent bf8fc0d0de
commit ce8b50caea
No known key found for this signature in database
GPG key ID: 0E2A37416A25EF58
3 changed files with 59 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

33
crates/compiler/constrain/src/expr.rs
View file

@ -1824,14 +1824,45 @@ fn constrain_when_branch_help(
for (i, loc_pattern) in when_branch.patterns.iter().enumerate() {
let pattern_expected = pattern_expected(HumanIndex::zero_based(i), loc_pattern.region);
let mut partial_state = PatternState::default();
constrain_pattern(
constraints,
env,
&loc_pattern.value,
loc_pattern.region,
pattern_expected,
&mut state,
&mut partial_state,
);
state.vars.extend(partial_state.vars);
state.constraints.extend(partial_state.constraints);
state
.delayed_is_open_constraints
.extend(partial_state.delayed_is_open_constraints);
if i == 0 {
state.headers.extend(partial_state.headers);
} else {
debug_assert!(
state.headers.keys().all(|sym| partial_state.headers.contains_key(sym)) &&
partial_state.headers.keys().all(|sym| state.headers.contains_key(sym)),
"State and partial state headers differ in bound symbols, should have been caught in canonicalization");
// Make sure the bound variables in the patterns on the same branch agree in their types.
for (sym, typ1) in state.headers.iter() {
let typ2 = partial_state
.headers
.get(sym)
.expect("bound variable in branch not bound in pattern!");
state.constraints.push(constraints.equal_types(
typ1.value.clone(),
Expected::NoExpectation(typ2.value.clone()),
Category::When,
typ2.region,
));
}
}
}
let (pattern_constraints, body_constraints) = if let Some(loc_guard) = &when_branch.guard {