integrate wat codegen

src/generator/index.ts

@@ -16,6 +16,7 @@ import { NoInfer } from "../../wazum/dist/utils";
 import { Node } from "../ast/nodes/Node";
 import { BranchStatement as BranchStatement } from "../ast/nodes/BranchStatement";
 import { BranchToStatement } from "../ast/nodes/BranchToStatement";
+import { IfStatement } from "../ast/nodes/IfStatement";
 
 let offset: number = 0;
 export class Generator {
@@ -70,6 +71,8 @@ export class Generator {
         body.push(this.parseReturnStatement(stmt));
       else if (stmt instanceof BranchStatement)
         body.push(this.parseBranch(stmt));
+      else if (stmt instanceof IfStatement) 
+        body.push(this.parseIfStmt(stmt));
       else
         throw new Error(
           "Could not parse body of FunctionDeclaration to wasm equivalent!",
@@ -98,6 +101,12 @@ export class Generator {
       w.branch("a"),
     ]);
   }
+  parseIfStmt(node: IfStatement): w.BranchIf {
+    return w.branchIf(
+      "block",
+      w.constant("i32", 1)
+    )
+  }
   parseVariable(node: VariableDeclaration): w.LocalSet {
     if (node.value instanceof StringLiteral) {
       const value = this.parseStringLiteral(node.value as StringLiteral);

src/test.ts

@@ -2,6 +2,7 @@ import { writeFileSync } from "fs";
 import { Parser } from "./parser";
 import { Tokenizer } from "./tokenizer";
 import { Transpile } from "./transpiler/transpiler";
+import { Generator } from "./generator";
 
 const start = Date.now();
 const tokenizer = new Tokenizer(`
@@ -21,9 +22,13 @@ fn main(a: i32, b: i32) -> i32 {
 console.dir(tokenizer.getAll(), { depth: 10 });
 const parser = new Parser(tokenizer, "test.zp");
 const program = parser.parseSource();
-console.dir(program, { depth: 10 });
+console.dir(program.topLevelStatements, { depth: 1 });
 
 const transpiled = Transpile.from(program);
 console.log("Transpiled:\n" + transpiled);
 
+const generator = new Generator();
+generator.parseFn(program.topLevelStatements[1])
+console.log("WAT:\n" + generator.toWat());
+
 writeFileSync("./test.ts", transpiled);