feat(jest-haste-map): add DependencyExtractor type

jestjs · Feb 23, 2022 · e9abd9f · e9abd9f
1 parent 4e7f6ec
commit e9abd9f
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/packages/jest-haste-map/src/index.ts b/packages/jest-haste-map/src/index.ts
@@ -30,6 +30,7 @@ import normalizePathSep from './lib/normalizePathSep';
 import type {
   ChangeEvent,
   CrawlerOptions,
+  DependencyExtractor,
   EventsQueue,
   FileData,
   FileMetaData,
@@ -316,7 +317,7 @@ export default class HasteMap extends EventEmitter {
     }
 
     if (options.dependencyExtractor) {
-      const dependencyExtractor = await requireOrImportModule<any>(
+      const dependencyExtractor = await requireOrImportModule<DependencyExtractor>(
         options.dependencyExtractor,
         false,
       );

diff --git a/packages/jest-haste-map/src/types.ts b/packages/jest-haste-map/src/types.ts
@@ -154,3 +154,8 @@ export type ChangeEvent = {
   hasteFS: HasteFS;
   moduleMap: ModuleMap;
 };
+
+export type DependencyExtractor = {
+  extract: (code: string, filePath: string, defaultExtract: DependencyExtractor['extract']) => Set<string>;
+  getCacheKey?: () => string;
+};
diff --git a/packages/jest-haste-map/src/worker.ts b/packages/jest-haste-map/src/worker.ts
@@ -12,7 +12,7 @@ import {requireOrImportModule} from 'jest-util';
 import blacklist from './blacklist';
 import H from './constants';
 import * as dependencyExtractor from './lib/dependencyExtractor';
-import type {HasteImpl, WorkerMessage, WorkerMetadata} from './types';
+import type {DependencyExtractor, HasteImpl, WorkerMessage, WorkerMetadata} from './types';
 
 const PACKAGE_JSON = path.sep + 'package.json';
 
@@ -75,7 +75,7 @@ export async function worker(data: WorkerMessage): Promise<WorkerMetadata> {
       dependencies = Array.from(
         data.dependencyExtractor
           ? (
-              await requireOrImportModule<any>(data.dependencyExtractor, false)
+              await requireOrImportModule<DependencyExtractor>(data.dependencyExtractor, false)
             ).extract(content, filePath, dependencyExtractor.extract)
           : dependencyExtractor.extract(content),
       );