implement visitors for define-syntax and syntax-rules

Author: s-kybound

src/transpiler/visitors/printer.ts

@@ -223,4 +223,33 @@ export class Printer implements Visitor {
     node.expression.accept(this.increment());
     this.display(") ");
   }
+
+  visitDefineSyntax(node: Atomic.DefineSyntax) {
+    // this.indent();
+    this.display("( define-syntax ");
+    node.name.accept(this.increment());
+    this.display(" ");
+    node.transformer.accept(this.increment());
+    this.display(") ");
+  }
+
+  visitSyntaxRules(node: Atomic.SyntaxRules) {
+    // this.indent();
+    this.display("( syntax-rules ");
+    this.display("( ");
+    node.literals.forEach(literal => {
+      literal.accept(this.increment());
+      this.display(" ");
+    });
+    this.display(") ");
+    node.rules.forEach(rule => {
+      this.display("\n");
+      this.display("( ");
+      rule[0].accept(this.increment());
+      this.display(" ");
+      rule[1].accept(this.increment());
+      this.display(") ");
+    });
+    this.display(") ");
+  }
 }

src/transpiler/visitors/redefiner.ts

@@ -226,4 +226,13 @@ export class Redefiner implements Visitor {
 
     return new Extended.Delay(location, newBody);
   }
+
+  // there are no redefinitions in the following nodes.
+  visitDefineSyntax(node: Atomic.DefineSyntax) {
+    return node;
+  }
+
+  visitSyntaxRules(node: Atomic.SyntaxRules) {
+    return node;
+  }
 }

src/transpiler/visitors/simplifier.ts

@@ -255,4 +255,13 @@ export class Simplifier implements Visitor {
     const makePromise = new Atomic.Identifier(location, "make-promise");
     return new Atomic.Application(location, makePromise, [delayedLambda]);
   }
+
+  // these nodes are already in their simplest form
+  visitDefineSyntax(node: Atomic.DefineSyntax) {
+    return node;
+  }
+
+  visitSyntaxRules(node: Atomic.SyntaxRules) {
+    return node;
+  }
 }

src/transpiler/visitors/transpiler.ts

@@ -410,4 +410,10 @@ export class Transpiler implements Visitor {
   visitDelay(node: Extended.Delay): [es.ArrowFunctionExpression] {
     throw new Error("The AST should be simplified!");
   }
+  visitDefineSyntax(node: Atomic.DefineSyntax) {
+    throw new Error("This should not be called!");
+  }
+  visitSyntaxRules(node: Atomic.SyntaxRules) {
+    throw new Error("This should not be called!");
+  }
 }

src/transpiler/visitors/visitor.ts

@@ -32,6 +32,9 @@ export interface Visitor {
 
   visitVector(node: Atomic.Vector): any;
 
+  visitSyntaxRules(node: Atomic.SyntaxRules): any;
+  visitDefineSyntax(node: Atomic.DefineSyntax): any;
+
   // Extended AST
   visitFunctionDefinition(node: Extended.FunctionDefinition): any;
   visitLet(node: Extended.Let): any;