temp work
This commit is contained in:
parent
afcb10bb72
commit
9ec1d78f1b
@ -1,5 +1,6 @@
|
||||
use std::rc::Rc;
|
||||
use crate::ast::*;
|
||||
use crate::util::deref_optional_box;
|
||||
|
||||
//TODO maybe these functions should take closures that return a KeepRecursing | StopHere type,
|
||||
//or a tuple of (T, <that type>)
|
||||
@ -39,3 +40,126 @@ pub trait ASTVisitor: Sized {
|
||||
fn prefix_exp(&mut self, _op: &PrefixOp, _arg: &Expression) {}
|
||||
fn pattern(&mut self, _pat: &Pattern) {}
|
||||
}
|
||||
|
||||
pub trait ASTVisitorNew: Sized {
|
||||
fn expression(&mut self, _expression: &Expression) -> VisitResult { VisitResult::default() }
|
||||
fn expression_kind(&mut self, _kind: &ExpressionKind) -> VisitResult { VisitResult::default() }
|
||||
|
||||
fn type_annotation(&mut self, _id: ItemId, _type_anno: Option<&TypeIdentifier>) -> VisitResult { VisitResult::default() }
|
||||
|
||||
fn named_struct(&mut self, _name: &QualifiedName, _fields: &Vec<(Rc<String>, Expression)>) -> VisitResult { VisitResult::default() }
|
||||
|
||||
fn nat_literal(&mut self, _id: ItemId, _n: u64)-> VisitResult { VisitResult::default() }
|
||||
fn float_literal(&mut self, _id: ItemId, _f: f64) -> VisitResult { VisitResult::default() }
|
||||
fn string_literal(&mut self, _id: ItemId, _s: &Rc<String>) -> VisitResult { VisitResult::default() }
|
||||
fn bool_literal(&mut self, _id: ItemId, _b: bool) -> VisitResult { VisitResult::default() }
|
||||
|
||||
fn qualified_name(&mut self, _id: ItemId, _name: &QualifiedName) {}
|
||||
}
|
||||
|
||||
enum VisitResult {
|
||||
KeepRecursing,
|
||||
Stop
|
||||
}
|
||||
|
||||
impl std::default::Default for VisitResult {
|
||||
fn default() -> VisitResult { VisitResult::KeepRecursing }
|
||||
}
|
||||
|
||||
pub fn traverse<V: ASTVisitorNew>(visitor: &mut V, expression: &Expression) {
|
||||
expresion(visitor, expresion)
|
||||
}
|
||||
|
||||
fn expression<V: ASTVisitorNew>(visitor: &mut V, expression: &Expression) {
|
||||
use ExpressionKind::*;
|
||||
|
||||
let Expression { id, kind, type_anno } = expression;
|
||||
|
||||
let output = match kind {
|
||||
NatLiteral(n) => v.nat_literal(id, *n),
|
||||
FloatLiteral(f) => v.float_literal(id, *f),
|
||||
StringLiteral(s) => v.string_literal(id, s),
|
||||
BoolLiteral(b) => v.bool_literal(d, *b),
|
||||
BinExp(op, lhs, rhs) => {
|
||||
expression(v, lhs);
|
||||
expression(v, rhs);
|
||||
},
|
||||
PrefixExp(op, arg) => {
|
||||
expression(v, arg);
|
||||
}
|
||||
TupleLiteral(exprs) => {
|
||||
for expr in exprs {
|
||||
expression(v, expr);
|
||||
}
|
||||
},
|
||||
Value(name) => v.qualified_name(id, name),
|
||||
NamedStruct { name, fields } => {
|
||||
for (_, expr) in fields.iter() {
|
||||
v.expression(expr);
|
||||
}
|
||||
}
|
||||
Call { f, arguments } => {
|
||||
expression(v, f);
|
||||
for arg in args.iter() {
|
||||
match arg {
|
||||
InvocationArgument::Positional(expr) => {
|
||||
expression(v, expr);
|
||||
},
|
||||
InvocationArgument::Keyword { expr, .. } => {
|
||||
expression(v, expr);
|
||||
},
|
||||
Ignored => (),
|
||||
}
|
||||
}
|
||||
},
|
||||
Index { indexee, indexers } => {
|
||||
expression(v, indexee);
|
||||
for i in indexers.iter() {
|
||||
expression(v, i);
|
||||
}
|
||||
},
|
||||
IfExpression { discriminator, body } => {
|
||||
discriminator.as_ref().map(|d| expression(v, d));
|
||||
match body {
|
||||
IfExpressionBody::SimpleConditional { then_case, else_case } => {
|
||||
walk_block(v, then_case);
|
||||
else_case.as_ref().map(|block| walk_block(v, block));
|
||||
},
|
||||
IfExpressionBody::SimplePatternMatch { pattern, then_case, else_case } => {
|
||||
v.pattern(pattern);
|
||||
walk_pattern(v, pattern);
|
||||
walk_block(v, then_case);
|
||||
else_case.as_ref().map(|block| walk_block(v, block));
|
||||
},
|
||||
IfExpressionBody::CondList(arms) => {
|
||||
for arm in arms {
|
||||
v.condition_arm(arm);
|
||||
condition_arm(v, arm);
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
WhileExpression { condition, body } => (),
|
||||
ForExpression { enumerators, body } => (),
|
||||
Lambda { params , type_anno, body } => {
|
||||
for param in params {
|
||||
formal_param(v, param);
|
||||
}
|
||||
v.type_annotation(type_anno);
|
||||
},
|
||||
ListLiteral(exprs) => {
|
||||
for expr in exprs {
|
||||
expression(v, expr);
|
||||
}
|
||||
},
|
||||
};
|
||||
output
|
||||
}
|
||||
|
||||
fn formal_param<V: ASTVisitorNew>(v: &mut V, param: &FormalParam) {
|
||||
param.default.as_ref().map(|p| {
|
||||
expression(v, p);
|
||||
});
|
||||
v.type_annotation(param.anno.as_ref());
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user