language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-28 14:24:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Lower top-level fn calls in home module with params

Browse source
This commit is contained in:
Agus Zubiaga 2024-08-17 13:44:19 -03:00
parent 07ec24502e
commit bcd8e7e28a
No known key found for this signature in database
4 changed files with 26 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -104,9 +104,7 @@ impl<'a> LowerParams<'a> {
*expr = self.call_params_var(*symbol, *var, *params_symbol, *params_var);
}
Var(symbol, var) => {
if symbol.module_id() == self.home_id
&& self.home_top_level_idents.contains(&symbol.ident_id())
{
if self.is_params_extended_home_symbol(symbol) {
// A reference to a top-level value def in the home module with params
let params = self.home_params.as_ref().unwrap();
*expr =
@ -130,6 +128,16 @@ impl<'a> LowerParams<'a> {
args.push((params_var, Loc::at_zero(Var(params_symbol, params_var))));
fun.1.value = Var(symbol, var);
}
Var(symbol, _var) => {
if self.is_params_extended_home_symbol(&symbol) {
// A call to a top-level function in the home module with params
let params = self.home_params.as_ref().unwrap();
args.push((
params.whole_var,
Loc::at_zero(Var(params.whole_symbol, params.whole_var)),
));
}
}
_ => self.lower_expr(&mut fun.1.value),
}
}
@ -151,6 +159,11 @@ impl<'a> LowerParams<'a> {
}
}
fn is_params_extended_home_symbol(&self, symbol: &Symbol) -> bool {
symbol.module_id() == self.home_id
&& self.home_top_level_idents.contains(&symbol.ident_id())
}
fn call_params_var(
&mut self,
symbol: Symbol,