Function handler

packages/ui/node/index.ts:26–117

(server)

    configureServer: {
      order: 'post',
      handler(server) {
        const uiOptions = ctx.config
        const base = uiOptions.uiBase

        // Serve coverage HTML at ./coverage if configured
        const coverageHtmlDir = ctx.config.coverage?.htmlDir
        if (coverageHtmlDir) {
          server.middlewares.use(
            join(base, 'coverage'),
            sirv(coverageHtmlDir, {
              single: true,
              dev: true,
              setHeaders: (res) => {
                res.setHeader(
                  'Cache-Control',
                  'public,max-age=0,must-revalidate',
                )
              },
            }),
          )
        }

        const clientDist = resolve(fileURLToPath(import.meta.url), '../client')
        const clientIndexHtml = fs.readFileSync(resolve(clientDist, 'index.html'), 'utf-8')

        // eslint-disable-next-line prefer-arrow-callback
        server.middlewares.use(function vitestAttachment(req, res, next) {
          if (!req.url) {
            return next()
          }

          const url = new URL(req.url, 'http://localhost')
          if (url.pathname === '/__vitest_attachment__') {
            const path = url.searchParams.get('path')
            const contentType = url.searchParams.get('contentType')

            // ignore invalid requests
            if (!isValidApiRequest(ctx.config, req) || !contentType || !path) {
              return next()
            }

            const fsPath = decodeURIComponent(path)

            if (!isFileServingAllowed(ctx.vite.config, fsPath)) {
              return next()
            }

            try {
              res.writeHead(200, {
                'content-type': contentType,
              })
              fs.createReadStream(fsPath)
                .pipe(res)
                .on('close', () => res.end())
            }
            catch (err) {
              next(err)
            }