fix(@ngtools/webpack): cleanup resources after modules are loaded

Author: clydin

packages/ngtools/webpack/src/angular_compiler_plugin.ts

@@ -134,7 +134,7 @@ export class AngularCompilerPlugin {
   private _program: (ts.Program | Program) | null;
   private _compilerHost: WebpackCompilerHost & CompilerHost;
   private _moduleResolutionCache: ts.ModuleResolutionCache;
-  private _resourceLoader: WebpackResourceLoader;
+  private _resourceLoader?: WebpackResourceLoader;
   // Contains `moduleImportPath#exportName` => `fullModulePath`.
   private _lazyRoutes: LazyRouteMap = Object.create(null);
   private _tsConfigPath: string;
@@ -580,10 +580,22 @@ export class AngularCompilerPlugin {
   }
 
   // Registration hook for webpack plugin.
+  // tslint:disable-next-line:no-big-function
   apply(compiler: Compiler & { watchMode?: boolean }) {
     // only present for webpack 4.23.0+, assume true otherwise
     const watchMode = compiler.watchMode === undefined ? true : compiler.watchMode;
 
+    // cleanup if not watching
+    if (!watchMode) {
+      compiler.hooks.thisCompilation.tap('angular-compiler', compilation => {
+        compilation.hooks.finishModules.tap('angular-compiler', () => {
+          this._program = null;
+          this._transformers = [];
+          this._resourceLoader = undefined;
+        });
+      });
+    }
+
     // Decorate inputFileSystem to serve contents of CompilerHost.
     // Use decorated inputFileSystem in watchFileSystem.
     compiler.hooks.environment.tap('angular-compiler', () => {
@@ -631,9 +643,11 @@ export class AngularCompilerPlugin {
         this._options.directTemplateLoading,
       );
 
-      // Create and set a new WebpackResourceLoader.
-      this._resourceLoader = new WebpackResourceLoader();
-      webpackCompilerHost.setResourceLoader(this._resourceLoader);
+      // Create and set a new WebpackResourceLoader in AOT
+      if (!this._JitMode) {
+        this._resourceLoader = new WebpackResourceLoader();
+        webpackCompilerHost.setResourceLoader(this._resourceLoader);
+      }
 
       // Use the WebpackCompilerHost with a resource loader to create an AngularCompilerHost.
       this._compilerHost = createCompilerHost({
@@ -787,7 +801,9 @@ export class AngularCompilerPlugin {
     (compilation as any)._ngToolsWebpackPluginInstance = this;
 
     // Update the resource loader with the new webpack compilation.
-    this._resourceLoader.update(compilation);
+    if (this._resourceLoader) {
+      this._resourceLoader.update(compilation);
+    }
 
     try {
       await this._update();
@@ -1012,6 +1028,10 @@ export class AngularCompilerPlugin {
   }
 
   getResourceDependencies(fileName: string): string[] {
+    if (!this._resourceLoader) {
+      return [];
+    }
+
     return this._resourceLoader.getResourceDependencies(fileName);
   }