@@ -8,77 +8,106 @@ import { angularJitApplicationTransform } from '../transformers/jit_transform';
88import { replaceResources } from '../transformers/replace-resources' ;
99
1010export class NgJestCompiler extends TsCompiler {
11- private readonly _defaultLibDirPath : string ;
12- private readonly _libSourceFileCache = new Map < string , ts . SourceFile > ( ) ;
13-
1411 constructor ( readonly configSet : ConfigSet , readonly jestCacheFS : Map < string , string > ) {
1512 super ( configSet , jestCacheFS ) ;
1613
1714 this . _logger . debug ( 'created NgJestCompiler' ) ;
18- this . _defaultLibDirPath = path . dirname ( this . _ts . getDefaultLibFilePath ( this . _compilerOptions ) ) ;
1915 }
2016
2117 /**
2218 * Copy from https://github.com/microsoft/TypeScript/blob/master/src/services/transpile.ts
2319 * This is required because the exposed function `transpileModule` from TypeScript doesn't allow to access `Program`
24- * and we need `Program` to be able to use Angular AST transformers .
20+ * and we need `Program` to be able to use Angular `replace-resources` transformer .
2521 */
2622 protected _transpileOutput ( fileContent : string , filePath : string ) : ts . TranspileOutput {
27- const scriptTarget = this . _compilerOptions . target ?? this . _ts . ScriptTarget . Latest ;
28- const sourceFile = this . _ts . createSourceFile ( filePath , fileContent , scriptTarget ) ;
23+ const diagnostics : ts . Diagnostic [ ] = [ ] ;
24+ const compilerOptions = { ...this . _compilerOptions } ;
25+ const options : ts . CompilerOptions = compilerOptions
26+ ? // @ts -expect-error internal TypeScript API
27+ this . _ts . fixupCompilerOptions ( compilerOptions , diagnostics )
28+ : { } ;
29+
30+ // mix in default options
31+ const defaultOptions = this . _ts . getDefaultCompilerOptions ( ) ;
32+ for ( const key in defaultOptions ) {
33+ if ( Object . prototype . hasOwnProperty . call ( defaultOptions , key ) && options [ key ] === undefined ) {
34+ options [ key ] = defaultOptions [ key ] ;
35+ }
36+ }
37+
38+ // @ts -expect-error internal TypeScript API
39+ for ( const option of this . _ts . transpileOptionValueCompilerOptions ) {
40+ options [ option . name ] = option . transpileOptionValue ;
41+ }
42+
43+ /**
44+ * transpileModule does not write anything to disk so there is no need to verify that there are no conflicts between
45+ * input and output paths.
46+ */
47+ options . suppressOutputPathCheck = true ;
48+
49+ // Filename can be non-ts file.
50+ options . allowNonTsExtensions = true ;
51+
52+ const sourceFile = this . _ts . createSourceFile (
53+ filePath ,
54+ fileContent ,
55+ options . target ?? this . _ts . ScriptTarget . Latest ,
56+ ) ;
57+
58+ let outputText : string | undefined ;
59+ let sourceMapText : string | undefined ;
60+
2961 const compilerHost : ts . CompilerHost = {
3062 getSourceFile : ( fileName ) => {
31- if ( fileName === path . normalize ( filePath ) ) {
32- return sourceFile ;
33- }
34-
35- let libSourceFile = this . _libSourceFileCache . get ( fileName ) ;
36- if ( ! libSourceFile ) {
37- const libFilePath = path . join ( this . _defaultLibDirPath , fileName ) ;
38- const libFileContent = this . _ts . sys . readFile ( libFilePath ) ?? '' ;
39- if ( libFileContent ) {
40- libSourceFile = this . _ts . createSourceFile ( fileName , libFileContent , scriptTarget ) ;
41- this . _libSourceFileCache . set ( fileName , libSourceFile ) ;
42- }
63+ return path . normalize ( fileName ) === path . normalize ( filePath ) ? sourceFile : undefined ;
64+ } ,
65+ writeFile : ( name , text ) => {
66+ if ( path . extname ( name ) === '.map' ) {
67+ sourceMapText = text ;
68+ } else {
69+ outputText = text ;
4370 }
44-
45- return libSourceFile ;
4671 } ,
47- // eslint-disable-next-line @typescript-eslint/no-empty-function
48- writeFile : ( ) => { } ,
4972 getDefaultLibFileName : ( ) => 'lib.d.ts' ,
5073 useCaseSensitiveFileNames : ( ) => false ,
5174 getCanonicalFileName : ( fileName ) => fileName ,
5275 getCurrentDirectory : ( ) => '' ,
5376 getNewLine : ( ) => os . EOL ,
54- fileExists : ( fileName ) => fileName === filePath ,
77+ fileExists : ( fileName ) => {
78+ return path . normalize ( fileName ) === path . normalize ( filePath ) ;
79+ } ,
5580 readFile : ( ) => '' ,
5681 directoryExists : ( ) => true ,
5782 getDirectories : ( ) => [ ] ,
5883 } ;
59- this . program = this . _ts . createProgram ( [ filePath ] , this . _compilerOptions , compilerHost ) ;
60-
61- return this . _ts . transpileModule ( fileContent , {
62- fileName : filePath ,
63- transformers : this . _makeTransformers ( this . configSet . resolvedTransformers ) ,
64- compilerOptions : this . _compilerOptions ,
65- reportDiagnostics : this . configSet . shouldReportDiagnostics ( filePath ) ,
66- } ) ;
84+
85+ this . program = this . _ts . createProgram ( [ filePath ] , options , compilerHost ) ;
86+ this . program . emit (
87+ undefined ,
88+ undefined ,
89+ undefined ,
90+ undefined ,
91+ this . _makeTransformers ( this . configSet . resolvedTransformers ) ,
92+ ) ;
93+
94+ return { outputText : outputText ?? '' , diagnostics, sourceMapText } ;
6795 }
6896
6997 protected _makeTransformers ( customTransformers : TsJestAstTransformer ) : ts . CustomTransformers {
70- const allTransformers = super . _makeTransformers ( customTransformers ) ;
98+ // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
99+ const program = this . program ! ;
71100
72101 return {
73- ...allTransformers . after ,
74- ...allTransformers . afterDeclarations ,
102+ ...super . _makeTransformers ( customTransformers ) . after ,
103+ ...super . _makeTransformers ( customTransformers ) . afterDeclarations ,
75104 before : [
76- ...( allTransformers . before ?? [ ] ) ,
77- // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
78- replaceResources ( this . program ! ) ,
79- // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
80- angularJitApplicationTransform ( this . program ! ) ,
81- ] ,
105+ ...customTransformers . before . map ( ( beforeTransformer ) =>
106+ beforeTransformer . factory ( this , beforeTransformer . options ) ,
107+ ) ,
108+ replaceResources ( program ) ,
109+ angularJitApplicationTransform ( program ) ,
110+ ] as Array < ts . TransformerFactory < ts . SourceFile > | ts . CustomTransformerFactory > ,
82111 } ;
83112 }
84113}
0 commit comments