import { resolve, relative } from 'node:path'
import type { Readable } from 'node:stream'
import resolveFiles, { resolveSync } from 'resolve-dependencies'

import type { NexeCompiler } from '../compiler.js'
import { NexeError } from '../compiler.js'
import { dequote, STDIN_FLAG } from '../util.js'

async function getStdIn(stdin: Readable): Promise<string> {
  let out = ''
  return await new Promise((resolve) => {
    stdin
      .setEncoding('utf8')
      .on('readable', () => {
        let current
        while ((current = stdin.read())) {
          out += current
        }
      })
      .on('end', () => resolve(out.trim()))
    setTimeout(() => {
      if (!out.trim()) {
        resolve(out.trim())
      }
    }, 1000)
  })
}

export default async function bundle(compiler: NexeCompiler, next: any) {
  const { bundle: doBundle, cwd, input: inputPath } = compiler.options
  let input = inputPath
  compiler.entrypoint = './' + relative(cwd, input)
  compiler.startup = ''

  if (!doBundle) {
    await compiler.addResource(resolve(cwd, input))
    return next()
  }

  let code = ''
  if (typeof doBundle === 'string') {
    // eslint-disable-next-line @typescript-eslint/no-var-requires
    code = await require(doBundle).createBundle(compiler.options)
  }

  if (input === STDIN_FLAG && (code = code || dequote(await getStdIn(process.stdin)))) {
    compiler.stdinUsed = true
    compiler.entrypoint = './__nexe_stdin.js'
    await compiler.addResource(resolve(cwd, compiler.entrypoint), code)
    return next()
  }

  if (input === STDIN_FLAG) {
    const maybeInput = resolveSync(cwd, '.')
    if (!maybeInput || !maybeInput.absPath) {
      throw new NexeError('No valid input detected')
    }
    input = maybeInput.absPath
    compiler.entrypoint = './' + relative(cwd, input)
  }

  const { files, warnings } = resolveFiles(
    input,
    ...Object.keys(compiler.bundle.list).filter((x) => x.endsWith('.js')),
    { cwd, expand: 'variable', loadContent: false }
  )

  if (
    warnings.filter((x) => x.startsWith('Error parsing file') && !x.includes('node_modules'))
      .length > 0
  ) {
    throw new NexeError('Parsing Error:\n' + warnings.join('\n'))
  }

  warnings.forEach((x: string) => compiler.log.step('[WARNING]: ' + x, 'stopAndPersist', 'yellow'))

  await Promise.all(
    Object.entries(files).map(async ([key, file]) => {
      return await compiler.addResource(key, file)
    })
  )

  return next()
}