Add function to get the latest program

Author: Zn4rK

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

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;