language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-29 14:54:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

Visit all declarations while lowering params

Browse source
This commit is contained in:
Agus Zubiaga 2024-08-17 15:23:28 -03:00
parent 6588a32195
commit c85c634387
No known key found for this signature in database
1 changed files with 19 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

32
crates/compiler/lower_params/src/lib.rs
View file

@ -1,3 +1,5 @@
use std::ops::Range;
use roc_can::{ use roc_can::{
expr::{ expr::{
AnnotatedMark, ClosureData, AnnotatedMark, ClosureData,
@ -44,14 +46,14 @@ pub fn lower(
var_store, var_store,
}; };
env.lower_decls(decls); env.lower_decls(decls, 0..decls.len());
} }
impl<'a> LowerParams<'a> { impl<'a> LowerParams<'a> {
fn lower_decls(&mut self, decls: &mut Declarations) { fn lower_decls(&mut self, decls: &mut Declarations, range: Range<usize>) {
let mut index = 0; let mut index = range.start;
while index < decls.len() { while index < range.end {
let tag = decls.declarations[index]; let tag = decls.declarations[index];
match tag { match tag {
@ -62,7 +64,7 @@ impl<'a> LowerParams<'a> {
decls.convert_value_to_function(index, vec![new_arg], self.var_store); decls.convert_value_to_function(index, vec![new_arg], self.var_store);
} }
} }
Function(fn_def_index) => { Function(fn_def_index) | Recursive(fn_def_index) | TailRecursive(fn_def_index) => {
if let Some((_, mark, pattern)) = self.home_params_argument() { if let Some((_, mark, pattern)) = self.home_params_argument() {
let var = self.var_store.fresh(); let var = self.var_store.fresh();
@ -81,15 +83,19 @@ impl<'a> LowerParams<'a> {
self.lower_expr(&mut decls.expressions[index].value) self.lower_expr(&mut decls.expressions[index].value)
} }
Recursive(_) => { /* todo */ } MutualRecursion {
MutualRecursion { length, .. } => { length,
/* todo */ cycle_mark: _,
index += length as usize; } => {
let length = length as usize;
self.lower_decls(decls, index + 1..index + 1 + length);
index += length;
}
Destructure(_) | Expectation | ExpectationFx => {
self.lower_expr(&mut decls.expressions[index].value)
} }
TailRecursive(_) => { /* todo */ }
Destructure(_) => { /* todo */ }
Expectation => { /* todo */ }
ExpectationFx => { /* todo */ }
} }
index += 1; index += 1;