feat: simple functions

This commit is contained in:
2025-05-04 14:42:22 +08:00
parent 6ecd20854a
commit bc50464db9
6 changed files with 77 additions and 27 deletions

View File

@@ -1,8 +1,11 @@
use ecow::EcoString;
use rpds::HashTrieMap;
use crate::bytecode::{OpCodes, ThunkIdx};
use crate::ty::internal::Value;
use crate::vm::{LockedEnv, VM};
#[derive(Debug, Clone)]
pub enum Param {
Ident(EcoString),
@@ -15,20 +18,31 @@ pub enum Param {
#[derive(Debug, Clone)]
pub struct Func {
pub env: LockedEnv,
pub param: Option<Param>,
pub opcodes: OpCodes,
}
impl Func {
pub fn new(opcodes: OpCodes) -> Func {
pub fn new(env: LockedEnv, opcodes: OpCodes) -> Func {
Func {
param: None,
env,
opcodes,
param: None,
}
}
pub fn call(self, _arg: Value) -> Value {
todo!()
pub fn call(self, vm: &VM, arg: Value) -> Value {
use Param::*;
let mut env = self.env.unlocked();
match self.param.unwrap() {
Ident(ident) => env.enter(HashTrieMap::new_sync().insert(ident.into(), arg)),
Formals { .. } => todo!()
}
vm.eval(self.opcodes, &mut env).unwrap()
}
pub fn push_ident_param(&mut self, param: EcoString) {