#!/usr/bin/env node

import { consola } from 'consola'
import { colorize } from 'consola/utils'
import { defineCommand } from 'citty'
import { createRequire } from 'node:module'
import {
  build,
  createDevServer,
  createNitro,
  prepare,
  prerender,
} from 'nitropack'
import { resolve } from 'pathe'
import { fileURLToPath } from 'node:url'
import { dirname } from 'pathe'
import { getNitroConfig } from '../../get-nitro-config.js'

const hmrKeyRe = /^runtimeConfig\.|routeRules\./
const __filename = fileURLToPath(import.meta.url)
const __dirname = dirname(__filename)

async function printInfo() {
  try {
    const _require = createRequire(import.meta.url)
    const mcflyPkg = await _require('@mcflyjs/core/package.json')
    const mcflyPkgVersion = `McFly ${colorize('bold', mcflyPkg.version)}`
    const nitroPkg = await _require('nitropack/package.json')
    const nitroPkgVersion = `Nitro ${nitroPkg.version}`
    consola.log(
      `${colorize('blue', mcflyPkgVersion)} ${colorize('dim', nitroPkgVersion)}`
    )
  } catch (e) {
    consola.error(e)
  }
}

async function serve(args) {
  try {
    /**
     * @type {string}
     */
    const rootDir = resolve(args.dir || args._dir || '.')

    /**
     * @typedef {import('nitropack').Nitro} Nitro
     * @type {Nitro}
     */
    let nitro
    const reload = async () => {
      // close existing nitro
      if (nitro) {
        consola.info('Restarting dev server...')
        if ('unwatch' in nitro.options._c12) {
          await nitro.options._c12.unwatch()
        }
        await nitro.close()
      }

      const nitroConfig = await getNitroConfig()

      // create new nitro
      nitro = await createNitro(
        {
          rootDir,
          dev: true,
          preset: 'nitro-dev',
          _cli: { command: 'dev' },
          ...nitroConfig,
        },
        {
          watch: true,
          c12: {
            async onUpdate({ getDiff, newConfig }) {
              const diff = getDiff()

              if (diff.length === 0) {
                return // No changes
              }

              consola.info(
                'Nitro config updated:\n' +
                  diff.map((entry) => `  ${entry.toString()}`).join('\n')
              )

              await (diff.every((e) => hmrKeyRe.test(e.key))
                ? nitro.updateConfig(newConfig.config || {}) // Hot reload
                : reload()) // Full reload
            },
          },
        }
      )
      nitro.hooks.hookOnce('restart', reload)

      nitro.options.handlers.push({
        middleware: true,
        handler: resolve(__dirname, '../../route-middleware.js'),
      })
      const server = createDevServer(nitro)

      // const listenOptions = parseArgs(args)
      await server.listen(1234)
      await prepare(nitro)
      await prerender(nitro)
      await build(nitro)
    }
    await reload()
  } catch (e) {
    consola.error(e)
  }
}

export default defineCommand({
  meta: {
    name: 'serve',
    description: 'Runs the dev server.',
  },
  async run({ args }) {
    await printInfo()
    await serve(args)
  },
})

export const exportedForTest = {
  serve,
  printInfo,
}