noteuclid/minilua
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add scope ops

Browse Source
This commit is contained in:
numzero 2025-04-11 20:29:40 +03:00
parent 4ead8e702e
commit 382cfdd903
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/run.rs
View File

@ -1,4 +1,4 @@
use std::{collections::HashMap, fmt::Debug, rc::Rc}; use std::{borrow::Cow, collections::HashMap, fmt::Debug, rc::Rc};
use crate::{ use crate::{
ast::{Constant, Expression, Location}, ast::{Constant, Expression, Location},
@ -36,6 +36,20 @@ pub struct Scope {
pub functions: HashMap<String, Function>, pub functions: HashMap<String, Function>,
} }
impl Scope {
pub fn get(&self, name: impl AsRef<str>) -> Value {
self.variables.get(name.as_ref()).cloned()
}
pub fn set(&mut self, name: Cow<str>, value: Value) {
if let Some(value) = value {
self.variables.insert(name.into(), value);
} else {
self.variables.remove(name.as_ref());
}
}
}
pub trait Eval { pub trait Eval {
fn eval(&self, scope: &mut Scope) -> Value; fn eval(&self, scope: &mut Scope) -> Value;
} }
@ -54,7 +68,7 @@ impl Eval for Constant {
impl Eval for Location { impl Eval for Location {
fn eval(&self, scope: &mut Scope) -> Value { fn eval(&self, scope: &mut Scope) -> Value {
match self { match self {
Location::Variable { name } => scope.variables.get(name).cloned(), Location::Variable { name } => scope.get(name),
Location::Field { table, index } => { Location::Field { table, index } => {
let table = table.eval(scope); let table = table.eval(scope);
let Some(ValueInner::Table(table)) = table else { let Some(ValueInner::Table(table)) = table else {