summaryrefslogtreecommitdiffhomepage
path: root/packages/opencode/src/server/adapter.node.ts
blob: 2f6b2787f50a23668c1cd7a7073d686eed0b6592 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import { createAdaptorServer, type ServerType } from "@hono/node-server"
import { createNodeWebSocket } from "@hono/node-ws"
import type { Hono } from "hono"
import type { Adapter, FetchApp, Opts } from "./adapter"

async function listen(app: FetchApp, opts: Opts, inject?: (server: ServerType) => void) {
  const start = (port: number) =>
    new Promise<ServerType>((resolve, reject) => {
      const server = createAdaptorServer({ fetch: app.fetch })
      inject?.(server)
      const fail = (err: Error) => {
        cleanup()
        reject(err)
      }
      const ready = () => {
        cleanup()
        resolve(server)
      }
      const cleanup = () => {
        server.off("error", fail)
        server.off("listening", ready)
      }
      server.once("error", fail)
      server.once("listening", ready)
      server.listen(port, opts.hostname)
    })

  const server = opts.port === 0 ? await start(4096).catch(() => start(0)) : await start(opts.port)
  const addr = server.address()
  if (!addr || typeof addr === "string") {
    throw new Error(`Failed to resolve server address for port ${opts.port}`)
  }

  let closing: Promise<void> | undefined
  return {
    port: addr.port,
    stop(close?: boolean) {
      closing ??= new Promise<void>((resolve, reject) => {
        server.close((err) => {
          if (err) {
            reject(err)
            return
          }
          resolve()
        })
        if (close) {
          if ("closeAllConnections" in server && typeof server.closeAllConnections === "function") {
            server.closeAllConnections()
          }
          if ("closeIdleConnections" in server && typeof server.closeIdleConnections === "function") {
            server.closeIdleConnections()
          }
        }
      })
      return closing
    },
  }
}

export const adapter: Adapter = {
  create(app: Hono) {
    const ws = createNodeWebSocket({ app })
    return {
      upgradeWebSocket: ws.upgradeWebSocket,
      listen: (opts) => listen(app, opts, ws.injectWebSocket),
    }
  },
  createFetch(app) {
    return {
      listen: (opts) => listen(app, opts),
    }
  },
}