teascade/ferrite-lua
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow returning expression lists

Browse Source
This commit is contained in:
Sofia 2026-03-15 15:23:37 +02:00
parent 219d4b4f7e
commit 3305d3c162
3 changed files with 19 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/test.lua
View File

@ -1,7 +1,5 @@
function add(x)
return function (y)
return x + y
end
return x
end
global c = print(add(5)(7))
global c = print(add(5))

2
src/ast.rs
View File

@ -207,7 +207,7 @@ impl Parse for Block {
#[derive(Debug, Clone)]
pub enum Statement {
Assignment(AccessModifier, Node<String>, Node<Expression>),
Return(Node<Expression>),
Return(ExpressionList),
If(Node<Expression>, Block),
}

22
src/compile.rs
View File

@ -77,7 +77,13 @@ impl Statement {
constants.extend(expr.kind.find_constants(scope));
constants
}
Statement::Return(expr) => expr.kind.find_constants(scope),
Statement::Return(expr_list) => {
let mut constants = HashSet::new();
for expr in &expr_list.0 {
constants.extend(expr.kind.find_constants(scope));
}
constants
}
Statement::If(cond, then) => {
let mut constants = HashSet::new();
constants.extend(cond.kind.find_constants(scope));
@ -106,12 +112,16 @@ impl Statement {
}
}
}
Statement::Return(expr) => {
let (instr, registers) = expr.kind.compile(state, scope, 1);
instructions.extend(instr);
Statement::Return(expr_list) => {
let mut ret_registers = Vec::new();
for expr in &expr_list.0 {
let (instr, registers) = expr.kind.compile(state, scope, 1);
instructions.extend(instr);
ret_registers.extend(registers);
}
instructions.push(Instruction::Return(
*registers.first().unwrap(),
*registers.last().unwrap(),
*ret_registers.first().unwrap(),
*ret_registers.last().unwrap(),
));
}
Statement::If(node, block) => todo!(),