Feature(compiler): Support more types as fn return by using NativeTypeResolver

ovr · ovr · commit 1847d804ef7a · 2018-11-17T23:13:39.000+03:00
diff --git a/src/backend/llvm/index.ts b/src/backend/llvm/index.ts
@@ -161,18 +161,7 @@ export function passFunctionDeclaration(parent: ts.FunctionDeclaration, ctx: Con
         throw Error('Function must be declared with return type');
     }
 
-    let returnType = llvm.Type.getVoidTy(ctx.llvmContext);
-
-    switch (parent.type.kind) {
-        case ts.SyntaxKind.NumberKeyword:
-            returnType = llvm.Type.getDoubleTy(ctx.llvmContext);
-            break;
-        default:
-            throw Error(
-                `Function declared with unsupported return type, unexpected "${parent.type.kind}"`
-            );
-    }
-
+    let returnType = NativeTypeResolver.getType(ctx.typeChecker.getTypeFromTypeNode(parent.type), ctx).getType();
     let fnType = llvm.FunctionType.get(returnType, false);
     let fn = llvm.Function.create(fnType, llvm.LinkageTypes.ExternalLinkage, <string>parent.name.escapedText, ctx.llvmModule);
 
