zaphar/ucg
1
0
Fork 0
mirror of https://github.com/zaphar/ucg.git synced 2025-07-22 18:19:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

DEV: Improve the Walk trait a bit.

Browse Source
This commit is contained in:
Jeremy Wall 2020-05-25 10:57:55 -04:00
parent c18ec8c3f7
commit 3052f7d7a8
1 changed files with 39 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/ast/walk.rs
View File

@ -104,10 +104,17 @@ pub trait Walker {
self.walk_fieldset(&mut def.tuple);
}
Expression::Simple(ref mut val) => {
self.visit_value(val);
self.walk_value(val);
}
Expression::Import(_) | Expression::Include(_) | Expression::Fail(_) => {
//noop
Expression::Import(i) => {
self.visit_import(i);
}
Expression::Include(i) => {
self.visit_include(i);
}
Expression::Fail(f) => {
self.visit_fail(f);
}
Expression::Not(ref mut def) => {
self.walk_expression(def.expr.as_mut());
@ -118,6 +125,35 @@ pub trait Walker {
}
}
fn walk_value(&mut self, val: &mut Value) {
match val {
Value::Empty(_)
| Value::Symbol(_)
| Value::Boolean(_)
| Value::Int(_)
| Value::Float(_)
| Value::Str(_) => self.visit_value(val),
Value::Tuple(fs) => self.walk_fieldset(&mut fs.val),
Value::List(vs) => {
for e in &mut vs.elems {
self.walk_expression(e);
}
}
}
}
fn visit_import(&mut self, i: &mut ImportDef) {
// noop by default;
}
fn visit_include(&mut self, i: &mut IncludeDef) {
// noop by default;
}
fn visit_fail(&mut self, f: &mut FailDef) {
// noop by default;
}
fn visit_value(&mut self, val: &mut Value);
fn visit_expression(&mut self, expr: &mut Expression);