use cel::common::ast::CallExpr;
use ferricel_types::functions::RuntimeFunction;
use walrus::InstrSeqBuilder;
use crate::compiler::{
context::{CompilerContext, CompilerEnv},
expr::compile_expr,
};
pub fn compile_ext_sets_function(
func_name: &str,
call_expr: &CallExpr,
body: &mut InstrSeqBuilder,
env: &CompilerEnv,
ctx: &CompilerContext,
module: &mut walrus::Module,
) -> Result<(), anyhow::Error> {
match func_name {
"contains" => compile_sets_contains(call_expr, body, env, ctx, module),
"intersects" => compile_sets_intersects(call_expr, body, env, ctx, module),
"equivalent" => compile_sets_equivalent(call_expr, body, env, ctx, module),
_ => anyhow::bail!("Unknown sets function: {}", func_name),
}
}
fn compile_sets_contains(
call_expr: &CallExpr,
body: &mut InstrSeqBuilder,
env: &CompilerEnv,
ctx: &CompilerContext,
module: &mut walrus::Module,
) -> Result<(), anyhow::Error> {
if call_expr.args.len() != 2 {
anyhow::bail!("sets.contains() expects exactly 2 arguments");
}
compile_expr(&call_expr.args[0].expr, body, env, ctx, module)?;
compile_expr(&call_expr.args[1].expr, body, env, ctx, module)?;
body.call(env.get(RuntimeFunction::SetsContains));
Ok(())
}
fn compile_sets_intersects(
call_expr: &CallExpr,
body: &mut InstrSeqBuilder,
env: &CompilerEnv,
ctx: &CompilerContext,
module: &mut walrus::Module,
) -> Result<(), anyhow::Error> {
if call_expr.args.len() != 2 {
anyhow::bail!("sets.intersects() expects exactly 2 arguments");
}
compile_expr(&call_expr.args[0].expr, body, env, ctx, module)?;
compile_expr(&call_expr.args[1].expr, body, env, ctx, module)?;
body.call(env.get(RuntimeFunction::SetsIntersects));
Ok(())
}
fn compile_sets_equivalent(
call_expr: &CallExpr,
body: &mut InstrSeqBuilder,
env: &CompilerEnv,
ctx: &CompilerContext,
module: &mut walrus::Module,
) -> Result<(), anyhow::Error> {
if call_expr.args.len() != 2 {
anyhow::bail!("sets.equivalent() expects exactly 2 arguments");
}
compile_expr(&call_expr.args[0].expr, body, env, ctx, module)?;
compile_expr(&call_expr.args[1].expr, body, env, ctx, module)?;
body.call(env.get(RuntimeFunction::SetsEquivalent));
Ok(())
}