Feature(compiler): Start working on Classes

ovr · ovr · commit 844243efcc3b · 2018-12-09T07:41:17.000+09:00
diff --git a/src/backend/llvm/code-generation/class-statement.ts b/src/backend/llvm/code-generation/class-statement.ts
@@ -0,0 +1,47 @@
+
+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";
+import {NativeTypeResolver} from "../native-type-resolver";
+
+export class ClassDeclarationGenerator implements NodeGenerateInterface<ts.ClassDeclaration, void> {
+    generate(node: ts.ClassDeclaration, ctx: Context, builder: llvm.IRBuilder): void {
+        if (node.typeParameters) {
+            throw new UnsupportedError(
+                node,
+                'Generic classes are unsupported'
+            )
+        }
+
+        const classType = ctx.typeChecker.getTypeAtLocation(node);
+
+        const properties = ctx.typeChecker.getPropertiesOfType(classType);
+
+        const struct = llvm.StructType.get(ctx.llvmContext, properties.map(
+            (property: ts.Symbol) => {
+                const nativeType = NativeTypeResolver.getType(
+                    ctx.typeChecker.getTypeOfSymbolAtLocation(property, node),
+                    ctx
+                );
+
+                return nativeType.getType();
+            }
+        ));
+
+        if (node.members) {
+            node.members.forEach(
+                (member) => {
+                    if (!ts.isPropertyDeclaration(member)) {
+                        throw new UnsupportedError(
+                            node,
+                            'Classes with methods are not supported'
+                        )
+                    }
+                }
+            );
+        }
+    }
+}
diff --git a/src/backend/llvm/index.ts b/src/backend/llvm/index.ts
@@ -17,6 +17,7 @@ import {DoStatementGenerator} from "./code-generation/do-statement";
 import {WhileStatementGenerator} from "./code-generation/while-statement";
 import {BreakStatementGenerator} from "./code-generation/break-statement";
 import {ContinueStatementGenerator} from "./code-generation/continue-statement";
+import {ClassDeclarationGenerator} from "./code-generation/class-statement";
 
 export function passIfStatement(parent: ts.IfStatement, ctx: Context, builder: llvm.IRBuilder) {
     const positiveBlock = llvm.BasicBlock.create(ctx.llvmContext, "if.true");
@@ -495,6 +496,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.ClassDeclaration:
+            new ClassDeclarationGenerator().generate(<any>stmt, ctx, builder);
+            break;
         case ts.SyntaxKind.BreakStatement:
             new BreakStatementGenerator().generate(<any>stmt, ctx, builder);
             break;
