import { createProxyMiddleware } from 'http-proxy-middleware'
import * as koaConnect from 'koa2-connect'
import { judgeFramework, judgeServerFramework } from '../cwd'
import { loadConfig } from '../loadConfig'

function onProxyReq (proxyReq: any, req: any) {
  Object.keys(req.headers).forEach(function (key) {
    proxyReq.setHeader(key, req.headers[key])
  })
}

const kc = koaConnect.default || koaConnect
const getDevProxyMiddlewaresArr = async () => {
  const { fePort, proxy, isDev, https, proxyKey, isVite } = loadConfig()
  const serverFramework = judgeServerFramework()
  const isExpress = serverFramework === 'ssr-plugin-nestjs'
  const proxyMiddlewaresArr: any[] = []

  function registerProxy (proxy: any) {
    for (const path in proxy) {
      const options = proxy[path]
      // 如果底层服务端框架是基于 express的。则不需要用 koaConnect 转换为 koa 中间件
      const middleware = isExpress ? createProxyMiddleware(path, options) : kc(createProxyMiddleware(path, options))
      proxyMiddlewaresArr.push(middleware)
    }
  }
  proxy && registerProxy(proxy)

  if (isDev) {
    if (isVite) {
      const framework = judgeFramework()
      if (!framework) {
        throw new Error('judgeFramework error')
      }

      // 本地开发请求走 vite 接管 前端文件夹请求
      const { createServer } = require('vite')
      const { clientConfig } = require(framework)
      const viteServer = await createServer(clientConfig)
      proxyMiddlewaresArr.push(isExpress ? viteServer.middlewares : kc(viteServer.middlewares))
    } else {
      // Webpack 场景 在本地开发阶段代理 serverPort 的资源到 fePort
      // 例如 http://localhost:3000/static/js/page.chunk.js -> http://localhost:8999/static/js/page.chunk.js
      const remoteStaticServerOptions = {
        target: `${https ? 'https' : 'http'}://127.0.0.1:${fePort}`,
        changeOrigin: true,
        secure: false,
        onProxyReq,
        logLevel: 'warn'
      }

      const proxyPathMap: Record<string, any> = {
        '/sockjs-node': remoteStaticServerOptions,
        '/__webpack_dev_server__': remoteStaticServerOptions,
        '/*.wasm': remoteStaticServerOptions
      }
      for (const key of proxyKey) {
        proxyPathMap[key] = remoteStaticServerOptions
      }
      registerProxy(proxyPathMap)
    }
  }

  return proxyMiddlewaresArr
}

export {
  getDevProxyMiddlewaresArr
}