From 5ed925439f3d267a71df78a6d75274bd6b7c958d Mon Sep 17 00:00:00 2001
From: kimamula <kenji.imamula@gmail.com>
Date: Thu, 21 Jun 2018 10:45:28 +0900
Subject: [PATCH 1/2] feat: pass ts.Program as an argument for
 getCustomTransformers

---
 src/__test__/custom-transformers.ts | 40 +++++++++++++++++++++++++++++
 src/checker/runtime.ts              |  7 ++---
 src/interfaces.ts                   |  2 +-
 3 files changed, 45 insertions(+), 4 deletions(-)
 create mode 100644 src/__test__/custom-transformers.ts

diff --git a/src/__test__/custom-transformers.ts b/src/__test__/custom-transformers.ts
new file mode 100644
index 0000000..3708ef2
--- /dev/null
+++ b/src/__test__/custom-transformers.ts
@@ -0,0 +1,40 @@
+import {src, webpackConfig, tsconfig, compile, checkOutput, expectErrors, spec, expect, query} from './utils'
+
+[true, false].forEach(transpileOnly => {
+	spec(`${__filename}-transpileOnly:${transpileOnly}`, async function() {
+		src(
+			'index.ts',
+			`
+			class HiThere {
+				constructor(a: number, b: string) {
+					const t = a + b;
+				}
+			}
+		`
+		)
+
+		tsconfig()
+		let getCustomTransformersCalled = false
+		let stats = await compile(webpackConfig(query({
+			transpileOnly,
+			getCustomTransformers: program => {
+				expect(typeof program.getTypeChecker === 'function').true
+				getCustomTransformersCalled = true
+				return {};
+			}
+		})))
+
+		expectErrors(stats, 0)
+		checkOutput(
+			'index.js',
+			`
+			class HiThere {
+				constructor(a, b) {
+					const t = a + b;
+				}
+			}
+		`
+		)
+		expect(getCustomTransformersCalled).true
+	})
+})
diff --git a/src/checker/runtime.ts b/src/checker/runtime.ts
index 006f57a..8af6e34 100644
--- a/src/checker/runtime.ts
+++ b/src/checker/runtime.ts
@@ -104,7 +104,7 @@ function createChecker(receive: (cb: (msg: Req) => void) => void, send: (msg: Re
 	let watchHost: WatchHost
 	let watch: ts.WatchOfFilesAndCompilerOptions<ts.SemanticDiagnosticsBuilderProgram>
 
-	let finalTransformers: undefined | (() => ts.CustomTransformers)
+	let finalTransformers: undefined | ((program: ts.Program) => ts.CustomTransformers)
 
 	function createWatchHost(): WatchHost {
 		return {
@@ -410,7 +410,7 @@ function createChecker(receive: (cb: (msg: Req) => void) => void, send: (msg: Re
 			writeFile,
 			/*cancellationToken*/ undefined,
 			/*emitOnlyDtsFiles*/ false,
-			finalTransformers && finalTransformers()
+			finalTransformers && finalTransformers(program.getProgram())
 		)
 		return outputFiles
 	}
@@ -433,7 +433,8 @@ function createChecker(receive: (cb: (msg: Req) => void) => void, send: (msg: Re
 		const trans = compiler.transpileModule(files.get(fileName).text, {
 			compilerOptions: compilerOptions,
 			fileName,
-			reportDiagnostics: false
+			reportDiagnostics: false,
+			transformers: finalTransformers ? finalTransformers(getProgram().getProgram()) : undefined
 		})
 
 		return {
diff --git a/src/interfaces.ts b/src/interfaces.ts
index 928ebc2..e49024b 100644
--- a/src/interfaces.ts
+++ b/src/interfaces.ts
@@ -28,7 +28,7 @@ export interface LoaderConfig {
 	debug?: boolean
 	reportFiles?: string[]
 	context?: string
-	getCustomTransformers?: string | (() => ts.CustomTransformers | undefined)
+	getCustomTransformers?: string | ((program: ts.Program) => ts.CustomTransformers | undefined)
 }
 
 export interface OutputFile {

From 374a068de1da39aeccb695a7f7e00341d1b42da4 Mon Sep 17 00:00:00 2001
From: kimamula <kenji.imamula@gmail.com>
Date: Thu, 21 Jun 2018 11:29:16 +0900
Subject: [PATCH 2/2] docs: add an explanation for getCustomTransformers

---
 README.md | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/README.md b/README.md
index 558e2f4..d9a1af8 100644
--- a/README.md
+++ b/README.md
@@ -222,6 +222,32 @@ reportFiles: [
 ]
 ```
 
+### getCustomTransformers *(string | ((program: ts.Program) => ts.CustomTransformers | undefined)) (default=undefined)*
+
+Provide custom transformers, TypeScript 2.4.1+. Example:
+
+```js
+const styledComponentsTransformer = require('typescript-plugin-styled-components').default;
+const keysTransformer = require('ts-transformer-keys/transformer').default;
+
+// ...
+rules: [
+    {
+        test: /\.tsx?$/,
+        loader: 'awesome-typescript-loader',
+        options: {
+            // ... other loader's options
+            getCustomTransformers: program => ({
+                before: [
+                    styledComponentsTransformer(),
+                    keysTransformer(program)
+                ]
+            })
+        }
+    }
+]
+```
+
 ## Compiler options
 
 You can pass compiler options inside the loader query string or in a TS config file.