Add function to get the latest program (#1352)

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## v9.2.5
+
+* [Add function to get the latest program](https://github.com/TypeStrong/ts-loader/pull/1352) - thanks @Zn4rK
+
 ## v9.2.4
 
 * [Fix undefined configPath now falls back to default](https://github.com/TypeStrong/ts-loader/pull/1346) - thanks @johnnyreilly

README.md

@@ -333,7 +333,7 @@ These options should be functions which will be used to resolve the import state
 #### getCustomTransformers
 | Type |
 |------|
-| ` (program: Program) => { before?: TransformerFactory<SourceFile>[]; after?: TransformerFactory<SourceFile>[]; afterDeclarations?: TransformerFactory<SourceFile>[]; } ` |
+| ` (program: Program, getProgram: () => Program) => { before?: TransformerFactory<SourceFile>[]; after?: TransformerFactory<SourceFile>[]; afterDeclarations?: TransformerFactory<SourceFile>[]; } ` |
 
 Provide custom transformers - only compatible with TypeScript 2.3+ (and 2.4 if using `transpileOnly` mode). For example usage take a look at [typescript-plugin-styled-components](https://github.com/Igorbek/typescript-plugin-styled-components) or our [test](test/comparison-tests/customTransformer).
 

package.json

@@ -1,6 +1,6 @@
 {
   "name": "ts-loader",
-  "version": "9.2.4",
+  "version": "9.2.5",
   "description": "TypeScript loader for webpack",
   "main": "index.js",
   "types": "dist",

src/instances.ts

@@ -377,7 +377,8 @@ export function initializeInstance(
           })
         : instance.compiler.createProgram([], instance.compilerOptions));
 
-    instance.transformers = getCustomTransformers(program);
+    const getProgram = () => program;
+    instance.transformers = getCustomTransformers(program, getProgram);
     // Setup watch run for solution building
     if (instance.solutionBuilderHost) {
       addAssetHooks(loader, instance);
@@ -407,9 +408,13 @@ export function initializeInstance(
         instance.compiler.createWatchProgram(instance.watchHost);
       instance.builderProgram =
         instance.watchOfFilesAndCompilerOptions.getProgram();
-      instance.program = instance.builderProgram.getProgram();
 
-      instance.transformers = getCustomTransformers(instance.program);
+      const getProgram = () => instance.builderProgram?.getProgram();
+      instance.program = getProgram();
+      instance.transformers = getCustomTransformers(
+        instance.program,
+        getProgram
+      );
     } else {
       instance.servicesHost = makeServicesHost(
         getScriptRegexp(instance),
@@ -423,9 +428,8 @@ export function initializeInstance(
         instance.compiler.createDocumentRegistry()
       );
 
-      instance.transformers = getCustomTransformers(
-        instance.languageService!.getProgram()
-      );
+      const getProgram = () => instance.languageService!.getProgram();
+      instance.transformers = getCustomTransformers(getProgram(), getProgram);
     }
 
     addAssetHooks(loader, instance);

src/interfaces.ts

@@ -270,7 +270,8 @@ export interface LoaderOptions {
   getCustomTransformers:
     | string
     | ((
-        program: typescript.Program
+        program: typescript.Program,
+        getProgram: () => typescript.Program
       ) => typescript.CustomTransformers | undefined);
   experimentalWatchApi: boolean;
   allowTsInNodeModules: boolean;