import webpack from 'webpack';
import WebpackDevServer from 'webpack-dev-server';
import type { Config as JestConfig } from '@jest/types';
import baseSetup from 'jest-playwright-preset/setup';
import { generateWebpackConfig } from '../generateWebpackConfig';
import { useDocker } from '../detectEnv';
import { startDocker } from '../docker';

let devServer: WebpackDevServer;
async function setupWebpack() {
  const webpackConfig = await generateWebpackConfig();
  const compiler = webpack(webpackConfig);
  const compilerDone = new Promise((resolve, reject) => {
    compiler.hooks.watchRun.tapAsync('@jest-playwright', (_, callback) => {
      console.log('\nBuilding browser bundle...');
      callback();
    });
    compiler.hooks.done.tapAsync('@jest-playwright', (stats, callback) => {
      if (stats.hasErrors()) {
        console.log('Build failed');
        reject(stats);
      } else {
        console.log('Build OK');
        resolve(stats);
      }
      callback();
    });
  });

  devServer = new WebpackDevServer(compiler, {
    noInfo: true,
    stats: 'minimal',
    disableHostCheck: true,
    ...webpackConfig.devServer,
  });
  devServer.listen(9000);
  (global as any)['__DEV_SERVER__'] = devServer;

  return compilerDone;
}

module.exports = async function setup(jestConfig: JestConfig.GlobalConfig) {
  // do not re-run webpack in watch mode
  // https://github.com/facebook/jest/issues/6800
  if (devServer) {
    // hack into middleware to wait for build to pass
    return new Promise((ok) => (devServer as any).middleware.waitUntilValid(ok));
  }

  await setupWebpack();
  if (useDocker) {
    console.log('Starting dockerized chrome...');
    await startDocker();
    console.log('Running E2E tests in docker');
  }
  await baseSetup(jestConfig);
};