import { cp } from 'node:fs/promises'
import { resolve } from 'node:path'
import serveStatic from 'serve-static'

const getPrefix = (s) => {
  const padEnd = s.endsWith('/') ? s : s + '/'
  return padEnd.startsWith('/') ? padEnd : '/' + padEnd
}

const copyPlugin = ({ inUrl, inFs }) => {
  const prefix = getPrefix(inUrl)
  const subdir = prefix.slice(1)
  let copyTarget
  const handler = serveStatic(inFs)

  return [
    {
      name: 'copy-plugin-serve',
      apply: 'serve',
      configureServer(server) {
        server.middlewares.use(prefix, handler)
      },
    },
    {
      name: 'copy-plugin-build',
      apply: 'build',
      configResolved(config) {
        copyTarget = resolve(config.root, config.build.outDir, subdir)
      },
      closeBundle: {
        order: 'post',
        sequential: true,
        async handler() {
          console.info(`Copying '${inFs}' to ${copyTarget}...`)
          await cp(inFs, copyTarget, { recursive: true })
          console.info('Done.')
        },
      },
    },
  ]
}

export default copyPlugin