import fs, { PathLike } from 'fs'
import path from 'path'
import {
  EXTNAME_VUE,
  isInHBuilderX,
  normalizePath,
  resolveBuiltIn,
} from '@dcloudio/uni-cli-shared'
import { SFCParseOptions } from '@vue/compiler-sfc'
import { Plugin } from 'vite'
import { isString } from '@vue/shared'
/**
 * TODO 临时重写，解决 @vitejs/plugin-vue 的 Bug
 */
export function rewriteCompilerSfcParse() {
  // @ts-ignore
  const compilerSfc = require(resolveBuiltIn('@vue/compiler-sfc'))
  const { parse } = compilerSfc
  compilerSfc.parse = (source: string, options: SFCParseOptions) => {
    if (options?.filename) {
      const extname = path.extname(options.filename)
      // wxs、filter、renderjs
      if (extname && !EXTNAME_VUE.includes(extname)) {
        const tag = extname.slice(1)
        source = `<${tag}>` + source + `</${tag}>`
      }
    }
    return parse(source, options)
  }
}

/**
 * 解决 HBuilderX 项目未包含 package.json 时，initTSConfck 可能导致查找过慢，或递归目录时权限不足报错
 * 即：未包含 package.json 时，直接移除 initTSConfck 相关逻辑
 * @param inputDir
 * @returns
 */
export function initFixedEsbuildInitTSConfck(inputDir: string): Plugin[] {
  if (!isInHBuilderX()) {
    return []
  }
  if (fs.existsSync(path.resolve(inputDir, 'package.json'))) {
    return []
  }
  // 'vite:esbuild', 'vite:esbuild-transpile' initTSConfck
  const existsSync = fs.existsSync
  // 根目录 lerna.json 路径
  const lernaJsonPath = normalizePath(path.resolve(inputDir, 'lerna.json'))
  return [
    {
      name: 'uni:fixed-HBuilderX-initTSConfck-before',
      enforce: 'pre',
      configResolved() {
        // 在 HBuilderX 项目中，缺少 package.json 时， 确保 searchForWorkspaceRoot 使用项目根目录， 否则 initTSConfck 会查找很费时，且可能访问目录权限报错
        // https://github.com/vitejs/vite/blob/43b7b78b1834a4c7128d8a5d987f66a4defcbd93/packages/vite/src/node/plugins/esbuild.ts#L407
        fs.existsSync = (p: PathLike) => {
          if (isString(p)) {
            // 访问根目录 lerna.json 时，直接返回 true
            // https://github.com/vitejs/vite/blob/43b7b78b1834a4c7128d8a5d987f66a4defcbd93/packages/vite/src/node/server/searchRoot.ts#L35
            if (
              p.endsWith('lerna.json') &&
              lernaJsonPath === normalizePath(p)
            ) {
              return true
            }
          }
          return existsSync(p)
        }
      },
    },
  ]
}