Add function to get the latest program

TypeStrong · Aug 3, 2021 · e41dc35 · e41dc35
1 parent 55d2a85
commit e41dc35
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 8 deletions.
diff --git a/CHANGELOG.md b/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

diff --git a/package.json b/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",

diff --git a/src/instances.ts b/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);

diff --git a/src/interfaces.ts b/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;