schala/schala-lang/language/src/ast/visitor.rs

use std::rc::Rc;
use std::error::Error;
use crate::ast::*;
use crate::ast::walker;

//TODO maybe these functions should take closures that return a KeepRecursing | StopHere type,
//or a tuple of (T, <that type>)

//TODO default implmentations should call walk methods - then I can test printing
pub trait ASTVisitor: Sized {
  fn visit(&mut self, ast: &AST) {
    println!("FUCK");
    walker::ast(self, ast);
  }

  fn block(&mut self, statements: &Vec<Statement>) {
    println!("oi");
    walker::block(self, statements);
  }

  fn statement(&mut self, statement: &Statement) {
    println!("stmt");
    walker::statement(self, statement);
  }

  fn declaration(&mut self, declaration: &Declaration) {
    walker::declaration(self, declaration);
  }

  fn signature(&mut self, signature: &Signature) {
  }

  fn binding(&mut self, name: &Rc<String>, constant: bool, type_anno: Option<&TypeIdentifier>, expr: &Expression) {
    walker::maybe_type_identifier(self, type_anno);
    walker::expression(self, expr);
  }

  fn expression(&mut self, expression: &Expression) {
    println!("expr yo");
    walker::expression(self, expression);
  }

  fn expression_kind(&mut self, kind: &ExpressionKind) {
    walker::expression_kind(self, kind);
  }

  fn maybe_type_identifier(&mut self, type_anno: Option<&TypeIdentifier>) {
    walker::maybe_type_identifier(self, type_anno);
  }

  fn named_struct(&mut self, name: &QualifiedName, fields: &Vec<(Rc<String>, Expression)>) {
    walker::named_struct(self, name, fields);
  }

  fn call(&mut self, f: &Expression, arguments: &Vec<InvocationArgument>) {
    walker::call(self, f, arguments);
  }

  fn index(&mut self, indexee: &Expression, indexers: &Vec<Expression>) {
    walker::index(self, indexee, indexers);
  }

  fn if_expression(&mut self, discrim: &Discriminator, body: &IfExpressionBody) {
  }

  fn while_expression(&mut self, condition: Option<&Expression>, body: &Block) {

  }
  fn for_expression(&mut self, enumerators: &Vec<Enumerator>, body: &ForBody) {

  }

  fn lambda(&mut self, params: &Vec<FormalParam>, type_anno: Option<&TypeIdentifier>, body: &Block) {
    walker::lambda(self, params, type_anno, body);
  }

  fn invocation_argument(&mut self, arg: &InvocationArgument) {
  }

  fn formal_param(&mut self, param: &FormalParam) {
    walker::formal_param(self, param);
  }

  fn type_anno(&mut self, anno: &TypeIdentifier) { }
  fn import(&mut self, import: &ImportSpecifier) {}
  fn qualified_name(&mut self, name: &QualifiedName) {}
  fn nat_literal(&mut self, n: u64) {}
  fn float_literal(&mut self, f: f64) {}
  fn string_literal(&mut self, s: &Rc<String>) {}
  fn bool_literal(&mut self, b: bool) {}
  fn binexp(&mut self, op: &BinOp, lhs: &Expression, rhs: &Expression) {
    walker::expression(self, lhs);
    walker::expression(self, rhs);
  }

  fn prefix_exp(&mut self, op: &PrefixOp, arg: &Expression) {
    walker::expression(self, arg);
  }
}