language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-10-04 00:54:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

typecheck toplevel expects

Browse source
This commit is contained in:
Folkert 2022-06-15 23:15:43 +02:00
parent 0dac546aba
commit f7246d2774
No known key found for this signature in database
GPG key ID: 1F17F6FFD112B97C
5 changed files with 60 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

34
compiler/constrain/src/expr.rs
View file

@ -4,7 +4,7 @@ use crate::builtins::{
use crate::pattern::{constrain_pattern, PatternState};
use roc_can::annotation::IntroducedVariables;
use roc_can::constraint::{Constraint, Constraints, OpportunisticResolve};
use roc_can::def::{Declaration, Def};
use roc_can::def::{Declaration, Def, Expects};
use roc_can::exhaustive::{sketch_pattern_to_rows, sketch_when_branches, ExhaustiveContext};
use roc_can::expected::Expected::{self, *};
use roc_can::expected::PExpected;
@ -1309,6 +1309,9 @@ pub fn constrain_decls(
constraint =
constrain_recursive_defs(constraints, &mut env, defs, constraint, *cycle_mark);
}
Declaration::Expects(expects) => {
constraint = constrain_expects(constraints, &mut env, expects, constraint);
}
Declaration::InvalidCycle(_) => {
// invalid cycles give a canonicalization error. we skip them here.
continue;
@ -1705,6 +1708,35 @@ fn attach_resolution_constraints(
constraints.and_constraint([constraint, resolution_constrs])
}
fn constrain_expects(
constraints: &mut Constraints,
env: &mut Env,
expects: &Expects,
body_con: Constraint,
) -> Constraint {
let expect_bool = |region| {
let bool_type = Type::Variable(Variable::BOOL);
Expected::ForReason(Reason::ExpectCondition, bool_type, region)
};
let mut expect_constraints = Vec::with_capacity(expects.conditions.len());
let it = expects.regions.iter().zip(expects.conditions.iter());
for (region, condition) in it {
let expected = expect_bool(*region);
expect_constraints.push(constrain_expr(
constraints,
env,
*region,
condition,
expected,
));
}
let defs_constraint = constraints.and_constraint(expect_constraints);
constraints.let_constraint([], [], [], defs_constraint, body_con)
}
fn constrain_def(
constraints: &mut Constraints,
env: &mut Env,