Nuxt/packages/nuxi/src/utils/server.ts

import type { RequestListener } from 'http'
import type { ListenOptions } from 'listhen'
import { loading } from '@nuxt/design'

export function createServer (defaultApp?) {
  const listener = createDynamicFunction(defaultApp || createLoadingHandler('Loading...'))

  async function listen (opts?: Partial<ListenOptions>) {
    const { listen } = await import('listhen')
    return listen(listener.call, opts)
  }

  return {
    setApp: (app: RequestListener) => listener.set(app),
    listen
  }
}

export function createLoadingHandler (message: string): RequestListener {
  return (_req, res) => {
    res.setHeader('Content-Type', 'text/html; charset=UTF-8')
    res.statusCode = 503 /* Service Unavailable */
    res.end(loading({ loading: message }))
  }
}

function createDynamicFunction<T extends (...args: any[]) => any> (initialValue: T) {
  let fn = initialValue
  return {
    set: (newFn: T) => { fn = newFn },
    call: ((...args: Parameters<T>) => fn(...args)) as T
  }
}