Feature(Backend\LLVM): Introduce SymbolTable

ovr · ovr · commit 35033ad64e4d · 2018-10-06T22:22:50.000+09:00
diff --git a/src/Backend/LLVM/index.ts b/src/Backend/LLVM/index.ts
@@ -11,7 +11,9 @@ import {
     CallExpression,
     SpreadElement,
     JSXNamespacedName,
-    FunctionDeclaration, Identifier, TSTypeAnnotation, ReturnStatement,
+    FunctionDeclaration,
+    Identifier,
+    ReturnStatement,
 } from '@babel/types';
 
 export function passBlockStatement(parent: BlockStatement, ctx: Context, builder: llvm.IRBuilder) {
@@ -97,6 +99,11 @@ function buildFromCallExpression(
 }
 
 function buildFromIdentifier(block: Identifier, ctx: Context, builder: llvm.IRBuilder): llvm.Value {
+    const variable = ctx.variables.get(block.name);
+    if (variable) {
+        return variable;
+    }
+
     return ctx.llvmModule.getFunction(block.name);
 }
 
@@ -125,6 +132,10 @@ export function passVariableDeclaration(block: VariableDeclaration, ctx: Context
     if (declaration.init) {
         const right = buildFromExpression(declaration.init, ctx, builder);
 
+        if (declaration.id.type === 'Identifier') {
+            ctx.variables.set(declaration.id.name, right);
+        }
+
         return;
     }
 
@@ -154,9 +165,14 @@ export function passStatement(stmt: Statement, ctx: Context, builder: llvm.IRBui
     }
 }
 
+class SymbolTable extends Map<string, llvm.Value> {
+
+}
+
 class Context {
     public llvmContext: llvm.LLVMContext;
     public llvmModule: llvm.Module;
+    public variables: SymbolTable = new SymbolTable();
 
     public constructor() {
         this.llvmContext = new llvm.LLVMContext();
diff --git a/src/Frontend/TypeScript/index.ts b/src/Frontend/TypeScript/index.ts
@@ -2,13 +2,11 @@ import {parse} from '@babel/parser'
 
 const example = `
 {
-    function returnVoid(): void {
-        return;
-    }
+    const a = 1;
+    const b = 2;
+    const c = a + b;
 
-    returnVoid();
-
-    puts("Hello World!");
+    puts(c);
 }
 `;
 
