import { join, resolve, parse } from 'node:path'

import * as colors from 'colorette'

import {
  applyDefaultArtifactsOptions,
  type ArtifactsOptions,
} from '../def/artifacts.js'
import {
  readNapiConfig,
  debugFactory,
  readFileAsync,
  writeFileAsync,
  UniArchsByPlatform,
  readdirAsync,
} from '../utils/index.js'

const debug = debugFactory('artifacts')

export async function collectArtifacts(userOptions: ArtifactsOptions) {
  const options = applyDefaultArtifactsOptions(userOptions)

  const resolvePath = (...paths: string[]) => resolve(options.cwd, ...paths)
  const packageJsonPath = resolvePath(options.packageJsonPath)
  const { targets, binaryName, packageName } = await readNapiConfig(
    packageJsonPath,
    options.configPath ? resolvePath(options.configPath) : undefined,
  )

  const distDirs = targets.map((platform) =>
    join(options.cwd, options.npmDir, platform.platformArchABI),
  )

  const universalSourceBins = new Set(
    targets
      .filter((platform) => platform.arch === 'universal')
      .flatMap((p) =>
        UniArchsByPlatform[p.platform]?.map((a) => `${p.platform}-${a}`),
      )
      .filter(Boolean) as string[],
  )

  await collectNodeBinaries(join(options.cwd, options.outputDir)).then(
    (output) =>
      Promise.all(
        output.map(async (filePath) => {
          debug.info(`Read [${colors.yellowBright(filePath)}]`)
          const sourceContent = await readFileAsync(filePath)
          const parsedName = parse(filePath)
          const terms = parsedName.name.split('.')
          const platformArchABI = terms.pop()!
          const _binaryName = terms.join('.')

          if (_binaryName !== binaryName) {
            debug.warn(
              `[${_binaryName}] is not matched with [${binaryName}], skip`,
            )
            return
          }
          const dir = distDirs.find((dir) => dir.includes(platformArchABI))
          if (!dir && universalSourceBins.has(platformArchABI)) {
            debug.warn(
              `[${platformArchABI}] has no dist dir but it is source bin for universal arch, skip`,
            )
            return
          }
          if (!dir) {
            throw new Error(`No dist dir found for ${filePath}`)
          }

          const distFilePath = join(dir, parsedName.base)
          debug.info(
            `Write file content to [${colors.yellowBright(distFilePath)}]`,
          )
          await writeFileAsync(distFilePath, sourceContent)
          const distFilePathLocal = join(
            parse(packageJsonPath).dir,
            parsedName.base,
          )
          debug.info(
            `Write file content to [${colors.yellowBright(distFilePathLocal)}]`,
          )
          await writeFileAsync(distFilePathLocal, sourceContent)
        }),
      ),
  )

  const wasiTarget = targets.find((t) => t.platform === 'wasi')
  if (wasiTarget) {
    const buildOutputDir = options.buildOutputDir
      ? resolve(options.cwd, options.buildOutputDir)
      : options.cwd
    const wasiDir = join(
      options.cwd,
      options.npmDir,
      wasiTarget.platformArchABI,
    )
    const cjsFile = join(buildOutputDir, `${binaryName}.wasi.cjs`)
    const workerFile = join(buildOutputDir, `wasi-worker.mjs`)
    const browserEntry = join(buildOutputDir, `${binaryName}.wasi-browser.js`)
    const browserWorkerFile = join(buildOutputDir, `wasi-worker-browser.mjs`)
    debug.info(
      `Move wasi binding file [${colors.yellowBright(
        cjsFile,
      )}] to [${colors.yellowBright(wasiDir)}]`,
    )
    await writeFileAsync(
      join(wasiDir, `${binaryName}.wasi.cjs`),
      await readFileAsync(cjsFile),
    )
    debug.info(
      `Move wasi worker file [${colors.yellowBright(
        workerFile,
      )}] to [${colors.yellowBright(wasiDir)}]`,
    )
    await writeFileAsync(
      join(wasiDir, `wasi-worker.mjs`),
      await readFileAsync(workerFile),
    )
    debug.info(
      `Move wasi browser entry file [${colors.yellowBright(
        browserEntry,
      )}] to [${colors.yellowBright(wasiDir)}]`,
    )
    await writeFileAsync(
      join(wasiDir, `${binaryName}.wasi-browser.js`),
      // https://github.com/vitejs/vite/issues/8427
      (await readFileAsync(browserEntry, 'utf8')).replace(
        `new URL('./wasi-worker-browser.mjs', import.meta.url)`,
        `new URL('${packageName}-wasm32-wasi/wasi-worker-browser.mjs', import.meta.url)`,
      ),
    )
    debug.info(
      `Move wasi browser worker file [${colors.yellowBright(
        browserWorkerFile,
      )}] to [${colors.yellowBright(wasiDir)}]`,
    )
    await writeFileAsync(
      join(wasiDir, `wasi-worker-browser.mjs`),
      await readFileAsync(browserWorkerFile),
    )
  }
}

async function collectNodeBinaries(root: string) {
  const files = await readdirAsync(root, { withFileTypes: true })
  const nodeBinaries = files
    .filter(
      (file) =>
        file.isFile() &&
        (file.name.endsWith('.node') || file.name.endsWith('.wasm')),
    )
    .map((file) => join(root, file.name))

  const dirs = files.filter((file) => file.isDirectory())
  for (const dir of dirs) {
    if (dir.name !== 'node_modules') {
      nodeBinaries.push(...(await collectNodeBinaries(join(root, dir.name))))
    }
  }
  return nodeBinaries
}