TypeScriptToLua
diff --git a/‎src/index.ts‎
Lines changed: 1 addition & 0 deletions b/‎src/index.ts‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎test/transpile/plugins/afterPrint.ts‎
Lines changed: 17 additions & 0 deletions b/‎test/transpile/plugins/afterPrint.ts‎
Lines changed: 17 additions & 0 deletions
diff --git a/‎test/transpile/plugins/beforeTransform.ts‎
Lines changed: 15 additions & 0 deletions b/‎test/transpile/plugins/beforeTransform.ts‎
Lines changed: 15 additions & 0 deletions
diff --git a/‎test/transpile/plugins/plugins.spec.ts‎
Lines changed: 33 additions & 1 deletion b/‎test/transpile/plugins/plugins.spec.ts‎
Lines changed: 33 additions & 1 deletion
@@ -8,3 +8,4 @@ export { LuaLibFeature } from "./LuaLib";
 export * from "./LuaPrinter";
 export * from "./transformation/context";
 export * from "./transpilation";
+export { ProcessedFile } from "./transpilation/utils";
@@ -0,0 +1,17 @@
+import * as ts from "typescript";
+import * as tstl from "../../../src";
+
+const plugin: tstl.Plugin = {
+    afterPrint(program: ts.Program, options: tstl.CompilerOptions, emitHost: tstl.EmitHost, result: tstl.ProcessedFile[]) {
+        void(program);
+        void(options);
+        void(emitHost);
+
+        for (const file of result) {
+            file.code = "-- Commented added by afterPrint plugin\n" + file.code;
+        }
+    }
+};
+
+// eslint-disable-next-line import/no-default-export
+export default plugin;
@@ -0,0 +1,15 @@
+import * as ts from "typescript";
+import * as tstl from "../../../src";
+
+const plugin: tstl.Plugin = {
+    beforeTransform(program: ts.Program, options: tstl.CompilerOptions, emitHost: tstl.EmitHost) {
+        void(program);
+        void(emitHost);
+
+        // Modify settings
+        options.outDir = "plugin/beforeTransform/outdir";
+    }
+};
+
+// eslint-disable-next-line import/no-default-export
+export default plugin;
@@ -78,8 +78,40 @@ test.each(["namespace", "module"])("%s with TS transformer plugin", moduleOrName
                     return false;
                 }
             }
-        `
+            `
         )
         .setOptions({ plugins: [{ transform: path.join(__dirname, "transformer-plugin.ts") }] })
         .expectNoExecutionError();
 });
+
+test("beforeTransform plugin", () => {
+    const { transpiledFiles } = util.testModule`
+        console.log("Hello, World!");
+    `
+        .setOptions({ luaPlugins: [{ name: path.join(__dirname, "beforeTransform.ts") }] })
+        .getLuaResult();
+
+    expect(transpiledFiles).toHaveLength(1);
+    // Expect emitted to output path set by the plugin
+    expect(transpiledFiles[0].outPath).toContain(path.join("plugin", "beforeTransform", "outdir"));
+});
+
+test("afterPrint plugin", () => {
+    const { transpiledFiles } = util.testModule`
+        console.log("Hello, World!");
+    `
+        .addExtraFile(
+            "extrafile.ts",
+            `
+                console.log("Hello, Mars!");
+            `
+        )
+        .setOptions({ luaPlugins: [{ name: path.join(__dirname, "afterPrint.ts") }] })
+        .getLuaResult();
+
+    expect(transpiledFiles).toHaveLength(2);
+    for (const f of transpiledFiles) {
+        // Expect plugin inserted extra lua at start of file
+        expect(f.lua).toContain("-- Commented added by afterPrint plugin");
+    }
+});