fix(babel): several massive memory leaks

packages/babel/src/cache.ts

@@ -10,15 +10,14 @@ function hashContent(content: string) {
 }
 
 interface ICaches {
-  contentHashes: Map<string, string>;
   entrypoints: Map<string, Entrypoint | IEvaluatedEntrypoint>;
 }
 
 type MapValue<T> = T extends Map<string, infer V> ? V : never;
 
 const cacheLogger = linariaLogger.extend('cache');
 
-const cacheNames = ['entrypoints', 'contentHashes'] as const;
+const cacheNames = ['entrypoints'] as const;
 type CacheNames = (typeof cacheNames)[number];
 
 const loggers = cacheNames.reduce(
@@ -30,10 +29,7 @@ const loggers = cacheNames.reduce(
 );
 
 export class TransformCacheCollection {
-  protected readonly entrypoints: Map<
-    string,
-    Entrypoint | IEvaluatedEntrypoint
-  >;
+  public readonly entrypoints: Map<string, Entrypoint | IEvaluatedEntrypoint>;
 
   private contentHashes = new Map<string, string>();
 

packages/babel/src/evaluators/index.ts

@@ -13,9 +13,10 @@ export default function evaluate(
   const m = new Module(entrypoint, cache);
 
   m.evaluate();
+  m.dispose();
 
   return {
-    value: m.exports,
+    value: entrypoint.exports,
     dependencies: m.dependencies,
   };
 }

packages/babel/src/index.ts

@@ -13,7 +13,10 @@ import type { PluginOptions } from './types';
 export { slugify } from '@linaria/utils';
 
 export { default as preeval } from './plugins/preeval';
-export { default as withLinariaMetadata } from './utils/withLinariaMetadata';
+export {
+  default as withLinariaMetadata,
+  getLinariaMetadata,
+} from './utils/withLinariaMetadata';
 export { default as Module, DefaultModuleImplementation } from './module';
 export { default as transform } from './transform';
 export * from './types';
@@ -30,11 +33,10 @@ export {
   syncResolveImports,
 } from './transform/generators/resolveImports';
 export { default as loadLinariaOptions } from './transform/helpers/loadLinariaOptions';
-export { StackOfMaps } from './transform/helpers/StackOfMaps';
 export { withDefaultServices } from './transform/helpers/withDefaultServices';
 export type { Services } from './transform/types';
 export { default as isNode } from './utils/isNode';
-export { default as getTagProcessor } from './utils/getTagProcessor';
+export { getTagProcessor } from './utils/getTagProcessor';
 export { default as getVisitorKeys } from './utils/getVisitorKeys';
 export type { VisitorKeys } from './utils/getVisitorKeys';
 export { default as peek } from './utils/peek';

packages/babel/src/module.ts

@@ -105,6 +105,12 @@ function resolve(
   return resolved;
 }
 
+function assertDisposed(
+  entrypoint: Entrypoint | null
+): asserts entrypoint is Entrypoint {
+  invariant(entrypoint, 'Module is disposed');
+}
+
 class Module {
   public readonly callstack: string[] = [];
 
@@ -181,6 +187,7 @@ class Module {
 
       const m = new Module(entrypoint, this.cache, this);
       m.evaluate();
+      m.dispose();
 
       return entrypoint.exports;
     },
@@ -192,12 +199,15 @@ class Module {
 
   public resolve = resolve.bind(this);
 
+  protected entrypoint: Entrypoint | null;
+
   constructor(
-    protected entrypoint: Entrypoint,
+    entrypoint: Entrypoint,
     private cache = new TransformCacheCollection(),
     parentModule?: Module,
     private moduleImpl: HiddenModuleMembers = DefaultModuleImplementation
   ) {
+    this.entrypoint = entrypoint;
     this.idx = entrypoint.idx;
     this.id = entrypoint.name;
     this.filename = entrypoint.name;
@@ -207,9 +217,9 @@ class Module {
     this.ignored = entrypoint.ignored ?? this.parentIsIgnored;
 
     if (parentModule) {
-      this.callstack = [...parentModule.callstack, parentModule.filename];
+      this.callstack = [entrypoint.name, ...parentModule.callstack];
     } else {
-      this.callstack = [];
+      this.callstack = [entrypoint.name];
     }
 
     this.extensions = entrypoint.pluginOptions.extensions;
@@ -218,19 +228,29 @@ class Module {
   }
 
   public get exports() {
+    assertDisposed(this.entrypoint);
     return this.entrypoint.exports;
   }
 
   public set exports(value) {
+    assertDisposed(this.entrypoint);
+
     this.entrypoint.exports = value;
 
-    this.debug(
-      'the whole exports was overridden with %O',
-      this.entrypoint.exportsValues
-    );
+    this.debug('the whole exports was overridden with %O', value);
+  }
+
+  dispose(): void {
+    assertDisposed(this.entrypoint);
+
+    this.debug('dispose');
+
+    this.entrypoint = null;
   }
 
   evaluate(): void {
+    assertDisposed(this.entrypoint);
+
     let { entrypoint } = this;
     // Evaluate could be called before the entrypoint was superseded.
     // In this case, we need to find the latest entrypoint.
@@ -314,6 +334,8 @@ class Module {
     only: string[],
     log: Debugger
   ): Entrypoint | IEvaluatedEntrypoint | null {
+    assertDisposed(this.entrypoint);
+
     const extension = path.extname(filename);
     if (extension !== '.json' && !this.extensions.includes(extension)) {
       return null;
@@ -421,6 +443,8 @@ class Module {
   }
 
   resolveDependency = (id: string): IEntrypointDependency => {
+    assertDisposed(this.entrypoint);
+
     const cached = this.entrypoint.getDependency(id);
     invariant(!(cached instanceof Promise), 'Dependency is not resolved yet');
 

packages/babel/src/plugins/collector.ts

@@ -9,7 +9,7 @@ import type { Identifier } from '@babel/types';
 
 import { debug } from '@linaria/logger';
 import type { StrictOptions, LinariaMetadata } from '@linaria/utils';
-import { removeWithRelated } from '@linaria/utils';
+import { invalidateTraversalCache, removeWithRelated } from '@linaria/utils';
 
 import type { Core } from '../babel';
 import type { IPluginState, ValueCache } from '../types';
@@ -86,5 +86,8 @@ export default function collectorPlugin(
       debug('collect:end', file.opts.filename);
     },
     visitor: {},
+    post(file: BabelFile) {
+      invalidateTraversalCache(file.path);
+    },
   };
 }

packages/babel/src/plugins/preeval.ts

@@ -12,6 +12,7 @@ import {
   addIdentifierToLinariaPreval,
   removeDangerousCode,
   isFeatureEnabled,
+  invalidateTraversalCache,
 } from '@linaria/utils';
 
 import type { Core } from '../babel';
@@ -88,6 +89,8 @@ export default function preeval(
 
       const log = createCustomDebug('preeval', getFileIdx(file.opts.filename!));
 
+      invalidateTraversalCache(file.path);
+
       if (this.processors.length === 0) {
         log('end', "We didn't find any Linaria template literals");