Added function that returns the AST and string

lolleko · lolleko · commit 85fc1da9c9cb · 2019-01-15T21:44:03.000+01:00
diff --git a/src/LuaTranspiler.ts b/src/LuaTranspiler.ts
@@ -2,6 +2,8 @@ import * as fs from "fs";
 import * as path from "path";
 import * as ts from "typescript";
 
+import * as tstl from "./LuaAST";
+
 import {CompilerOptions, LuaLibImportKind} from "./CompilerOptions";
 import {LuaPrinter} from "./LuaPrinter";
 import {LuaTransformer} from "./LuaTransformer";
@@ -111,6 +113,13 @@ export class LuaTranspiler {
         return this.luaPrinter.print(luaAST, lualibFeatureSet);
     }
 
+    public transpileSourceFileKeepAST(sourceFile: ts.SourceFile): [tstl.Block, string] {
+        // Transform AST
+        const [luaAST, lualibFeatureSet] = this.luaTransformer.transformSourceFile(sourceFile);
+        // Print AST
+        return [luaAST, this.luaPrinter.print(luaAST, lualibFeatureSet)];
+    }
+
     public reportDiagnostic(diagnostic: ts.Diagnostic): void {
         if (diagnostic.file) {
             const {line, character} = diagnostic.file.getLineAndCharacterOfPosition(diagnostic.start!);
