import { DevServer, Plugin } from '@web/dev-server-core';
import chokidar from 'chokidar';

import { RunSessions, watchFilesMiddleware } from './middleware/watchFilesMiddleware.js';
import { cacheMiddleware } from './middleware/cacheMiddleware.js';
import { serveTestRunnerHtmlPlugin } from './plugins/serveTestRunnerHtmlPlugin.js';
import { serveTestFrameworkPlugin } from './plugins/serveTestFrameworkPlugin.js';
import { testRunnerApiPlugin } from './plugins/api/testRunnerApiPlugin.js';
import { TestRunnerCoreConfig } from '../config/TestRunnerCoreConfig';
import { TestSessionManager } from '../test-session/TestSessionManager';
import { TestRunner } from '../runner/TestRunner';

const CACHED_PATTERNS = [
  'node_modules/@web/test-runner-',
  'node_modules/@esm-bundle/chai',
  'node_modules/mocha/',
  'node_modules/chai/',
];
const isDefined = (_: unknown) => Boolean(_);

export class TestRunnerServer {
  private devServer: DevServer;
  private fileWatcher = chokidar.watch([]);

  constructor(
    config: TestRunnerCoreConfig,
    testRunner: TestRunner,
    sessions: TestSessionManager,
    testFiles: string[],
    runSessions: RunSessions,
  ) {
    const { plugins = [], testFramework, rootDir } = config;

    const { testFrameworkImport, testFrameworkPlugin } = testFramework
      ? serveTestFrameworkPlugin(testFramework)
      : ({} as { testFrameworkImport?: string; testFrameworkPlugin?: Plugin });
    const serverConfig = {
      port: config.port,
      hostname: config.hostname,
      rootDir,
      injectWebSocket: true,

      http2: config.http2,
      sslKey: config.sslKey,
      sslCert: config.sslCert,

      mimeTypes: config.mimeTypes,

      disableFileWatcher: !config.watch && !config.manual,

      middleware: [
        watchFilesMiddleware({ runSessions, sessions, rootDir, fileWatcher: this.fileWatcher }),
        cacheMiddleware(CACHED_PATTERNS, config.watch),
        ...(config.middleware || []),
      ],

      plugins: [
        testRunnerApiPlugin(config, testRunner, sessions, plugins),
        serveTestRunnerHtmlPlugin(config, testFiles, sessions, testFrameworkImport),
        testFrameworkPlugin,
        ...(config.plugins || []),
      ].filter(isDefined) as Plugin[],
    };

    this.devServer = new DevServer(serverConfig, config.logger, this.fileWatcher);
  }

  async start() {
    await this.devServer.start();
  }

  async stop() {
    await this.devServer.stop();
  }
}