playwright/packages/playwright-test/src/experimentalLoader.ts

/**
 * Copyright (c) Microsoft Corporation.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import fs from 'fs';
import path from 'path';
import { tsConfigLoader, TsConfigLoaderResult } from './third_party/tsconfig-loader';
import { transformHook } from './transform';

const tsConfigCache = new Map<string, TsConfigLoaderResult>();

async function resolve(specifier: string, context: { parentURL: string }, defaultResolve: any) {
  if (specifier.endsWith('.js') || specifier.endsWith('.ts') || specifier.endsWith('.mjs'))
    return defaultResolve(specifier, context, defaultResolve);
  let url = new URL(specifier, context.parentURL).toString();
  url = url.substring('file://'.length);
  if (fs.existsSync(url + '.ts'))
    return defaultResolve(specifier + '.ts', context, defaultResolve);
  if (fs.existsSync(url + '.js'))
    return defaultResolve(specifier + '.js', context, defaultResolve);
  return defaultResolve(specifier, context, defaultResolve);
}

async function load(url: string, context: any, defaultLoad: any) {
  if (url.endsWith('.ts')) {
    const filename = url.substring('file://'.length);
    const cwd = path.dirname(filename);
    let tsconfig = tsConfigCache.get(cwd);
    if (!tsconfig) {
      tsconfig = tsConfigLoader({
        getEnv: (name: string) => process.env[name],
        cwd
      });
      tsConfigCache.set(cwd, tsconfig);
    }
    const code = fs.readFileSync(filename, 'utf-8');
    const source = transformHook(code, filename, tsconfig, true);
    return { format: 'module', source };
  }
  return defaultLoad(url, context, defaultLoad);
}

module.exports = { resolve, load };