import type { Connect } from 'vite'
import type { ParentBrowserProject } from '../projectParent'
import { resolveTester } from '../serverTester'
import { allowIframes, disableCache } from './utils'

export function createTesterMiddleware(browserServer: ParentBrowserProject): Connect.NextHandleFunction {
  return async function vitestTesterMiddleware(req, res, next) {
    if (!req.url) {
      return next()
    }
    const url = new URL(req.url, 'http://localhost')
    if (!url.pathname.startsWith(browserServer.prefixTesterUrl) || !url.searchParams.has('sessionId')) {
      return next()
    }

    const html = await resolveTester(browserServer, url, res, next)
    if (html) {
      disableCache(res)
      allowIframes(res)
      res.write(html, 'utf-8')
      res.end()
    }
  }
}