Add Lua AST to TranspiledFile interface

TypeScriptToLua · May 2, 2019 · be51ac2 · be51ac2
1 parent d5b8e65
commit be51ac2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/Transpile.ts b/src/Transpile.ts
@@ -36,6 +36,7 @@ function getCustomTransformers(
 }
 
 export interface TranspiledFile {
+    luaAst?: Block;
     lua?: string;
     sourceMap?: string;
     declaration?: string;
@@ -101,14 +102,14 @@ export function transpile({
 
     const processSourceFile = (sourceFile: ts.SourceFile) => {
         try {
-            const [luaAST, lualibFeatureSet] = transformer.transformSourceFile(sourceFile);
+            const [luaAst, lualibFeatureSet] = transformer.transformSourceFile(sourceFile);
             if (!options.noEmit && !options.emitDeclarationOnly) {
                 const [lua, sourceMap] = printer.print(
-                    luaAST,
+                    luaAst,
                     lualibFeatureSet,
                     sourceFile.fileName
                 );
-                updateTranspiledFile(sourceFile.fileName, { lua, sourceMap });
+                updateTranspiledFile(sourceFile.fileName, { luaAst, lua, sourceMap });
             }
         } catch (err) {
             if (!(err instanceof TranspileError)) throw err;