Feature(compiler): BreakStatement - initial support

ovr · ovr · commit 219061f59950 · 2018-12-08T00:10:55.000+09:00
diff --git a/src/backend/llvm/code-generation/break-statement.ts b/src/backend/llvm/code-generation/break-statement.ts
@@ -0,0 +1,26 @@
+
+import * as ts from "typescript";
+import * as llvm from 'llvm-node';
+import {NodeGenerateInterface} from "../node-generate.interface";
+import {Context} from "../context";
+import UnsupportedError from "../../error/unsupported.error";
+
+export class BreakStatementGenerator implements NodeGenerateInterface<ts.BreakStatement, void> {
+    generate(node: ts.BreakStatement, ctx: Context, builder: llvm.IRBuilder): void {
+        if (node.label) {
+            throw new UnsupportedError(
+                node,
+                'Break with label is unsupported'
+            );
+        }
+
+        if (!ctx.scope.breakBlock) {
+            throw new UnsupportedError(
+                node,
+                'There is no pointer to jump outside this block to break'
+            );
+        }
+
+        builder.createBr(ctx.scope.breakBlock);
+    }
+}
diff --git a/src/backend/llvm/index.ts b/src/backend/llvm/index.ts
@@ -15,6 +15,7 @@ import {ReturnStatementCodeGenerator} from "./code-generation/return-statement";
 import {ForStatementGenerator} from "./code-generation/for-statement";
 import {DoStatementGenerator} from "./code-generation/do-statement";
 import {WhileStatementGenerator} from "./code-generation/while-statement";
+import {BreakStatementGenerator} from "./code-generation/break-statement";
 
 export function passIfStatement(parent: ts.IfStatement, ctx: Context, builder: llvm.IRBuilder) {
     const positiveBlock = llvm.BasicBlock.create(ctx.llvmContext, "if.true");
@@ -493,6 +494,9 @@ export function passStatement(stmt: ts.Statement, ctx: Context, builder: llvm.IR
         case ts.SyntaxKind.ReturnStatement:
             new ReturnStatementCodeGenerator().generate(<any>stmt, ctx, builder);
             break;
+        case ts.SyntaxKind.BreakStatement:
+            new BreakStatementGenerator().generate(<any>stmt, ctx, builder);
+            break;
         case ts.SyntaxKind.IfStatement:
             passIfStatement(<any>stmt, ctx, builder);
             break;
