@wyw-in-js/transform 2.0.0-alpha.0 → 2.0.0

package/esm/utils/oxcPreevalStage/staticOverlay.js

@@ -0,0 +1,18 @@
+import { usesStaticEvaluation } from "./evalStrategy.js";
+export const createStaticPreevalOverlay = (processed, dependencyNames, options) => {
+	const staticValuesEnabled = usesStaticEvaluation(options);
+	const staticValueNames = staticValuesEnabled ? new Set(processed.staticValues.map((item) => item.name)) : null;
+	const evalDependencyNames = staticValuesEnabled ? dependencyNames.filter((name) => !staticValueNames.has(name)) : dependencyNames;
+	const staticValueCache = new Map();
+	if (staticValuesEnabled) {
+		processed.staticValues.forEach(({ name, value }) => {
+			staticValueCache.set(name, value);
+		});
+	}
+	return {
+		evalDependencyNames,
+		staticValueCache,
+		staticValueCandidates: staticValuesEnabled ? processed.staticValueCandidates : []
+	};
+};
+//# sourceMappingURL=staticOverlay.js.map

package/esm/utils/oxcPreevalStage/staticOverlay.js.map

@@ -0,0 +1 @@
+{"mappings":"AACA,SAAS,4BAA4B;AAGrC,OAAO,MAAM,8BACX,WACA,iBACA,YACyB;CACzB,MAAM,sBAAsB,qBAAqB,QAAQ;CACzD,MAAM,mBAAmB,sBACrB,IAAI,IAAI,UAAU,aAAa,KAAK,SAAS,KAAK,KAAK,CAAC,GACxD;CACJ,MAAM,sBAAsB,sBACxB,gBAAgB,QAAQ,SAAS,CAAC,iBAAkB,IAAI,KAAK,CAAC,GAC9D;CACJ,MAAM,mBAAmB,IAAI,KAAsB;AAEnD,KAAI,qBAAqB;AACvB,YAAU,aAAa,SAAS,EAAE,MAAM,YAAY;AAClD,oBAAiB,IAAI,MAAM,MAAM;IACjC;;AAGJ,QAAO;EACL;EACA;EACA,uBAAuB,sBACnB,UAAU,wBACV,EAAE;EACP","names":[],"sources":["../../../src/utils/oxcPreevalStage/staticOverlay.ts"],"version":3,"sourcesContent":["import type { ApplyOxcProcessorsResult } from '../applyOxcProcessors/types';\nimport { usesStaticEvaluation } from './evalStrategy';\nimport type { OxcPreevalOptions, StaticPreevalOverlay } from './types';\n\nexport const createStaticPreevalOverlay = (\n  processed: ApplyOxcProcessorsResult,\n  dependencyNames: string[],\n  options: OxcPreevalOptions\n): StaticPreevalOverlay => {\n  const staticValuesEnabled = usesStaticEvaluation(options);\n  const staticValueNames = staticValuesEnabled\n    ? new Set(processed.staticValues.map((item) => item.name))\n    : null;\n  const evalDependencyNames = staticValuesEnabled\n    ? dependencyNames.filter((name) => !staticValueNames!.has(name))\n    : dependencyNames;\n  const staticValueCache = new Map<string, unknown>();\n\n  if (staticValuesEnabled) {\n    processed.staticValues.forEach(({ name, value }) => {\n      staticValueCache.set(name, value);\n    });\n  }\n\n  return {\n    evalDependencyNames,\n    staticValueCache,\n    staticValueCandidates: staticValuesEnabled\n      ? processed.staticValueCandidates\n      : [],\n  };\n};\n"],"file":"staticOverlay.js"}

package/esm/utils/oxcPreevalStage/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/esm/utils/oxcPreevalStage/types.js.map

@@ -0,0 +1 @@
+{"mappings":"","names":[],"sources":["../../../src/utils/oxcPreevalStage/types.ts"],"version":3,"sourcesContent":["import type { StrictOptions } from '@wyw-in-js/shared';\n\nimport type { EventEmitter } from '../EventEmitter';\nimport type { WYWTransformMetadata } from '../TransformMetadata';\nimport type { OxcStaticValueCandidate } from '../collectOxcTemplateDependencies';\n\nexport type OxcPreevalOptions = Pick<\n  StrictOptions,\n  | 'classNameSlug'\n  | 'codeRemover'\n  | 'displayName'\n  | 'eval'\n  | 'extensions'\n  | 'features'\n  | 'staticBindings'\n  | 'tagResolver'\n> & { eventEmitter?: EventEmitter };\n\nexport type OxcPreevalResult = {\n  baseCode: string;\n  code: string;\n  dependencyNames: string[];\n  metadata: WYWTransformMetadata | null;\n  processorClassNames: Record<string, string>;\n  staticDependencies: string[];\n  staticValueCache: Map<string, unknown>;\n  staticValueCandidates: OxcStaticValueCandidate[];\n};\n\nexport type StaticPreevalOverlay = {\n  evalDependencyNames: string[];\n  staticValueCache: Map<string, unknown>;\n  staticValueCandidates: OxcStaticValueCandidate[];\n};\n"],"file":"types.js"}

package/esm/utils/oxcPreevalStage.js

@@ -1,97 +1,41 @@
-import { parseSync } from "oxc-parser";
-import { isFeatureEnabled } from "@wyw-in-js/shared";
 import { EventEmitter } from "./EventEmitter.js";
-import { applyOxcProcessors } from "./applyOxcProcessors.js";
-import { removeDangerousCodeWithOxc, replaceImportMetaEnvWithOxc, rewriteDynamicImportsAndAddRequireFallbackWithOxc } from "./oxcPreevalTransforms.js";
-const DYNAMIC_IMPORT_RE = /\bimport(?:\s|\/\*[\s\S]*?\*\/)*\(/;
-const REQUIRE_CALL_RE = /\brequire(?:\s|\/\*[\s\S]*?\*\/)*\(/;
-const isEnvDisabled = (value) => value === "0" || value === "false" || value === "no" || value === "off";
-const isStaticImportValuesEnabled = (features, filename) => {
-	const envValue = process.env.WYW_STATIC_IMPORT_VALUES?.trim().toLowerCase();
-	if (envValue) {
-		return !isEnvDisabled(envValue);
-	}
-	return isFeatureEnabled(features, "staticImportValues", filename);
-};
-const parseSourceType = (code, filename) => {
-	const parsed = parseSync(filename, code, {
-		astType: filename.endsWith(".ts") || filename.endsWith(".tsx") ? "ts" : "js",
-		range: true,
-		sourceType: "unambiguous"
-	});
-	const fatalError = parsed.errors.find((error) => error.severity === "Error");
-	if (fatalError) {
-		throw new Error(fatalError.message);
-	}
-	return parsed.program.sourceType === "script" ? "script" : "module";
-};
-export const appendOxcWywPreval = (code, filename, dependencyNames) => {
-	const uniqueNames = [...new Set(dependencyNames)];
-	const properties = uniqueNames.map((name) => `${name}: ${name}`).join(", ");
-	const object = uniqueNames.length > 0 ? `{ ${properties} }` : "{}";
-	if (parseSourceType(code, filename) === "script") {
-		return `${code}\nexports.__wywPreval = ${object};`;
-	}
-	return `${code}\nexport const __wywPreval = ${object};`;
-};
+import { appendOxcWywPreval } from "./oxcPreevalStage/prevalExport.js";
+import { prepareOxcPreevalCode } from "./oxcPreevalStage/prepareCode.js";
+import { collectPreevalProcessors } from "./oxcPreevalStage/processors.js";
+import { createStaticPreevalOverlay } from "./oxcPreevalStage/staticOverlay.js";
+export { appendOxcWywPreval } from "./oxcPreevalStage/prevalExport.js";
 export const runOxcPreevalStage = (code, fileContext, options) => {
 	const filename = fileContext.filename ?? "unknown.js";
-	const dependencyNames = [];
 	const eventEmitter = options.eventEmitter ?? EventEmitter.dummy;
-	const processed = eventEmitter.perf("transform:preeval:processTemplate", () => applyOxcProcessors(code, fileContext, options, (processor) => {
-		processor.dependencies.forEach((dependency) => {
-			if (dependency.ex.type === "Identifier") {
-				dependencyNames.push(dependency.ex.name);
-			}
-		});
-		processor.doEvaltimeReplacement();
-	}));
-	const staticValuesEnabled = isStaticImportValuesEnabled(options.features, filename);
-	const staticValueNames = staticValuesEnabled ? new Set(processed.staticValues.map((item) => item.name)) : null;
-	const evalDependencyNames = staticValuesEnabled ? dependencyNames.filter((name) => !staticValueNames.has(name)) : dependencyNames;
-	let nextCode = eventEmitter.perf("transform:preeval:importMetaEnv", () => replaceImportMetaEnvWithOxc(processed.code, filename));
-	if (isFeatureEnabled(options.features, "dangerousCodeRemover", filename)) {
-		nextCode = eventEmitter.perf("transform:preeval:removeDangerousCode", () => removeDangerousCodeWithOxc(nextCode, filename, options.codeRemover));
-	}
-	const shouldRewriteDynamicImports = DYNAMIC_IMPORT_RE.test(nextCode);
-	const shouldAddRequireFallback = REQUIRE_CALL_RE.test(nextCode);
-	if (shouldRewriteDynamicImports || shouldAddRequireFallback) {
-		nextCode = rewriteDynamicImportsAndAddRequireFallbackWithOxc(nextCode, filename, {
-			addRequireFallback: shouldAddRequireFallback,
-			eventEmitter,
-			rewriteDynamicImports: shouldRewriteDynamicImports
-		});
-	}
+	const { dependencyNames, processed } = collectPreevalProcessors(code, fileContext, options, eventEmitter);
+	const staticOverlay = createStaticPreevalOverlay(processed, dependencyNames, options);
+	const baseCode = prepareOxcPreevalCode(processed.code, filename, options, eventEmitter);
 	if (processed.processors.length === 0) {
 		return {
-			baseCode: nextCode,
-			code: nextCode,
+			baseCode,
+			code: baseCode,
 			dependencyNames: [],
 			metadata: null,
+			processorClassNames: {},
 			staticDependencies: [],
 			staticValueCandidates: [],
 			staticValueCache: new Map()
 		};
 	}
-	const staticValueCache = new Map();
-	if (staticValuesEnabled) {
-		processed.staticValues.forEach(({ name, value }) => {
-			staticValueCache.set(name, value);
-		});
-	}
 	return {
-		baseCode: nextCode,
-		code: appendOxcWywPreval(nextCode, filename, evalDependencyNames),
-		dependencyNames: evalDependencyNames,
+		baseCode,
+		code: appendOxcWywPreval(baseCode, filename, staticOverlay.evalDependencyNames),
+		dependencyNames: staticOverlay.evalDependencyNames,
 		metadata: {
 			dependencies: [],
 			processors: processed.processors,
 			replacements: [],
 			rules: {}
 		},
+		processorClassNames: Object.fromEntries(processed.processorClassNamesByLocal),
 		staticDependencies: [],
-		staticValueCandidates: staticValuesEnabled ? processed.staticValueCandidates : [],
-		staticValueCache
+		staticValueCache: staticOverlay.staticValueCache,
+		staticValueCandidates: staticOverlay.staticValueCandidates
 	};
 };
 //# sourceMappingURL=oxcPreevalStage.js.map

package/esm/utils/oxcPreevalStage.js.map

@@ -1 +1 @@
-{"mappings":"AAAA,SAAS,iBAAiB;AAI1B,SAAS,wBAAwB;AAEjC,SAAS,oBAAoB;AAG7B,SAAS,0BAA0B;AACnC,SACE,4BACA,6BACA,yDACK;AAuBP,MAAM,oBAAoB;AAC1B,MAAM,kBAAkB;AAExB,MAAM,iBAAiB,UACrB,UAAU,OAAO,UAAU,WAAW,UAAU,QAAQ,UAAU;AAEpE,MAAM,+BACJ,UACA,aACY;CACZ,MAAM,WAAW,QAAQ,IAAI,0BAA0B,MAAM,CAAC,aAAa;AAC3E,KAAI,UAAU;AACZ,SAAO,CAAC,cAAc,SAAS;;AAGjC,QAAO,iBAAiB,UAAU,sBAAsB,SAAS;;AAGnE,MAAM,mBACJ,MACA,aACwB;CACxB,MAAM,SAAS,UAAU,UAAU,MAAM;EACvC,SACE,SAAS,SAAS,MAAM,IAAI,SAAS,SAAS,OAAO,GAAG,OAAO;EACjE,OAAO;EACP,YAAY;EACb,CAAC;CACF,MAAM,aAAa,OAAO,OAAO,MAAM,UAAU,MAAM,aAAa,QAAQ;AAC5E,KAAI,YAAY;AACd,QAAM,IAAI,MAAM,WAAW,QAAQ;;AAGrC,QAAO,OAAO,QAAQ,eAAe,WAAW,WAAW;;AAG7D,OAAO,MAAM,sBACX,MACA,UACA,oBACW;CACX,MAAM,cAAc,CAAC,GAAG,IAAI,IAAI,gBAAgB,CAAC;CACjD,MAAM,aAAa,YAAY,KAAK,SAAS,GAAG,KAAK,IAAI,OAAO,CAAC,KAAK,KAAK;CAC3E,MAAM,SAAS,YAAY,SAAS,IAAI,KAAK,WAAW,MAAM;AAE9D,KAAI,gBAAgB,MAAM,SAAS,KAAK,UAAU;AAChD,SAAO,GAAG,KAAK,0BAA0B,OAAO;;AAGlD,QAAO,GAAG,KAAK,+BAA+B,OAAO;;AAGvD,OAAO,MAAM,sBACX,MACA,aACA,YACqB;CACrB,MAAM,WAAW,YAAY,YAAY;CACzC,MAAM,kBAA4B,EAAE;CACpC,MAAM,eAAe,QAAQ,gBAAgB,aAAa;CAE1D,MAAM,YAAY,aAAa,KAAK,2CAClC,mBAAmB,MAAM,aAAa,UAAU,cAAc;AAC5D,YAAU,aAAa,SAAS,eAAe;AAC7C,OAAI,WAAW,GAAG,SAAS,cAAc;AACvC,oBAAgB,KAAK,WAAW,GAAG,KAAK;;IAE1C;AAEF,YAAU,uBAAuB;GACjC,CACH;CACD,MAAM,sBAAsB,4BAC1B,QAAQ,UACR,SACD;CACD,MAAM,mBAAmB,sBACrB,IAAI,IAAI,UAAU,aAAa,KAAK,SAAS,KAAK,KAAK,CAAC,GACxD;CACJ,MAAM,sBAAsB,sBACxB,gBAAgB,QAAQ,SAAS,CAAC,iBAAkB,IAAI,KAAK,CAAC,GAC9D;CAEJ,IAAI,WAAW,aAAa,KAAK,yCAC/B,4BAA4B,UAAU,MAAM,SAAS,CACtD;AAED,KAAI,iBAAiB,QAAQ,UAAU,wBAAwB,SAAS,EAAE;AACxE,aAAW,aAAa,KAAK,+CAC3B,2BAA2B,UAAU,UAAU,QAAQ,YAAY,CACpE;;CAGH,MAAM,8BAA8B,kBAAkB,KAAK,SAAS;CACpE,MAAM,2BAA2B,gBAAgB,KAAK,SAAS;AAC/D,KAAI,+BAA+B,0BAA0B;AAC3D,aAAW,kDACT,UACA,UACA;GACE,oBAAoB;GACpB;GACA,uBAAuB;GACxB,CACF;;AAGH,KAAI,UAAU,WAAW,WAAW,GAAG;AACrC,SAAO;GACL,UAAU;GACV,MAAM;GACN,iBAAiB,EAAE;GACnB,UAAU;GACV,oBAAoB,EAAE;GACtB,uBAAuB,EAAE;GACzB,kBAAkB,IAAI,KAAK;GAC5B;;CAGH,MAAM,mBAAmB,IAAI,KAAsB;AACnD,KAAI,qBAAqB;AACvB,YAAU,aAAa,SAAS,EAAE,MAAM,YAAY;AAClD,oBAAiB,IAAI,MAAM,MAAM;IACjC;;AAGJ,QAAO;EACL,UAAU;EACV,MAAM,mBAAmB,UAAU,UAAU,oBAAoB;EACjE,iBAAiB;EACjB,UAAU;GACR,cAAc,EAAE;GAChB,YAAY,UAAU;GACtB,cAAc,EAAE;GAChB,OAAO,EAAE;GACV;EACD,oBAAoB,EAAE;EACtB,uBAAuB,sBACnB,UAAU,wBACV,EAAE;EACN;EACD","names":[],"sources":["../../src/utils/oxcPreevalStage.ts"],"version":3,"sourcesContent":["import { parseSync } from 'oxc-parser';\n\nimport type { IFileContext } from '@wyw-in-js/processor-utils';\nimport type { StrictOptions } from '@wyw-in-js/shared';\nimport { isFeatureEnabled } from '@wyw-in-js/shared';\n\nimport { EventEmitter } from './EventEmitter';\nimport type { WYWTransformMetadata } from './TransformMetadata';\nimport type { OxcStaticValueCandidate } from './collectOxcTemplateDependencies';\nimport { applyOxcProcessors } from './applyOxcProcessors';\nimport {\n  removeDangerousCodeWithOxc,\n  replaceImportMetaEnvWithOxc,\n  rewriteDynamicImportsAndAddRequireFallbackWithOxc,\n} from './oxcPreevalTransforms';\n\ntype OxcPreevalOptions = Pick<\n  StrictOptions,\n  | 'classNameSlug'\n  | 'codeRemover'\n  | 'displayName'\n  | 'evaluate'\n  | 'extensions'\n  | 'features'\n  | 'tagResolver'\n> & { eventEmitter?: EventEmitter };\n\ntype OxcPreevalResult = {\n  baseCode: string;\n  code: string;\n  dependencyNames: string[];\n  metadata: WYWTransformMetadata | null;\n  staticDependencies: string[];\n  staticValueCache: Map<string, unknown>;\n  staticValueCandidates: OxcStaticValueCandidate[];\n};\n\nconst DYNAMIC_IMPORT_RE = /\\bimport(?:\\s|\\/\\*[\\s\\S]*?\\*\\/)*\\(/;\nconst REQUIRE_CALL_RE = /\\brequire(?:\\s|\\/\\*[\\s\\S]*?\\*\\/)*\\(/;\n\nconst isEnvDisabled = (value: string): boolean =>\n  value === '0' || value === 'false' || value === 'no' || value === 'off';\n\nconst isStaticImportValuesEnabled = (\n  features: StrictOptions['features'],\n  filename: string\n): boolean => {\n  const envValue = process.env.WYW_STATIC_IMPORT_VALUES?.trim().toLowerCase();\n  if (envValue) {\n    return !isEnvDisabled(envValue);\n  }\n\n  return isFeatureEnabled(features, 'staticImportValues', filename);\n};\n\nconst parseSourceType = (\n  code: string,\n  filename: string\n): 'module' | 'script' => {\n  const parsed = parseSync(filename, code, {\n    astType:\n      filename.endsWith('.ts') || filename.endsWith('.tsx') ? 'ts' : 'js',\n    range: true,\n    sourceType: 'unambiguous',\n  });\n  const fatalError = parsed.errors.find((error) => error.severity === 'Error');\n  if (fatalError) {\n    throw new Error(fatalError.message);\n  }\n\n  return parsed.program.sourceType === 'script' ? 'script' : 'module';\n};\n\nexport const appendOxcWywPreval = (\n  code: string,\n  filename: string,\n  dependencyNames: string[]\n): string => {\n  const uniqueNames = [...new Set(dependencyNames)];\n  const properties = uniqueNames.map((name) => `${name}: ${name}`).join(', ');\n  const object = uniqueNames.length > 0 ? `{ ${properties} }` : '{}';\n\n  if (parseSourceType(code, filename) === 'script') {\n    return `${code}\\nexports.__wywPreval = ${object};`;\n  }\n\n  return `${code}\\nexport const __wywPreval = ${object};`;\n};\n\nexport const runOxcPreevalStage = (\n  code: string,\n  fileContext: IFileContext,\n  options: OxcPreevalOptions\n): OxcPreevalResult => {\n  const filename = fileContext.filename ?? 'unknown.js';\n  const dependencyNames: string[] = [];\n  const eventEmitter = options.eventEmitter ?? EventEmitter.dummy;\n\n  const processed = eventEmitter.perf('transform:preeval:processTemplate', () =>\n    applyOxcProcessors(code, fileContext, options, (processor) => {\n      processor.dependencies.forEach((dependency) => {\n        if (dependency.ex.type === 'Identifier') {\n          dependencyNames.push(dependency.ex.name);\n        }\n      });\n\n      processor.doEvaltimeReplacement();\n    })\n  );\n  const staticValuesEnabled = isStaticImportValuesEnabled(\n    options.features,\n    filename\n  );\n  const staticValueNames = staticValuesEnabled\n    ? new Set(processed.staticValues.map((item) => item.name))\n    : null;\n  const evalDependencyNames = staticValuesEnabled\n    ? dependencyNames.filter((name) => !staticValueNames!.has(name))\n    : dependencyNames;\n\n  let nextCode = eventEmitter.perf('transform:preeval:importMetaEnv', () =>\n    replaceImportMetaEnvWithOxc(processed.code, filename)\n  );\n\n  if (isFeatureEnabled(options.features, 'dangerousCodeRemover', filename)) {\n    nextCode = eventEmitter.perf('transform:preeval:removeDangerousCode', () =>\n      removeDangerousCodeWithOxc(nextCode, filename, options.codeRemover)\n    );\n  }\n\n  const shouldRewriteDynamicImports = DYNAMIC_IMPORT_RE.test(nextCode);\n  const shouldAddRequireFallback = REQUIRE_CALL_RE.test(nextCode);\n  if (shouldRewriteDynamicImports || shouldAddRequireFallback) {\n    nextCode = rewriteDynamicImportsAndAddRequireFallbackWithOxc(\n      nextCode,\n      filename,\n      {\n        addRequireFallback: shouldAddRequireFallback,\n        eventEmitter,\n        rewriteDynamicImports: shouldRewriteDynamicImports,\n      }\n    );\n  }\n\n  if (processed.processors.length === 0) {\n    return {\n      baseCode: nextCode,\n      code: nextCode,\n      dependencyNames: [],\n      metadata: null,\n      staticDependencies: [],\n      staticValueCandidates: [],\n      staticValueCache: new Map(),\n    };\n  }\n\n  const staticValueCache = new Map<string, unknown>();\n  if (staticValuesEnabled) {\n    processed.staticValues.forEach(({ name, value }) => {\n      staticValueCache.set(name, value);\n    });\n  }\n\n  return {\n    baseCode: nextCode,\n    code: appendOxcWywPreval(nextCode, filename, evalDependencyNames),\n    dependencyNames: evalDependencyNames,\n    metadata: {\n      dependencies: [],\n      processors: processed.processors,\n      replacements: [],\n      rules: {},\n    },\n    staticDependencies: [],\n    staticValueCandidates: staticValuesEnabled\n      ? processed.staticValueCandidates\n      : [],\n    staticValueCache,\n  };\n};\n"],"file":"oxcPreevalStage.js"}
+{"mappings":"AAEA,SAAS,oBAAoB;AAC7B,SAAS,0BAA0B;AACnC,SAAS,6BAA6B;AACtC,SAAS,gCAAgC;AACzC,SAAS,kCAAkC;AAM3C,SAAS,0BAA0B;AAEnC,OAAO,MAAM,sBACX,MACA,aACA,YACqB;CACrB,MAAM,WAAW,YAAY,YAAY;CACzC,MAAM,eAAe,QAAQ,gBAAgB,aAAa;CAC1D,MAAM,EAAE,iBAAiB,cAAc,yBACrC,MACA,aACA,SACA,aACD;CACD,MAAM,gBAAgB,2BACpB,WACA,iBACA,QACD;CACD,MAAM,WAAW,sBACf,UAAU,MACV,UACA,SACA,aACD;AAED,KAAI,UAAU,WAAW,WAAW,GAAG;AACrC,SAAO;GACL;GACA,MAAM;GACN,iBAAiB,EAAE;GACnB,UAAU;GACV,qBAAqB,EAAE;GACvB,oBAAoB,EAAE;GACtB,uBAAuB,EAAE;GACzB,kBAAkB,IAAI,KAAK;GAC5B;;AAGH,QAAO;EACL;EACA,MAAM,mBACJ,UACA,UACA,cAAc,oBACf;EACD,iBAAiB,cAAc;EAC/B,UAAU;GACR,cAAc,EAAE;GAChB,YAAY,UAAU;GACtB,cAAc,EAAE;GAChB,OAAO,EAAE;GACV;EACD,qBAAqB,OAAO,YAC1B,UAAU,2BACX;EACD,oBAAoB,EAAE;EACtB,kBAAkB,cAAc;EAChC,uBAAuB,cAAc;EACtC","names":[],"sources":["../../src/utils/oxcPreevalStage.ts"],"version":3,"sourcesContent":["import type { IFileContext } from '@wyw-in-js/processor-utils';\n\nimport { EventEmitter } from './EventEmitter';\nimport { appendOxcWywPreval } from './oxcPreevalStage/prevalExport';\nimport { prepareOxcPreevalCode } from './oxcPreevalStage/prepareCode';\nimport { collectPreevalProcessors } from './oxcPreevalStage/processors';\nimport { createStaticPreevalOverlay } from './oxcPreevalStage/staticOverlay';\nimport type {\n  OxcPreevalOptions,\n  OxcPreevalResult,\n} from './oxcPreevalStage/types';\n\nexport { appendOxcWywPreval } from './oxcPreevalStage/prevalExport';\n\nexport const runOxcPreevalStage = (\n  code: string,\n  fileContext: IFileContext,\n  options: OxcPreevalOptions\n): OxcPreevalResult => {\n  const filename = fileContext.filename ?? 'unknown.js';\n  const eventEmitter = options.eventEmitter ?? EventEmitter.dummy;\n  const { dependencyNames, processed } = collectPreevalProcessors(\n    code,\n    fileContext,\n    options,\n    eventEmitter\n  );\n  const staticOverlay = createStaticPreevalOverlay(\n    processed,\n    dependencyNames,\n    options\n  );\n  const baseCode = prepareOxcPreevalCode(\n    processed.code,\n    filename,\n    options,\n    eventEmitter\n  );\n\n  if (processed.processors.length === 0) {\n    return {\n      baseCode,\n      code: baseCode,\n      dependencyNames: [],\n      metadata: null,\n      processorClassNames: {},\n      staticDependencies: [],\n      staticValueCandidates: [],\n      staticValueCache: new Map(),\n    };\n  }\n\n  return {\n    baseCode,\n    code: appendOxcWywPreval(\n      baseCode,\n      filename,\n      staticOverlay.evalDependencyNames\n    ),\n    dependencyNames: staticOverlay.evalDependencyNames,\n    metadata: {\n      dependencies: [],\n      processors: processed.processors,\n      replacements: [],\n      rules: {},\n    },\n    processorClassNames: Object.fromEntries(\n      processed.processorClassNamesByLocal\n    ),\n    staticDependencies: [],\n    staticValueCache: staticOverlay.staticValueCache,\n    staticValueCandidates: staticOverlay.staticValueCandidates,\n  };\n};\n"],"file":"oxcPreevalStage.js"}

package/esm/utils/oxcPreevalTransforms.js

@@ -1,5 +1,6 @@
 import { collectOxcExportsAndImports } from "./collectOxcExportsAndImports.js";
 import { EventEmitter } from "./EventEmitter.js";
+import { getOxcNodeChildren } from "./oxc/ast.js";
 import { parseOxcProgramCached } from "./parseOxc.js";
 const ssrCheckFields = new Set([
 	"document",
@@ -40,6 +41,7 @@ const defaultReactComponentTypes = [
 	"MemoExoticComponent",
 	"NamedExoticComponent"
 ];
+const defaultReactHocs = ["forwardRef", "memo"];
 const generatedProcessorHelperNameRe = /^_exp\d*$/;
 const requireCallRe = /\brequire\s*\(/;
 const windowTokenRe = /\bwindow\b/;
@@ -96,28 +98,11 @@ const getStaticBinding = (scope, name) => {
 };
 const isFileLikeRequireSpecifier = (value) => value.startsWith(".") || value.startsWith("/") || value.startsWith("file:");
 const isNode = (value) => !!value && typeof value === "object" && "type" in value && typeof value.type === "string";
-const getChildren = (node) => {
-	const result = [];
-	const record = node;
-	Object.keys(record).forEach((key) => {
-		if (key === "type" || key === "start" || key === "end" || key === "range") {
-			return;
-		}
-		const value = record[key];
-		if (isNode(value)) {
-			result.push(value);
-			return;
-		}
-		if (Array.isArray(value)) {
-			value.forEach((item) => {
-				if (isNode(item)) {
-					result.push(item);
-				}
-			});
-		}
-	});
-	return result;
-};
+// Reuses the per-node.type visitor-key cache in utils/oxc/ast.ts. This file's
+// getChildren shape historically diverged from the canonical one (a smaller
+// metadata-key skip set), but the produced children list is identical in
+// practice because oxc-parser nodes don't carry the extra metadata fields.
+const getChildren = getOxcNodeChildren;
 const parseOxc = (code, filename) => {
 	return parseOxcProgramCached(filename, code, "unambiguous");
 };
@@ -313,57 +298,92 @@ const predeclareScopeNames = (node, scope) => {
 		});
 	}
 	const visitScopeDescendants = (child) => {
-		if (child.type === "VariableDeclarator") {
-			collectBindingNames(child.id).forEach((name) => {
-				scope.names.add(name);
-			});
-		} else if (child.type === "FunctionDeclaration" && child.id) {
-			scope.names.add(child.id.name);
-		} else if (child.type === "ClassDeclaration" && child.id) {
-			scope.names.add(child.id.name);
-		} else if (child.type === "ImportDefaultSpecifier" || child.type === "ImportNamespaceSpecifier" || child.type === "ImportSpecifier") {
-			scope.names.add(child.local.name);
+		switch (child.type) {
+			case "VariableDeclarator": {
+				const names = collectBindingNames(child.id);
+				for (let i = 0; i < names.length; i += 1) {
+					scope.names.add(names[i]);
+				}
+				break;
+			}
+			case "FunctionDeclaration":
+			case "ClassDeclaration":
+				if (child.id) {
+					scope.names.add(child.id.name);
+				}
+				break;
+			case "ImportDefaultSpecifier":
+			case "ImportNamespaceSpecifier":
+			case "ImportSpecifier":
+				scope.names.add(child.local.name);
+				break;
+			default: break;
 		}
 		if (createsScope(child)) {
 			return;
 		}
-		getChildren(child).forEach(visitScopeDescendants);
+		const children = getChildren(child);
+		for (let i = 0; i < children.length; i += 1) {
+			visitScopeDescendants(children[i]);
+		}
 	};
-	getChildren(node).forEach(visitScopeDescendants);
+	const rootChildren = getChildren(node);
+	for (let i = 0; i < rootChildren.length; i += 1) {
+		visitScopeDescendants(rootChildren[i]);
+	}
 };
 const declareBindings = (node, scope) => {
-	if (node.type === "VariableDeclarator") {
-		const names = collectBindingNames(node.id);
-		names.forEach((name) => {
-			scope.names.add(name);
-			scope.bindings.set(name, null);
-		});
-		if (node.id.type === "Identifier" && node.init) {
-			scope.bindings.set(node.id.name, node.init);
-		}
-		return;
-	}
-	if (node.type === "FunctionDeclaration" && node.id) {
-		scope.names.add(node.id.name);
-		scope.bindings.set(node.id.name, null);
-	}
-	if (node.type === "ClassDeclaration" && node.id) {
-		scope.names.add(node.id.name);
-		scope.bindings.set(node.id.name, null);
-		return;
-	}
-	if (node.type === "ImportDefaultSpecifier" || node.type === "ImportNamespaceSpecifier" || node.type === "ImportSpecifier") {
-		scope.names.add(node.local.name);
-		scope.bindings.set(node.local.name, null);
-		return;
-	}
-	if (node.type === "FunctionDeclaration" || node.type === "FunctionExpression" || node.type === "ArrowFunctionExpression") {
-		node.params.forEach((param) => {
-			collectBindingNames(param).forEach((name) => {
+	// Called for every visited AST node, so the dispatch is hot. A switch lets
+	// V8 generate a jump table on node.type; the previous chained `if`s walked
+	// each branch's string-compare for every non-matching node.
+	switch (node.type) {
+		case "VariableDeclarator": {
+			const names = collectBindingNames(node.id);
+			for (let i = 0; i < names.length; i += 1) {
+				const name = names[i];
 				scope.names.add(name);
 				scope.bindings.set(name, null);
-			});
-		});
+			}
+			if (node.id.type === "Identifier" && node.init) {
+				scope.bindings.set(node.id.name, node.init);
+			}
+			return;
+		}
+		case "ClassDeclaration": {
+			if (node.id) {
+				scope.names.add(node.id.name);
+				scope.bindings.set(node.id.name, null);
+			}
+			return;
+		}
+		case "ImportDefaultSpecifier":
+		case "ImportNamespaceSpecifier":
+		case "ImportSpecifier": {
+			scope.names.add(node.local.name);
+			scope.bindings.set(node.local.name, null);
+			return;
+		}
+		case "FunctionDeclaration": {
+			if (node.id) {
+				scope.names.add(node.id.name);
+				scope.bindings.set(node.id.name, null);
+			}
+		}
+		// eslint-disable-next-line no-fallthrough
+		case "FunctionExpression":
+		case "ArrowFunctionExpression": {
+			const { params } = node;
+			for (let i = 0; i < params.length; i += 1) {
+				const names = collectBindingNames(params[i]);
+				for (let j = 0; j < names.length; j += 1) {
+					const name = names[j];
+					scope.names.add(name);
+					scope.bindings.set(name, null);
+				}
+			}
+			break;
+		}
+		default: break;
 	}
 };
 const visit = (node, scope, enter, parent = null, ancestors = []) => {
@@ -374,7 +394,17 @@ const visit = (node, scope, enter, parent = null, ancestors = []) => {
 	}
 	declareBindings(node, currentScope);
 	enter(node, currentScope, parent, ancestors);
-	getChildren(node).forEach((child) => visit(child, currentScope, enter, node, [...ancestors, node]));
+	// Push onto a shared ancestors stack instead of allocating `[...ancestors,
+	// node]` per child step (O(n × depth) extra allocation on deep ASTs).
+	// Every consumer of `ancestors` in this file reads it synchronously inside
+	// the enter callback; future callers that need to retain the reference
+	// must .slice() it themselves.
+	ancestors.push(node);
+	const children = getChildren(node);
+	for (let i = 0; i < children.length; i += 1) {
+		visit(children[i], currentScope, enter, node, ancestors);
+	}
+	ancestors.pop();
 };
 export const replaceImportMetaEnvWithOxc = (code, filename) => {
 	if (!importMetaEnvRe.test(code)) {
@@ -785,6 +815,7 @@ const isHocCall = (call, hocs, imports) => {
 	const matched = getImportForExpression(getInnermostCallee(call), imports);
 	return !!matched && hocs[matched[0]]?.includes(matched[1]);
 };
+const isInsideHocCall = (ancestors, hocs, imports) => ancestors.some((ancestor) => ancestor.type === "CallExpression" && isHocCall(ancestor, hocs, imports));
 const getComponentTypes = (options) => {
 	const componentTypes = { ...options?.componentTypes ?? { react: [defaultPlaceholder] } };
 	const reactTypes = componentTypes.react;
@@ -795,6 +826,12 @@ const getComponentTypes = (options) => {
 	}
 	return componentTypes;
 };
+const getHocs = (options) => {
+	const hocs = { ...options?.hocs ?? {} };
+	const reactHocs = new Set([...defaultReactHocs, ...hocs.react ?? []]);
+	hocs.react = [...reactHocs];
+	return hocs;
+};
 const getTypeImport = (node, imports) => {
 	if (node.type === "Identifier") {
 		const matched = getImportBinding(imports, node.name);
@@ -914,7 +951,7 @@ export const removeDangerousCodeWithOxc = (code, filename, options) => {
 	const program = parseOxc(code, filename);
 	const imports = collectImportBindings(code, filename, program);
 	const componentTypes = getComponentTypes(options);
-	const hocs = options?.hocs ?? {};
+	const hocs = getHocs(options);
 	const hasHocs = Object.keys(hocs).length > 0;
 	const windowScopedNames = windowTokenRe.test(code) ? collectWindowScopedNames(program) : new Set();
 	const reExportedLocalNames = collectReExportedLocalNames(program);
@@ -1003,6 +1040,9 @@ export const removeDangerousCodeWithOxc = (code, filename, options) => {
 		if (!isAlwaysForbidden && isInDeferredFunctionScope(ancestors)) {
 			return;
 		}
+		if (hasHocs && isInsideHocCall(ancestors, hocs, imports)) {
+			return;
+		}
 		if (isBindingPosition(node, parent) && !isAlwaysForbidden) {
 			return;
 		}