fix: make generateTrace work with newer TypeScript versions

Closes: TypeStrong#722

Author: mkantor

README.md

@@ -347,7 +347,7 @@ yarn add --dev @types/webpack
 
 ## Profiling types resolution
 
-When using TypeScript 4.1.x in `build` mode you can profile long type checks by
+When using TypeScript 4.3.0 or newer in `build` mode you can profile long type checks by
 setting "generateTrace" compiler option. This is an instruction from [microsoft/TypeScript#40063](https://github.com/microsoft/TypeScript/pull/40063):
 
 1. Set "generateTrace": "{folderName}" in your `tsconfig.json` (under `compilerOptions`)

src/typescript/worker/lib/tracing.ts

@@ -4,37 +4,48 @@ import { getConfigFilePathFromCompilerOptions } from './config';
 import { typescript } from './typescript';
 import { config } from './worker-config';
 
-// write this type as it's available only starting from TypeScript 4.1.0
+// these types are internal in TypeScript, so reproduce them here
+type Mode = 'project' | 'build' | 'server';
 interface Tracing {
-  startTracing(configFilePath: string, traceDirPath: string, isBuildMode: boolean): void;
-  stopTracing(typeCatalog: unknown): void;
-  dumpLegend(): void;
+  startTracing:
+    | undefined
+    | ((tracingMode: Mode, traceDir: string, configFilePath?: string) => void);
+
+  tracing:
+    | undefined
+    | {
+        stopTracing(): void;
+        dumpLegend(): void;
+      };
 }
 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-const tracing: Tracing | undefined = (typescript as any).tracing;
+const traceableTypescript: Tracing = typescript as any;
 
 export function startTracingIfNeeded(compilerOptions: ts.CompilerOptions) {
-  if (compilerOptions.generateTrace && tracing) {
-    tracing.startTracing(
-      getConfigFilePathFromCompilerOptions(compilerOptions),
-      compilerOptions.generateTrace as string,
-      config.build
+  if (
+    typeof compilerOptions.generateTrace === 'string' &&
+    typeof traceableTypescript.startTracing === 'function'
+  ) {
+    traceableTypescript.startTracing(
+      config.build ? 'build' : 'project',
+      compilerOptions.generateTrace,
+      getConfigFilePathFromCompilerOptions(compilerOptions)
     );
   }
 }
 
 export function stopTracingIfNeeded(program: ts.BuilderProgram) {
   const compilerOptions = program.getCompilerOptions();
 
-  if (compilerOptions.generateTrace && tracing) {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    tracing.stopTracing((program.getProgram() as any).getTypeCatalog());
+  if (
+    typeof compilerOptions.generateTrace === 'string' &&
+    typeof traceableTypescript.tracing?.stopTracing === 'function'
+  ) {
+    traceableTypescript.tracing.stopTracing();
   }
 }
 
 export function dumpTracingLegendIfNeeded() {
-  if (tracing) {
-    tracing.dumpLegend();
-  }
+  traceableTypescript.tracing?.dumpLegend();
 }

test/e2e/type-script-tracing.spec.ts

@@ -3,36 +3,38 @@ import path from 'path';
 import { extractWebpackErrors } from './driver/webpack-errors-extractor';
 
 describe('TypeScript Tracing', () => {
-  it.each([{ build: true, typescript: '~4.1.0' }])(
-    'can generate trace files for %p',
-    async ({ build, ...dependencies }) => {
-      await sandbox.load(path.join(__dirname, 'fixtures/typescript-basic'));
-      await sandbox.install('yarn', { ...dependencies });
+  it.each([
+    { build: true, typescript: '~4.3.0' },
+    { build: true, typescript: '~4.4.0' },
+    { build: true, typescript: '~4.5.0' },
+    { build: true, typescript: '~4.6.0' },
+  ])('can generate trace files for %p', async ({ build, ...dependencies }) => {
+    await sandbox.load(path.join(__dirname, 'fixtures/typescript-basic'));
+    await sandbox.install('yarn', { ...dependencies });
 
-      // enable tracing
-      await sandbox.patch(
-        'tsconfig.json',
-        '"outDir": "./dist"',
-        '"outDir": "./dist",\n"generateTrace": "./traces"'
-      );
+    // enable tracing
+    await sandbox.patch(
+      'tsconfig.json',
+      '"outDir": "./dist"',
+      '"outDir": "./dist",\n"generateTrace": "./traces"'
+    );
 
-      await sandbox.write(
-        'fork-ts-checker.config.js',
-        `module.exports = ${JSON.stringify({ typescript: { build } })};`
-      );
+    await sandbox.write(
+      'fork-ts-checker.config.js',
+      `module.exports = ${JSON.stringify({ typescript: { build } })};`
+    );
 
-      const webpackResult = await sandbox.exec('yarn webpack --mode=development');
-      const errors = extractWebpackErrors(webpackResult);
-      expect(errors).toEqual([]);
+    const webpackResult = await sandbox.exec('yarn webpack --mode=development');
+    const errors = extractWebpackErrors(webpackResult);
+    expect(errors).toEqual([]);
 
-      expect(await sandbox.exists('dist')).toEqual(true);
+    expect(await sandbox.exists('dist')).toEqual(true);
 
-      expect(await sandbox.list('./traces')).toEqual(
-        expect.arrayContaining([
-          expect.objectContaining({ name: expect.stringMatching(/types.*\.json/) }),
-          expect.objectContaining({ name: expect.stringMatching(/trace.*\.json/) }),
-        ])
-      );
-    }
-  );
+    expect(await sandbox.list('./traces')).toEqual(
+      expect.arrayContaining([
+        expect.objectContaining({ name: expect.stringMatching(/types.*\.json/) }),
+        expect.objectContaining({ name: expect.stringMatching(/trace.*\.json/) }),
+      ])
+    );
+  });
 });