123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177
import { path, serve as stdServe, ws } from '../deps.ts'import { RouteModule } from '../framework/core/routing.ts'import { existsFileSync } from '../shared/fs.ts'import log from '../shared/log.ts'import util from '../shared/util.ts'import type { ServerRequest } from '../types.ts'import { Request } from './api.ts'import { Application } from './app.ts'import { getContentType } from './mime.ts'import { createHtml, reHashJs } from './util.ts'
/** The Aleph Server class. */export class Server {  #app: Application  #ready: boolean
  constructor(app: Application) {    this.#app = app    this.#ready = false  }
  async handle(r: ServerRequest) {    if (!this.#ready) {      await this.#app.ready      this.#ready = true    }
    const app = this.#app    const url = new URL('http://localhost' + r.url)    const pathname = util.cleanPath(decodeURI(url.pathname))    const req = new Request(r, pathname, {}, url.searchParams)
    try {      // serve hmr ws      if (pathname === '/_hmr') {        const { conn, r: bufReader, w: bufWriter, headers } = r        ws.acceptWebSocket({ conn, bufReader, bufWriter, headers }).then(async socket => {          const watcher = app.createFSWatcher()          watcher.on('add', (mod: RouteModule) => socket.send(JSON.stringify({ ...mod, type: 'add' })))          watcher.on('remove', (url: string) => {            watcher.removeAllListeners('modify-' + url)            socket.send(JSON.stringify({ type: 'remove', url }))          })          for await (const e of socket) {            if (util.isNEString(e)) {              try {                const data = JSON.parse(e)                if (data.type === 'hotAccept' && util.isNEString(data.url)) {                  const mod = app.getModule(data.url)                  if (mod) {                    watcher.on('modify-' + mod.url, (hash: string) => socket.send(JSON.stringify({                      type: 'update',                      url: mod.url,                      updateUrl: util.cleanPath(`${app.config.baseUrl}/_aleph/${util.trimModuleExt(mod.url)}.${util.shortHash(hash!)}.js`),                      hash,                    })))                  }                }              } catch (e) { }            } else if (ws.isWebSocketCloseEvent(e)) {              break            }          }          app.removeFSWatcher(watcher)        })        return      }
      // serve public files      const filePath = path.join(app.workingDir, 'public', pathname)      if (existsFileSync(filePath)) {        const info = Deno.lstatSync(filePath)        const lastModified = info.mtime?.toUTCString() ?? new Date().toUTCString()        if (lastModified === r.headers.get('If-Modified-Since')) {          req.status(304).send('')          return        }
        const body = Deno.readFileSync(filePath)        req.setHeader('Last-Modified', lastModified)        req.send(body, getContentType(filePath))        return      }
      // serve APIs      if (pathname.startsWith('/api/')) {        app.handleAPI(r, { pathname, search: url.search })        return      }
      // serve dist files      if (pathname.startsWith('/_aleph/')) {        if (pathname.startsWith('/_aleph/data/') && pathname.endsWith('.json')) {          let p = util.trimSuffix(util.trimPrefix(pathname, '/_aleph/data'), '.json')          if (p === '/index') {            p = '/'          }          const [status, data] = await app.getSSRData({ pathname: p })          if (status === 200) {            req.send(JSON.stringify(data), 'application/json; charset=utf-8')          } else {            req.status(status).send('')          }          return        } else if (reHashJs.test(pathname) && ['main', 'main.bundle'].includes(util.trimPrefix(pathname, '/_aleph/').replace(reHashJs, ''))) {          req.send(app.getMainJS(pathname.startsWith('/_aleph/main.bundle')), 'application/javascript; charset=utf-8')          return        } else {          const filePath = path.join(app.buildDir, util.trimPrefix(pathname, '/_aleph/'))          if (existsFileSync(filePath)) {            const info = Deno.lstatSync(filePath)            const lastModified = info.mtime?.toUTCString() ?? new Date().toUTCString()            if (lastModified === r.headers.get('If-Modified-Since')) {              req.status(304).send('')              return            }
            let content = await Deno.readTextFile(filePath)
            if (reHashJs.test(filePath)) {              const metaFile = filePath.replace(reHashJs, '') + '.meta.json'              if (existsFileSync(metaFile)) {                try {                  const { url } = JSON.parse(await Deno.readTextFile(metaFile))                  const mod = app.getModule(url)                  if (mod && app.isHMRable(mod.url)) {                    content = app.injectHMRCode(mod, content)                  }                } catch (e) { }              }            }
            req.setHeader('Last-Modified', lastModified)            req.send(content, getContentType(filePath))            return          }        }
        req.status(404).send('file not found')        return      }
      // ssr      const [status, html] = await app.getPageHtml({ pathname, search: url.search })      req.status(status).send(html, 'text/html; charset=utf-8')    } catch (err) {      req.status(500).send(createHtml({        lang: 'en',        head: ['<title>500 - internal server error</title>'],        body: `<p><strong><code>500</code></strong><small> - </small><span>${err.message}</span></p>`      }), 'text/html; charset=utf-8')    }  }}
/** start a standard aleph server. */export async function serve(hostname: string, port: number, app: Application) {  const server = new Server(app)  await app.ready  while (true) {    try {      const s = stdServe({ hostname, port })      log.info(`Aleph server ready on http://${hostname}:${port}`)      for await (const r of s) {        server.handle(r)      }    } catch (err) {      if (err instanceof Deno.errors.AddrInUse) {        log.warn(`port ${port} already in use, try ${port + 1}`)        port++      } else {        log.fatal(err.message)      }    }  }}