kjuulh/rhai
1
0
Fork 0
Code Issues 1 Pull Requests 1 Packages Projects Releases Wiki Activity

Code cleanup.

Browse Source
This commit is contained in:
Stephen Chung 2020-03-06 23:49:52 +08:00
parent be615cea27
commit c917aa0a5a
1 changed files with 54 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

116
src/engine.rs
View File

@ -19,7 +19,7 @@ const KEYWORD_DEBUG: &'static str = "debug";
const KEYWORD_TYPE_OF: &'static str = "type_of"; const KEYWORD_TYPE_OF: &'static str = "type_of";
#[derive(Copy, Clone, Debug, Eq, Hash, PartialEq, PartialOrd, Ord)] #[derive(Copy, Clone, Debug, Eq, Hash, PartialEq, PartialOrd, Ord)]
enum VariableType { enum IndexSourceType {
Array, Array,
String, String,
Expression, Expression,
@ -283,25 +283,27 @@ impl Engine<'_> {
val: Dynamic, val: Dynamic,
idx: i64, idx: i64,
pos: Position, pos: Position,
) -> Result<(Dynamic, VariableType), EvalAltResult> { ) -> Result<(Dynamic, IndexSourceType), EvalAltResult> {
if val.is::<Array>() { if val.is::<Array>() {
// val_array[idx]
let arr = val.downcast::<Array>().expect("array expected"); let arr = val.downcast::<Array>().expect("array expected");
if idx >= 0 { if idx >= 0 {
arr.get(idx as usize) arr.get(idx as usize)
.cloned() .cloned()
.map(|v| (v, VariableType::Array)) .map(|v| (v, IndexSourceType::Array))
.ok_or_else(|| EvalAltResult::ErrorArrayBounds(arr.len(), idx, pos)) .ok_or_else(|| EvalAltResult::ErrorArrayBounds(arr.len(), idx, pos))
} else { } else {
Err(EvalAltResult::ErrorArrayBounds(arr.len(), idx, pos)) Err(EvalAltResult::ErrorArrayBounds(arr.len(), idx, pos))
} }
} else if val.is::<String>() { } else if val.is::<String>() {
// val_string[idx]
let s = val.downcast::<String>().expect("string expected"); let s = val.downcast::<String>().expect("string expected");
if idx >= 0 { if idx >= 0 {
s.chars() s.chars()
.nth(idx as usize) .nth(idx as usize)
.map(|ch| (ch.into_dynamic(), VariableType::String)) .map(|ch| (ch.into_dynamic(), IndexSourceType::String))
.ok_or_else(|| EvalAltResult::ErrorStringBounds(s.chars().count(), idx, pos)) .ok_or_else(|| EvalAltResult::ErrorStringBounds(s.chars().count(), idx, pos))
} else { } else {
Err(EvalAltResult::ErrorStringBounds( Err(EvalAltResult::ErrorStringBounds(
@ -311,6 +313,7 @@ impl Engine<'_> {
)) ))
} }
} else { } else {
// Error - cannot be indexed
Err(EvalAltResult::ErrorIndexingType(pos)) Err(EvalAltResult::ErrorIndexingType(pos))
} }
} }
@ -321,7 +324,7 @@ impl Engine<'_> {
scope: &mut Scope, scope: &mut Scope,
lhs: &Expr, lhs: &Expr,
idx_expr: &Expr, idx_expr: &Expr,
) -> Result<(VariableType, Option<(String, usize)>, usize, Dynamic), EvalAltResult> { ) -> Result<(IndexSourceType, Option<(String, usize)>, usize, Dynamic), EvalAltResult> {
let idx = self.eval_index_value(scope, idx_expr)?; let idx = self.eval_index_value(scope, idx_expr)?;
match lhs { match lhs {
@ -332,58 +335,57 @@ impl Engine<'_> {
|val| Self::get_indexed_value(val, idx, idx_expr.position()), |val| Self::get_indexed_value(val, idx, idx_expr.position()),
lhs.position(), lhs.position(),
) )
.map(|(src_idx, (val, source_type))| { .map(|(src_idx, (val, src_type))| {
(source_type, Some((id.clone(), src_idx)), idx as usize, val) (src_type, Some((id.clone(), src_idx)), idx as usize, val)
}), }),
// (expr)[idx_expr] // (expr)[idx_expr]
expr => Self::get_indexed_value(self.eval_expr(scope, expr)?, idx, idx_expr.position()) expr => Self::get_indexed_value(self.eval_expr(scope, expr)?, idx, idx_expr.position())
.map(|(val, _)| (VariableType::Expression, None, idx as usize, val)), .map(|(val, _)| (IndexSourceType::Expression, None, idx as usize, val)),
} }
} }
/// Replace a character at an index position in a mutable string /// Replace a character at an index position in a mutable string
fn str_replace_char(s: &mut String, idx: usize, new_ch: char) { fn str_replace_char(s: &mut String, idx: usize, new_ch: char) {
// The new character let mut chars: Vec<char> = s.chars().collect();
let ch = s.chars().nth(idx).expect("string index out of bounds"); let ch = *chars.get(idx).expect("string index out of bounds");
// See if changed - if so, update the String // See if changed - if so, update the String
if ch == new_ch { if ch != new_ch {
return;
}
// Collect all the characters after the index
let mut chars: Vec<char> = s.chars().collect();
chars[idx] = new_ch; chars[idx] = new_ch;
s.clear(); s.clear();
chars.iter().for_each(|&ch| s.push(ch)); chars.iter().for_each(|&ch| s.push(ch));
} }
}
/// Update the value at an index position in a variable inside the scope /// Update the value at an index position in a variable inside the scope
fn update_indexed_variable_in_scope( fn update_indexed_variable_in_scope(
source_type: VariableType, src_type: IndexSourceType,
scope: &mut Scope, scope: &mut Scope,
id: &str, id: &str,
src_idx: usize, src_idx: usize,
idx: usize, idx: usize,
val: Dynamic, val: Dynamic,
) -> Option<Dynamic> { ) -> Dynamic {
match source_type { match src_type {
VariableType::Array => { // array_id[idx] = val
IndexSourceType::Array => {
let arr = scope.get_mut_by_type::<Array>(id, src_idx); let arr = scope.get_mut_by_type::<Array>(id, src_idx);
Some((arr[idx as usize] = val).into_dynamic()) (arr[idx as usize] = val).into_dynamic()
} }
VariableType::String => { // string_id[idx] = val
IndexSourceType::String => {
let s = scope.get_mut_by_type::<String>(id, src_idx); let s = scope.get_mut_by_type::<String>(id, src_idx);
// Value must be a character // Value must be a character
let ch = *val let ch = *val
.downcast::<char>() .downcast::<char>()
.expect("char value expected to update an index position in a string"); .expect("char value expected to update an index position in a string");
Some(Self::str_replace_char(s, idx as usize, ch).into_dynamic()) Self::str_replace_char(s, idx as usize, ch).into_dynamic()
} }
_ => None, // All other variable types should be an error
_ => panic!("array or string source type expected for indexing"),
} }
} }
@ -397,19 +399,19 @@ impl Engine<'_> {
match dot_lhs { match dot_lhs {
// xxx.??? // xxx.???
Expr::Identifier(id, pos) => { Expr::Identifier(id, pos) => {
let (sc_idx, mut target) = Self::search_scope(scope, id, Ok, *pos)?; let (src_idx, mut target) = Self::search_scope(scope, id, Ok, *pos)?;
let value = self.get_dot_val_helper(scope, target.as_mut(), dot_rhs); let value = self.get_dot_val_helper(scope, target.as_mut(), dot_rhs);
// In case the expression mutated `target`, we need to reassign it because // In case the expression mutated `target`, we need to reassign it because
// of the above `clone`. // of the above `clone`.
*scope.get_mut(id, sc_idx) = target; *scope.get_mut(id, src_idx) = target;
value value
} }
// lhs[idx_expr].??? // lhs[idx_expr].???
Expr::Index(lhs, idx_expr) => { Expr::Index(lhs, idx_expr) => {
let (source_type, src, idx, mut target) = let (src_type, src, idx, mut target) =
self.eval_index_expr(scope, lhs, idx_expr)?; self.eval_index_expr(scope, lhs, idx_expr)?;
let value = self.get_dot_val_helper(scope, target.as_mut(), dot_rhs); let value = self.get_dot_val_helper(scope, target.as_mut(), dot_rhs);
@ -417,14 +419,8 @@ impl Engine<'_> {
// of the above `clone`. // of the above `clone`.
if let Some((id, src_idx)) = src { if let Some((id, src_idx)) = src {
Self::update_indexed_variable_in_scope( Self::update_indexed_variable_in_scope(
source_type, src_type, scope, &id, src_idx, idx, target,
scope, );
&id,
src_idx,
idx,
target,
)
.expect("array or string source type expected for indexing");
} }
value value
@ -435,8 +431,6 @@ impl Engine<'_> {
let mut target = self.eval_expr(scope, expr)?; let mut target = self.eval_expr(scope, expr)?;
self.get_dot_val_helper(scope, target.as_mut(), dot_rhs) self.get_dot_val_helper(scope, target.as_mut(), dot_rhs)
} }
// Syntax error
//_ => Err(EvalAltResult::ErrorDotExpr(dot_lhs.position())),
} }
} }
@ -495,19 +489,19 @@ impl Engine<'_> {
match dot_lhs { match dot_lhs {
// id.??? // id.???
Expr::Identifier(id, pos) => { Expr::Identifier(id, pos) => {
let (sc_idx, mut target) = Self::search_scope(scope, id, Ok, *pos)?; let (src_idx, mut target) = Self::search_scope(scope, id, Ok, *pos)?;
let value = self.set_dot_val_helper(target.as_mut(), dot_rhs, source_val); let value = self.set_dot_val_helper(target.as_mut(), dot_rhs, source_val);
// In case the expression mutated `target`, we need to reassign it because // In case the expression mutated `target`, we need to reassign it because
// of the above `clone`. // of the above `clone`.
*scope.get_mut(id, sc_idx) = target; *scope.get_mut(id, src_idx) = target;
value value
} }
// lhs[idx_expr].??? // lhs[idx_expr].???
Expr::Index(lhs, idx_expr) => { Expr::Index(lhs, idx_expr) => {
let (source_type, src, idx, mut target) = let (src_type, src, idx, mut target) =
self.eval_index_expr(scope, lhs, idx_expr)?; self.eval_index_expr(scope, lhs, idx_expr)?;
let value = self.set_dot_val_helper(target.as_mut(), dot_rhs, source_val); let value = self.set_dot_val_helper(target.as_mut(), dot_rhs, source_val);
@ -516,14 +510,8 @@ impl Engine<'_> {
if let Some((id, src_idx)) = src { if let Some((id, src_idx)) = src {
Self::update_indexed_variable_in_scope( Self::update_indexed_variable_in_scope(
source_type, src_type, scope, &id, src_idx, idx, target,
scope, );
&id,
src_idx,
idx,
target,
)
.expect("array or string source_type expected for indexing");
} }
value value
@ -565,24 +553,18 @@ impl Engine<'_> {
// idx_lhs[idx_expr] = rhs // idx_lhs[idx_expr] = rhs
Expr::Index(idx_lhs, idx_expr) => { Expr::Index(idx_lhs, idx_expr) => {
let (source_type, src, idx, _) = let (src_type, src, idx, _) =
self.eval_index_expr(scope, idx_lhs, idx_expr)?; self.eval_index_expr(scope, idx_lhs, idx_expr)?;
if let Some((id, src_idx)) = src { if let Some((id, src_idx)) = src {
Self::update_indexed_variable_in_scope( Ok(Self::update_indexed_variable_in_scope(
source_type, src_type, scope, &id, src_idx, idx, rhs_val,
scope, ))
&id,
src_idx,
idx,
rhs_val,
)
} else { } else {
None Err(EvalAltResult::ErrorAssignmentToUnknownLHS(
idx_lhs.position(),
))
} }
.ok_or_else(|| {
EvalAltResult::ErrorAssignmentToUnknownLHS(idx_lhs.position())
})
} }
// dot_lhs.dot_rhs = rhs // dot_lhs.dot_rhs = rhs
@ -670,8 +652,10 @@ impl Engine<'_> {
stmt: &Stmt, stmt: &Stmt,
) -> Result<Dynamic, EvalAltResult> { ) -> Result<Dynamic, EvalAltResult> {
match stmt { match stmt {
// Expression as statement
Stmt::Expr(expr) => self.eval_expr(scope, expr), Stmt::Expr(expr) => self.eval_expr(scope, expr),
// Block scope
Stmt::Block(block) => { Stmt::Block(block) => {
let prev_len = scope.len(); let prev_len = scope.len();
let mut last_result: Result<Dynamic, EvalAltResult> = Ok(().into_dynamic()); let mut last_result: Result<Dynamic, EvalAltResult> = Ok(().into_dynamic());
@ -685,13 +669,12 @@ impl Engine<'_> {
} }
} }
while scope.len() > prev_len { scope.rewind(prev_len);
scope.pop();
}
last_result last_result
} }
// If-else statement
Stmt::IfElse(guard, body, else_body) => self Stmt::IfElse(guard, body, else_body) => self
.eval_expr(scope, guard)? .eval_expr(scope, guard)?
.downcast::<bool>() .downcast::<bool>()
@ -706,6 +689,7 @@ impl Engine<'_> {
} }
}), }),
// While loop
Stmt::While(guard, body) => loop { Stmt::While(guard, body) => loop {
match self.eval_expr(scope, guard)?.downcast::<bool>() { match self.eval_expr(scope, guard)?.downcast::<bool>() {
Ok(guard_val) => { Ok(guard_val) => {
@ -723,6 +707,7 @@ impl Engine<'_> {
} }
}, },
// Loop statement
Stmt::Loop(body) => loop { Stmt::Loop(body) => loop {
match self.eval_stmt(scope, body) { match self.eval_stmt(scope, body) {
Err(EvalAltResult::LoopBreak) => return Ok(().into_dynamic()), Err(EvalAltResult::LoopBreak) => return Ok(().into_dynamic()),
@ -731,6 +716,7 @@ impl Engine<'_> {
} }
}, },
// For loop
Stmt::For(name, expr, body) => { Stmt::For(name, expr, body) => {
let arr = self.eval_expr(scope, expr)?; let arr = self.eval_expr(scope, expr)?;
let tid = Any::type_id(&*arr); let tid = Any::type_id(&*arr);
@ -755,6 +741,7 @@ impl Engine<'_> {
} }
} }
// Break statement
Stmt::Break(_) => Err(EvalAltResult::LoopBreak), Stmt::Break(_) => Err(EvalAltResult::LoopBreak),
// Empty return // Empty return
@ -783,6 +770,7 @@ impl Engine<'_> {
)) ))
} }
// Let statement
Stmt::Let(name, init, _) => { Stmt::Let(name, init, _) => {
if let Some(v) = init { if let Some(v) = init {
let val = self.eval_expr(scope, v)?; let val = self.eval_expr(scope, v)?;