Function createLoadHook

packages/vitest/src/runtime/workers/native.ts:152–234

(_worker: WorkerSetupContext)

])

function createLoadHook(_worker: WorkerSetupContext): module.LoadHookSync {
  return (url, context, nextLoad) => {
    const result: module.LoadFnOutput = url.includes('mock=') && isBuiltin(cleanUrl(url))
      ? { format: 'commonjs' } // avoid resolving the builtin module that is supposed to be mocked
      : nextLoad(url, context)
    if (
      (result.format && ignoreFormats.has(result.format))
      || url.includes(distDir)
    ) {
      return result
    }

    const mocker = getNativeMocker()

    mocker?.checkCircularManualMock(url)

    if (url.includes('mock=automock') || url.includes('mock=autospy')) {
      const automockedResult = mocker?.loadAutomock(url, result)
      if (automockedResult != null) {
        return automockedResult
      }
      return result
    }

    if (url.includes('mock=manual')) {
      const mockedResult = mocker?.loadManualMock(url, result)
      if (mockedResult != null) {
        return mockedResult
      }
      return result
    }

    // ignore non-vitest modules for performance reasons,
    // vi.hoisted and vi.mock won't work outside of test files or setup files
    if (!result.source || !url.includes('vitest=')) {
      return result
    }

    const filename = url.startsWith('file://') ? fileURLToPath(url) : url
    const source = result.source.toString()
    const transformedCode = result.format?.includes('typescript')
      ? module.stripTypeScriptTypes(source)
      : source

    let _ms: MagicString | undefined
    const ms = () => _ms || (_ms = new MagicString(source))

    if (source.includes('import.meta.vitest')) {
      replaceInSourceMarker(url, source, ms)
    }

    hoistMocks(
      transformedCode,
      filename,
      code => parse(code, {
        ecmaVersion: 'latest',
        sourceType: result.format === 'module' || result.format === 'module-typescript' || result.format === 'typescript'
          ? 'module'